add generate Entity code tool

agents/code_generate_agent.py

@@ -8,7 +8,7 @@ from langchain.agents import initialize_agent
 from langchain.prompts import StringPromptTemplate
 from agents.promopts import code_generate_agent_template
 from agents.tools.api_layer_code_tool import apiLayerCodeGenerator
-from agents.tools.domain_layer_code_tool import domainLayerCodeGenerator
+from agents.tools.domain_layer_code_tool import domainLayerCodeGenerator, entityCodeGenerator
 from agents.tools.persistent_layer_code_tool import persistentLayerCodeGenerator
 
 
@@ -66,7 +66,7 @@ class CustomOutputParser(AgentOutputParser):
 
 # agent = initialize_agent(
 #     tools=tools, llm=llm_chain, template=AGENT_PROMPT, stop=["\nObservation:"], agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
-code_agent_tools = [domainLayerCodeGenerator, persistentLayerCodeGenerator, apiLayerCodeGenerator]
+code_agent_tools = [domainLayerCodeGenerator, entityCodeGenerator, persistentLayerCodeGenerator, apiLayerCodeGenerator]
 
 def code_agent_executor() -> AgentExecutor:
     output_parser = CustomOutputParser()

agents/tools/domain_layer_code_tool.py

@@ -22,7 +22,9 @@ the domain layer inclue 2 componets:
             }}
 
             @Builder
-            public record FeatureDescription(String name, String description, Boolean isEnable))) {{
+            public record FeatureDescription(String name, 
+                            String description, 
+                            Boolean isEnable))) {{
             
             }}
         }}
@@ -82,4 +84,115 @@ domainLayerChain = LLMChain(llm = ChatOpenAI(temperature=0.1), prompt=DOMAIN_LAY
 def domainLayerCodeGenerator(input: str) -> str:
     '''useful for when you need to generate domain layer code'''
     response = domainLayerChain.run(input)
+    return response
+
+
+
+
+ENTITY = """You are a software developer. Your task is to generate the Enity of domain layer tests and product code.
+
+===TechStack
+Java17、reactor、lombok、Junit5、reactor test、Mockito
+===END OF TechStack
+
+===Architecture
+Entity is a kine of component of the domain layer, which is use to represents business concepts and encapsulates business rules.
+Entity may include 3 parts: 
+- id(identity of entity)
+- description (properties package of entity represent the value of entity), 
+- associations (collection of associated entiy)
+
+---example code: 
+    public class Feature {{
+        // id
+        private FeatureId id;
+        
+        // description
+        private FeatureDescription description;
+
+        // associations
+        private FeatureConfigs configs;
+
+        public record FeatureId(String featureKey) {{
+        
+        }}
+
+        @Builder
+        public record FeatureDescription(String name, 
+                            String description, 
+                            Boolean isEnable,
+                            LocalDateTime updatedAt,
+                            LocalDateTime createdAt))) {{
+        
+        }}
+
+        public Feature update(Feature newFeature) {{
+             this.description = FeatureDescription.builder()
+                .name(newFeature.description.name())
+                .description(newFeature.description.description())
+                .isEnable(this.description.isEnable())
+                .updatedAt(LocalDateTime.now())
+                .createdAt(this.description.createdAt());
+
+            return this;
+        }}
+
+        public interface FeatureConfigs() {{
+            Flux<FeatureConfig> findAll();
+            Flux<FeatureConfig> subCollection(long from, long to);
+            Mono<FeatureConfig> findById(FeatureConfigId id);
+        }}
+    }}
+---end of example code
+===END OF Architecture
+
+===TestStrategy
+For the Entity, we can write unit test to ensure that the business rules related to Merchandise are correctly encapsulated.
+---example code
+    class FeatureTest {{
+        @Test
+        void should_update_feature_description() {{
+            // given
+            Feature feature = Feature.builder()
+                .id(new FeatureId("featureKey"))
+                .description(new FeatureDescription("name", "description", true, LocalDateTime.now(), LocalDateTime.now()))
+                .build();
+            Feature newFeature = Feature.builder()
+                .id(new FeatureId("featureKey"))
+                .description(new FeatureDescription("newName", "newDescription", true, LocalDateTime.now(), LocalDateTime.now()))
+                .build();
+            // when
+            feature.update(newFeature);
+            // then
+            assertThat(feature.description().name()).isEqualTo("newName");
+            assertThat(feature.description().description()).isEqualTo("newDescription");
+        }}
+    }}
+---end of example code
+===END OF TestStrategy
+
+Use the following format:
+request: the request that you need to fulfill
+
+Entity:
+```
+the Entity code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Test:
+```
+the test code that you write to fulfill the request, follow TechStack Architecture and TestStrategy
+```
+
+request: {input}"""
+
+ENTITY_PROMPT = PromptTemplate(input_variables=["input"], template=ENTITY,)
+
+entityChain = LLMChain(llm = ChatOpenAI(temperature=0.1), prompt=ENTITY_PROMPT)
+
+
+@tool("Generate Entity Code", return_direct=True)
+def entityCodeGenerator(input: str) -> str:
+    '''useful for when you need to generate entity code'''
+    response = entityChain.run(input)
     return response

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import TextLoader
-from agents.tools.python_util_tool import generate_and_excute_python_code
+from agents.tools.python_code_tool import generate_and_excute_python_code
 from chains import HumanFeedBackChain, contextRewriteChain
 from embedding import CustomEmbedding
 from memories import HumenFeedbackBufferMemory
@@ -146,7 +146,7 @@ with gr.Blocks() as demo:
                                     sendMessageByMultiPart, [code_chatbot, code_context, relateCode, toolTextBox[index]], [code_chatbot]).then(
                                     generateCodeByMultiPart, [code_context, relateCode, toolTextBox[index], code_chatbot], [code_chatbot, code])
             with gr.Tab("FAQ"):
-                faq_chatbot = gr.Chatbot().style(max_width="50%")
+                faq_chatbot = gr.Chatbot().style()
                 with gr.Row():
                     faq = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
                         container=False)