⬆️ Upgrade openapi generator

clients/python/Makefile

@@ -116,8 +116,8 @@ _check_venv_active:
 
 .PHONY: install-dev
 install-dev: _check_venv_active .install-dev-reqs python-client  ## installs osparc_client and osparc in edit mode
-	uv pip install -e artifacts/client
-	uv pip install -e .
+	# for some reason this command refuses to run with uv. Looks related to https://github.com/astral-sh/uv/issues/1661
+	pip install --editable artifacts/client --editable .
 	uv pip list
 
 .PHONY: install-unit-test

clients/python/requirements/dev.txt

@@ -1,4 +1,3 @@
 -r ../../../requirements.txt
 -r unit-test.txt
--r e2e-test.txt
 pylint

clients/python/requirements/unit-test.txt

@@ -1,8 +1,11 @@
 -r ../../../requirements.txt
 black
 faker
+openapi-spec-validator #installed as backend for prance
+parse
 pipdeptree
 pipreqs
+prance
 pytest
 pytest-asyncio
 pytest-mock

clients/python/setup.py

@@ -30,6 +30,7 @@
     "tenacity",
     "tqdm>=4.48.0",
     f"osparc_client=={VERSION}",
+    "urllib3",
 ]
 
 SETUP = dict(

clients/python/src/osparc/_api_client.py

@@ -3,9 +3,8 @@
 from typing import Optional
 
 from osparc_client.api_client import ApiClient as _ApiClient
-from osparc_client import Configuration
+from ._configuration import Configuration
 from pydantic import ValidationError
-
 from ._settings import ConfigurationEnvVars
 
 
@@ -16,7 +15,6 @@ def __init__(
         header_name=None,
         header_value=None,
         cookie=None,
-        pool_threads=1,
     ):
         if configuration is None:
             try:
@@ -36,4 +34,4 @@ def __init__(
                     "osparc.Configuration object explicitly"
                 ) from exc
 
-        super().__init__(configuration, header_name, header_value, cookie, pool_threads)
+        super().__init__(configuration, header_name, header_value, cookie)

clients/python/src/osparc/_api_files_api.py

@@ -9,10 +9,12 @@
 import string
 from pathlib import Path
 from typing import Any, Iterator, List, Optional, Tuple, Union
+from tempfile import NamedTemporaryFile
 
 import httpx
 from httpx import Response
 from osparc_client.api.files_api import FilesApi as _FilesApi
+from osparc_client import Filesize as _Filesize
 from tqdm.asyncio import tqdm
 from tqdm.contrib.logging import logging_redirect_tqdm
 
@@ -71,7 +73,10 @@ def download_file(
             raise RuntimeError(
                 f"destination_folder: {destination_folder} must be a directory"
             )
-        downloaded_file: Path = Path(super().download_file(file_id, **kwargs))
+        with NamedTemporaryFile(delete=False) as tmp_file:
+            downloaded_file = Path(tmp_file.name)
+        data = super().download_file(file_id, **kwargs)
+        downloaded_file.write_bytes(data)
         if destination_folder is not None:
             dest_file: Path = destination_folder / downloaded_file.name
             while dest_file.is_file():
@@ -115,7 +120,7 @@ async def upload_file_async(
                 return file_result
         client_file: ClientFile = ClientFile(
             filename=file.name,
-            filesize=file.stat().st_size,
+            filesize=_Filesize(file.stat().st_size),
             sha256_checksum=checksum,
         )
         client_upload_schema: ClientFileUploadData = super().get_upload_links(

clients/python/src/osparc/_api_solvers_api.py

@@ -4,7 +4,17 @@
 
 import httpx
 from osparc_client.api.solvers_api import SolversApi as _SolversApi
-from .models import JobInputs, OnePageSolverPort, SolverPort
+from .models import (
+    JobInputs,
+    OnePageSolverPort,
+    SolverPort,
+    Job,
+    JobOutputs,
+    JobMetadata,
+    JobMetadataUpdate,
+)
+from osparc_client import JobInputs as _JobInputs
+from osparc_client import JobMetadataUpdate as _JobMetadataUpdate
 
 from ._api_client import ApiClient
 from ._settings import ParentProjectInfo
@@ -15,8 +25,12 @@
     dev_feature,
     dev_features_enabled,
 )
-
 import warnings
+from tempfile import NamedTemporaryFile
+from pathlib import Path
+from pydantic import validate_call
+from pydantic import Field, StrictStr
+from typing import Annotated
 
 
 class SolversApi(_SolversApi):
@@ -97,8 +111,59 @@ def jobs(self, solver_key: str, version: str, **kwargs) -> PaginationGenerator:
         )
         return self.iter_jobs(solver_key, version, **kwargs)
 
+    @validate_call
     def create_job(
         self, solver_key: str, version: str, job_inputs: JobInputs, **kwargs
-    ):
+    ) -> Job:
+        _job_inputs = _JobInputs.from_json(job_inputs.model_dump_json())
+        assert _job_inputs is not None
         kwargs = {**kwargs, **ParentProjectInfo().model_dump(exclude_none=True)}
-        return super().create_job(solver_key, version, job_inputs, **kwargs)
+        return super().create_job(solver_key, version, _job_inputs, **kwargs)
+
+    def get_job_output_logfile(
+        self,
+        solver_key: Annotated[str, Field(strict=True)],
+        version: Annotated[str, Field(strict=True)],
+        job_id: StrictStr,
+        **kwargs,
+    ):
+        data = super().get_job_output_logfile(
+            solver_key=solver_key, version=version, job_id=job_id, **kwargs
+        )
+        with NamedTemporaryFile(delete=False) as tmp_file:
+            log_file = Path(tmp_file.name)
+            log_file.write_bytes(data)
+            return log_file
+
+    def get_job_outputs(
+        self,
+        solver_key: Annotated[str, Field(strict=True)],
+        version: Annotated[str, Field(strict=True)],
+        job_id: StrictStr,
+        **kwargs,
+    ) -> JobOutputs:
+        _osparc_client_outputs = super().get_job_outputs(
+            solver_key=solver_key, version=version, job_id=job_id, **kwargs
+        )
+        return JobOutputs.model_validate_json(_osparc_client_outputs.to_json())
+
+    def get_job_custom_metadata(self, *args, **kwargs) -> JobMetadata:
+        metadata = super().get_job_custom_metadata(*args, **kwargs)
+        return JobMetadata.model_validate_json(metadata.to_json())
+
+    @validate_call
+    def replace_job_custom_metadata(
+        self,
+        solver_key: str,
+        version: str,
+        job_id: str,
+        job_metadata_update: JobMetadataUpdate,
+    ) -> JobMetadata:
+        _job_metadata_update = _JobMetadataUpdate.from_json(
+            job_metadata_update.model_dump_json()
+        )
+        assert _job_metadata_update is not None
+        _job_custom_metadata = super().replace_job_custom_metadata(
+            solver_key, version, job_id, _job_metadata_update
+        )
+        return JobMetadata.model_validate_json(_job_custom_metadata.to_json())

clients/python/src/osparc/_api_studies_api.py

@@ -7,8 +7,19 @@
 from typing import Any, Optional
 
 import httpx
-from .models import JobInputs, JobLogsMap, PageStudy
+from pydantic import StrictStr
+
+from .models import (
+    JobInputs,
+    JobLogsMap,
+    PageStudy,
+    JobOutputs,
+    JobMetadata,
+    JobMetadataUpdate,
+)
 from osparc_client.api.studies_api import StudiesApi as _StudiesApi
+from osparc_client import JobInputs as _JobInputs
+from osparc_client import JobMetadataUpdate as _JobMetadataUpdate
 from tqdm.asyncio import tqdm_asyncio
 
 from ._api_client import ApiClient
@@ -64,8 +75,18 @@ def __getattr__(self, name: str) -> Any:
         return super().__getattribute__(name)
 
     def create_study_job(self, study_id: str, job_inputs: JobInputs, **kwargs):
+        _job_inputs = _JobInputs.from_json(job_inputs.model_dump_json())
+        assert _job_inputs is not None
         kwargs = {**kwargs, **ParentProjectInfo().model_dump(exclude_none=True)}
-        return super().create_study_job(study_id, job_inputs, **kwargs)
+        return super().create_study_job(study_id, _job_inputs, **kwargs)
+
+    def get_study_job_outputs(
+        self, study_id: StrictStr, job_id: StrictStr, **kwargs
+    ) -> JobOutputs:
+        _job_outputs = super().get_study_job_outputs(
+            study_id=study_id, job_id=job_id, **kwargs
+        )
+        return JobOutputs.model_validate_json(_job_outputs.to_json())
 
     def clone_study(self, study_id: str, **kwargs):
         kwargs = {**kwargs, **ParentProjectInfo().model_dump(exclude_none=True)}
@@ -146,3 +167,22 @@ async def _download(unique_node_name: str, download_link: str) -> None:
             )
 
         return folder
