test/agents-flex
1
0
Fork 0
mirror of https://gitee.com/agents-flex/agents-flex.git synced 2024-11-29 18:38:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: refactor BaseLlmClientListener.java

Browse Source
This commit is contained in:
Michael Yang 2024-10-30 11:41:18 +08:00
parent 50d6d7bea8
commit 074d1f03ec
1 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
agents-flex-core/src/main/java/com/agentsflex/core/llm/client/BaseLlmClientListener.java
View File

@ -79,22 +79,24 @@ public class BaseLlmClientListener implements LlmClientListener {
return;
}
JSONObject jsonObject = JSON.parseObject(response);
if (isFunctionCalling) {
FunctionMessage functionMessage = functionMessageParser.parse(jsonObject);
List<Function> functions = ((FunctionPrompt) prompt).getFunctions();
FunctionMessageResponse functionMessageResponse = new FunctionMessageResponse(response, functions, functionMessage);
//noinspection unchecked
streamResponseListener.onMessage(context, functionMessageResponse);
} else {
lastAiMessage = messageParser.parse(jsonObject);
fullMessage.append(lastAiMessage.getContent());
lastAiMessage.setFullContent(fullMessage.toString());
AiMessageResponse aiMessageResponse = new AiMessageResponse(response, lastAiMessage);
//noinspection unchecked
streamResponseListener.onMessage(context, aiMessageResponse);
try {
JSONObject jsonObject = JSON.parseObject(response);
if (isFunctionCalling) {
FunctionMessage functionMessage = functionMessageParser.parse(jsonObject);
List<Function> functions = ((FunctionPrompt) prompt).getFunctions();
FunctionMessageResponse functionMessageResponse = new FunctionMessageResponse(response, functions, functionMessage);
//noinspection unchecked
streamResponseListener.onMessage(context, functionMessageResponse);
} else {
lastAiMessage = messageParser.parse(jsonObject);
fullMessage.append(lastAiMessage.getContent());
lastAiMessage.setFullContent(fullMessage.toString());
AiMessageResponse aiMessageResponse = new AiMessageResponse(response, lastAiMessage);
//noinspection unchecked
streamResponseListener.onMessage(context, aiMessageResponse);
}
} catch (Exception err) {
streamResponseListener.onFailure(context, err);
}
}