test: fix test error

Change-Id: I3997dc53a2cfaa35501a1fbda5902ee15528124e
This commit is contained in:
laundry
2025-05-19 15:47:08 +08:00
parent e927b556d6
commit 3b1db26507

View File

@@ -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 = [