env variable
Browse files
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("
|
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("
|
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(),
|