package cn.iocoder.yudao.module.ai.framework.ai.core.model;

import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
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;

/**
 * AI Model 模型工厂的接口类
 */
public interface AiModelFactory {

    /**
     * 获取（或创建并缓存）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);

    EmbeddingModel getOrCreateEmbeddingModel(AiPlatformEnum platform, String apiKey, String url, String model);

    VectorStore getOrCreateVectorStore(Class<? extends VectorStore> type,
                                       EmbeddingModel embeddingModel,
                                       Map<String, Class<?>> metadataFields);

}