UncountablePythonSDK 0.0.75__py3-none-any.whl → 0.0.76__py3-none-any.whl

uncountable/types/api/entity/create_entity.py

@@ -12,6 +12,7 @@ from pkgs.serialization import serial_class
 from ... import base_t
 from ... import entity_t
 from ... import field_values_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",
@@ -43,8 +44,9 @@ class EntityFieldInitialValue:
 )
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
-    definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.CONDITION_PARAMETER], typing.Literal[entity_t.EntityType.OUTPUT]]
+    entity_type: entity_t.LimitedEntityType
+    definition_id: typing.Optional[base_t.ObjectId] = None
+    definition_key: typing.Optional[identifier_t.IdentifierKey] = None
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None
 
 

uncountable/types/api/entity/grant_entity_permissions.py

@@ -0,0 +1,48 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import entity_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/entity/grant_entity_permissions"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.grant_entity_permissions.Arguments",
+)
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    entity_type: entity_t.LimitedEntityType
+    entity_key: identifier_t.IdentifierKey
+    permission_types: list[entity_t.GrantableEntityPermissionType]
+    user_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None
+    user_group_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None
+    all_users: typing.Optional[bool] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.grant_entity_permissions.Data",
+)
+@dataclasses.dataclass(kw_only=True)
+class Data(async_batch_t.AsyncBatchActionReturn):
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/get_column_calculation_values.py

@@ -0,0 +1,58 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import base_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "RecipeColumnCalculationValue",
+]
+
+ENDPOINT_METHOD = "GET"
+ENDPOINT_PATH = "api/external/recipes/get_column_calculation_values"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.Arguments",
+)
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    recipe_keys: list[identifier_t.IdentifierKey]
+    calculation_key: identifier_t.IdentifierKey
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.RecipeColumnCalculationValue",
+    to_string_values={"calculation_quantity"},
+)
+@dataclasses.dataclass(kw_only=True)
+class RecipeColumnCalculationValue:
+    recipe_id: base_t.ObjectId
+    ingredient_id: base_t.ObjectId
+    recipe_step_id: base_t.ObjectId
+    recipe_input_id: base_t.ObjectId
+    calculation_quantity: typing.Optional[Decimal] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.Data",
+)
+@dataclasses.dataclass(kw_only=True)
+class Data:
+    values: list[RecipeColumnCalculationValue]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch_processor.py

@@ -15,7 +15,9 @@ from uncountable.types import base_t
 import uncountable.types.api.recipes.clear_recipe_outputs as clear_recipe_outputs_t
 import uncountable.types.api.recipes.create_recipe as create_recipe_t
 import uncountable.types.api.recipes.edit_recipe_inputs as edit_recipe_inputs_t
+from uncountable.types import entity_t
 from uncountable.types import generic_upload_t
+import uncountable.types.api.entity.grant_entity_permissions as grant_entity_permissions_t
 from uncountable.types import identifier_t
 import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
 from uncountable.types import recipe_identifiers_t
@@ -230,6 +232,47 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def grant_entity_permissions(
+        self,
+        *,
+        entity_type: entity_t.LimitedEntityType,
+        entity_key: identifier_t.IdentifierKey,
+        permission_types: list[entity_t.GrantableEntityPermissionType],
+        user_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        user_group_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        all_users: typing.Optional[bool] = None,
+        depends_on: typing.Optional[list[str]] = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Grant entity permissions to a list of users or user groups or to all users.
+
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = grant_entity_permissions_t.Arguments(
+            entity_type=entity_type,
+            entity_key=entity_key,
+            permission_types=permission_types,
+            user_keys=user_keys,
+            user_group_keys=user_group_keys,
+            all_users=all_users,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.GRANT_ENTITY_PERMISSIONS,
+            data=json_data,
+            depends_on=depends_on,
+            batch_reference=batch_reference,
+        )
+
+        self._enqueue(req)
+
+        return async_batch_t.QueuedAsyncBatchRequest(
+            path=req.path,
+            batch_reference=req.batch_reference,
+        )
+
     def invoke_uploader(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -34,6 +34,7 @@ class AsyncBatchRequestPath(StrEnum):
     INVOKE_UPLOADER = "uploader/invoke_uploader"
     ASSOCIATE_RECIPE_AS_INPUT = "recipes/associate_recipe_as_input"
     CLEAR_RECIPE_OUTPUTS = "recipes/clear_recipe_outputs"
+    GRANT_ENTITY_PERMISSIONS = "entity/grant_entity_permissions"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/auth_retrieval.py

@@ -0,0 +1,12 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .auth_retrieval_t import AuthRetrievalType as AuthRetrievalType
+from .auth_retrieval_t import AuthRetrievalBase as AuthRetrievalBase
+from .auth_retrieval_t import AuthRetrievalOAuth as AuthRetrievalOAuth
+from .auth_retrieval_t import AuthRetrievalBasic as AuthRetrievalBasic
+from .auth_retrieval_t import AuthRetrieval as AuthRetrieval
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/auth_retrieval_t.py

@@ -0,0 +1,75 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from . import secret_retrieval_t
+
+__all__: list[str] = [
+    "AuthRetrieval",
+    "AuthRetrievalBase",
+    "AuthRetrievalBasic",
+    "AuthRetrievalOAuth",
+    "AuthRetrievalType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class AuthRetrievalType(StrEnum):
+    OAUTH = "oauth"
+    BASIC = "basic"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalBase",
+)
+@dataclasses.dataclass(kw_only=True)
+class AuthRetrievalBase:
+    type: AuthRetrievalType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalOAuth",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class AuthRetrievalOAuth(AuthRetrievalBase):
+    type: typing.Literal[AuthRetrievalType.OAUTH] = AuthRetrievalType.OAUTH
+    refresh_token_secret: secret_retrieval_t.SecretRetrieval
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalBasic",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class AuthRetrievalBasic(AuthRetrievalBase):
+    type: typing.Literal[AuthRetrievalType.BASIC] = AuthRetrievalType.BASIC
+    api_id_secret: secret_retrieval_t.SecretRetrieval
+    api_key_secret: secret_retrieval_t.SecretRetrieval
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+AuthRetrieval = typing.Annotated[
+    typing.Union[AuthRetrievalOAuth, AuthRetrievalBasic],
+    serial_union_annotation(
+    named_type_path="sdk.auth_retrieval.AuthRetrieval",
+        discriminator="type",
+        discriminator_map={
+            "oauth": AuthRetrievalOAuth,
+            "basic": AuthRetrievalBasic,
+        },
+    ),
+]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -31,6 +31,7 @@ import uncountable.types.api.batch.execute_batch as execute_batch_t
 import uncountable.types.api.batch.execute_batch_load_async as execute_batch_load_async_t
 from uncountable.types import field_values_t
 from uncountable.types import generic_upload_t
+import uncountable.types.api.recipes.get_column_calculation_values as get_column_calculation_values_t
 import uncountable.types.api.recipes.get_curve as get_curve_t
 import uncountable.types.api.entity.get_entities_data as get_entities_data_t
 import uncountable.types.api.inputs.get_input_data as get_input_data_t
@@ -47,6 +48,7 @@ import uncountable.types.api.recipe_metadata.get_recipe_metadata_data as get_rec
 import uncountable.types.api.recipes.get_recipe_names as get_recipe_names_t
 import uncountable.types.api.recipes.get_recipe_output_metadata as get_recipe_output_metadata_t
 import uncountable.types.api.recipes.get_recipes_data as get_recipes_data_t
+import uncountable.types.api.entity.grant_entity_permissions as grant_entity_permissions_t
 from uncountable.types import id_source_t
 from uncountable.types import identifier_t
 import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
@@ -256,18 +258,21 @@ class ClientMethods(ABC):
     def create_entities(
         self,
         *,
-        definition_id: base_t.ObjectId,
-        entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.CONDITION_PARAMETER], typing.Literal[entity_t.EntityType.OUTPUT]],
+        entity_type: entity_t.LimitedEntityType,
         entities_to_create: list[create_entities_t.EntityToCreate],
+        definition_id: typing.Optional[base_t.ObjectId] = None,
+        definition_key: typing.Optional[identifier_t.IdentifierKey] = None,
     ) -> create_entities_t.Data:
         """Creates new Uncountable entities
 
         :param definition_id: Definition id for the entities to create
+        :param definition_key: Key for the definition of the entities to create
         :param entity_type: The type of the entities to create
         :param entities_to_create: A list of the entities to create
         """
         args = create_entities_t.Arguments(
             definition_id=definition_id,
+            definition_key=definition_key,
             entity_type=entity_type,
             entities_to_create=entities_to_create,
         )
