Amalia commited on
Commit
8ddb639
·
1 Parent(s): d0c1441

include the tools in the codeagent....

Browse files
Files changed (3) hide show
  1. .gitignore +1 -0
  2. app.py +15 -1
  3. requirements.txt +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .env
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import gradio as gr
3
  import requests
4
  import inspect
@@ -7,6 +8,9 @@ from smolagents import CodeAgent, tool, InferenceClientModel
7
  from langchain_community.tools.tavily_search import TavilySearchResults
8
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
9
 
 
 
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
12
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -38,6 +42,16 @@ def subtract(a: int, b: int) -> int:
38
  """
39
  return a - b
40
 
 
 
 
 
 
 
 
 
 
 
41
  @tool
42
  def web_search(query: str) -> str:
43
  """Search Tavily for a query and return maximum 3 results.
@@ -86,7 +100,7 @@ class BasicAgent:
86
  model_id = "Qwen/Qwen3-32B"
87
  self.agent = CodeAgent(
88
  model=InferenceClientModel(model_id=model_id, token=os.getenv("HF_TOKEN")),
89
- tools=[multiply, add, subtract],
90
  max_steps=10,
91
  )
92
  print("BasicAgent initialized.")
 
1
  import os
2
+ from dotenv import load_dotenv
3
  import gradio as gr
4
  import requests
5
  import inspect
 
8
  from langchain_community.tools.tavily_search import TavilySearchResults
9
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
10
 
11
+ # Load environment variables from .env file
12
+ load_dotenv()
13
+
14
  # (Keep Constants as is)
15
  # --- Constants ---
16
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
42
  """
43
  return a - b
44
 
45
+ @tool
46
+ def modulus(a: int, b: int) -> int:
47
+ """Get the modulus of two numbers.
48
+
49
+ Args:
50
+ a: first int
51
+ b: second int
52
+ """
53
+ return a % b
54
+
55
  @tool
56
  def web_search(query: str) -> str:
57
  """Search Tavily for a query and return maximum 3 results.
 
100
  model_id = "Qwen/Qwen3-32B"
101
  self.agent = CodeAgent(
102
  model=InferenceClientModel(model_id=model_id, token=os.getenv("HF_TOKEN")),
103
+ tools=[multiply, add, subtract, modulus, web_search, arxiv_search, wikipedia_search],
104
  max_steps=10,
105
  )
106
  print("BasicAgent initialized.")
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  gradio
2
  requests
3
  smolagents
4
- langchain-community
 
 
1
  gradio
2
  requests
3
  smolagents
4
+ langchain-community
5
+ python-dotenv