Update agent.py

agent.py

@@ -6,7 +6,7 @@ from transformers import (
     AutoProcessor,
     AutoModelForSpeechSeq2Seq
 )
-from ddgs import DDGS
+from duckduckgo_search import DDGS
 import pandas as pd
 import os
 
@@ -37,7 +37,7 @@ class GaiaAgent:
     def search(self, query: str) -> str:
         try:
             with DDGS() as ddgs:
-                results = list(ddgs.text(query, safesearch="off"))
+                results = ddgs.text("your query", max_results=1)
                 if results:
                     return results[0]['body']
         except Exception: