uapi

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,