wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-10 17:24:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add unit tests of llms (#299)

Browse Source
This commit is contained in:
Willem Jiang
2025-06-11 19:46:01 +08:00
committed by GitHub
parent 397ac57235
commit 2554e4ba63
2 changed files with 70 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llms/llm.py
View File

@@ -70,9 +70,3 @@ def get_llm_by_type(
# In the future, we will use reasoning_llm and vl_llm for different purposes
# reasoning_llm = get_llm_by_type("reasoning")
# vl_llm = get_llm_by_type("vision")
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"))