yumoqing/rag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

uapi

Browse Source
This commit is contained in:
wangmeihua 2025-08-12 18:30:46 +08:00
parent de76d6a261
commit 149a63a574
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rag/uapi_service.py
View File

@ -55,13 +55,13 @@ class APIService:
params_kw = {"text": text} params_kw = {"text": text}
d = await uapi.request(upappid, apiname, user, params_kw) d = await uapi.request(upappid, apiname, user, params_kw)
if d.get("object") != "list": if d.get("object") != "list":
error(f"Request #{request_id} invalid response format: {d}") error(f"request #{request_id} invalid response format: {d}")
raise RuntimeError("三元组抽取服务响应格式错误") raise RuntimeError("三元组抽取服务响应格式错误")
triples = d["data"] triples = d["data"]
debug(f"Request #{request_id} extracted {len(triples)} triples") debug(f"request #{request_id} extracted {len(triples)} triples")
return triples return triples
except Exception as e: except Exception as e:
error(f"Request #{request_id} failed to extract triples: {str(e)}") error(f"request #{request_id} failed to extract triples: {str(e)}")
raise RuntimeError(f"三元组抽取服务调用失败: {str(e)}") raise RuntimeError(f"三元组抽取服务调用失败: {str(e)}")
# 重排序服务 (BAAI/bge-reranker-v2-m3) # 重排序服务 (BAAI/bge-reranker-v2-m3)
@ -183,7 +183,7 @@ class APIService:
payload = json.dumps(params_kw) # 转换为 JSON 字符串 payload = json.dumps(params_kw) # 转换为 JSON 字符串
payload_bytes = payload.encode() # 编码为字节 payload_bytes = payload.encode() # 编码为字节
payload_size = len(payload_bytes) # 获取字节数 payload_size = len(payload_bytes) # 获取字节数
debug(f"Request payload size for insertdocument: {payload_size} bytes") debug(f"request payload size for insertdocument: {payload_size} bytes")
return await uapi.request(upappid, apiname, user, params_kw) return await uapi.request(upappid, apiname, user, params_kw)
async def milvus_delete_document(self, request, userid: str, file_path: str, knowledge_base_id: str, document_id: str, upappid: str, apiname: str, user: str, db_type: str = "") -> Dict[str, Any]: async def milvus_delete_document(self, request, userid: str, file_path: str, knowledge_base_id: str, document_id: str, upappid: str, apiname: str, user: str, db_type: str = "") -> Dict[str, Any]: