Spaces:
Build error
Build error
Changed the agent model
Browse files
agent.py
CHANGED
|
@@ -13,7 +13,7 @@ web_page_visit_tool = VisitWebpageTool()
|
|
| 13 |
python_script_reader_tool = ReadPythonScript()
|
| 14 |
final_answer = FinalAnswerTool()
|
| 15 |
# --- Basic Agent Definition ---
|
| 16 |
-
def build_agent(model_id:str='
|
| 17 |
model = InferenceClientModel(
|
| 18 |
max_tokens=4192,
|
| 19 |
temperature=0.5,
|
|
@@ -23,6 +23,7 @@ def build_agent(model_id:str='google/gemma-3-4b-it'):
|
|
| 23 |
agent = CodeAgent(
|
| 24 |
model=model,
|
| 25 |
tools=[search_tool,web_page_visit_tool,python_script_reader_tool],
|
| 26 |
-
prompt_templates=prompt_templates
|
|
|
|
| 27 |
)
|
| 28 |
return agent
|
|
|
|
| 13 |
python_script_reader_tool = ReadPythonScript()
|
| 14 |
final_answer = FinalAnswerTool()
|
| 15 |
# --- Basic Agent Definition ---
|
| 16 |
+
def build_agent(model_id:str='meta-llama/Llama-3.3-70B-Instruct'):
|
| 17 |
model = InferenceClientModel(
|
| 18 |
max_tokens=4192,
|
| 19 |
temperature=0.5,
|
|
|
|
| 23 |
agent = CodeAgent(
|
| 24 |
model=model,
|
| 25 |
tools=[search_tool,web_page_visit_tool,python_script_reader_tool],
|
| 26 |
+
prompt_templates=prompt_templates,
|
| 27 |
+
name= "Alex"
|
| 28 |
)
|
| 29 |
return agent
|