修改搜索节点
This commit is contained in:
@@ -6,7 +6,35 @@ def check_status(status_code: float, body: str):
|
||||
raise Exception(f"注册任务失败,状态码:{status_code}")
|
||||
if "code" not in body or body["code"] != 1:
|
||||
raise Exception(f"注册任务失败,返回值:{body}")
|
||||
|
||||
def format_rag_context(data: list) -> str:
|
||||
'''
|
||||
将检索到的 data 列表格式化为 Markdown 文本
|
||||
'''
|
||||
if not data:
|
||||
return "未找到相关的参考资料。"
|
||||
|
||||
formatted_parts = []
|
||||
|
||||
for i, item in enumerate(data):
|
||||
# 提取字段(对应你数据库和 API 返回的字段)
|
||||
title = item.get("title") or "无标题"
|
||||
url = item.get("source_url") or "未知来源"
|
||||
content = item.get("content", "").strip()
|
||||
c_idx = item.get("chunk_index", 0)
|
||||
|
||||
# 构造 Markdown 块
|
||||
block = (
|
||||
f"### [资料 {i+1}] {title}\n"
|
||||
f"**来源**: {url}\n"
|
||||
f"**切片索引**: {c_idx}\n"
|
||||
f"**内容**: {content}"
|
||||
)
|
||||
formatted_parts.append(block)
|
||||
|
||||
# 使用分隔符连接多个资料块
|
||||
return "\n\n---\n\n".join(formatted_parts)
|
||||
|
||||
def main(status_code: float, body: str):
|
||||
try:
|
||||
check_status(status_code, body)
|
||||
@@ -14,7 +42,8 @@ def main(status_code: float, body: str):
|
||||
raise e
|
||||
|
||||
data = body["data"]
|
||||
|
||||
rag_context = format_rag_context(data)
|
||||
|
||||
return {
|
||||
"RAG_results": data
|
||||
"RAG_results": rag_context
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user