Commit 5e8b27f3 authored by alex yao's avatar alex yao

fix: 转换数组

parent 3a0a412d
......@@ -24,7 +24,11 @@ public class KnowledgeDefaultCache {
if (StringUtils.isNotBlank(param)) {
String[] knowledgeId = param.split(";");
if (ArrayUtils.isNotEmpty(knowledgeId)) {
return (Integer[]) Arrays.stream(knowledgeId).map(Integer::valueOf).toArray();
Integer[] res = new Integer[knowledgeId.length];
for (int i = 0; i < knowledgeId.length; i++) {
res[i] = Integer.valueOf(knowledgeId[i]);
}
return res;
}
}
return ArrayUtils.EMPTY_INTEGER_OBJECT_ARRAY;
......
package cn.com.poc.dify;
import cn.com.gsst.dify_client.DifyChatClient;
import cn.com.gsst.dify_client.DifyClientFactory;
import cn.com.gsst.dify_client.callback.ChatStreamCallback;
import cn.com.gsst.dify_client.enums.ResponseMode;
import cn.com.gsst.dify_client.event.MessageEndEvent;
import cn.com.gsst.dify_client.event.MessageEvent;
import cn.com.gsst.dify_client.exception.DifyApiException;
import cn.com.gsst.dify_client.model.chat.ChatMessage;
import cn.com.gsst.dify_client.model.chat.ChatMessageResponse;
import cn.com.yict.framemax.core.spring.SingleContextInitializer;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.concurrent.CountDownLatch;
/**
* @author alex.yao
* @date 2025/9/15
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(initializers = SingleContextInitializer.class)
@WebAppConfiguration
public class DifyTest {
@Test
public void test_chat() throws DifyApiException, IOException {
// 创建聊天客户端
DifyChatClient chatClient = DifyClientFactory.createChatClient("http://192.168.21.174:6060/v1", "app-pv9FlrZCMfYogqzmbSxwg16s");
// 创建聊天消息
ChatMessage message = ChatMessage.builder()
.query("你好,请介绍一下自己")
.user("user-123")
.inputs(new LinkedHashMap<>())
.responseMode(ResponseMode.BLOCKING)
.build();
// 发送消息并获取响应
ChatMessageResponse response = chatClient.sendChatMessage(message);
System.out.println("回复: " + response.getAnswer());
System.out.println("会话ID: " + response.getConversationId());
System.out.println("消息ID: " + response.getMessageId());
}
@Test
public void test_chatStream() throws DifyApiException, IOException, InterruptedException {
// 创建聊天客户端
DifyChatClient chatClient = DifyClientFactory.createChatClient("http://192.168.21.174:6060/v1", "app-pv9FlrZCMfYogqzmbSxwg16s");
// 创建聊天消息
ChatMessage message = ChatMessage.builder()
.query("你好,请介绍一下自己")
.user("user-123")
.inputs(new LinkedHashMap<>())
.responseMode(ResponseMode.STREAMING)
.build();
CountDownLatch countDownLatch = new CountDownLatch(1);
// 发送消息并获取响应
chatClient.sendChatMessageStream(message, new ChatStreamCallback() {
@Override
public void onMessage(MessageEvent event) {
ChatStreamCallback.super.onMessage(event);
System.out.println("answer:" + event.getAnswer());
}
@Override
public void onMessageEnd(MessageEndEvent event) {
ChatStreamCallback.super.onMessageEnd(event);
countDownLatch.countDown();
}
});
countDownLatch.await();
}
}
......@@ -3,6 +3,7 @@ package cn.com.poc.knowledge;
import cn.com.poc.common.utils.StringUtils;
import cn.com.poc.common.utils.UUIDTool;
import cn.com.poc.knowledge.aggregate.KnowledgeService;
import cn.com.poc.knowledge.cache.KnowledgeDefaultCache;
import cn.com.yict.framemax.core.i18n.I18nMessageException;
import cn.com.yict.framemax.core.spring.SingleContextInitializer;
import cn.hutool.poi.excel.ExcelReader;
......@@ -51,6 +52,12 @@ public class KnowledgeServiceTest {
MockitoAnnotations.initMocks(this);
}
@Test
public void test_default() {
KnowledgeDefaultCache knowledgeDefaultCache = new KnowledgeDefaultCache();
System.out.println(knowledgeDefaultCache.getDefaultKnowledgeId());
}
@Test
public void test_delDocument() {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment