iQuentin commited on
Commit
9ca7ada
·
verified ·
1 Parent(s): 7fca18f

api_key in a secret

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -1,5 +1,7 @@
 
1
  from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, tool, InferenceClientModel
2
 
 
3
  class QAgent:
4
  def __init__(self):
5
  print("BasicAgent initialized.")
@@ -7,7 +9,7 @@ class QAgent:
7
  print(f"Agent received question (first 50 chars): {question[:50]}...")
8
  # provider = "nebius"
9
  model = "Qwen/Qwen2.5-Coder-32B-Instruct"
10
- api_key = SP_HF_TOK
11
 
12
  agent = CodeAgent(tools=[DuckDuckGoSearchTool, VisitWebpageTool],
13
  model=InferenceClientModel(
 
1
+ import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, tool, InferenceClientModel
3
 
4
+
5
  class QAgent:
6
  def __init__(self):
7
  print("BasicAgent initialized.")
 
9
  print(f"Agent received question (first 50 chars): {question[:50]}...")
10
  # provider = "nebius"
11
  model = "Qwen/Qwen2.5-Coder-32B-Instruct"
12
+ api_key = os.environ.get('SP_HF_TOK')
13
 
14
  agent = CodeAgent(tools=[DuckDuckGoSearchTool, VisitWebpageTool],
15
  model=InferenceClientModel(