ddgs another adjustment

tools.py

@@ -86,6 +86,7 @@ nltk.download('words', quiet=True)
 english_words = set(words.words())
 
 logger = logging.getLogger(__name__)
+logging.basicConfig(filename='app.log', level=logging.INFO)
 
 # --- Model Configuration ---
 def create_llm_pipeline():
@@ -895,7 +896,22 @@ Snippet: {snippet}
 """
         return formatted
     
-    def run(self, query: str) -> str:
+    def run(self, tool_input: Union[str, Dict]) -> str:
+        query_str: Optional[str] = None
+
+        if isinstance(tool_input, dict):
+            # Try common keys where the actual query string might be stored
+            if "query" in tool_input and isinstance(tool_input["query"], str):
+                query_str = tool_input["query"]
+            elif "input" in tool_input and isinstance(tool_input["input"], str):
+                query_str = tool_input["input"]
+            # Add more checks if other dictionary structures are possible
+            else:
+                return "Invalid input: Dictionary received, but does not contain a recognizable string query under 'query' or 'input' keys."
+        elif isinstance(tool_input, str):
+            query_str = tool_input
+        else:
+            return f"Invalid input type: Expected a string or a dictionary, but got {type(tool_input).__name__}."
 
         """Execute the enhanced search."""
         if not query or not query.strip():