Niraya666 commited on
Commit
9d90e54
·
verified ·
1 Parent(s): 4c70715

Upload 7 files

Browse files

aft phase2.5 test

Files changed (2) hide show
  1. agent/gaia_agent.py +7 -3
  2. 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=model_id or "gpt-4o",
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