From a114838a9f84a4973c224e90c1324a15ba3cd076 Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期日, 23 三月 2025 23:06:38 +0800
Subject: [PATCH] feat: 默认使用openai向量化模型
---
ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java | 12 ++----------
1 files 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 764c2c1..f15c492 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 @@
@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);
--
Gitblit v1.9.3