mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
Merge branch 'v2' of https://github.com/maxkb-dev/maxkb into v2
This commit is contained in:
commit
9a6be423ac
|
|
@ -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())
|
||||
|
||||
|
|
|
|||
|
|
@ -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"]
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue