Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in
Toggle navigation
P
poc-api
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
1
Merge Requests
1
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
poc
poc-api
Commits
5e8b27f3
Commit
5e8b27f3
authored
Sep 17, 2025
by
alex yao
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: 转换数组
parent
3a0a412d
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
94 additions
and
1 deletion
+94
-1
KnowledgeDefaultCache.java
...ava/cn/com/poc/knowledge/cache/KnowledgeDefaultCache.java
+5
-1
DifyTest.java
src/test/java/cn/com/poc/dify/DifyTest.java
+82
-0
KnowledgeServiceTest.java
src/test/java/cn/com/poc/knowledge/KnowledgeServiceTest.java
+7
-0
No files found.
src/main/java/cn/com/poc/knowledge/cache/KnowledgeDefaultCache.java
View file @
5e8b27f3
...
...
@@ -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
;
...
...
src/test/java/cn/com/poc/dify/DifyTest.java
0 → 100644
View file @
5e8b27f3
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
();
}
}
src/test/java/cn/com/poc/knowledge/KnowledgeServiceTest.java
View file @
5e8b27f3
...
...
@@ -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
()
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment