This commit is contained in:
liqiang-fit2cloud 2025-07-21 14:10:27 +08:00
commit 9a6be423ac
5 changed files with 11 additions and 8 deletions

View File

@ -213,10 +213,10 @@ class KnowledgeView(APIView):
'workspace_id': workspace_id,
'knowledge_id': knowledge_id,
'user_id': request.user.id,
"query_text": request.query_params.get("query_text"),
"top_number": request.query_params.get("top_number"),
'similarity': request.query_params.get('similarity'),
'search_mode': request.query_params.get('search_mode')
"query_text": request.data.get("query_text"),
"top_number": request.data.get("top_number"),
'similarity': request.data.get('similarity'),
'search_mode': request.data.get('search_mode')
}
).hit_test())

View File

@ -11,6 +11,7 @@ from typing import Dict
import requests
from langchain_community.embeddings import OpenAIEmbeddings
from common.utils.logger import maxkb_logger
from models_provider.base_model_provider import MaxKBBaseModel
@ -35,7 +36,9 @@ class SiliconCloudEmbeddingModel(MaxKBBaseModel, OpenAIEmbeddings):
response = requests.post(self.openai_api_base + '/embeddings', json=payload, headers=headers)
data = response.json()
# print(data)
if data['data'] is None or 'code' in data:
raise ValueError(f"Embedding API returned no data: {data}")
# 假设返回结构中有 'data[0].embedding'
return data["data"][0]["embedding"]

View File

@ -163,7 +163,7 @@ const putKnowledgeHitTest: (
data: any,
loading?: Ref<boolean>,
) => Promise<Result<Array<any>>> = (knowledge_id, data, loading) => {
return put(`${prefix.value}/${knowledge_id}/hit_test`, undefined, data, loading)
return put(`${prefix.value}/${knowledge_id}/hit_test`, data, undefined, loading)
}
/**

View File

@ -154,7 +154,7 @@ const putKnowledgeHitTest: (
data: any,
loading?: Ref<boolean>,
) => Promise<Result<Array<any>>> = (knowledge_id, data, loading) => {
return put(`${prefix}/${knowledge_id}/hit_test`, undefined, data, loading)
return put(`${prefix}/${knowledge_id}/hit_test`, data, undefined, loading)
}

View File

@ -155,7 +155,7 @@ const putKnowledgeHitTest: (
data: any,
loading?: Ref<boolean>,
) => Promise<Result<Array<any>>> = (knowledge_id, data, loading) => {
return put(`${prefix}/${knowledge_id}/hit_test`, undefined, data, loading)
return put(`${prefix}/${knowledge_id}/hit_test`, data, undefined, loading)
}