From 28a01dfe0ed7a8e39458dfe392ac3f6f071afc79 Mon Sep 17 00:00:00 2001
From: laundry <1561693140@qq.com>
Date: Tue, 20 May 2025 11:58:39 +0800
Subject: [PATCH] test: fix unit test error

Change-Id: If4c4cd10673e76a30945674c7cda198aeabf28d0
---
 src/llms/llm.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/llms/llm.py b/src/llms/llm.py
index 5ac7402..cc05f91 100644
--- a/src/llms/llm.py
+++ b/src/llms/llm.py
@@ -44,8 +44,7 @@ def get_llm_by_type(
     return llm
 
 
-# Initialize LLMs for different purposes - now these will be cached
-basic_llm = get_llm_by_type("basic")
+
 
 # In the future, we will use reasoning_llm and vl_llm for different purposes
 # reasoning_llm = get_llm_by_type("reasoning")
@@ -53,4 +52,6 @@ basic_llm = get_llm_by_type("basic")
 
 
 if __name__ == "__main__":
+    # Initialize LLMs for different purposes - now these will be cached
+    basic_llm = get_llm_by_type("basic")
     print(basic_llm.invoke("Hello"))