Spaces:
Sleeping
Sleeping
meirk-brd
commited on
Commit
·
92b8ebf
1
Parent(s):
87f5e0a
add url validation
Browse files
tool.py
CHANGED
|
@@ -216,10 +216,6 @@ class BrightDataDatasetTool(Tool):
|
|
| 216 |
return self._ensure_scheme(raw)
|
| 217 |
|
| 218 |
if isinstance(raw, dict):
|
| 219 |
-
url_value = raw.get("url")
|
| 220 |
-
if isinstance(url_value, str) and url_value.startswith(("http://", "https://")):
|
| 221 |
-
return url_value
|
| 222 |
-
|
| 223 |
orig_name = raw.get("orig_name")
|
| 224 |
if (
|
| 225 |
isinstance(orig_name, str)
|
|
@@ -228,6 +224,10 @@ class BrightDataDatasetTool(Tool):
|
|
| 228 |
):
|
| 229 |
return self._ensure_scheme(orig_name)
|
| 230 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 231 |
return None
|
| 232 |
|
| 233 |
def _ensure_scheme(self, url: str) -> str:
|
|
|
|
| 216 |
return self._ensure_scheme(raw)
|
| 217 |
|
| 218 |
if isinstance(raw, dict):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 219 |
orig_name = raw.get("orig_name")
|
| 220 |
if (
|
| 221 |
isinstance(orig_name, str)
|
|
|
|
| 224 |
):
|
| 225 |
return self._ensure_scheme(orig_name)
|
| 226 |
|
| 227 |
+
url_value = raw.get("url")
|
| 228 |
+
if isinstance(url_value, str) and url_value.startswith(("http://", "https://")):
|
| 229 |
+
return url_value
|
| 230 |
+
|
| 231 |
return None
|
| 232 |
|
| 233 |
def _ensure_scheme(self, url: str) -> str:
|