mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-03 06:12:14 +08:00
test: fix test error
Change-Id: I3997dc53a2cfaa35501a1fbda5902ee15528124e
This commit is contained in:
@@ -2,11 +2,12 @@ import json
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
from langgraph.types import Command
|
||||
|
||||
from src.graph.nodes import background_investigation_node
|
||||
from src.config import SearchEngine
|
||||
from langchain_core.messages import HumanMessage
|
||||
# 在这里 mock 掉 get_llm_by_type,避免 ValueError
|
||||
with patch("src.llms.llm.get_llm_by_type", return_value=MagicMock()):
|
||||
from langgraph.types import Command
|
||||
from src.graph.nodes import background_investigation_node
|
||||
from src.config import SearchEngine
|
||||
from langchain_core.messages import HumanMessage
|
||||
|
||||
# Mock data
|
||||
MOCK_SEARCH_RESULTS = [
|
||||
|
||||
Reference in New Issue
Block a user