From 3b1db26507cc8aad8b786954931a6f1e421703d7 Mon Sep 17 00:00:00 2001
From: laundry <1561693140@qq.com>
Date: Mon, 19 May 2025 15:47:08 +0800
Subject: [PATCH] test: fix test error

Change-Id: I3997dc53a2cfaa35501a1fbda5902ee15528124e
---
 tests/integration/test_nodes.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/tests/integration/test_nodes.py b/tests/integration/test_nodes.py
index 6a1c8f6..978acaa 100644
--- a/tests/integration/test_nodes.py
+++ b/tests/integration/test_nodes.py
@@ -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 = [