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(),
|