daqc commited on
Commit
721d7b1
·
verified ·
1 Parent(s): 98dcf3c

Delete try-litellm.py

Browse files
Files changed (1) hide show
  1. try-litellm.py +0 -25
try-litellm.py DELETED
@@ -1,25 +0,0 @@
1
- import os
2
-
3
- from smolagents import (
4
- # MANAGED_AGENT_PROMPT,
5
- CodeAgent,
6
- HfApiModel,
7
- LiteLLMModel,
8
- Model,
9
- ToolCallingAgent,
10
- )
11
-
12
- custom_role_conversions = {"tool-call": "assistant", "tool-response": "user"}
13
-
14
- model = LiteLLMModel(
15
- # "gpt-4o",
16
- # os.getenv("MODEL_ID", "gpt-4o-mini"),
17
- os.getenv("MODEL_ID", "deepseek-ai/DeepSeek-V3"),
18
- custom_role_conversions=custom_role_conversions,
19
- api_base=os.getenv("OPENAI_API_BASE"),
20
- api_key=os.getenv("OPENAI_API_KEY"),
21
- )
22
-
23
- print(model)
24
-
25
- print(model.invoke("Say this is a test"))