6 小时以前 9bad721754fe8bbe2e5f459d0706e0fefac569f3
yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/AiModelFactory.java
@@ -4,6 +4,7 @@
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.embedding.EmbeddingModel;
import org.springframework.ai.vectorstore.VectorStore;
import java.time.Duration;
import java.util.Map;
/**
@@ -11,7 +12,15 @@
 */
public interface AiModelFactory {
    ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url, String model);
    /**
     * 获取(或创建并缓存)Chat 模型
     *
     * @param temperature 采样温度,null 表示用默认值
     * @param maxTokens   单次回复最大 token 数,null 表示沿用模型默认值
     * @param timeout     单次调用超时
     */
    ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url, String model,
                                   Double temperature, Integer maxTokens, Duration timeout);
    ChatModel getDefaultChatModel(AiPlatformEnum platform);