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

修改api

Browse Source
This commit is contained in:
wangmeihua 2025-08-06 11:42:53 +08:00
parent 64e42705bf
commit 296e5a739a
1 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
rag/api_service.py
View File

@ -48,7 +48,7 @@ class APIService:
try:
async with ClientSession() as session:
async with session.post(
"http://localhost:9998/v1/embeddings",
"http://222.81.167.136:9998/v1/embeddings",
headers={"Content-Type": "application/json"},
json={"input": texts if isinstance(texts, list) else [texts]}
) as response:
@ -74,7 +74,7 @@ class APIService:
raise ValueError("查询文本不能为空")
async with ClientSession() as session:
async with session.post(
"http://localhost:9990/v1/entities",
"http://222.81.167.136:9990/v1/entities",
headers={"Content-Type": "application/json"},
json={"query": query}
) as response:
@ -104,7 +104,7 @@ class APIService:
timeout=ClientTimeout(total=None)
) as session:
async with session.post(
"http://localhost:9991/v1/triples",
"http://222.81.167.136:9991/v1/triples",
headers={"Content-Type": "application/json; charset=utf-8"},
json={"text": text}
) as response:
@ -140,7 +140,7 @@ class APIService:
documents = [result.get("text", str(result)) for result in results]
async with ClientSession() as session:
async with session.post(
"http://localhost:9997/v1/rerank",
"http://222.81.167.136:9997/v1/rerank",
headers={"Content-Type": "application/json"},
json={
"model": "rerank-001",
@ -172,11 +172,11 @@ class APIService:
# Neo4j 服务
async def neo4j_docs(self) -> Dict[str, Any]:
"""获取 Neo4j 文档"""
return await self._make_request("http://localhost:8885/docs", "docs", {})
return await self._make_request("http://222.81.167.136:8885/docs", "docs", {})
async def neo4j_initialize(self) -> Dict[str, Any]:
"""初始化 Neo4j 服务"""
return await self._make_request("http://localhost:8885/v1/initialize", "initialize", {})
return await self._make_request("http://222.81.167.136:8885/v1/initialize", "initialize", {})
async def neo4j_insert_triples(self, triples: list, document_id: str, knowledge_base_id: str, userid: str) -> Dict[str, Any]:
"""插入三元组到 Neo4j"""
@ -186,15 +186,15 @@ class APIService:
"knowledge_base_id": knowledge_base_id,
"userid": userid
}
return await self._make_request("http://localhost:8885/v1/inserttriples", "inserttriples", params)
return await self._make_request("http://222.81.167.136:8885/v1/inserttriples", "inserttriples", params)
async def neo4j_delete_document(self, document_id: str) -> Dict[str, Any]:
"""删除指定文档"""
return await self._make_request("http://localhost:8885/v1/deletedocument", "deletedocument", {"document_id": document_id})
return await self._make_request("http://222.81.167.136:8885/v1/deletedocument", "deletedocument", {"document_id": document_id})
async def neo4j_delete_knowledgebase(self, userid: str, knowledge_base_id: str) -> Dict[str, Any]:
"""删除用户知识库"""
return await self._make_request("http://localhost:8885/v1/deleteknowledgebase", "deleteknowledgebase",
return await self._make_request("http://222.81.167.136:8885/v1/deleteknowledgebase", "deleteknowledgebase",
{"userid": userid, "knowledge_base_id": knowledge_base_id})
async def neo4j_match_triplets(self, query: str, query_entities: list, userid: str, knowledge_base_id: str) -> Dict[str, Any]:
@ -205,18 +205,18 @@ class APIService:
"userid": userid,
"knowledge_base_id": knowledge_base_id
}
return await self._make_request("http://localhost:8885/v1/matchtriplets", "matchtriplets", params)
return await self._make_request("http://222.81.167.136:8885/v1/matchtriplets", "matchtriplets", params)
# RAG 服务
async def rag_create_collection(self, db_type: str = "") -> Dict[str, Any]:
"""创建集合"""
params = {"db_type": db_type} if db_type else {}
return await self._make_request("http://localhost:8888/v1/createcollection", "createcollection", params)
return await self._make_request("http://222.81.167.136:8888/v1/createcollection", "createcollection", params)
async def rag_delete_collection(self, db_type: str = "") -> Dict[str, Any]:
"""删除集合"""
params = {"db_type": db_type} if db_type else {}
return await self._make_request("http://localhost:8888/v1/deletecollection", "deletecollection", params)
return await self._make_request("http://222.81.167.136:8888/v1/deletecollection", "deletecollection", params)
async def rag_insert_file(self, file_path: str, userid: str, knowledge_base_id: str, document_id: str) -> Dict[str, Any]:
"""添加记录"""
@ -226,7 +226,7 @@ class APIService:
"knowledge_base_id": knowledge_base_id,
"document_id": document_id
}
return await self._make_request("http://localhost:8888/v1/insertfile", "insertfile", params)
return await self._make_request("http://222.81.167.136:8888/v1/insertfile", "insertfile", params)
async def rag_delete_file(self, userid: str, file_path: str, knowledge_base_id: str, document_id: str) -> Dict[str, Any]:
"""删除记录"""
@ -236,11 +236,11 @@ class APIService:
"knowledge_base_id": knowledge_base_id,
"document_id": document_id
}
return await self._make_request("http://localhost:8888/v1/deletefile", "deletefile", params)
return await self._make_request("http://222.81.167.136:8888/v1/deletefile", "deletefile", params)
async def rag_delete_knowledgebase(self, userid: str, knowledge_base_id: str) -> Dict[str, Any]:
"""删除知识库"""
return await self._make_request("http://localhost:8888/v1/deleteknowledgebase", "deleteknowledgebase",
return await self._make_request("http://222.81.167.136:8888/v1/deleteknowledgebase", "deleteknowledgebase",
{"userid": userid, "knowledge_base_id": knowledge_base_id})
async def rag_search_query(self, query: str, userid: str, knowledge_base_ids: list, limit: int, offset: int,
@ -254,7 +254,7 @@ class APIService:
"offset": offset,
"use_rerank": use_rerank
}
return await self._make_request("http://localhost:8888/v1/searchquery", "searchquery", params)
return await self._make_request("http://222.81.167.136:8888/v1/searchquery", "searchquery", params)
async def rag_fused_search_query(self, query: str, userid: str, knowledge_base_ids: list, limit: int, offset: int,
use_rerank: bool) -> Dict[str, Any]:
@ -267,16 +267,16 @@ class APIService:
"offset": offset,
"use_rerank": use_rerank
}
return await self._make_request("http://localhost:8888/v1/fusedsearchquery", "fusedsearchquery", params)
return await self._make_request("http://222.81.167.136:8888/v1/fusedsearchquery", "fusedsearchquery", params)
async def rag_list_user_files(self, userid: str) -> Dict[str, Any]:
"""列出用户知识库列表"""
return await self._make_request("http://localhost:8888/v1/listuserfiles", "listuserfiles", {"userid": userid})
return await self._make_request("http://222.81.167.136:8888/v1/listuserfiles", "listuserfiles", {"userid": userid})
async def rag_list_all_knowledgebases(self) -> Dict[str, Any]:
"""列出数据库中所有数据"""
return await self._make_request("http://localhost:8888/v1/listallknowledgebases", "listallknowledgebases", {})
return await self._make_request("http://222.81.167.136:8888/v1/listallknowledgebases", "listallknowledgebases", {})
async def rag_docs(self) -> Dict[str, Any]:
"""获取 RAG 帮助文档"""
return await self._make_request("http://localhost:8888/v1/docs", "docs", {})
return await self._make_request("http://222.81.167.136:8888/v1/docs", "docs", {})