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