diff --git a/ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java b/ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java index 764c2c16..f15c4927 100644 --- a/ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java +++ b/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> batchVectorization(List chunkList, String kid) { - List> vectorList = new ArrayList<>(); - + List> 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);