DTStudios commited on
Commit
b2c820a
·
verified ·
1 Parent(s): 9928479

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -24,6 +24,13 @@ def __init__(self, max_results=10, **kwargs):
24
  "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
25
  ) from e
26
  self.ddgs = DDGS(**kwargs)
 
 
 
 
 
 
 
27
 
28
  @tool
29
  def get_current_time_in_timezone(timezone: str) -> str:
 
24
  "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
25
  ) from e
26
  self.ddgs = DDGS(**kwargs)
27
+
28
+ def forward(self, query: str) -> str:
29
+ results = self.ddgs.text(query, max_results=self.max_results)
30
+ if len(results) == 0:
31
+ raise Exception("No results found! Try a less restrictive/shorter query.")
32
+ postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
33
+ return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
34
 
35
  @tool
36
  def get_current_time_in_timezone(timezone: str) -> str: