From 3285f3bca7e3db93db43ba15cec414c0c3925e10 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=B7=E9=9B=A8?= Date: Mon, 29 Sep 2025 19:29:26 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=9Aembeding=E6=94=B9=E6=88=90?= =?UTF-8?q?=E6=94=AF=E6=8C=81=E7=94=B5=E4=BF=A1=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .env | 4 ++-- service/cus_vanna_srevice.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.env b/.env index 99b4a00..3739ba7 100644 --- a/.env +++ b/.env @@ -3,8 +3,8 @@ IS_FIRST_LOAD=True CHAT_MODEL_BASE_URL=https://api.siliconflow.cn CHAT_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole CHAT_MODEL_NAME=zai-org/GLM-4.5 - -EMBEDDING_MODEL_BASE_URL=https://api.siliconflow.cn +#使用ai中台的模型 +EMBEDDING_MODEL_BASE_URL=http://10.225.128.2:13206/member1/small-model/bge/encode EMBEDDING_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole EMBEDDING_MODEL_NAME=BAAI/bge-m3 diff --git a/service/cus_vanna_srevice.py b/service/cus_vanna_srevice.py index f4dcfe0..f4da618 100644 --- a/service/cus_vanna_srevice.py +++ b/service/cus_vanna_srevice.py @@ -270,21 +270,21 @@ class CustomQdrant_VectorStore(Qdrant_VectorStore): request_body = { "model": self.embedding_model_name, - "input": data, + "sentences": [data], } request_body.update(kwargs) response = requests.post( - url=f"{self.embedding_api_base}/v1/embeddings", + url=f"{self.embedding_api_base}", json=request_body, - headers={"Authorization": f"Bearer {self.embedding_api_key}"}, + headers={"Authorization": f"Bearer {self.embedding_api_key}", 'Content-Type': 'application/json'}, ) if response.status_code != 200: raise RuntimeError( f"Failed to create the embeddings, detail: {_get_error_string(response)}" ) result = response.json() - embeddings = [d["embedding"] for d in result["data"]] + embeddings = result['embeddings'] return embeddings[0] class CustomVanna(CustomQdrant_VectorStore, OpenAICompatibleLLM):