update smart domain promopt

agents/code_generate_agent.py

@@ -7,7 +7,11 @@ 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 asociationCodeGenerator, domainLayerCodeGenerator, entityCodeGenerator
+from agents.tools.domain_layer_code_tool import domainLayerCodeGenerator
+from agents.tools.smart_domain.entity import entityCodeGenerator
+from agents.tools.smart_domain.association import associationCodeGenerator
+from agents.tools.smart_domain.db_entity_repository import dbEntityRepositoryCodeGenerator
+from agents.tools.smart_domain.association_impl import asociationImplCodeGenerator
 from agents.tools.persistent_layer_code_tool import persistentLayerCodeGenerator
 from models import llm
 
@@ -66,7 +70,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, entityCodeGenerator, asociationCodeGenerator, persistentLayerCodeGenerator, apiLayerCodeGenerator]
+code_agent_tools = [domainLayerCodeGenerator, entityCodeGenerator, associationCodeGenerator, persistentLayerCodeGenerator, dbEntityRepositoryCodeGenerator, asociationImplCodeGenerator, apiLayerCodeGenerator]
 
 def code_agent_executor() -> AgentExecutor:
     output_parser = CustomOutputParser()

agents/tools/domain_layer_code_tool.py

@@ -1,115 +1,26 @@
 from langchain import LLMChain, PromptTemplate
 from langchain.agents import tool
+from agents.tools.smart_domain.common import getPrefix
 
 from models import llm
+from agents.tools.smart_domain.entity import entity_architecture, entity_test_strategy, entity_tech_stack
+from agents.tools.smart_domain.association import association_architecture, association_test_strategy, association_teck_stack
 
 
-DOMAIN_LAYER = """You are a software developer. Your task is to generate the domain layer tests and product code.
 
-===TechStack
-Java17、reactor、lombok、Junit5、reactor test、Mockito
-===END OF TechStack
 
-===Architecture
-the domain layer inclue 2 componets:
-* Entity: This component 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:
-        @Getter
-        @AllArgsConstructor 
-        public class Feature {{
-            // id
-            private FeatureId id;
-            
-            // description
-            private FeatureDescription description;
+domain_task = """Your task is to generate the domain layer tests and product code."""
+domain_teck_stack = """Java17、reactor、lombok、Junit5、reactor test、Mockito"""
+domain_architecture = f"""the domain layer inclue 2 componets:
+* {entity_architecture}
+* {association_architecture}"""
 
-            // associations
-            private FeatureConfigs configs;
+domain_test_strategy = f"""{entity_test_strategy}
+{association_test_strategy}"""
 
-            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
-* Association: This component is use to define association between entities, which can represents the concept of a set of entity. 
-    ---eaxmple code: 
-        public interface Features {{
-            Flux<Feature> findAll();
-
-            Mono<Long> size();
-
-            Flux<Feature> subCollection(long from, long to);
-
-            Mono<Feature> findById(FeatureId id);
-
-            Mono<Feature> save(Feature feature);
-
-            Mono<Void> update(FeatureId id, FeatureDescription description);
-
-            Mono<Void> delete(FeatureId id);
-
-            Mono<Void> publish(FeatureId id);
-
-            Mono<Void> disable(FeatureId id);
-        }}
-    ---end of eaxmple 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
-For the Association，do not write tests because it is has no impletation.
-===END OF TestStrategy
+DOMAIN_LAYER = getPrefix(domain_task, domain_teck_stack, domain_architecture, domain_test_strategy) + """
 
 Use the following format:
 request: the request that you need to fulfill
