mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-19 12:24:46 +08:00
test: fix unit test error
Change-Id: If4c4cd10673e76a30945674c7cda198aeabf28d0
This commit is contained in:
@@ -44,8 +44,7 @@ def get_llm_by_type(
|
|||||||
return llm
|
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
|
# In the future, we will use reasoning_llm and vl_llm for different purposes
|
||||||
# reasoning_llm = get_llm_by_type("reasoning")
|
# reasoning_llm = get_llm_by_type("reasoning")
|
||||||
@@ -53,4 +52,6 @@ basic_llm = get_llm_by_type("basic")
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
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"))
|
print(basic_llm.invoke("Hello"))
|
||||||
|
|||||||
Reference in New Issue
Block a user