jjjulllesss commited on
Commit
ca345fd
·
verified ·
1 Parent(s): 8f43b5e

env variable

Browse files
Files changed (1) hide show
  1. agent.py +3 -5
agent.py CHANGED
@@ -100,7 +100,7 @@ class ImageAnalysisTool(Tool):
100
  def forward(self, task_id: str) -> str:
101
  client = InferenceClient(
102
  provider="nebius",
103
- api_key=os.getenv("HF_API_KEY"),
104
  )
105
 
106
  completion = client.chat.completions.create(
@@ -130,7 +130,7 @@ class ImageAnalysisTool(Tool):
130
  def final_answer_formatting(answer, question):
131
  client = InferenceClient(
132
  provider="nebius",
133
- api_key=os.getenv("HF_API_KEY"),
134
  )
135
 
136
  prompt = f"""
@@ -159,9 +159,7 @@ def final_answer_formatting(answer, question):
159
 
160
 
161
  web_agent = CodeAgent(
162
- model=HfApiModel(
163
- model_id='Qwen/Qwen3-32B'
164
- ),
165
  tools=[
166
  WikipediaSearchTool(),
167
  DuckDuckGoSearchTool(),
 
100
  def forward(self, task_id: str) -> str:
101
  client = InferenceClient(
102
  provider="nebius",
103
+ api_key=os.getenv("HF_TOKEN"),
104
  )
105
 
106
  completion = client.chat.completions.create(
 
130
  def final_answer_formatting(answer, question):
131
  client = InferenceClient(
132
  provider="nebius",
133
+ api_key=os.getenv("HF_TOKEN"),
134
  )
135
 
136
  prompt = f"""
 
159
 
160
 
161
  web_agent = CodeAgent(
162
+ model=HfApiModel(),
 
 
163
  tools=[
164
  WikipediaSearchTool(),
165
  DuckDuckGoSearchTool(),