langchain4j （三）

PromptTemplate提示词模板Testpublic void testChatModelQwenChatStructuredPromptTemplete() {String apiKey System.getenv(“aliQwen-key”);OpenAiChatModel model OpenAiChatModel.builder().baseUrl(“https://dashscope.aliyuncs.com/compatible-mode/v1”).apiKey(apiKey).modelName(“qwen-long”).build();String role “外科医生”;String question “牙疼”;PromptTemplate template PromptTemplate.from(“你是一个{{it}}助手,{{question}}怎么办”);MapString, Object variables new java.util.HashMap();variables.put(“it”, role);variables.put(“question”, question);Prompt prompt template.apply(variables);//3 Prompt提示词变成UserMessageUserMessage userMessage prompt.toUserMessage();//4 调用大模型ChatResponse chatResponse model.chat(userMessage);//4.1 后台打印System.out.println(chatResponse.aiMessage().text());}package org.folio.rest.util;import dev.langchain4j.data.message.ChatMessage;import dev.langchain4j.data.message.ChatMessageDeserializer;import dev.langchain4j.data.message.ChatMessageSerializer;import dev.langchain4j.memory.chat.ChatMemoryProvider;import dev.langchain4j.memory.chat.MessageWindowChatMemory;import dev.langchain4j.store.memory.chat.ChatMemoryStore;import com.hazelcast.core.HazelcastInstance;import com.hazelcast.map.IMap;import java.util.List;import java.util.concurrent.ConcurrentHashMap;import java.util.concurrent.ConcurrentMap;public class HazelcastChatMemoryStore implements ChatMemoryStore {private final IMapObject, String memoryMap;private final ConcurrentMapObject, Object locks new ConcurrentHashMap();public HazelcastChatMemoryStore(HazelcastInstance hazelcastInstance) {this.memoryMap hazelcastInstance.getMap(“chat-memories”);}Overridepublic List getMessages(Object memoryId) {String value memoryMap.get(memoryId);return ChatMessageDeserializer.messagesFromJson(value);// return value ! null ? ChatMessageDeserializer.messagesFromJson(value) : null;}Overridepublic void updateMessages(Object memoryId, List messages) {memoryMap.set(memoryId, ChatMessageSerializer.messagesToJson(messages));}Overridepublic void deleteMessages(Object memoryId) {memoryMap.delete(memoryId);locks.remove(memoryId);}// 获取锁用于并发控制public Object getLock(Object memoryId) {return locks.computeIfAbsent(memoryId, k - new Object());}}package org.folio.rest.util;import com.hazelcast.core.HazelcastInstance;import dev.langchain4j.memory.chat.ChatMemoryProvider;import dev.langchain4j.memory.chat.MessageWindowChatMemory;import dev.langchain4j.store.memory.chat.ChatMemoryStore;public class ChatMemoryManager {private final ChatMemoryStore store;private final int maxMessages;public ChatMemoryManager(HazelcastInstance hazelcastInstance, int maxMessages) {this.store new HazelcastChatMemoryStore(hazelcastInstance);this.maxMessages maxMessages;}public ChatMemoryProvider createMemoryProvider() {return memoryId - MessageWindowChatMemory.builder().id(memoryId).maxMessages(maxMessages).chatMemoryStore(store).build();}public void clearMemory(String sessionId) {store.deleteMessages(sessionId);}public boolean hasMemory(String sessionId) {return store.getMessages(sessionId) ! null !store.getMessages(sessionId).isEmpty();}}tools调用Testpublic void testChatModelQwenChatFunction() {String apiKey System.getenv(“aliQwen-key”);QwenChatModel model QwenChatModel.builder().apiKey(apiKey).modelName(“qwen-plus”).build();// ChatMemoryManager memoryManager new ChatMemoryManager(HazecastUtil.getClient(), 20);FunctionAssistant assistant AiServices.builder(FunctionAssistant.class).chatModel(model).tools(new InvoiceHandler())// .chatMemoryProvider(memoryManager.createMemoryProvider()).build();// 避免 InvalidRequestException 错误确保输入格式正确并非空LangChain4j 需要至少有用户输入消息且内容非空// 可以尝试先用 UserMessage.from 创建消息对象String response1 assistant.chat(“请帮我计算 1 加 2 等于多少”);System.out.println(AI Response 1: response1);String response2 assistant.chat(“陈培林是谁”);System.out.println(AI Response 2: response2);}Tool(“计算两个数字的和返回 a b 的结果”)public double sum(double a, double b) {double result a b ;System.out.printf(“工具调用: sum(%.2f, %.2f) %.2f%n”, a, b, result);return result;}Tool(“回答用户提问陈培林是谁”)public String handle(P(“人名”) String name) {return “陈培林是深圳市四海智慧科技有限公司的全职后端”;}/**rag