package com.xly.milvus.config;

import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.model.embedding.onnx.allminilml6v2.AllMiniLmL6V2EmbeddingModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * LangChain4j 嵌入模型配置类
 */
@Configuration
public class EmbeddingConfig {

    @Bean
    public EmbeddingModel embeddingModel() {
        // 使用 All-MiniLM-L6-v2 嵌入模型
        // 这是一个轻量级的模型，维度为384，适合在本地运行
        return new AllMiniLmL6V2EmbeddingModel();
    }
}