@@ -281,17 +286,20 @@ class ClientMethods(ABC):
     def create_entity(
         self,
         *,
-        definition_id: base_t.ObjectId,
-        entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.CONDITION_PARAMETER], typing.Literal[entity_t.EntityType.OUTPUT]],
+        entity_type: entity_t.LimitedEntityType,
+        definition_id: typing.Optional[base_t.ObjectId] = None,
+        definition_key: typing.Optional[identifier_t.IdentifierKey] = None,
         field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None,
     ) -> create_entity_t.Data:
         """Creates a new Uncountable entity
 
         :param definition_id: Definition id of the entity to create
+        :param definition_key: Key for the definition of the entity to create
         :param entity_type: The type of the entities requested
         """
         args = create_entity_t.Arguments(
             definition_id=definition_id,
+            definition_key=definition_key,
             entity_type=entity_type,
             field_values=field_values,
         )
@@ -491,6 +499,28 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=execute_batch_load_async_t.Data)
 
+    def get_column_calculation_values(
+        self,
+        *,
+        recipe_keys: list[identifier_t.IdentifierKey],
+        calculation_key: identifier_t.IdentifierKey,
+    ) -> get_column_calculation_values_t.Data:
+        """Get the values for a column calculations for recipes
+
+        :param recipe_keys: Required: The recipe keys to load values for
+        :param calculation_key: Required: The calculation key to get the values for
+        """
+        args = get_column_calculation_values_t.Arguments(
+            recipe_keys=recipe_keys,
+            calculation_key=calculation_key,
+        )
+        api_request = APIRequest(
+            method=get_column_calculation_values_t.ENDPOINT_METHOD,
+            endpoint=get_column_calculation_values_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=get_column_calculation_values_t.Data)
+
     def get_curve(
         self,
         *,
@@ -873,6 +903,34 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=get_recipes_data_t.Data)
 