@@ -132,6 +43,7 @@ the test code that you write to fulfill the request, follow TechStack Architectu
 request: {input}"""
 
 
+
 DOMAIN_LAYER_PROMPT = PromptTemplate(input_variables=["input"], template=DOMAIN_LAYER,)
 
 domainLayerChain = LLMChain(llm = llm(temperature=0.1), prompt=DOMAIN_LAYER_PROMPT)
@@ -141,173 +53,4 @@ domainLayerChain = LLMChain(llm = llm(temperature=0.1), prompt=DOMAIN_LAYER_PROM
 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:
-    @Getter
-    @AllArgsConstructor
-    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 = llm(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
-
-ASSOCIATION = """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
-Association is a kine of component of the domain layer, is use to define association between entities, which can represents the concept of a collection of entity, so it can include same business logic of entity collection.
----eaxmple code: 
-    public interface Features {{
-        Flux<Feature> findAll();
-
-        Mono<Long> size();
-
-        Flux<Feature> subCollection(long from, long to);
-
-        Mono<Feature> findById(FeatureId id);
-
-        Mono<Feature> save(Feature feature);
-
-        Mono<Void> update(FeatureId id, FeatureDescription description);
-
-        Mono<Void> delete(FeatureId id);
-
-        Mono<Void> publish(FeatureId id);
-
-        Mono<Void> disable(FeatureId id);
-    }}
----end of eaxmple code
-===END OF Architecture
-
-===TestStrategy
-For the Association，do not write tests because it is has no impletation.
-===END OF TestStrategy
-
-Use the following format:
-request: the request （whitch may include Enity existed in the domain layer）that you need to fulfill, 
-
-Association:
-```
-the Association code that you write to fulfill the request, follow TechStack and Architecture
-```
-
-request: {input}"""
-
-ASSOCIATION_PROMPT = PromptTemplate(input_variables=["input"], template=ASSOCIATION,)
-
-asociationChain = LLMChain(llm = llm(temperature=0.1), prompt=ASSOCIATION_PROMPT)
-
-
-@tool("Generate Asociation Code", return_direct=True)
-def asociationCodeGenerator(input: str) -> str:
-    '''useful for when you need to generate asociation code'''
-    response = asociationChain.run(input)
     return response

agents/tools/persistent_layer_code_tool.py

@@ -2,139 +2,21 @@ from langchain import LLMChain, PromptTemplate
 from langchain.agents import tool
 
 from models import llm
+from agents.tools.smart_domain.common import getPrefix
+from agents.tools.smart_domain.db_entity_repository import db_entity_architecture, db_entity_test_strategy
+from agents.tools.smart_domain.association_impl import association_impl_architecture, association_impl_test_strategy
 
 
-PERSISTENT_LAYER = """You are a software developer. Your task is to generate the persistent layer tests and product code.
+persistent_task = """"Your task is to generate the persistent layer tests and product code."""
+persistent_tech_stack = """Java17、reactor、lombok、Junit5、reactor test、Mockito、 Spring Data Reactive Couchbase、Testcontainers、Couchbase、WebClient"""
+persistent_architecture = f"""the persistent layer inclue 3 componets:
+{db_entity_architecture}
+{association_impl_architecture}"""
 
-===TechStack
-Java17、reactor、lombok、Junit5、reactor test、Mockito、 Spring Data Reactive Couchbase、Testcontainers、Couchbase、WebClient
-===END OF TechStack
+persistent_test_strategy = f"""{db_entity_test_strategy}
+{association_impl_test_strategy}"""
 
