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

fix: search_result.dspy 用 StreamHttpClient + try/except 替代裸 aiohttp

Browse Source
This commit is contained in:
yumoqing 2026-07-31 14:59:18 +08:00
parent 0d6acfb4ef
commit 0cb6f34135
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
wwwroot/knowledge_bases_list/search_result.dspy
View File

@ -29,25 +29,32 @@ if not query:
] ]
}, ensure_ascii=False) }, ensure_ascii=False)
import aiohttp # 1. Embed
vec = []
async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=20)) as s: try:
# 1. Embed client = StreamHttpClient()
r = await s.post('https://embedding.opencomputing.net:10443/api/embed', resp = await client.request('POST', 'https://embedding.opencomputing.net:10443/api/embed',
json={"texts": [query], "model": "CLIP-ViT-H-14"}) json={"texts": [query], "model": "CLIP-ViT-H-14"})
emb = await r.json() if r.status == 200 else {} emb = json.loads(resp)
vec = emb.get("text_embeddings", emb.get("embeddings", [[]]))[0] vec = emb.get("text_embeddings", emb.get("embeddings", [[]]))[0]
if not vec: except:
return json.dumps({"widgettype": "Text", "options": {"text": "向量化失败", "cfontsize": 14, "color": "#e74c3c"}}, ensure_ascii=False) pass
if not vec:
return json.dumps({"widgettype": "Text", "options": {"text": "向量化失败", "cfontsize": 14, "color": "#e74c3c"}}, ensure_ascii=False)
# 2. VDB search # 2. VDB search
raw_rows = []
try:
recall_n = top_k * 3 recall_n = top_k * 3
r2 = await s.post('https://vectordb.opencomputing.net:10443/v1/query', client2 = StreamHttpClient()
resp2 = await client2.request('POST', 'https://vectordb.opencomputing.net:10443/v1/query',
json={"colname": kb_id, "vector": vec, "pagerows": recall_n, "output_fields": ["*"]}) json={"colname": kb_id, "vector": vec, "pagerows": recall_n, "output_fields": ["*"]})
vdb = await r2.json() if r2.status == 200 else {} vdb = json.loads(resp2)
raw_rows = vdb.get("data", {}).get("rows", []) raw_rows = vdb.get("data", {}).get("rows", [])
if not isinstance(raw_rows, list): if not isinstance(raw_rows, list):
raw_rows = [] raw_rows = []
except:
pass
hits = [] hits = []
for row in raw_rows: for row in raw_rows: