From 1385b165c9a62a486b4494ebb11320d3c1117a5e Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期日, 02 三月 2025 11:19:29 +0800
Subject: [PATCH] feat(知识库): 增加知识库模块
---
ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/openai/OpenAiStreamClient.java | 13 +++++++++++++
1 files changed, 13 insertions(+), 0 deletions(-)
diff --git a/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/openai/OpenAiStreamClient.java b/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/openai/OpenAiStreamClient.java
index e293aa4..cad1bbb 100644
--- a/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/openai/OpenAiStreamClient.java
+++ b/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/openai/OpenAiStreamClient.java
@@ -21,6 +21,8 @@
import org.ruoyi.common.chat.entity.chat.ChatCompletion;
import org.ruoyi.common.chat.entity.chat.ChatCompletionResponse;
import org.ruoyi.common.chat.entity.chat.ChatCompletionWithPicture;
+import org.ruoyi.common.chat.entity.embeddings.Embedding;
+import org.ruoyi.common.chat.entity.embeddings.EmbeddingResponse;
import org.ruoyi.common.chat.entity.files.UploadFileResponse;
import org.ruoyi.common.chat.entity.images.Image;
import org.ruoyi.common.chat.entity.images.ImageResponse;
@@ -293,6 +295,17 @@
}
/**
+ * 鏂囨湰杞崲鍚戦噺
+ *
+ * @param embedding 鍏ュ弬
+ * @return EmbeddingResponse
+ */
+ public EmbeddingResponse embeddings(Embedding embedding) {
+ Single<EmbeddingResponse> embeddings = this.openAiApi.embeddings(embedding);
+ return embeddings.blockingGet();
+ }
+
+ /**
* 璐︽埛淇℃伅鏌ヨ锛氶噷闈㈠寘鍚�婚噾棰濈瓑淇℃伅
*
* @return 璐︽埛淇℃伅
--
Gitblit v1.9.3