cosmotech-api 3.2.6.dev0__py3-none-any.whl → 3.3.4__py3-none-any.whl

cosmotech_api/__init__.py

@@ -7,7 +7,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    The version of the OpenAPI document: 3.3.4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -23,7 +23,10 @@ from cosmotech_api.api.dataset_api import DatasetApi
 from cosmotech_api.api.organization_api import OrganizationApi
 from cosmotech_api.api.run_api import RunApi
 from cosmotech_api.api.runner_api import RunnerApi
+from cosmotech_api.api.scenario_api import ScenarioApi
+from cosmotech_api.api.scenariorun_api import ScenariorunApi
 from cosmotech_api.api.solution_api import SolutionApi
+from cosmotech_api.api.twingraph_api import TwingraphApi
 from cosmotech_api.api.workspace_api import WorkspaceApi
 
 # import ApiClient
@@ -44,7 +47,6 @@ from cosmotech_api.models.connector_parameter import ConnectorParameter
 from cosmotech_api.models.connector_parameter_group import ConnectorParameterGroup
 from cosmotech_api.models.container_resource_size_info import ContainerResourceSizeInfo
 from cosmotech_api.models.container_resource_sizing import ContainerResourceSizing
-from cosmotech_api.models.created_run import CreatedRun
 from cosmotech_api.models.dataset import Dataset
 from cosmotech_api.models.dataset_access_control import DatasetAccessControl
 from cosmotech_api.models.dataset_compatibility import DatasetCompatibility
@@ -67,19 +69,26 @@ from cosmotech_api.models.organization import Organization
 from cosmotech_api.models.organization_access_control import OrganizationAccessControl
 from cosmotech_api.models.organization_role import OrganizationRole
 from cosmotech_api.models.organization_security import OrganizationSecurity
+from cosmotech_api.models.organization_service import OrganizationService
+from cosmotech_api.models.organization_services import OrganizationServices
 from cosmotech_api.models.query_result import QueryResult
 from cosmotech_api.models.resource_size_info import ResourceSizeInfo
 from cosmotech_api.models.run import Run
 from cosmotech_api.models.run_container import RunContainer
+from cosmotech_api.models.run_container_artifact import RunContainerArtifact
+from cosmotech_api.models.run_container_logs import RunContainerLogs
 from cosmotech_api.models.run_data import RunData
 from cosmotech_api.models.run_data_query import RunDataQuery
 from cosmotech_api.models.run_logs import RunLogs
-from cosmotech_api.models.run_logs_entry import RunLogsEntry
 from cosmotech_api.models.run_resource_requested import RunResourceRequested
+from cosmotech_api.models.run_search import RunSearch
+from cosmotech_api.models.run_search_state import RunSearchState
+from cosmotech_api.models.run_start_containers import RunStartContainers
 from cosmotech_api.models.run_state import RunState
 from cosmotech_api.models.run_status import RunStatus
 from cosmotech_api.models.run_status_node import RunStatusNode
 from cosmotech_api.models.run_template import RunTemplate
+from cosmotech_api.models.run_template_handler_id import RunTemplateHandlerId
 from cosmotech_api.models.run_template_orchestrator import RunTemplateOrchestrator
 from cosmotech_api.models.run_template_parameter import RunTemplateParameter
 from cosmotech_api.models.run_template_parameter_group import RunTemplateParameterGroup
@@ -88,11 +97,44 @@ from cosmotech_api.models.run_template_resource_sizing import RunTemplateResourc
 from cosmotech_api.models.run_template_step_source import RunTemplateStepSource
 from cosmotech_api.models.runner import Runner
 from cosmotech_api.models.runner_access_control import RunnerAccessControl
+from cosmotech_api.models.runner_changed_parameter_value import RunnerChangedParameterValue
+from cosmotech_api.models.runner_comparison_result import RunnerComparisonResult
+from cosmotech_api.models.runner_data_download_info import RunnerDataDownloadInfo
+from cosmotech_api.models.runner_data_download_job import RunnerDataDownloadJob
+from cosmotech_api.models.runner_job_state import RunnerJobState
+from cosmotech_api.models.runner_last_run import RunnerLastRun
+from cosmotech_api.models.runner_parent_last_run import RunnerParentLastRun
 from cosmotech_api.models.runner_resource_sizing import RunnerResourceSizing
 from cosmotech_api.models.runner_role import RunnerRole
+from cosmotech_api.models.runner_root_last_run import RunnerRootLastRun
 from cosmotech_api.models.runner_run_template_parameter_value import RunnerRunTemplateParameterValue
 from cosmotech_api.models.runner_security import RunnerSecurity
 from cosmotech_api.models.runner_validation_status import RunnerValidationStatus
