Ramja commited on
Commit
e04f68d
·
verified ·
1 Parent(s): aa87af0

Update agents.py

Browse files

Updated HFApi to InferenctClient models

Files changed (1) hide show
  1. agents.py +2 -2
agents.py CHANGED
@@ -1,11 +1,11 @@
1
- from smolagents import CodeAgent, HfApiModel, LiteLLMModel, InferenceClientModel
2
  from tools import duck_search_tool, visit_web_page_tool, youtube_transcript_search
3
 
4
 
5
 
6
  # Initialize the model for agent_1
7
  agent_1_instructions = "Search for and retrieve relevant information about the question asked. Do not respond with any thought just the final answer:\n"
8
- model_1 = HfApiModel(
9
  "Qwen/Qwen2.5-Coder-32B-Instruct", provider="together", max_tokens=8096
10
  )
11
 
 
1
+ from smolagents import CodeAgent, LiteLLMModel, InferenceClientModel
2
  from tools import duck_search_tool, visit_web_page_tool, youtube_transcript_search
3
 
4
 
5
 
6
  # Initialize the model for agent_1
7
  agent_1_instructions = "Search for and retrieve relevant information about the question asked. Do not respond with any thought just the final answer:\n"
8
+ model_1 = InferenceClientModel(
9
  "Qwen/Qwen2.5-Coder-32B-Instruct", provider="together", max_tokens=8096
10
  )
11