Преглед на файлове

: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);
         }