+
+    def get_study_job_custom_metadata(self, study_id: str, job_id: str) -> JobMetadata:
+        _job_metadata = super().get_study_job_custom_metadata(study_id, job_id)
+        return JobMetadata.model_validate_json(_job_metadata.to_json())
+
+    def replace_study_job_custom_metadata(
+        self,
+        study_id: StrictStr,
+        job_id: StrictStr,
+        job_metadata_update: JobMetadataUpdate,
+    ) -> JobMetadata:
+        _job_metadata_update = _JobMetadataUpdate.from_json(
+            job_metadata_update.model_dump_json()
+        )
+        assert _job_metadata_update is not None
+        _job_metadata = super().replace_study_job_custom_metadata(
+            study_id, job_id, _job_metadata_update
+        )
+        return JobMetadata.model_validate_json(_job_metadata.to_json())

clients/python/src/osparc/_configuration.py

@@ -0,0 +1,45 @@
+from osparc_client import Configuration as _Configuration
+from typing import Set
+from urllib3 import Retry
+
+
+class Configuration(_Configuration):
+    def __init__(
+        self,
+        host="https://api.osparc.io",
+        api_key=None,
+        api_key_prefix=None,
+        username=None,
+        password=None,
+        *,
+        retry_max_count: int = 4,
+        retry_methods: Set[str] = {
+            "DELETE",
+            "GET",
+            "HEAD",
+            "OPTIONS",
+            "PUT",
+            "TRACE",
+            "POST",
+            "PATCH",
+            "CONNECT",
+        },
+        retry_status_codes: Set[int] = {429, 503, 504},
+        retry_backoff_factor=4.0,
+    ):
+        retries = Retry(
+            total=retry_max_count,
+            backoff_factor=retry_backoff_factor,
+            status_forcelist=retry_status_codes,
+            allowed_methods=retry_methods,
+            respect_retry_after_header=True,
+            raise_on_status=True,
+        )
+        super().__init__(
+            host=host,
+            api_key=api_key,
+            api_key_prefix=api_key_prefix,
+            username=username,
+            password=password,
+            retries=retries,
+        )

