mirror of
https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow
synced 2025-08-20 17:19:09 +08:00
test: fix unit test error
Change-Id: If4c4cd10673e76a30945674c7cda198aeabf28d0
This commit is contained in:
parent
3b1db26507
commit
28a01dfe0e
@ -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"))
|
||||
|
Loading…
x
Reference in New Issue
Block a user