Explorar o código

add:添加系统消息注解以定义AI角色

xingbo hai 1 día
pai
achega
283ff20d34

+ 2 - 0
nb-core/src/main/java/com/nb/core/ai/service/StreamAIService.java

@@ -1,5 +1,6 @@
 package com.nb.core.ai.service;
 
+import dev.langchain4j.service.SystemMessage;
 import reactor.core.publisher.Flux;
 
 
@@ -10,6 +11,7 @@ public interface StreamAIService {
      * @param message 用户消息
      * @return 流式响应
      */
+    @SystemMessage("你是驼人的专用医疗专家")
     Flux<String> generateStream(String message);
 
 }