Upload 7 files
Browse filesaft phase2.5 test
- agent/gaia_agent.py +7 -3
- requirements.txt +10 -1
agent/gaia_agent.py
CHANGED
|
@@ -39,8 +39,13 @@ class GaiaAgent:
|
|
| 39 |
if custom_base:
|
| 40 |
model_kwargs["api_base"] = custom_base
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
model = LiteLLMModel(
|
| 43 |
-
model_id=
|
| 44 |
**model_kwargs
|
| 45 |
)
|
| 46 |
else:
|
|
@@ -74,8 +79,7 @@ class GaiaAgent:
|
|
| 74 |
"itertools",
|
| 75 |
"collections",
|
| 76 |
"statistics",
|
| 77 |
-
"bs4"
|
| 78 |
-
"BeautifulSoup"
|
| 79 |
],
|
| 80 |
max_steps=10,
|
| 81 |
verbosity_level=1
|
|
|
|
| 39 |
if custom_base:
|
| 40 |
model_kwargs["api_base"] = custom_base
|
| 41 |
|
| 42 |
+
# Fix: Add provider prefix for LiteLLM
|
| 43 |
+
model_id_with_prefix = model_id or "gpt-4o"
|
| 44 |
+
if custom_base and "moonshot" in custom_base and not model_id_with_prefix.startswith("moonshot/"):
|
| 45 |
+
model_id_with_prefix = f"moonshot/{model_id_with_prefix}"
|
| 46 |
+
|
| 47 |
model = LiteLLMModel(
|
| 48 |
+
model_id=model_id_with_prefix,
|
| 49 |
**model_kwargs
|
| 50 |
)
|
| 51 |
else:
|
|
|
|
| 79 |
"itertools",
|
| 80 |
"collections",
|
| 81 |
"statistics",
|
| 82 |
+
"bs4"
|
|
|
|
| 83 |
],
|
| 84 |
max_steps=10,
|
| 85 |
verbosity_level=1
|
requirements.txt
CHANGED
|
@@ -1,2 +1,11 @@
|
|
| 1 |
gradio
|
| 2 |
-
requests
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
gradio
|
| 2 |
+
requests
|
| 3 |
+
pandas
|
| 4 |
+
smolagents
|
| 5 |
+
duckduckgo-search
|
| 6 |
+
openai
|
| 7 |
+
pillow
|
| 8 |
+
openpyxl
|
| 9 |
+
youtube-transcript-api
|
| 10 |
+
beautifulsoup4
|
| 11 |
+
yt-dlp
|