From a114838a9f84a4973c224e90c1324a15ba3cd076 Mon Sep 17 00:00:00 2001 From: ageer Date: Sun, 23 Mar 2025 23:06:38 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E9=BB=98=E8=AE=A4=E4=BD=BF=E7=94=A8ope?= =?UTF-8?q?nai=E5=90=91=E9=87=8F=E5=8C=96=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../chain/vectorizer/OpenAiVectorization.java | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) 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);