Shreyas094 commited on
Commit
e6fae54
·
verified ·
1 Parent(s): ac6819a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -17
app.py CHANGED
@@ -272,23 +272,9 @@ def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temp
272
  return final_response
273
 
274
  class SimpleDDGSearch:
275
- def search(self, query: str, use_chat: bool = False, model: str = "claude-3-haiku", num_results: int = 10):
276
- if use_chat:
277
- result = DDGS().chat(
278
- keywords=query,
279
- model=model,
280
- timeout=30
281
- )
282
- return [{"content": result, "url": "AI-generated response"}]
283
- else:
284
- results = DDGS().news(
285
- keywords=query,
286
- region='wt-wt',
287
- safesearch='off',
288
- timelimit='m',
289
- max_results=num_results
290
- )
291
- return [{"content": res["body"], "url": res["url"]} for res in results]
292
 
293
  class TrafilaturaWebCrawler:
294
  def get_website_content_from_url(self, url: str) -> str:
 
272
  return final_response
273
 
274
  class SimpleDDGSearch:
275
+ def search(self, query: str, num_results: int = 10):
276
+ results = DDGS().news(query, region='wt-wt', safesearch='off', max_results=num_results)
277
+ return [res["url"] for res in results]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
278
 
279
  class TrafilaturaWebCrawler:
280
  def get_website_content_from_url(self, url: str) -> str: