From 82e5417926439c3046fe941c3a75f15f2779e764 Mon Sep 17 00:00:00 2001 From: zhangzepin Date: Tue, 22 Jul 2025 11:18:48 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=91=BD=E4=BB=A4=E8=A1=8C=E6=89=A7?= =?UTF-8?q?=E8=A1=8C=E6=97=B6=E6=89=93=E5=8D=B0=E4=BF=A1=E6=81=AF=E8=B0=83?= =?UTF-8?q?=E6=95=B4=E5=8F=8A=E6=97=A5=E5=BF=97=E8=A1=A5=E5=85=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 7 ++++++- src/tools/crawler/html_parser_crawler.py | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9f347bf..f68a4f4 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,7 @@ import argparse import asyncio +import json from src.manager.workflow import Workflow @@ -17,7 +18,11 @@ async def run_workflow(query: str): workflow = Workflow() workflow.build_graph() async for msg in workflow.run(query, "default_session_id", []): - print(f"{msg}\n") + msgObj = json.loads(msg) + if "message_type" in msgObj and msgObj["message_type"] == "AIMessageChunk" and "content" in msgObj: + print(msgObj["content"], end="") + else: + print(f"\n{msg}") if __name__ == "__main__": parser = argparse.ArgumentParser(description="Run deepsearch project") diff --git a/src/tools/crawler/html_parser_crawler.py b/src/tools/crawler/html_parser_crawler.py index 79a511a..b428543 100644 --- a/src/tools/crawler/html_parser_crawler.py +++ b/src/tools/crawler/html_parser_crawler.py @@ -56,7 +56,7 @@ class BasicWebCrawler(BaseModel): "images": images, } else: - logger.error("Crawl Tool: Html request failed.") + logger.error(f"Crawl Tool: Html request failed, {url}, {response.content}.") if __name__ == "__main__": -- Gitee