+    def grant_entity_permissions(
+        self,
+        *,
+        entity_type: entity_t.LimitedEntityType,
+        entity_key: identifier_t.IdentifierKey,
+        permission_types: list[entity_t.GrantableEntityPermissionType],
+        user_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        user_group_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        all_users: typing.Optional[bool] = None,
+    ) -> grant_entity_permissions_t.Data:
+        """Grant entity permissions to a list of users or user groups or to all users.
+
+        """
+        args = grant_entity_permissions_t.Arguments(
+            entity_type=entity_type,
+            entity_key=entity_key,
+            permission_types=permission_types,
+            user_keys=user_keys,
+            user_group_keys=user_group_keys,
+            all_users=all_users,
+        )
+        api_request = APIRequest(
+            method=grant_entity_permissions_t.ENDPOINT_METHOD,
+            endpoint=grant_entity_permissions_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=grant_entity_permissions_t.Data)
+
     def invoke_uploader(
         self,
         *,

uncountable/types/entity.py

@@ -5,5 +5,8 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # Kept only for SDK backwards compatibility
 from .entity_t import EntityType as EntityType
+from .entity_t import LimitedEntityType as LimitedEntityType
+from .entity_t import EntityPermissionType as EntityPermissionType
+from .entity_t import GrantableEntityPermissionType as GrantableEntityPermissionType
 from .entity_t import Entity as Entity
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/entity_t.py

@@ -15,7 +15,10 @@ from . import base_t
 
 __all__: list[str] = [
     "Entity",
+    "EntityPermissionType",
     "EntityType",
+    "GrantableEntityPermissionType",
+    "LimitedEntityType",
 ]
 
 
@@ -331,6 +334,23 @@ class EntityType(StrEnum):
     UPLOADER = "uploader"
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+LimitedEntityType = typing.Union[typing.Literal[EntityType.LAB_REQUEST], typing.Literal[EntityType.APPROVAL], typing.Literal[EntityType.CUSTOM_ENTITY], typing.Literal[EntityType.INVENTORY_AMOUNT], typing.Literal[EntityType.TASK], typing.Literal[EntityType.PROJECT], typing.Literal[EntityType.EQUIPMENT], typing.Literal[EntityType.INV_LOCAL_LOCATIONS], typing.Literal[EntityType.FIELD_OPTION_SET], typing.Literal[EntityType.WEBHOOK], typing.Literal[EntityType.SPECS], typing.Literal[EntityType.GOAL], typing.Literal[EntityType.INGREDIENT_TAG_MAP], typing.Literal[EntityType.INGREDIENT_TAG], typing.Literal[EntityType.CONDITION_PARAMETER], typing.Literal[EntityType.OUTPUT]]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class EntityPermissionType(StrEnum):
+    READ = "read"
+    WRITE = "write"
+    CREATE = "create"
+    REOPEN = "reopen"
+    ADMIN = "admin"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+GrantableEntityPermissionType = typing.Union[typing.Literal[EntityPermissionType.READ], typing.Literal[EntityPermissionType.WRITE]]
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.entity.Entity",

uncountable/types/integration_server.py

@@ -0,0 +1,9 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .integration_server_t import IntegrationEnvironment as IntegrationEnvironment
+from .integration_server_t import EnvironmentConfig as EnvironmentConfig
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/integration_server_t.py

@@ -0,0 +1,37 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from . import auth_retrieval_t
+from . import client_config_t
+
+__all__: list[str] = [
+    "EnvironmentConfig",
+    "IntegrationEnvironment",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class IntegrationEnvironment(StrEnum):
+    LOCAL = "local"
+    PROD = "prod"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.integration_server.EnvironmentConfig",
+)
+@dataclasses.dataclass(kw_only=True)
+class EnvironmentConfig:
+    auth_retrieval: auth_retrieval_t.AuthRetrieval
+    base_url: str
+    client_options: typing.Optional[client_config_t.ClientConfigOptions] = None
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/job_definition.py

@@ -6,7 +6,6 @@
 # Kept only for SDK backwards compatibility
 from .job_definition_t import JobDefinitionType as JobDefinitionType
 from .job_definition_t import JobExecutorType as JobExecutorType
-from .job_definition_t import AuthRetrievalType as AuthRetrievalType
 from .job_definition_t import GenericUploadDataSourceType as GenericUploadDataSourceType
 from .job_definition_t import JobExecutorBase as JobExecutorBase
 from .job_definition_t import JobExecutorScript as JobExecutorScript
@@ -17,14 +16,11 @@ from .job_definition_t import GenericUploadDataSourceS3 as GenericUploadDataSour
 from .job_definition_t import GenericUploadDataSource as GenericUploadDataSource
 from .job_definition_t import JobExecutorGenericUpload as JobExecutorGenericUpload
 from .job_definition_t import JobExecutor as JobExecutor
+from .job_definition_t import JobLoggingSettings as JobLoggingSettings
 from .job_definition_t import JobDefinitionBase as JobDefinitionBase
 from .job_definition_t import CronJobDefinition as CronJobDefinition
 from .job_definition_t import WebhookJobDefinition as WebhookJobDefinition
 from .job_definition_t import JobDefinition as JobDefinition
-from .job_definition_t import AuthRetrievalBase as AuthRetrievalBase
-from .job_definition_t import AuthRetrievalOAuth as AuthRetrievalOAuth
-from .job_definition_t import AuthRetrievalBasic as AuthRetrievalBasic
-from .job_definition_t import AuthRetrieval as AuthRetrieval
 from .job_definition_t import ProfileDefinition as ProfileDefinition
 from .job_definition_t import ProfileMetadata as ProfileMetadata
 from .job_definition_t import JobResult as JobResult

uncountable/types/job_definition_t.py

@@ -11,16 +11,14 @@ from pkgs.strenum_compat import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
+from . import auth_retrieval_t
+from . import base_t
 from . import client_config_t
 from . import generic_upload_t
+from . import integration_server_t
 from . import secret_retrieval_t
 
 __all__: list[str] = [
-    "AuthRetrieval",
-    "AuthRetrievalBase",
-    "AuthRetrievalBasic",
-    "AuthRetrievalOAuth",
-    "AuthRetrievalType",
     "CronJobDefinition",
     "GenericUploadDataSource",
     "GenericUploadDataSourceBase",
@@ -35,6 +33,7 @@ __all__: list[str] = [
     "JobExecutorGenericUpload",
     "JobExecutorScript",
     "JobExecutorType",
+    "JobLoggingSettings",
     "JobResult",
     "ProfileDefinition",
     "ProfileMetadata",
@@ -55,12 +54,6 @@ class JobExecutorType(StrEnum):
     GENERIC_UPLOAD = "generic_upload"
 
 
-# DO NOT MODIFY -- This file is generated by type_spec
-class AuthRetrievalType(StrEnum):
-    OAUTH = "oauth"
-    BASIC = "basic"
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
 class GenericUploadDataSourceType(StrEnum):
     SFTP = "sftp"
@@ -172,6 +165,16 @@ JobExecutor = typing.Annotated[
 ]
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.job_definition.JobLoggingSettings",
+)
+@dataclasses.dataclass(kw_only=True)
+class JobLoggingSettings:
+    enabled: bool = False
+    share_with_user_groups: typing.Optional[list[base_t.ObjectId]] = None
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.job_definition.JobDefinitionBase",
@@ -182,6 +185,7 @@ class JobDefinitionBase:
     name: str
     executor: JobExecutor
     enabled: bool = True
+    logging_settings: typing.Optional[JobLoggingSettings] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -220,62 +224,17 @@ JobDefinition = typing.Annotated[
 ]
 
 
-# DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    named_type_path="sdk.job_definition.AuthRetrievalBase",
-)
-@dataclasses.dataclass(kw_only=True)
-class AuthRetrievalBase:
-    type: AuthRetrievalType
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    named_type_path="sdk.job_definition.AuthRetrievalOAuth",
-    parse_require={"type"},
-)
-@dataclasses.dataclass(kw_only=True)
-class AuthRetrievalOAuth(AuthRetrievalBase):
-    type: typing.Literal[AuthRetrievalType.OAUTH] = AuthRetrievalType.OAUTH
-    refresh_token_secret: secret_retrieval_t.SecretRetrieval
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    named_type_path="sdk.job_definition.AuthRetrievalBasic",
-    parse_require={"type"},
-)
-@dataclasses.dataclass(kw_only=True)
-class AuthRetrievalBasic(AuthRetrievalBase):
-    type: typing.Literal[AuthRetrievalType.BASIC] = AuthRetrievalType.BASIC
-    api_id_secret: secret_retrieval_t.SecretRetrieval
-    api_key_secret: secret_retrieval_t.SecretRetrieval
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-AuthRetrieval = typing.Annotated[
-    typing.Union[AuthRetrievalOAuth, AuthRetrievalBasic],
-    serial_union_annotation(
-    named_type_path="sdk.job_definition.AuthRetrieval",
-        discriminator="type",
-        discriminator_map={
-            "oauth": AuthRetrievalOAuth,
-            "basic": AuthRetrievalBasic,
-        },
-    ),
-]
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.job_definition.ProfileDefinition",
 )
 @dataclasses.dataclass(kw_only=True)
 class ProfileDefinition:
-    auth_retrieval: AuthRetrieval
-    base_url: str
     jobs: list[JobDefinition]
+    auth_retrieval: typing.Optional[auth_retrieval_t.AuthRetrieval] = None
+    base_url: typing.Optional[str] = None
     client_options: typing.Optional[client_config_t.ClientConfigOptions] = None
+    environments: typing.Optional[dict[integration_server_t.IntegrationEnvironment, integration_server_t.EnvironmentConfig]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -286,8 +245,9 @@ class ProfileDefinition:
 class ProfileMetadata:
     name: str
     base_url: str
-    auth_retrieval: AuthRetrieval
+    auth_retrieval: auth_retrieval_t.AuthRetrieval
     client_options: typing.Optional[client_config_t.ClientConfigOptions]
+    jobs: list[JobDefinition]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec