yesdopepe commited on
Commit
8ce363a
·
1 Parent(s): a533d72

Update .gitignore, enhance README title, and modify app.py for environment variable support and model change

Browse files
Files changed (3) hide show
  1. .gitignore +8 -0
  2. README.md +1 -1
  3. app.py +13 -6
.gitignore CHANGED
@@ -1 +1,9 @@
 
 
 
 
 
 
 
 
1
  /venv
 
1
+ # Python
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # Environment variables
7
+ .env
8
+
9
  /venv
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Basic Agent with SMOLagents
3
  emoji: ⚡
4
  colorFrom: pink
5
  colorTo: yellow
 
1
  ---
2
+ title: Basic websearch Agent with SMOLagents
3
  emoji: ⚡
4
  colorFrom: pink
5
  colorTo: yellow
app.py CHANGED
@@ -3,6 +3,8 @@ import datetime
3
  import requests
4
  import pytz
5
  import yaml
 
 
6
  from tools.final_answer import FinalAnswerTool
7
  from tools.visit_webpage import VisitWebpageTool
8
  from tools.web_search import DuckDuckGoSearchTool
@@ -39,23 +41,28 @@ def get_current_time_in_timezone(timezone: str) -> str:
39
  final_answer = FinalAnswerTool()
40
  web_search = DuckDuckGoSearchTool()
41
  visit_webpage = VisitWebpageTool()
 
 
 
42
  model = HfApiModel(
43
- max_tokens=2096,
44
- temperature=0.5,
45
- model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
46
- custom_role_conversions=None,
 
 
47
  )
48
 
49
 
50
  # Import tool from Hub
51
- image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer, image_generation_tool, web_search, visit_webpage, get_current_time_in_timezone, my_cutom_tool], ## added get_current_time_in_timezone and my_cutom_tool
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,
 
3
  import requests
4
  import pytz
5
  import yaml
6
+ import os
7
+ from dotenv import load_dotenv
8
  from tools.final_answer import FinalAnswerTool
9
  from tools.visit_webpage import VisitWebpageTool
10
  from tools.web_search import DuckDuckGoSearchTool
 
41
  final_answer = FinalAnswerTool()
42
  web_search = DuckDuckGoSearchTool()
43
  visit_webpage = VisitWebpageTool()
44
+
45
+ load_dotenv() # Load environment variables from .env file
46
+
47
  model = HfApiModel(
48
+ max_tokens=2096,
49
+ temperature=0.5,
50
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
51
+ custom_role_conversions=None,
52
+ provider="together",
53
+ token=os.getenv('TOGETHER_TOKEN') # Get token from environment variable
54
  )
55
 
56
 
57
  # Import tool from Hub
58
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True, token=os.getenv('HUGGINGFACE_TOKEN'))
59
 
60
  with open("prompts.yaml", 'r') as stream:
61
  prompt_templates = yaml.safe_load(stream)
62
 
63
  agent = CodeAgent(
64
  model=model,
65
+ tools=[final_answer, web_search, visit_webpage, get_current_time_in_timezone], ## added get_current_time_in_timezone and my_cutom_tool
66
  max_steps=6,
67
  verbosity_level=1,
68
  grammar=None,