public/liteFlow
1
0
Fork 0
mirror of https://gitee.com/dromara/liteFlow.git synced 2026-06-10 11:17:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test: 测试结构化输出

Browse Source
This commit is contained in:
LuanY77
2025-07-31 13:02:02 +08:00
parent 5cb64025bd
commit 4326ab2da1
4 changed files with 44 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
liteflow-ai/liteflow-ai-core/src/main/java/com/yomahub/liteflow/ai/proxy/invocation/ChatAIInvocationHandler.java
View File

@@ -64,7 +64,9 @@ public class ChatAIInvocationHandler extends AbstractAIInvocationHandler<ChatPro
Object aiService = AiServiceFactory.createAiService(wrapBean.getEntityClass(), chatModel);
try {
return AiServiceFactory.chat(aiService, wrapBean.getUserPrompt(), wrapBean.getSystemPrompt());
Object result = AiServiceFactory.chat(aiService, wrapBean.getUserPrompt(), wrapBean.getSystemPrompt());
LOG.info("Chat response: {}", result);
return result;
} catch (Throwable e) {
throw new LiteFlowAIException("Error during blocking chat processing", e);
}

14
liteflow-ai/liteflow-ai-ollama/src/main/java/com/yomahub/liteflow/ai/ollama/model/OllamaModelProvider.java
View File

@@ -2,13 +2,17 @@ package com.yomahub.liteflow.ai.ollama.model;
import com.yomahub.liteflow.ai.domain.ModelConfig;
import com.yomahub.liteflow.ai.domain.constant.ProviderName;
import com.yomahub.liteflow.ai.domain.enums.ResponseType;
import com.yomahub.liteflow.ai.model.ModelProviderRegistrar;
import com.yomahub.liteflow.ai.proxy.wrap.AIProxyWrapBean;
import dev.langchain4j.model.chat.Capability;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import dev.langchain4j.model.chat.request.ResponseFormat;
import dev.langchain4j.model.ollama.OllamaChatModel;
import dev.langchain4j.model.ollama.OllamaStreamingChatModel;
import java.util.Objects;
import java.util.Optional;
/**
@@ -33,8 +37,14 @@ public class OllamaModelProvider extends ModelProviderRegistrar {
.builder()
.baseUrl(modelConfig.getBaseUrl())
.modelName(modelConfig.getModel())
.build()
);
).map(ollamaChatModelBuilder -> {
// 开启结构化输出
if (Objects.equals(ResponseType.JSON, wrapBean.getResponseType())) {
ollamaChatModelBuilder.responseFormat(ResponseFormat.JSON)
.supportedCapabilities(Capability.RESPONSE_FORMAT_JSON_SCHEMA);
}
return ollamaChatModelBuilder.build();
});
}
@Override

2
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/AICmp.java
View File

@@ -29,7 +29,7 @@ import com.yomahub.liteflow.ai.domain.enums.ResponseType;
)
@AIOutput(
responseType = ResponseType.JSON,
entityClass = Integer.class,
entityClass = Output.class,
methodExpress = "setData",
useKeyIndex = true,
key = "result"

28
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/cmp/Output.java Normal file
View File

@@ -0,0 +1,28 @@
package com.yomahub.liteflow.test.cmp;
/**
* 测试结构化输出使用
*
* @author 苍镜月
* @since TODO
*/
public class Output {
private String content;
public Output() {
}
public Output(String content) {
this.content = content;
}
public String getContent() {
return content;
}
public void setContent(String content) {
this.content = content;
}
}