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

test: fix test error

Browse Source 
Change-Id: I3997dc53a2cfaa35501a1fbda5902ee15528124e
This commit is contained in:
laundry
2025-05-19 15:47:08 +08:00
parent e927b556d6
commit 3b1db26507
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration/test_nodes.py
View File

@@ -2,11 +2,12 @@ import json
import pytest import pytest
from unittest.mock import patch, MagicMock from unittest.mock import patch, MagicMock
from langgraph.types import Command # 在这里 mock 掉 get_llm_by_type，避免 ValueError
with patch("src.llms.llm.get_llm_by_type", return_value=MagicMock()):
from src.graph.nodes import background_investigation_node from langgraph.types import Command
from src.config import SearchEngine from src.graph.nodes import background_investigation_node
from langchain_core.messages import HumanMessage from src.config import SearchEngine
from langchain_core.messages import HumanMessage
# Mock data # Mock data
MOCK_SEARCH_RESULTS = [ MOCK_SEARCH_RESULTS = [