Skip to content

Commit d59e040

Browse files
committed
refactor and clean the litellm class
1 parent 5982b21 commit d59e040

File tree

6 files changed

+133
-332
lines changed

6 files changed

+133
-332
lines changed

cookbook/models/litellm/basic_gpt.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
from agno.agent import Agent
2-
from agno.models.litellm import LiteLLMSDK
2+
from agno.models.litellm import LiteLLM
33

44
openai_agent = Agent(
5-
model=LiteLLMSDK(
5+
model=LiteLLM(
66
id="gpt-4o",
77
name="LiteLLM",
88
),

cookbook/models/litellm/basic_hf.py

-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from agno.agent import Agent
22
from agno.models.litellm import LiteLLMSDK
3-
from agno.tools.duckduckgo import DuckDuckGoTools
43

54
openai_agent = Agent(
65
model=LiteLLMSDK(

cookbook/models/litellm/tool_use.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
from agno.agent import Agent
2-
from agno.models.litellm import LiteLLMSDK
2+
from agno.models.litellm import LiteLLM
33
from agno.tools.yfinance import YFinanceTools
44

55
openai_agent = Agent(
6-
model=LiteLLMSDK(
6+
model=LiteLLM(
77
id="gpt-4o",
88
name="LiteLLM",
99
),
+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
from agno.models.litellm.litellm_chat import LiteLLMSDK
1+
from agno.models.litellm.litellm_chat import LiteLLM

0 commit comments

Comments
 (0)