George Sergia commited on
Commit
b216a70
·
1 Parent(s): 62603ce

Add HF token

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. agent.py +4 -1
  3. app.py +3 -0
  4. requirements.txt +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .env
agent.py CHANGED
@@ -1,4 +1,5 @@
1
  from smolagents import DuckDuckGoSearchTool, CodeAgent, HfApiModel
 
2
  import yaml
3
 
4
  class BasicAgent:
@@ -11,7 +12,9 @@ class BasicAgent:
11
  with open("prompts.yaml", 'r') as stream:
12
  prompt_templates = yaml.safe_load(stream)
13
 
14
- model = HfApiModel(model_id="Qwen/Qwen3-235B-A22B")
 
 
15
  search_tool = DuckDuckGoSearchTool()
16
 
17
  agent = CodeAgent(
 
1
  from smolagents import DuckDuckGoSearchTool, CodeAgent, HfApiModel
2
+ import os
3
  import yaml
4
 
5
  class BasicAgent:
 
12
  with open("prompts.yaml", 'r') as stream:
13
  prompt_templates = yaml.safe_load(stream)
14
 
15
+ HF_TOKEN = os.getenv("HF_TOKEN")
16
+
17
+ model = HfApiModel(model_id="Qwen/Qwen3-235B-A22B", token=HF_TOKEN)
18
  search_tool = DuckDuckGoSearchTool()
19
 
20
  agent = CodeAgent(
app.py CHANGED
@@ -4,6 +4,9 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  from agent import BasicAgent
 
 
 
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
 
4
  import inspect
5
  import pandas as pd
6
  from agent import BasicAgent
7
+ from dotenv.main import load_dotenv
8
+
9
+ load_dotenv()
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  gradio
2
  requests
3
  smolagents
4
- pyyaml
 
 
1
  gradio
2
  requests
3
  smolagents
4
+ pyyaml
5
+ dotenv