소스 검색

:zap: fix chat prompt build

tycoding 8 달 전
부모
커밋
37d52b9aed
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      langchat-ai/langchat-ai-core/src/main/java/cn/tycoding/langchat/ai/core/service/impl/LangChatServiceImpl.java

+ 3 - 1
langchat-ai/langchat-ai-core/src/main/java/cn/tycoding/langchat/ai/core/service/impl/LangChatServiceImpl.java

@@ -63,12 +63,14 @@ public class LangChatServiceImpl implements LangChatService {
 
     private AiServices<Agent> build(StreamingChatLanguageModel streamModel, ChatLanguageModel model, ChatReq req) {
         AiServices<Agent> aiServices = AiServices.builder(Agent.class)
-                .systemMessageProvider(memoryId -> req.getPromptText())
                 .chatMemoryProvider(memoryId -> MessageWindowChatMemory.builder()
                         .id(req.getConversationId())
                         .chatMemoryStore(new PersistentChatMemoryStore())
                         .maxMessages(chatProps.getMemoryMaxMessage())
                         .build());
+        if (StrUtil.isNotBlank(req.getPromptText())) {
+            aiServices.systemMessageProvider(memoryId -> req.getPromptText());
+        }
         if (streamModel != null) {
             aiServices.streamingChatLanguageModel(streamModel);
         }