diff --git a/job_executor/adapter/datastore_api/__init__.py b/job_executor/adapter/datastore_api/__init__.py index 709b844..bdae84b 100644 --- a/job_executor/adapter/datastore_api/__init__.py +++ b/job_executor/adapter/datastore_api/__init__.py @@ -8,7 +8,6 @@ from urllib3 import Retry from job_executor.adapter.datastore_api.models import ( - DatastoreResponse, Job, JobQueryResult, JobStatus, @@ -132,9 +131,17 @@ def get_datastores() -> list[str]: def get_datastore_directory(rdn: str) -> Path: - request_url = f"{DATASTORE_API_URL}/datastores/{rdn}" - response = execute_request("GET", request_url, True) - return Path(DatastoreResponse.model_validate(response.json()).directory) + request_url = f"{DATASTORE_API_URL}/datastores/{rdn}/directory" + response = execute_request( + "GET", + request_url, + True, + headers={ + "X-API-Key": DATASTORE_API_SERVICE_KEY, + }, + ) + directory = response.json() + return Path(directory) def post_public_key(datastore_rdn: str, public_key_pem: bytes) -> None: diff --git a/job_executor/adapter/datastore_api/models.py b/job_executor/adapter/datastore_api/models.py index 74b8b95..4ed81fa 100644 --- a/job_executor/adapter/datastore_api/models.py +++ b/job_executor/adapter/datastore_api/models.py @@ -9,15 +9,6 @@ from job_executor.common.models import CamelModel -class DatastoreResponse(CamelModel): - datastore_id: int - name: str - rdn: str - description: str - directory: str - bump_enabled: bool - - class MaintenanceStatus(CamelModel): paused: bool msg: str