wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-18 20:14:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: convert crawl_tool dict return to JSON string for type consistency (#636)

Browse Source 
Keep fixing #631
This pull request updates the crawl_tool function to return its results as a JSON string instead of a dictionary, and adjusts the unit tests accordingly to handle the new return type. The changes ensure consistent serialization of output and proper validation in tests.
This commit is contained in:
Willem Jiang
2025-10-21 10:00:33 +08:00
committed by GitHub
parent e2ff765460
commit d30c4d00d3
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/unit/tools/test_crawl.py
View File

@@ -1,3 +1,4 @@
import json
from unittest.mock import Mock, patch
from src.tools.crawl import crawl_tool
@@ -21,10 +22,11 @@ class TestCrawlTool:
result = crawl_tool(url)
# Assert
assert isinstance(result, dict)
assert result["url"] == url
assert "crawled_content" in result
assert len(result["crawled_content"]) <= 1000
assert isinstance(result, str)
result_dict = json.loads(result)
assert result_dict["url"] == url
assert "crawled_content" in result_dict
assert len(result_dict["crawled_content"]) <= 1000
mock_crawler_class.assert_called_once()
mock_crawler.crawl.assert_called_once_with(url)
mock_article.to_markdown.assert_called_once()
@@ -45,7 +47,8 @@ class TestCrawlTool:
result = crawl_tool(url)
# Assert
assert result["crawled_content"] == short_content
result_dict = json.loads(result)
assert result_dict["crawled_content"] == short_content
@patch("src.tools.crawl.Crawler")
@patch("src.tools.crawl.logger")