clients/python/src/osparc/_models.py

@@ -0,0 +1,45 @@
+from osparc_client import JobInputs as _JobInputs
+from osparc_client import JobOutputs as _JobOutputs
+from osparc_client import JobMetadata as _JobMetadata
+from osparc_client import JobMetadataUpdate as _JobMetadataUpdate
+from .models import File
+from typing import Dict, Union, List
+from pydantic import BaseModel, StrictStr, Field
+
+
+class JobInputs(BaseModel):
+    values: Dict[str, Union[File, List[object], bool, float, int, str, None]]
+
+    def __init__(
+        self, values: Dict[str, Union[File, List[object], bool, float, int, str, None]]
+    ):
+        super().__init__(values=values)
+
+
+assert set(_JobInputs.model_fields.keys()) == set(JobInputs.model_fields.keys())
+
+
+class JobOutputs(BaseModel):
+    job_id: StrictStr = Field(description="Job that produced this output")
+    results: Dict[str, Union[File, List[object], bool, float, int, str, None]]
+
+
+assert set(_JobOutputs.model_fields.keys()) == set(JobOutputs.model_fields.keys())
+
+
+class JobMetadata(BaseModel):
+    job_id: StrictStr
+    metadata: Dict[str, Union[bool, float, int, str, None]]
+    url: str | None
+
+
+assert set(_JobMetadata.model_fields.keys()) == set(JobMetadata.model_fields.keys())
+
+
+class JobMetadataUpdate(BaseModel):
+    metadata: Dict[str, Union[bool, float, int, str, None]]
+
+
+assert set(_JobMetadataUpdate.model_fields.keys()) == set(
+    JobMetadataUpdate.model_fields.keys()
+)

clients/python/src/osparc/api.py

@@ -4,7 +4,7 @@
 # NOTE: this is an interface. Keep it clean!
 
 
-from osparc_client.configuration import Configuration as Configuration
+from ._configuration import Configuration as Configuration
 from ._api_client import ApiClient as ApiClient
 
 

clients/python/src/osparc/models.py

@@ -12,9 +12,6 @@
 from osparc_client.models.body_complete_multipart_upload_v0_files_file_id_complete_post import (
     BodyCompleteMultipartUploadV0FilesFileIdCompletePost as BodyCompleteMultipartUploadV0FilesFileIdCompletePost,
 )
-from osparc_client.models.body_upload_file_v0_files_content_put import (
-    BodyUploadFileV0FilesContentPut as BodyUploadFileV0FilesContentPut,
-)
 from osparc_client.models.client_file import ClientFile as ClientFile
 from osparc_client.models.client_file_upload_data import (
     ClientFileUploadData as ClientFileUploadData,
@@ -31,13 +28,8 @@
     HTTPValidationError as HTTPValidationError,
 )
 from osparc_client.models.job import Job as Job
-from osparc_client.models.job_inputs import JobInputs as JobInputs
 from osparc_client.models.job_logs_map import JobLogsMap as JobLogsMap
-from osparc_client.models.job_metadata import JobMetadata as JobMetadata
-from osparc_client.models.job_metadata_update import (
-    JobMetadataUpdate as JobMetadataUpdate,
-)
-from osparc_client.models.job_outputs import JobOutputs as JobOutputs
+from osparc_client.models.values_value import ValuesValue as ValuesValue
 from osparc_client.models.job_status import JobStatus as JobStatus
 from osparc_client.models.links import Links as Links
 from osparc_client.models.log_link import LogLink as LogLink
@@ -48,6 +40,7 @@
 from osparc_client.models.one_page_study_port import (
     OnePageStudyPort as OnePageStudyPort,
 )
+from osparc_client import MetadataValue as MetadataValue
 from osparc_client.models.page_file import PageFile as PageFile
 from osparc_client.models.page_job import PageJob as PageJob
 from osparc_client.models.page_study import PageStudy as PageStudy
@@ -75,6 +68,10 @@
 )
 from osparc_client.models.wallet_status import WalletStatus as WalletStatus
 
+from ._models import JobInputs as JobInputs
+from ._models import JobOutputs as JobOutputs
+from ._models import JobMetadata as JobMetadata
+from ._models import JobMetadataUpdate as JobMetadataUpdate
 
 # renames
 TaskStates = _RunningState