name

src/autolabel/transforms/custom_api.py

@@ -30,16 +30,16 @@ def __init__(
         self,
         cache: BaseCache,
         output_columns: Dict[str, Any],
-        base_url: str,
+        api_url: str,
         request_columns: List[str],
         headers: Dict[str, str] = HEADERS,
         timeout: int = 60,
     ) -> None:
         super().__init__(cache, output_columns)
         self.request_columns = request_columns
-        if not urlparse(base_url).scheme:
-            base_url = f"https://{base_url}"
-        self.base_url = base_url
+        if not urlparse(api_url).scheme:
+            api_url = f"https://{api_url}"
+        self.api_url = api_url
         self.headers = headers
         self.max_retries = MAX_RETRIES
         try:
@@ -119,15 +119,15 @@ async def _apply(self, row: Dict[str, Any]) -> Dict[str, Any]:
                 logger.warning(
                     f"Missing request column: {col} in row {row}",
                 )
-        url = self.base_url.format_map(defaultdict(str, row))
+        url = self.api_url.format_map(defaultdict(str, row))
         result = await self._get_result(url)
         transformed_row = {self.output_columns["result"]: result}
         return self._return_output_row(transformed_row)
 
     def params(self):
         return {
             "output_columns": self.output_columns,
-            "base_url": self.base_url,
+            "api_url": self.api_url,
             "request_columns": self.request_columns,
         }