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

rag

Browse Source
This commit is contained in:
wangmeihua 2025-10-10 11:49:20 +08:00
parent 5bb3180cc4
commit f61bb22316
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
rag/uapi_service.py
View File

@ -30,6 +30,7 @@ class APIService:
request_id = str(uuid.uuid4()) request_id = str(uuid.uuid4())
debug(f"Request #{request_id} started for embeddings, texts={texts[:2]}") debug(f"Request #{request_id} started for embeddings, texts={texts[:2]}")
try: try:
debug(f"get_embeddings request: {request}")
uapi = UAPI(request, DictObject(**globals())) uapi = UAPI(request, DictObject(**globals()))
params_kw = {"input": texts} params_kw = {"input": texts}
b = await uapi.call(upappid, apiname, user, params_kw) b = await uapi.call(upappid, apiname, user, params_kw)
@ -101,8 +102,10 @@ class APIService:
top_n = len(results) top_n = len(results)
else: else:
top_n = min(top_n, len(results)) top_n = min(top_n, len(results))
debug(f"top_n={top_n}")
uapi = UAPI(request, DictObject(**globals())) uapi = UAPI(request, DictObject(**globals()))
documents = [result.get("text", str(result)) for result in results] documents = [result.get("text", str(result)) for result in results]
debug(f"文档块长度：{len(documents)},\ndocuments: {documents}")
params_kw = { params_kw = {
"model": "rerank-001", "model": "rerank-001",
"query": query, "query": query,