-===Architecture
-the persistent layer inclue 3 componets:
-* DbEntity: This component is use to define data structure that save to DB.
-    ---eaxmple code:
-        @Document 
-        public class FeatureDb {{
-            @Version
-            private long version;
-
-            @Id
-            @GeneratedValue(strategy = GenerationStrategy.UNIQUE)
-            private String id;
-            
-            private String featureKey;
-
-            private Feature.FeatureDescription description;
-        }}
-    ---end of eaxmple code
-* Repository: This component is use to define the interface to access DB.
-    ---eaxmple code:
-        public interface FeatureDbRepository extends ReactiveCrudRepository<FeatureDb, String> {{
-            Mono<FeatureDb> findByFeatureKey(String featureKey);
-        }}
-    ---end of eaxmple code
-* Association Impletation: This component provide implementation of Association of domain layer, rely on Repository. 
-    ---eaxmple code: 
-        @Component
-        @RequiredArgsConstructor
-        public class FeaturesImpl implements Features{{
-            private final FeatureDbRepository featureDbRepository;
-
-            Flux<Feature> findAll() {{
-                return featureDbRepository.findAll().map(FeatureDb::toFeature);
-            }}
-
-            Mono<Feature> save(Feature feature) {{
-                return featureDbRepository.save(FeatureDb.fromFeature(feature)).map(FeatureDb::toFeature);
-            }}
-        }}
-    ---end of eaxmple code
-===END OF Architecture
-
-===TestStrategy
-For the DbEntity And Repository, we can write component test to test the actual implementation of database operations, test class should extends RepositoryTestBase to use Testcontainers ability.
-    ---eaxmple code: 
-        class FeatureDbRepositoryTest extends RepositoryTestBase {{
-            @Autowired
-            FeatureDbRepository repository;
-
-            @BeforeEach
-            void setUp() {{
-                repository.deleteAll().block();
-            }}
-
-            @AfterEach
-            void tearDown() {{
-                repository.deleteAll().block();
-            }}
-
-            @Test
-            void should_save_Feature_success() {{
-                var featureKey = "featureKey1";
-                repository.save(FeatureTestUtil.createFeatureDb(featureKey))
-                        .as(StepVerifier::create)
-                        .expectNextCount(1)
-                        .verifyComplete();
-            }}
-
-            @Test
-            void should_add_same_featureKey_fail() {{
-                var featureKey = "featureKey1";
-                repository.save(FeatureTestUtil.createFeatureDb(featureKey)).block();
-
-                repository.save(FeatureTestUtil.createFeatureDb(featureKey))
-                        .as(StepVerifier::create)
-                        .expectError()
-                        .verify();
-            }}
-        }}
-    ---end of eaxmple code
-For Association Impletation，we writ unit test, and stub repository method with Mockito.
-    ---eaxmple code:
-        @ExtendWith(MockitoExtension.class)
-        class FeatureImplTest {{
-            @Mock
-            FeatureDbRepository repository;
-
-            Features features;
-
-            @BeforeEach
-            void setUp() {{
-                features = new FeaturesImpl(repository);
-            }}
-
-            @Test
-            void should_add_success() {{
-                when(repository.save(any(FeatureDb.class))).thenAnswer(invocation -> {{
-                    FeatureDb featureDb = invocation.getArgument(0);
-                    return Mono.just(featureDb);
-                }});
-
-                features.add(createFeature("featureKey1"))
-                        .as(StepVerifier::create)
-                        .expectNextMatches(config -> config.getId().featureKey().equals("featureKey1")
-                                && config.getDescription().updatedAt() != null
-                                && config.getDescription().createdAt() != null
-                        )
-                        .verifyComplete();
-            }}
-
-            @Test
-            void should_add_return_error_when_repository_save_error() {{
-                Feature feature = createFeature("featureKey1");
-
-                when(repository.save(any(FeatureDb.class))).thenReturn(Mono.error(new DuplicateKeyException("save error")));
-
-                features.add(feature)
-                        .as(StepVerifier::create)
-                        .expectError()
-                        .verify();
-            }}
-        }}
-            
-
-===END OF TestStrategy
+PERSISTENT_LAYER = getPrefix(persistent_task, persistent_tech_stack, persistent_architecture, persistent_test_strategy) + """
 
 Use the following format:
 request: the request that you need to fulfill include Entity and Association of domain layer

agents/tools/smart_domain/association.py

@@ -0,0 +1,62 @@
+from langchain import LLMChain, PromptTemplate
+from langchain.agents import tool
+
+from agents.tools.smart_domain.common import getPrefix
+from models import llm
+
+
+association_architecture = """
+Association: This component is use to define association between entities, which can represents the concept of a collection of entity, so it can include same business logic of entity collection. 
+---eaxmple code: 
+    public interface Features {{
+        Flux<Feature> findAll();
+
+        Mono<Long> size();
+
+        Flux<Feature> subCollection(long from, long to);
+
+        Mono<Feature> findById(FeatureId id);
+
+        Mono<Feature> save(Feature feature);
+
+        Mono<Void> update(FeatureId id, FeatureDescription description);
+
+        Mono<Void> delete(FeatureId id);
+
+        Mono<Void> publish(FeatureId id);
+
+        Mono<Void> disable(FeatureId id);
+    }}
+---end of eaxmple code
+"""
+
+association_test_strategy = """
+For the Association，do not write tests because it is has no impletation.
+"""
+
+association_teck_stack = """Java17、reactor、lombok、Junit5、reactor test、Mockito"""
+
+association_task = """Your task is to generate the Association of domain layer tests and product code."""
+
+ASSOCIATION = getPrefix(association_task, association_teck_stack, association_architecture, association_test_strategy) + """
+
+Use the following format:
+request: the request （whitch may include Enity existed in the domain layer）that you need to fulfill, 
+
+Association:
+```
+the Association code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+request: {input}"""
+
+ASSOCIATION_PROMPT = PromptTemplate(input_variables=["input"], template=ASSOCIATION,)
+
+asociationChain = LLMChain(llm = llm(temperature=0.1), prompt=ASSOCIATION_PROMPT)
+
+
+@tool("Generate Association Code", return_direct=True)
+def associationCodeGenerator(input: str) -> str:
+    '''useful for when you need to generate asociation code'''
+    response = asociationChain.run(input)
+    return response

agents/tools/smart_domain/association_impl.py

@@ -0,0 +1,97 @@
+from langchain import LLMChain, PromptTemplate
+from agents.tools.smart_domain.common import getPrefix
+from models import llm
+from langchain.agents import tool
+
+association_impl_architecture = """* Association Impletation: This component provide implementation of Association of domain layer, rely on Repository. 
+---eaxmple code: 
+    @Component
+    @RequiredArgsConstructor
+    public class FeaturesImpl implements Features{{
+        private final FeatureDbRepository featureDbRepository;
+
+        Flux<Feature> findAll() {{
+            return featureDbRepository.findAll().map(FeatureDb::toFeature);
+        }}
+
+        Mono<Feature> save(Feature feature) {{
+            return featureDbRepository.save(FeatureDb.fromFeature(feature)).map(FeatureDb::toFeature);
+        }}
+    }}
+---end of eaxmple code"""
+
+association_impl_test_strategy="""For Association Impletation，we writ unit test, and stub repository method with Mockito.
+    ---eaxmple code:
+        @ExtendWith(MockitoExtension.class)
+        class FeatureImplTest {{
+            @Mock
+            FeatureDbRepository repository;
+
+            Features features;
+
+            @BeforeEach
+            void setUp() {{
+                features = new FeaturesImpl(repository);
+            }}
+
+            @Test
+            void should_add_success() {{
+                when(repository.save(any(FeatureDb.class))).thenAnswer(invocation -> {{
+                    FeatureDb featureDb = invocation.getArgument(0);
+                    return Mono.just(featureDb);
+                }});
+
+                features.add(createFeature("featureKey1"))
+                        .as(StepVerifier::create)
+                        .expectNextMatches(config -> config.getId().featureKey().equals("featureKey1")
+                                && config.getDescription().updatedAt() != null
+                                && config.getDescription().createdAt() != null
+                        )
+                        .verifyComplete();
+            }}
+
+            @Test
+            void should_add_return_error_when_repository_save_error() {{
+                Feature feature = createFeature("featureKey1");
+
+                when(repository.save(any(FeatureDb.class))).thenReturn(Mono.error(new DuplicateKeyException("save error")));
+
+                features.add(feature)
+                        .as(StepVerifier::create)
+                        .expectError()
+                        .verify();
+            }}
+        }}"""
+
+association_impl_task = """Your task is to generate the Association Impletation tests and product code."""
+
+association_impl_teck_stack = """Java17、reactor、lombok、Junit5、reactor test、Mockito、 Spring Data Reactive Couchbase"""
+
+ASSOCIATION_IMPL = getPrefix(association_impl_task, association_impl_teck_stack, association_impl_architecture, association_impl_test_strategy) + """
+
+Use the following format:
+request: the request that you need to fulfill
+
+Association Impletation:
+```
+the Association Impletation code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Test:
+```
+the test code of Association Impletation that you write to fulfill the request, follow TechStack Architecture and TestStrategy
+```
+
+request: {input}
+"""
+
+ASSOCIATION_IMPL_PROMOPT = PromptTemplate(input_variables=["input"], template=ASSOCIATION_IMPL,)
+
+asociationChain = LLMChain(llm = llm(temperature=0.1), prompt=ASSOCIATION_IMPL_PROMOPT)
+
+
+@tool("Generate Asociation Impletation Code", return_direct=True)
+def asociationImplCodeGenerator(input: str) -> str:
+    '''useful for when you need to generate Asociation Impletation code'''
+    response = asociationChain.run(input)
+    return response

agents/tools/smart_domain/common.py

@@ -0,0 +1,18 @@
+tech_prefix = """You are a software developer. {task}
+
+===TechStack
+{tech_stack}
+===END OF TechStack
+
+===Architecture
+{architecture}
+===END OF Architecture
+
+===TestStrategy
+{test_strategy}
+===END OF TestStrategy
+
+"""
+
+def getPrefix(task, tech_stack, architecture, test_strategy):
+    return tech_prefix.format(task=task, tech_stack=tech_stack, architecture=architecture, test_strategy=test_strategy)

agents/tools/smart_domain/db_entity_repository.py

@@ -0,0 +1,101 @@
+from langchain.prompts import PromptTemplate
+from agents.tools.smart_domain.common import getPrefix
+from langchain.chains import LLMChain
+from langchain.agents import tool
+from models import llm
+
+db_entity_tech_stack = """Java17、reactor、lombok、Junit5、reactor test、Mockito、 Spring Data Reactive Couchbase、Couchbase"""
+
+db_entity_architecture = """
+* DbEntity: This component is use to define data structure that save to DB.
+---eaxmple code:
+    @Document 
+    public class FeatureDb {{
+        @Version
+        private long version;
+
+        @Id
+        @GeneratedValue(strategy = GenerationStrategy.UNIQUE)
+        private String id;
+        
+        private String featureKey;
+
+        private Feature.FeatureDescription description;
+    }}
+---end of eaxmple code
+* Repository: This component is use to define the interface to access DB.
+    ---eaxmple code:
+        public interface FeatureDbRepository extends ReactiveCrudRepository<FeatureDb, String> {{
+            Mono<FeatureDb> findByFeatureKey(String featureKey);
+        }}
+    ---end of eaxmple code
+"""
+
+db_entity_test_strategy = """For the DbEntity And Repository, we can write component test to test the actual implementation of database operations, test class should extends RepositoryTestBase to use Testcontainers ability.
+---eaxmple code: 
+    class FeatureDbRepositoryTest extends RepositoryTestBase {{
+        @Autowired
+        FeatureDbRepository repository;
+
+        @BeforeEach
+        void setUp() {{
+            repository.deleteAll().block();
+        }}
+
+        @AfterEach
+        void tearDown() {{
+            repository.deleteAll().block();
+        }}
+
+        @Test
+        void should_save_Feature_success() {{
+            var featureKey = "featureKey1";
+            repository.save(FeatureTestUtil.createFeatureDb(featureKey))
+                    .as(StepVerifier::create)
+                    .expectNextCount(1)
+                    .verifyComplete();
+        }}
+
+        @Test
+        void should_add_same_featureKey_fail() {{
+            var featureKey = "featureKey1";
+            repository.save(FeatureTestUtil.createFeatureDb(featureKey)).block();
+
+            repository.save(FeatureTestUtil.createFeatureDb(featureKey))
+                    .as(StepVerifier::create)
+                    .expectError()
+                    .verify();
+        }}
+    }}
+---end of eaxmple code
+"""
+
+db_entity_task = """Your task is to generate the DbEntity and Repository tests and product code."""
+
+DB_ENTITY = getPrefix(db_entity_task, db_entity_tech_stack, db_entity_architecture, db_entity_test_strategy) + """
+
+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}"""
+
+DB_ENTITY_PROMPT = PromptTemplate(input_variables=["input"], template=DB_ENTITY,)
+
+db_entity_Repository_chain = LLMChain(llm = llm(temperature=0.1), prompt=DB_ENTITY_PROMPT)
+
+
+@tool("Generate DBEntity and Repository Code", return_direct=True)
+def dbEntityRepositoryCodeGenerator(input: str) -> str:
+    '''useful for when you need to generate DBEntity and Repository code'''
+    response = db_entity_Repository_chain.run(input)
+    return response

agents/tools/smart_domain/entity.py

@@ -0,0 +1,115 @@
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from langchain.agents import tool
+from agents.tools.smart_domain.common import getPrefix
+from models import llm
+
+entity_architecture = """
+Entity: This component is use to represents business concepts and encapsulates business rules.
+It 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:
+    @Getter
+    @AllArgsConstructor 
+    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
+"""
+
+entity_test_strategy = """
+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
+"""
+
+entity_tech_stack = """
+Java17、reactor、lombok、Junit5、reactor test、Mockito
+"""
+
+entity_task = """Your task is to generate the Enity of domain layer tests and product code."""
+ENTITY = getPrefix(entity_task, entity_tech_stack, entity_architecture, entity_test_strategy) + """
+
+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 = llm(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