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