diff --git a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/LLMRagWithSpringBoot.java b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/LLMRagWithSpringBoot.java index b3ca12d..8b824b8 100644 --- a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/LLMRagWithSpringBoot.java +++ b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/LLMRagWithSpringBoot.java @@ -1,6 +1,6 @@ package com.learning.ai; -import com.learning.ai.config.CustomerSupportAgent; +import com.learning.ai.config.AICustomerSupportAgent; import java.util.Map; import org.springframework.boot.ApplicationRunner; import org.springframework.boot.SpringApplication; @@ -15,7 +15,7 @@ public static void main(String[] args) { } @Bean - ApplicationRunner interactiveChatRunner(CustomerSupportAgent agent) { + ApplicationRunner interactiveChatRunner(AICustomerSupportAgent agent) { return args -> { var response = agent.chat("what should I know about the transition to consumer direct care network washington?"); diff --git a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AIConfig.java b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AIConfig.java index b544d58..70ce611 100644 --- a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AIConfig.java +++ b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AIConfig.java @@ -32,11 +32,11 @@ public class AIConfig { @Bean - CustomerSupportAgent customerSupportAgent( + AICustomerSupportAgent customerSupportAgent( ChatLanguageModel chatLanguageModel, // ChatTools bookingTools, ContentRetriever contentRetriever) { - return AiServices.builder(CustomerSupportAgent.class) + return AiServices.builder(AICustomerSupportAgent.class) .chatLanguageModel(chatLanguageModel) .chatMemory(MessageWindowChatMemory.withMaxMessages(20)) // .tools(bookingTools) diff --git a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/CustomerSupportAgent.java b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AICustomerSupportAgent.java similarity index 87% rename from llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/CustomerSupportAgent.java rename to llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AICustomerSupportAgent.java index a9f8f12..abcc982 100644 --- a/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/CustomerSupportAgent.java +++ b/llm-rag-with-langchain4j-spring-boot/src/main/java/com/learning/ai/config/AICustomerSupportAgent.java @@ -1,8 +1,9 @@ package com.learning.ai.config; import dev.langchain4j.service.SystemMessage; +import dev.langchain4j.service.V; -public interface CustomerSupportAgent { +public interface AICustomerSupportAgent { @SystemMessage({ """ @@ -19,5 +20,5 @@ public interface CustomerSupportAgent { {documents} """ }) - String chat(String documents); + String chat(@V("documents") String documents); }