Update app.py
Browse files
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,
|
| 276 |
-
|
| 277 |
-
|
| 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:
|