meirk-brd commited on
Commit
92b8ebf
·
1 Parent(s): 87f5e0a

add url validation

Browse files
Files changed (1) hide show
  1. tool.py +4 -4
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: