Bastati commited on
Commit
657f4cd
·
1 Parent(s): 5364a76

Changed the agent model

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
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='google/gemma-3-4b-it'):
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