+from cosmotech_api.models.scenario import Scenario
+from cosmotech_api.models.scenario_access_control import ScenarioAccessControl
+from cosmotech_api.models.scenario_changed_parameter_value import ScenarioChangedParameterValue
+from cosmotech_api.models.scenario_comparison_result import ScenarioComparisonResult
+from cosmotech_api.models.scenario_data_download_info import ScenarioDataDownloadInfo
+from cosmotech_api.models.scenario_data_download_job import ScenarioDataDownloadJob
+from cosmotech_api.models.scenario_job_state import ScenarioJobState
+from cosmotech_api.models.scenario_last_run import ScenarioLastRun
+from cosmotech_api.models.scenario_resource_sizing import ScenarioResourceSizing
+from cosmotech_api.models.scenario_role import ScenarioRole
+from cosmotech_api.models.scenario_run import ScenarioRun
+from cosmotech_api.models.scenario_run_container import ScenarioRunContainer
+from cosmotech_api.models.scenario_run_container_artifact import ScenarioRunContainerArtifact
+from cosmotech_api.models.scenario_run_container_logs import ScenarioRunContainerLogs
+from cosmotech_api.models.scenario_run_logs import ScenarioRunLogs
+from cosmotech_api.models.scenario_run_resource_requested import ScenarioRunResourceRequested
+from cosmotech_api.models.scenario_run_search import ScenarioRunSearch
+from cosmotech_api.models.scenario_run_search_state import ScenarioRunSearchState
+from cosmotech_api.models.scenario_run_start_containers import ScenarioRunStartContainers
+from cosmotech_api.models.scenario_run_state import ScenarioRunState
+from cosmotech_api.models.scenario_run_status import ScenarioRunStatus
+from cosmotech_api.models.scenario_run_status_node import ScenarioRunStatusNode
+from cosmotech_api.models.scenario_run_template_parameter_value import ScenarioRunTemplateParameterValue
+from cosmotech_api.models.scenario_security import ScenarioSecurity
+from cosmotech_api.models.scenario_validation_status import ScenarioValidationStatus
 from cosmotech_api.models.send_run_data_request import SendRunDataRequest
 from cosmotech_api.models.solution import Solution
 from cosmotech_api.models.solution_access_control import SolutionAccessControl
@@ -101,11 +143,15 @@ from cosmotech_api.models.solution_security import SolutionSecurity
 from cosmotech_api.models.source_info import SourceInfo
 from cosmotech_api.models.sub_dataset_graph_query import SubDatasetGraphQuery
 from cosmotech_api.models.twin_graph_batch_result import TwinGraphBatchResult
+from cosmotech_api.models.twin_graph_hash import TwinGraphHash
+from cosmotech_api.models.twin_graph_query import TwinGraphQuery
 from cosmotech_api.models.twincache_status_enum import TwincacheStatusEnum
+from cosmotech_api.models.validator import Validator
 from cosmotech_api.models.workspace import Workspace
 from cosmotech_api.models.workspace_access_control import WorkspaceAccessControl
 from cosmotech_api.models.workspace_file import WorkspaceFile
 from cosmotech_api.models.workspace_role import WorkspaceRole
+from cosmotech_api.models.workspace_secret import WorkspaceSecret
 from cosmotech_api.models.workspace_security import WorkspaceSecurity
 from cosmotech_api.models.workspace_solution import WorkspaceSolution
 from cosmotech_api.models.workspace_web_app import WorkspaceWebApp

cosmotech_api/api/__init__.py

@@ -6,6 +6,9 @@ from cosmotech_api.api.dataset_api import DatasetApi
 from cosmotech_api.api.organization_api import OrganizationApi
 from cosmotech_api.api.run_api import RunApi
 from cosmotech_api.api.runner_api import RunnerApi
+from cosmotech_api.api.scenario_api import ScenarioApi
+from cosmotech_api.api.scenariorun_api import ScenariorunApi
 from cosmotech_api.api.solution_api import SolutionApi
+from cosmotech_api.api.twingraph_api import TwingraphApi
 from cosmotech_api.api.workspace_api import WorkspaceApi
 

cosmotech_api/api/connector_api.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    The version of the OpenAPI document: 3.3.4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
 from pydantic import Field, StrictInt, StrictStr
-from typing import Optional
+from typing import List, Optional
 from typing_extensions import Annotated
 from cosmotech_api.models.connector import Connector
 
