diff --git a/src/intelligence_layer/connectors/argilla/argilla_client.py b/src/intelligence_layer/connectors/argilla/argilla_client.py index 28ac0387c..acc180a72 100644 --- a/src/intelligence_layer/connectors/argilla/argilla_client.py +++ b/src/intelligence_layer/connectors/argilla/argilla_client.py @@ -285,7 +285,7 @@ def split_dataset(self, dataset_id: str, n_splits: int) -> None: def _create_metadata_property(self, dataset_id: str, n_splits: int) -> None: response = self.session.get( - f"http://localhost:6900/api/v1/me/datasets/{dataset_id}/metadata-properties" + f"{self.api_url}api/v1/me/datasets/{dataset_id}/metadata-properties" ) response.raise_for_status() existing_split_id = [ @@ -295,7 +295,7 @@ def _create_metadata_property(self, dataset_id: str, n_splits: int) -> None: ] if len(existing_split_id) > 0: self.session.delete( - f"http://localhost:6900/api/v1/metadata-properties/{existing_split_id[0]}" + f"{self.api_url}api/v1/metadata-properties/{existing_split_id[0]}" ) data = { @@ -307,7 +307,7 @@ def _create_metadata_property(self, dataset_id: str, n_splits: int) -> None: } response = self.session.post( - f"http://localhost:6900/api/v1/datasets/{dataset_id}/metadata-properties", + f"{self.api_url}api/v1/datasets/{dataset_id}/metadata-properties", json=data, ) response.raise_for_status() @@ -340,7 +340,7 @@ def chunks( ] } response = self.session.patch( - f"http://localhost:6900/api/v1/datasets/{dataset_id}/records", + f"{self.api_url}api/v1/datasets/{dataset_id}/records", json=data, ) response.raise_for_status()