diff --git a/rag/folderinfo.py b/rag/folderinfo.py index 809f354..04f7e77 100644 --- a/rag/folderinfo.py +++ b/rag/folderinfo.py @@ -1,5 +1,3 @@ -from urllib.request import Request - from appPublic.timeUtils import curDateString from filemgr.filemgr import FileMgr from rag.uapi_service import APIService @@ -191,7 +189,7 @@ where a.orgid = b.orgid for i in range(0, len(texts), 10): # 每次处理 10 个文本块 batch_texts = texts[i:i + 10] batch_embeddings = await api_service.get_embeddings( - request=Request, + request=request, texts=batch_texts, upappid=service_params['embedding'], apiname="BAAI/bge-m3", @@ -246,7 +244,7 @@ where a.orgid = b.orgid debug(f"处理 {len(chunk_texts)} 个分片进行三元组抽取") tasks = [ api_service.extract_triples( - request=Request, + request=request, text=chunk, upappid=service_params['triples'], apiname="Babelscape/mrebel-large", # 固定 apiname @@ -289,7 +287,7 @@ where a.orgid = b.orgid for i in range(0, len(unique_triples), 30): # 每次插入 30 个三元组 batch_triples = unique_triples[i:i + 30] neo4j_result = await api_service.neo4j_insert_triples( - request=Request, + request=request, triples=batch_triples, document_id=id, knowledge_base_id=fiid,