diff --git a/rag/api_service.py b/rag/api_service.py index e1f8cf5..a50593f 100644 --- a/rag/api_service.py +++ b/rag/api_service.py @@ -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", {}) \ No newline at end of file + return await self._make_request("http://222.81.167.136:8888/v1/docs", "docs", {}) \ No newline at end of file