package com.ard.agent.config;

import org.springframework.ai.embedding.EmbeddingModel;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

/**
 * Embedding 模型配置类，解决 dashscopeEmbeddingModel 和 ollamaEmbeddingModel 的冲突
 */
@Configuration
public class EmbeddingModelConfig {

    /**
     * 默认使用 Ollama 的 Embedding 模型（用于 Milvus 向量存储）
     */
    @Primary
    @Bean
    public EmbeddingModel primaryOllamaEmbeddingModel(
            @Qualifier("ollamaEmbeddingModel") EmbeddingModel ollamaEmbeddingModel) {
        return ollamaEmbeddingModel;
    }

    /**
     * DashScope 的 Embedding 模型（需要时通过 @Qualifier("dashscopeEmbeddingModel") 注入）
     */
    //@Primary
    @Bean
    public EmbeddingModel primaryDashscopeEmbeddingModel(
            @Qualifier("dashscopeEmbeddingModel") EmbeddingModel dashscopeEmbeddingModel) {
        return dashscopeEmbeddingModel;
    }
}