xservice/xservice-ai-center
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat:智能体创建

Browse Source
This commit is contained in:
Ttt
2026-03-17 22:49:00 +08:00
parent 24d01bc3e1
commit 5267d29d97
8 changed files with 61 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
xservice-service/src/main/java/com/xiang/xservice/ai/agent/SimpleChatAgent.java
View File

@@ -49,7 +49,7 @@ public class SimpleChatAgent implements BaseAgent {
.modelName(openAIConfig.getModelName())
.temperature(openAIConfig.getTemperature())
.build();
StreamingChatModel chat = router.route(ModelStrategyEnum.CHAT, modelType.getModelType(), modelConfig);
StreamingChatModel chat = router.routeStream(ModelStrategyEnum.CHAT, modelType.getModelType(), modelConfig);
if (chat == null) {
throw new RuntimeException("chat model route failed");
}

10
xservice-service/src/main/java/com/xiang/xservice/ai/core/provider/BaseProvider.java
View File

@@ -2,7 +2,6 @@ package com.xiang.xservice.ai.core.provider;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.enums.ModelTypeEnum;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
@@ -14,11 +13,18 @@ public interface BaseProvider {
*/
String providerName();
/**
* 创建model
* 创建流式model
* @param config model配置文件
* @return ChatModel
*/
StreamingChatModel build(ModelConfig config);
/**
* 同步聊天模型
* @param config
* @return
*/
ChatModel buildChatModel(ModelConfig config);
/**
* 用于标记这个 Provider 是否适合某个 TaskType
* @param taskType ModelStrategyEnum

9
xservice-service/src/main/java/com/xiang/xservice/ai/core/provider/OllamaLlmProvider.java
View File

@@ -26,6 +26,15 @@ public class OllamaLlmProvider implements BaseProvider{
.build();
}
@Override
public ChatModel buildChatModel(ModelConfig config) {
return OllamaChatModel.builder()
.baseUrl(config.getBaseUrl())
.modelName(config.getModelName())
.temperature(config.getTemperature())
.build();
}
@Override
public boolean support(ModelStrategyEnum taskType) {
return true;

19
xservice-service/src/main/java/com/xiang/xservice/ai/core/provider/OpenAILlmProvider.java
View File

@@ -3,13 +3,14 @@ package com.xiang.xservice.ai.core.provider;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.enums.ModelTypeEnum;
import com.xiang.xservice.ai.core.storage.MemoryPersistentStore;
import dev.langchain4j.memory.chat.ChatMemoryProvider;
import dev.langchain4j.memory.chat.MessageWindowChatMemory;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import dev.langchain4j.model.openai.OpenAiChatModel;
import dev.langchain4j.model.openai.OpenAiStreamingChatModel;
import org.springframework.stereotype.Component;
import java.time.Duration;
@Component
public class OpenAILlmProvider implements BaseProvider {
@@ -33,6 +34,18 @@ public class OpenAILlmProvider implements BaseProvider {
.build();
}
@Override
public ChatModel buildChatModel(ModelConfig config) {
return OpenAiChatModel.builder()
.baseUrl(config.getBaseUrl())
.apiKey(config.getApiKey())
.modelName(config.getModelName())
.temperature(config.getTemperature())
.maxTokens(config.getMaxTokens())
.timeout(Duration.ofSeconds(300))
.build();
}
@Override
public boolean support(ModelStrategyEnum taskType) {
return true;

12
xservice-service/src/main/java/com/xiang/xservice/ai/core/route/ModelRouter.java
View File

@@ -3,10 +3,8 @@ package com.xiang.xservice.ai.core.route;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.provider.BaseProvider;
import com.xiang.xservice.ai.core.strategy.BaseStrategy;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import java.util.List;
@@ -29,7 +27,7 @@ public class ModelRouter {
/**
* 根据 providerName 选择 Provider，并用 config 构建模型
*/
public StreamingChatModel route(String providerName, ModelConfig config) {
public StreamingChatModel routeStream(String providerName, ModelConfig config) {
BaseProvider provider = providerMap.get(providerName);
if (provider == null) {
throw new RuntimeException("Provider " + providerName + " not found");
@@ -37,6 +35,14 @@ public class ModelRouter {
return provider.build(config);
}
public ChatModel route(String providerName, ModelConfig config) {
BaseProvider provider = providerMap.get(providerName);
if (provider == null) {
throw new RuntimeException("Provider " + providerName + " not found");
}
return provider.buildChatModel(config);
}
/**
* 获取某个 TaskType 支持的所有 Provider 名称
*/

13
xservice-service/src/main/java/com/xiang/xservice/ai/core/route/TaskRouter.java
View File

@@ -3,6 +3,7 @@ package com.xiang.xservice.ai.core.route;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.provider.BaseProvider;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import org.springframework.stereotype.Component;
@@ -29,7 +30,17 @@ public class TaskRouter {
/**
* 第一层路由 + 第二层 ProviderRouter
*/
public StreamingChatModel route(ModelStrategyEnum taskType, String providerName, ModelConfig config) {
public StreamingChatModel routeStream(ModelStrategyEnum taskType, String providerName, ModelConfig config) {
// 如果没有传 providerName，使用默认 Provider
if (providerName == null || providerName.isEmpty()) {
providerName = taskDefaultProviderMap.get(taskType).get(0);
}
// 第二层路由
return providerRouter.routeStream(providerName, config);
}
public ChatModel route(ModelStrategyEnum taskType, String providerName, ModelConfig config) {
// 如果没有传 providerName，使用默认 Provider
if (providerName == null || providerName.isEmpty()) {
providerName = taskDefaultProviderMap.get(taskType).get(0);

4
xservice-service/src/main/java/com/xiang/xservice/ai/core/strategy/ChatStrategy.java
View File

@@ -2,9 +2,7 @@ package com.xiang.xservice.ai.core.strategy;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.provider.BaseProvider;
import com.xiang.xservice.ai.core.route.TaskRouter;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
@@ -22,6 +20,6 @@ public class ChatStrategy implements BaseStrategy {
@Override
public StreamingChatModel createProvider(String provider, ModelConfig config) {
return taskRouter.route(ModelStrategyEnum.CHAT, provider, config);
return taskRouter.routeStream(ModelStrategyEnum.CHAT, provider, config);
}
}