Spaces:
Runtime error
Runtime error
Upload agent
Browse files- agent.json +5 -2
- app.py +6 -3
agent.json
CHANGED
@@ -6,9 +6,12 @@
|
|
6 |
"final_answer"
|
7 |
],
|
8 |
"model": {
|
9 |
-
"class": "
|
10 |
"data": {
|
11 |
-
"
|
|
|
|
|
|
|
12 |
}
|
13 |
},
|
14 |
"managed_agents": {},
|
|
|
6 |
"final_answer"
|
7 |
],
|
8 |
"model": {
|
9 |
+
"class": "LiteLLMModel",
|
10 |
"data": {
|
11 |
+
"temperature": 0.2,
|
12 |
+
"max_tokens": 10,
|
13 |
+
"model_id": "anthropic/claude-3-5-sonnet-latest",
|
14 |
+
"api_base": null
|
15 |
}
|
16 |
},
|
17 |
"managed_agents": {},
|
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import yaml
|
2 |
import os
|
3 |
-
from smolagents import GradioUI, CodeAgent,
|
4 |
|
5 |
# Get current directory path
|
6 |
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
@@ -11,8 +11,11 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
|
|
11 |
|
12 |
|
13 |
|
14 |
-
model =
|
15 |
-
|
|
|
|
|
|
|
16 |
)
|
17 |
|
18 |
web_search = WebSearch()
|
|
|
1 |
import yaml
|
2 |
import os
|
3 |
+
from smolagents import GradioUI, CodeAgent, LiteLLMModel
|
4 |
|
5 |
# Get current directory path
|
6 |
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
11 |
|
12 |
|
13 |
|
14 |
+
model = LiteLLMModel(
|
15 |
+
temperature=0.2,
|
16 |
+
max_tokens=10,
|
17 |
+
model_id='anthropic/claude-3-5-sonnet-latest',
|
18 |
+
api_base=None,
|
19 |
)
|
20 |
|
21 |
web_search = WebSearch()
|