fix dataset request

tool.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import ast
 import json
 import os
 import time
@@ -70,14 +71,16 @@ class BrightDataDatasetTool(Tool):
     def forward(
         self,
         dataset: str,
-        url: Optional[str] = None,
-        keyword: Optional[str] = None,
-        first_name: Optional[str] = None,
-        last_name: Optional[str] = None,
-        days_limit: Optional[str] = None,
-        num_of_reviews: Optional[str] = None,
-        num_of_comments: Optional[str] = None,
-    ) -> str:
+            url: Optional[str] = None,
+            keyword: Optional[str] = None,
+            first_name: Optional[str] = None,
+            last_name: Optional[str] = None,
+            days_limit: Optional[str] = None,
+            num_of_reviews: Optional[str] = None,
+            num_of_comments: Optional[str] = None,
+        ) -> str:
+        url = self._coerce_url_input(url)
+
         api_token = os.getenv("BRIGHT_DATA_API_TOKEN")
         if not api_token:
             raise ValueError("BRIGHT_DATA_API_TOKEN not found in environment variables")
@@ -197,6 +200,45 @@ class BrightDataDatasetTool(Tool):
 
         raise TimeoutError(f"Timeout waiting for snapshot {snapshot_id} after {max_attempts} seconds")
 
+    def _coerce_url_input(self, raw: Optional[Any]) -> Optional[str]:
+        if raw is None:
+            return None
+
+        if isinstance(raw, str):
+            if raw.strip().startswith("{") and "orig_name" in raw:
+                parsed = self._parse_file_dict_string(raw)
+                if parsed:
+                    raw = parsed
+                else:
+                    return self._ensure_scheme(raw)
+            else:
+                return self._ensure_scheme(raw)
+
+        if isinstance(raw, dict):
+            orig_name = raw.get("orig_name")
+            if isinstance(orig_name, str) and orig_name:
+                return self._ensure_scheme(orig_name)
+
+            url_value = raw.get("url")
+            if isinstance(url_value, str):
+                if url_value.startswith(("http://", "https://")):
+                    return url_value
+                return None
+
+        return None
+
+    def _ensure_scheme(self, url: str) -> str:
+        if url.startswith(("http://", "https://")):
+            return url
+        return f"https://{url}"
+
+    def _parse_file_dict_string(self, value: str) -> Optional[dict]:
+        try:
+            parsed = ast.literal_eval(value)
+            return parsed if isinstance(parsed, dict) else None
+        except (ValueError, SyntaxError):
+            return None
+
     def _get_gradio_app_code(self, tool_module_name: str = "tool") -> str:
         choices = sorted(self.datasets.keys())
         dataset_fields = {key: value["inputs"] for key, value in self.datasets.items()}