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.util.Map;

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

    ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url, String model);

    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);

}