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