GitHub-Proxy/deer-flow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow synced 2025-08-20 21:39:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: fix unit test error

Browse Source 
Change-Id: If4c4cd10673e76a30945674c7cda198aeabf28d0
This commit is contained in:
laundry 2025-05-20 11:58:39 +08:00
parent 3b1db26507
commit 28a01dfe0e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/llms/llm.py
View File

@ -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"))