瀏覽代碼

feat: 默认使用openai向量化模型

ageer 2 月之前
父節點
當前提交
a114838a9f

+ 2 - 10
ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java

@@ -39,19 +39,11 @@ public class OpenAiVectorization implements Vectorization {
 
     @Override
     public List<List<Double>> batchVectorization(List<String> chunkList, String kid) {
-        List<List<Double>> vectorList = new ArrayList<>();
-
+        List<List<Double>> vectorList;
+        openAiStreamClient = chatConfig.getOpenAiStreamClient();
         // 获取知识库信息
         KnowledgeInfoVo knowledgeInfoVo = knowledgeInfoService.queryById(Long.valueOf(kid));
 
-        // 如果使用本地模型
-        try {
-            return localModelsVectorization.batchVectorization(chunkList, kid);
-        } catch (Exception e) {
-            log.error("Local models vectorization failed, falling back to OpenAI embeddings", e);
-        }
-
-        // 如果本地模型失败,则调用 OpenAI 服务进行向量化
         Embedding embedding = buildEmbedding(chunkList, knowledgeInfoVo);
         EmbeddingResponse embeddings = openAiStreamClient.embeddings(embedding);