@@ -43,7 +43,7 @@ class ConnectorApi:
     @validate_call
     def find_all_connectors(
         self,
-        page: Annotated[Optional[StrictInt], Field(description="page number to query (first page is at index 0)")] = None,
+        page: Annotated[Optional[StrictInt], Field(description="page number to query")] = None,
         size: Annotated[Optional[StrictInt], Field(description="amount of result by page")] = None,
         _request_timeout: Union[
             None,
@@ -61,7 +61,7 @@ class ConnectorApi:
         """List all Connectors
 
 
-        :param page: page number to query (first page is at index 0)
+        :param page: page number to query
         :type page: int
         :param size: amount of result by page
         :type size: int
@@ -113,7 +113,7 @@ class ConnectorApi:
     @validate_call
     def find_all_connectors_with_http_info(
         self,
-        page: Annotated[Optional[StrictInt], Field(description="page number to query (first page is at index 0)")] = None,
+        page: Annotated[Optional[StrictInt], Field(description="page number to query")] = None,
         size: Annotated[Optional[StrictInt], Field(description="amount of result by page")] = None,
         _request_timeout: Union[
             None,
@@ -131,7 +131,7 @@ class ConnectorApi:
         """List all Connectors
 
 
-        :param page: page number to query (first page is at index 0)
+        :param page: page number to query
         :type page: int
         :param size: amount of result by page
         :type size: int
@@ -183,7 +183,7 @@ class ConnectorApi:
     @validate_call
     def find_all_connectors_without_preload_content(
         self,
-        page: Annotated[Optional[StrictInt], Field(description="page number to query (first page is at index 0)")] = None,
+        page: Annotated[Optional[StrictInt], Field(description="page number to query")] = None,
         size: Annotated[Optional[StrictInt], Field(description="amount of result by page")] = None,
         _request_timeout: Union[
             None,
@@ -201,7 +201,7 @@ class ConnectorApi:
         """List all Connectors
 
 
-        :param page: page number to query (first page is at index 0)
+        :param page: page number to query
         :type page: int
         :param size: amount of result by page
         :type size: int
@@ -265,7 +265,7 @@ class ConnectorApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, str] = {}
+        _files: Dict[str, Union[str, bytes]] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -284,11 +284,12 @@ class ConnectorApi:
 
 
         # set the HTTP header `Accept`
-        _header_params['Accept'] = self.api_client.select_header_accept(
-            [
-                'application/json'
-            ]
-        )
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
 
 
         # authentication setting
@@ -529,7 +530,7 @@ class ConnectorApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, str] = {}
+        _files: Dict[str, Union[str, bytes]] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -542,11 +543,268 @@ class ConnectorApi:
 
 
         # set the HTTP header `Accept`
-        _header_params['Accept'] = self.api_client.select_header_accept(
-            [
-                'application/json'
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'oAuth2AuthCode'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/connectors/{connector_id}',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
+    @validate_call
+    def find_connector_by_name(
+        self,
+        connector_name: Annotated[StrictStr, Field(description="the Connector name")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> Connector:
+        """Get the details of a connector
+
+
+        :param connector_name: the Connector name (required)
+        :type connector_name: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._find_connector_by_name_serialize(
+            connector_name=connector_name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "Connector",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def find_connector_by_name_with_http_info(
+        self,
+        connector_name: Annotated[StrictStr, Field(description="the Connector name")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[Connector]:
+        """Get the details of a connector
+
+
+        :param connector_name: the Connector name (required)
+        :type connector_name: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._find_connector_by_name_serialize(
+            connector_name=connector_name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "Connector",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def find_connector_by_name_without_preload_content(
+        self,
+        connector_name: Annotated[StrictStr, Field(description="the Connector name")],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
             ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Get the details of a connector
+
+
+        :param connector_name: the Connector name (required)
+        :type connector_name: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._find_connector_by_name_serialize(
+            connector_name=connector_name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "Connector",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
         )
+        return response_data.response
+
+
+    def _find_connector_by_name_serialize(
+        self,
+        connector_name,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[str, Union[str, bytes]] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if connector_name is not None:
+            _path_params['connector_name'] = connector_name
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
 
 
         # authentication setting
@@ -556,7 +814,7 @@ class ConnectorApi:
 
         return self.api_client.param_serialize(
             method='GET',
-            resource_path='/connectors/{connector_id}',
+            resource_path='/connectors/name/{connector_name}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -787,7 +1045,7 @@ class ConnectorApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, str] = {}
+        _files: Dict[str, Union[str, bytes]] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -800,11 +1058,12 @@ class ConnectorApi:
 
 
         # set the HTTP header `Accept`
-        _header_params['Accept'] = self.api_client.select_header_accept(
-            [
-                'application/json'
-            ]
-        )
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
 
         # set the HTTP header `Content-Type`
         if _content_type:
@@ -1059,7 +1318,7 @@ class ConnectorApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, str] = {}
+        _files: Dict[str, Union[str, bytes]] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters