xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-17 13:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
ageer
2025-03-23 23:06:38 +08:00
parent 9062d47b99
commit a114838a9f
1 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ruoyi-modules/ruoyi-knowledge/src/main/java/org/ruoyi/knowledge/chain/vectorizer/OpenAiVectorization.java
View File

@@ -39,19 +39,11 @@ public class OpenAiVectorization implements Vectorization {
@Override @Override
public List<List<Double>> batchVectorization(List<String> chunkList, String kid) { 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)); 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); Embedding embedding = buildEmbedding(chunkList, knowledgeInfoVo);
EmbeddingResponse embeddings = openAiStreamClient.embeddings(embedding); EmbeddingResponse embeddings = openAiStreamClient.embeddings(embedding);