Hanson commited on
Commit
c5aaba5
·
1 Parent(s): 3f8dd4c

fix: add ddgs dependency for search tool

Browse files
Files changed (2) hide show
  1. app.py +1 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -12,7 +12,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
14
  import os
15
- import spaces # 💡 1. 引入 Hugging Face 的 GPU 空間套件
16
 
17
  class BasicAgent:
18
  def __init__(self):
@@ -25,7 +25,6 @@ class BasicAgent:
25
  self.search_tool = DuckDuckGoSearchTool()
26
  self.agent = CodeAgent(tools=[self.search_tool], model=self.model)
27
 
28
- # 💡 2. 在實際執行的函式上方加上這個裝飾器,滿足 Zero GPU 的啟動檢查
29
  @spaces.GPU
30
  def __call__(self, question: str) -> str:
31
  try:
 
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool
14
  import os
15
+ import spaces
16
 
17
  class BasicAgent:
18
  def __init__(self):
 
25
  self.search_tool = DuckDuckGoSearchTool()
26
  self.agent = CodeAgent(tools=[self.search_tool], model=self.model)
27
 
 
28
  @spaces.GPU
29
  def __call__(self, question: str) -> str:
30
  try:
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  gradio
2
  requests
3
- smolagents[litellm]
 
 
1
  gradio
2
  requests
3
+ smolagents[litellm]
4
+ ddgs