superposition-sdk 0.86.0__py3-none-any.whl → 0.87.0__py3-none-any.whl

superposition_sdk/_private/schemas.py

@@ -12253,6 +12253,197 @@ LIST_WORKSPACE = Schema(
 
 )
 
+MIGRATE_WORKSPACE_SCHEMA_INPUT = Schema.collection(
+    id=ShapeID("io.superposition#MigrateWorkspaceSchemaInput"),
+
+    traits=[
+        Trait.new(id=ShapeID("smithy.synthetic#originalShapeId"), value="io.superposition#MigrateWorkspaceSchemaRequest"),
+        Trait.new(id=ShapeID("smithy.api#input")),
+
+    ],
+    members={
+        "org_id": {
+            "target": STRING,
+            "index": 0,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#default"), value="juspay"),
+                Trait.new(id=ShapeID("smithy.api#httpHeader"), value="x-org-id"),
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "workspace_name": {
+            "target": STRING,
+            "index": 1,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+                Trait.new(id=ShapeID("smithy.api#httpLabel")),
+
+            ],
+        },
+
+    }
+)
+
+MIGRATE_WORKSPACE_SCHEMA_OUTPUT = Schema.collection(
+    id=ShapeID("io.superposition#MigrateWorkspaceSchemaOutput"),
+
+    traits=[
+        Trait.new(id=ShapeID("smithy.synthetic#originalShapeId"), value="io.superposition#WorkspaceResponse"),
+        Trait.new(id=ShapeID("smithy.api#output")),
+
+    ],
+    members={
+        "workspace_name": {
+            "target": STRING,
+            "index": 0,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "organisation_id": {
+            "target": STRING,
+            "index": 1,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "organisation_name": {
+            "target": STRING,
+            "index": 2,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "workspace_schema_name": {
+            "target": STRING,
+            "index": 3,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "workspace_status": {
+            "target": WORKSPACE_STATUS,
+            "index": 4,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "workspace_admin_email": {
+            "target": STRING,
+            "index": 5,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "config_version": {
+            "target": STRING,
+            "index": 6,
+        },
+
+        "created_by": {
+            "target": STRING,
+            "index": 7,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "last_modified_by": {
+            "target": STRING,
+            "index": 8,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "last_modified_at": {
+            "target": DATE_TIME,
+            "index": 9,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "created_at": {
+            "target": DATE_TIME,
+            "index": 10,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "mandatory_dimensions": {
+            "target": LIST_MANDATORY_DIMENSIONS,
+            "index": 11,
+        },
+
+        "strict_mode": {
+            "target": BOOLEAN,
+            "index": 12,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "metrics": {
+            "target": DOCUMENT,
+            "index": 13,
+        },
+
+        "allow_experiment_self_approval": {
+            "target": BOOLEAN,
+            "index": 14,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+        "auto_populate_control": {
+            "target": BOOLEAN,
+            "index": 15,
+            "traits": [
+                Trait.new(id=ShapeID("smithy.api#required")),
+
+            ],
+        },
+
+    }
+)
+
+MIGRATE_WORKSPACE_SCHEMA = Schema(
+    id=ShapeID("io.superposition#MigrateWorkspaceSchema"),
+    shape_type=ShapeType.OPERATION,
+    traits=[
+        Trait.new(id=ShapeID("smithy.api#http"), value=MappingProxyType({
+                "method": "POST",
+                "uri": "/workspaces/{workspace_name}/db/migrate",
+            })),
+        Trait.new(id=ShapeID("smithy.api#readonly")),
+
+    ],
+
+)
+
 UPDATE_ORGANISATION_INPUT = Schema.collection(
     id=ShapeID("io.superposition#UpdateOrganisationInput"),
 

superposition_sdk/client.py

@@ -78,6 +78,7 @@ from .deserialize import (
     _deserialize_list_versions,
     _deserialize_list_webhook,
     _deserialize_list_workspace,
+    _deserialize_migrate_workspace_schema,
     _deserialize_move_context,
     _deserialize_pause_experiment,
     _deserialize_publish,
@@ -230,7 +231,10 @@ from .models import (
     ListWebhookOutput,
     ListWorkspaceInput,
     ListWorkspaceOutput,
+    MIGRATE_WORKSPACE_SCHEMA,
     MOVE_CONTEXT,
+    MigrateWorkspaceSchemaInput,
+    MigrateWorkspaceSchemaOutput,
     MoveContextInput,
     MoveContextOutput,
     PAUSE_EXPERIMENT,
@@ -331,6 +335,7 @@ from .serialize import (
     _serialize_list_versions,
     _serialize_list_webhook,
     _serialize_list_workspace,
+    _serialize_migrate_workspace_schema,
     _serialize_move_context,
     _serialize_pause_experiment,
     _serialize_publish,
@@ -1521,6 +1526,32 @@ class Superposition:
             operation=LIST_WORKSPACE,
         )
 
+    async def migrate_workspace_schema(self, input: MigrateWorkspaceSchemaInput, plugins: list[Plugin] | None = None) -> MigrateWorkspaceSchemaOutput:
+        """
+        Invokes the MigrateWorkspaceSchema operation.
+
+        :param input: The operation's input.
+
+        :param plugins: A list of callables that modify the configuration dynamically.
+            Changes made by these plugins only apply for the duration of the operation
+            execution and will not affect any other operation invocations.
+
+        """
+        operation_plugins: list[Plugin] = [
+
+        ]
+        if plugins:
+            operation_plugins.extend(plugins)
+
+        return await self._execute_operation(
+            input=input,
+            plugins=operation_plugins,
+            serialize=_serialize_migrate_workspace_schema,
+            deserialize=_deserialize_migrate_workspace_schema,
+            config=self._config,
+            operation=MIGRATE_WORKSPACE_SCHEMA,
+        )
+
     async def move_context(self, input: MoveContextInput, plugins: list[Plugin] | None = None) -> MoveContextOutput:
         """
         Invokes the MoveContext operation.

superposition_sdk/config.py

@@ -104,6 +104,8 @@ from .models import (
     ListWebhookOutput,
     ListWorkspaceInput,
     ListWorkspaceOutput,
+    MigrateWorkspaceSchemaInput,
+    MigrateWorkspaceSchemaOutput,
     MoveContextInput,
     MoveContextOutput,
     PauseExperimentInput,
@@ -143,7 +145,7 @@ from .models import (
 )
 
 
-_ServiceInterceptor = Union[Interceptor[AddMembersToGroupInput, AddMembersToGroupOutput, Any, Any], Interceptor[ApplicableVariantsInput, ApplicableVariantsOutput, Any, Any], Interceptor[BulkOperationInput, BulkOperationOutput, Any, Any], Interceptor[ConcludeExperimentInput, ConcludeExperimentOutput, Any, Any], Interceptor[CreateContextInput, CreateContextOutput, Any, Any], Interceptor[CreateDefaultConfigInput, CreateDefaultConfigOutput, Any, Any], Interceptor[CreateDimensionInput, CreateDimensionOutput, Any, Any], Interceptor[CreateExperimentInput, CreateExperimentOutput, Any, Any], Interceptor[CreateExperimentGroupInput, CreateExperimentGroupOutput, Any, Any], Interceptor[CreateFunctionInput, CreateFunctionOutput, Any, Any], Interceptor[CreateOrganisationInput, CreateOrganisationOutput, Any, Any], Interceptor[CreateTypeTemplatesInput, CreateTypeTemplatesOutput, Any, Any], Interceptor[CreateWebhookInput, CreateWebhookOutput, Any, Any], Interceptor[CreateWorkspaceInput, CreateWorkspaceOutput, Any, Any], Interceptor[DeleteContextInput, DeleteContextOutput, Any, Any], Interceptor[DeleteDefaultConfigInput, DeleteDefaultConfigOutput, Any, Any], Interceptor[DeleteDimensionInput, DeleteDimensionOutput, Any, Any], Interceptor[DeleteExperimentGroupInput, DeleteExperimentGroupOutput, Any, Any], Interceptor[DeleteFunctionInput, DeleteFunctionOutput, Any, Any], Interceptor[DeleteTypeTemplatesInput, DeleteTypeTemplatesOutput, Any, Any], Interceptor[DiscardExperimentInput, DiscardExperimentOutput, Any, Any], Interceptor[GetConfigInput, GetConfigOutput, Any, Any], Interceptor[GetConfigFastInput, GetConfigFastOutput, Any, Any], Interceptor[GetContextInput, GetContextOutput, Any, Any], Interceptor[GetContextFromConditionInput, GetContextFromConditionOutput, Any, Any], Interceptor[GetDimensionInput, GetDimensionOutput, Any, Any], Interceptor[GetExperimentInput, GetExperimentOutput, Any, Any], Interceptor[GetExperimentGroupInput, GetExperimentGroupOutput, Any, Any], Interceptor[GetFunctionInput, GetFunctionOutput, Any, Any], Interceptor[GetOrganisationInput, GetOrganisationOutput, Any, Any], Interceptor[GetResolvedConfigInput, GetResolvedConfigOutput, Any, Any], Interceptor[GetTypeTemplatesListInput, GetTypeTemplatesListOutput, Any, Any], Interceptor[GetWebhookInput, GetWebhookOutput, Any, Any], Interceptor[ListAuditLogsInput, ListAuditLogsOutput, Any, Any], Interceptor[ListContextsInput, ListContextsOutput, Any, Any], Interceptor[ListDefaultConfigsInput, ListDefaultConfigsOutput, Any, Any], Interceptor[ListDimensionsInput, ListDimensionsOutput, Any, Any], Interceptor[ListExperimentInput, ListExperimentOutput, Any, Any], Interceptor[ListExperimentGroupsInput, ListExperimentGroupsOutput, Any, Any], Interceptor[ListFunctionInput, ListFunctionOutput, Any, Any], Interceptor[ListOrganisationInput, ListOrganisationOutput, Any, Any], Interceptor[ListVersionsInput, ListVersionsOutput, Any, Any], Interceptor[ListWebhookInput, ListWebhookOutput, Any, Any], Interceptor[ListWorkspaceInput, ListWorkspaceOutput, Any, Any], Interceptor[MoveContextInput, MoveContextOutput, Any, Any], Interceptor[PauseExperimentInput, PauseExperimentOutput, Any, Any], Interceptor[PublishInput, PublishOutput, Any, Any], Interceptor[RampExperimentInput, RampExperimentOutput, Any, Any], Interceptor[RemoveMembersFromGroupInput, RemoveMembersFromGroupOutput, Any, Any], Interceptor[ResumeExperimentInput, ResumeExperimentOutput, Any, Any], Interceptor[TestInput, TestOutput, Any, Any], Interceptor[UpdateDefaultConfigInput, UpdateDefaultConfigOutput, Any, Any], Interceptor[UpdateDimensionInput, UpdateDimensionOutput, Any, Any], Interceptor[UpdateExperimentGroupInput, UpdateExperimentGroupOutput, Any, Any], Interceptor[UpdateFunctionInput, UpdateFunctionOutput, Any, Any], Interceptor[UpdateOrganisationInput, UpdateOrganisationOutput, Any, Any], Interceptor[UpdateOverrideInput, UpdateOverrideOutput, Any, Any], Interceptor[UpdateOverridesExperimentInput, UpdateOverridesExperimentOutput, Any, Any], Interceptor[UpdateTypeTemplatesInput, UpdateTypeTemplatesOutput, Any, Any], Interceptor[UpdateWebhookInput, UpdateWebhookOutput, Any, Any], Interceptor[UpdateWorkspaceInput, UpdateWorkspaceOutput, Any, Any], Interceptor[WeightRecomputeInput, WeightRecomputeOutput, Any, Any]]
+_ServiceInterceptor = Union[Interceptor[AddMembersToGroupInput, AddMembersToGroupOutput, Any, Any], Interceptor[ApplicableVariantsInput, ApplicableVariantsOutput, Any, Any], Interceptor[BulkOperationInput, BulkOperationOutput, Any, Any], Interceptor[ConcludeExperimentInput, ConcludeExperimentOutput, Any, Any], Interceptor[CreateContextInput, CreateContextOutput, Any, Any], Interceptor[CreateDefaultConfigInput, CreateDefaultConfigOutput, Any, Any], Interceptor[CreateDimensionInput, CreateDimensionOutput, Any, Any], Interceptor[CreateExperimentInput, CreateExperimentOutput, Any, Any], Interceptor[CreateExperimentGroupInput, CreateExperimentGroupOutput, Any, Any], Interceptor[CreateFunctionInput, CreateFunctionOutput, Any, Any], Interceptor[CreateOrganisationInput, CreateOrganisationOutput, Any, Any], Interceptor[CreateTypeTemplatesInput, CreateTypeTemplatesOutput, Any, Any], Interceptor[CreateWebhookInput, CreateWebhookOutput, Any, Any], Interceptor[CreateWorkspaceInput, CreateWorkspaceOutput, Any, Any], Interceptor[DeleteContextInput, DeleteContextOutput, Any, Any], Interceptor[DeleteDefaultConfigInput, DeleteDefaultConfigOutput, Any, Any], Interceptor[DeleteDimensionInput, DeleteDimensionOutput, Any, Any], Interceptor[DeleteExperimentGroupInput, DeleteExperimentGroupOutput, Any, Any], Interceptor[DeleteFunctionInput, DeleteFunctionOutput, Any, Any], Interceptor[DeleteTypeTemplatesInput, DeleteTypeTemplatesOutput, Any, Any], Interceptor[DiscardExperimentInput, DiscardExperimentOutput, Any, Any], Interceptor[GetConfigInput, GetConfigOutput, Any, Any], Interceptor[GetConfigFastInput, GetConfigFastOutput, Any, Any], Interceptor[GetContextInput, GetContextOutput, Any, Any], Interceptor[GetContextFromConditionInput, GetContextFromConditionOutput, Any, Any], Interceptor[GetDimensionInput, GetDimensionOutput, Any, Any], Interceptor[GetExperimentInput, GetExperimentOutput, Any, Any], Interceptor[GetExperimentGroupInput, GetExperimentGroupOutput, Any, Any], Interceptor[GetFunctionInput, GetFunctionOutput, Any, Any], Interceptor[GetOrganisationInput, GetOrganisationOutput, Any, Any], Interceptor[GetResolvedConfigInput, GetResolvedConfigOutput, Any, Any], Interceptor[GetTypeTemplatesListInput, GetTypeTemplatesListOutput, Any, Any], Interceptor[GetWebhookInput, GetWebhookOutput, Any, Any], Interceptor[ListAuditLogsInput, ListAuditLogsOutput, Any, Any], Interceptor[ListContextsInput, ListContextsOutput, Any, Any], Interceptor[ListDefaultConfigsInput, ListDefaultConfigsOutput, Any, Any], Interceptor[ListDimensionsInput, ListDimensionsOutput, Any, Any], Interceptor[ListExperimentInput, ListExperimentOutput, Any, Any], Interceptor[ListExperimentGroupsInput, ListExperimentGroupsOutput, Any, Any], Interceptor[ListFunctionInput, ListFunctionOutput, Any, Any], Interceptor[ListOrganisationInput, ListOrganisationOutput, Any, Any], Interceptor[ListVersionsInput, ListVersionsOutput, Any, Any], Interceptor[ListWebhookInput, ListWebhookOutput, Any, Any], Interceptor[ListWorkspaceInput, ListWorkspaceOutput, Any, Any], Interceptor[MigrateWorkspaceSchemaInput, MigrateWorkspaceSchemaOutput, Any, Any], Interceptor[MoveContextInput, MoveContextOutput, Any, Any], Interceptor[PauseExperimentInput, PauseExperimentOutput, Any, Any], Interceptor[PublishInput, PublishOutput, Any, Any], Interceptor[RampExperimentInput, RampExperimentOutput, Any, Any], Interceptor[RemoveMembersFromGroupInput, RemoveMembersFromGroupOutput, Any, Any], Interceptor[ResumeExperimentInput, ResumeExperimentOutput, Any, Any], Interceptor[TestInput, TestOutput, Any, Any], Interceptor[UpdateDefaultConfigInput, UpdateDefaultConfigOutput, Any, Any], Interceptor[UpdateDimensionInput, UpdateDimensionOutput, Any, Any], Interceptor[UpdateExperimentGroupInput, UpdateExperimentGroupOutput, Any, Any], Interceptor[UpdateFunctionInput, UpdateFunctionOutput, Any, Any], Interceptor[UpdateOrganisationInput, UpdateOrganisationOutput, Any, Any], Interceptor[UpdateOverrideInput, UpdateOverrideOutput, Any, Any], Interceptor[UpdateOverridesExperimentInput, UpdateOverridesExperimentOutput, Any, Any], Interceptor[UpdateTypeTemplatesInput, UpdateTypeTemplatesOutput, Any, Any], Interceptor[UpdateWebhookInput, UpdateWebhookOutput, Any, Any], Interceptor[UpdateWorkspaceInput, UpdateWorkspaceOutput, Any, Any], Interceptor[WeightRecomputeInput, WeightRecomputeOutput, Any, Any]]
 @dataclass(init=False)
 class Config:
     """Configuration for Superposition."""

superposition_sdk/deserialize.py

@@ -62,6 +62,7 @@ from .models import (
     ListVersionsOutput,
     ListWebhookOutput,
     ListWorkspaceOutput,
+    MigrateWorkspaceSchemaOutput,
     MoveContextOutput,
     OrganisationNotFound,
     PauseExperimentOutput,
@@ -1247,6 +1248,31 @@ async def _deserialize_error_list_workspace(http_response: HTTPResponse, config:
         case _:
             return UnknownApiError(f"{code}: {message}")
 
+async def _deserialize_migrate_workspace_schema(http_response: HTTPResponse, config: Config) -> MigrateWorkspaceSchemaOutput:
+    if http_response.status != 200 and http_response.status >= 300:
+        raise await _deserialize_error_migrate_workspace_schema(http_response, config)
+
+    kwargs: dict[str, Any] = {}
+
+    body = await http_response.consume_body_async()
+    if body:
+        codec = JSONCodec(default_timestamp_format=TimestampFormat.EPOCH_SECONDS)
+        deserializer = codec.create_deserializer(body)
+        body_kwargs = MigrateWorkspaceSchemaOutput.deserialize_kwargs(deserializer)
+        kwargs.update(body_kwargs)
+
+    return MigrateWorkspaceSchemaOutput(**kwargs)
+
+async def _deserialize_error_migrate_workspace_schema(http_response: HTTPResponse, config: Config) -> ApiError:
+    code, message, parsed_body = await parse_rest_json_error_info(http_response)
+
+    match code.lower():
+        case "internalservererror":
+            return await _deserialize_error_internal_server_error(http_response, config, parsed_body, message)
+
+        case _:
+            return UnknownApiError(f"{code}: {message}")
+
 async def _deserialize_move_context(http_response: HTTPResponse, config: Config) -> MoveContextOutput:
     if http_response.status != 200 and http_response.status >= 300:
         raise await _deserialize_error_move_context(http_response, config)

superposition_sdk/models.py

@@ -167,6 +167,9 @@ from ._private.schemas import (
     LIST_WORKSPACE as _SCHEMA_LIST_WORKSPACE,
     LIST_WORKSPACE_INPUT as _SCHEMA_LIST_WORKSPACE_INPUT,
     LIST_WORKSPACE_OUTPUT as _SCHEMA_LIST_WORKSPACE_OUTPUT,
+    MIGRATE_WORKSPACE_SCHEMA as _SCHEMA_MIGRATE_WORKSPACE_SCHEMA,
+    MIGRATE_WORKSPACE_SCHEMA_INPUT as _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT,
+    MIGRATE_WORKSPACE_SCHEMA_OUTPUT as _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT,
     MOVE_CONTEXT as _SCHEMA_MOVE_CONTEXT,
     MOVE_CONTEXT_INPUT as _SCHEMA_MOVE_CONTEXT_INPUT,
     MOVE_CONTEXT_OUTPUT as _SCHEMA_MOVE_CONTEXT_OUTPUT,
@@ -2983,8 +2986,8 @@ DELETE_CONTEXT = APIOperation(
         input_schema = _SCHEMA_DELETE_CONTEXT_INPUT,
         output_schema = _SCHEMA_DELETE_CONTEXT_OUTPUT,
         error_registry = TypeRegistry({
-            ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
-ShapeID("io.superposition#InternalServerError"): InternalServerError,
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#httpBearerAuth")
@@ -5946,8 +5949,8 @@ DELETE_DEFAULT_CONFIG = APIOperation(
         input_schema = _SCHEMA_DELETE_DEFAULT_CONFIG_INPUT,
         output_schema = _SCHEMA_DELETE_DEFAULT_CONFIG_OUTPUT,
         error_registry = TypeRegistry({
-            ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
-ShapeID("io.superposition#InternalServerError"): InternalServerError,
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#httpBearerAuth")
@@ -6168,8 +6171,8 @@ LIST_DEFAULT_CONFIGS = APIOperation(
         input_schema = _SCHEMA_LIST_DEFAULT_CONFIGS_INPUT,
         output_schema = _SCHEMA_LIST_DEFAULT_CONFIGS_OUTPUT,
         error_registry = TypeRegistry({
-            ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
-ShapeID("io.superposition#InternalServerError"): InternalServerError,
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#httpBearerAuth")
@@ -6439,8 +6442,8 @@ DELETE_DIMENSION = APIOperation(
         input_schema = _SCHEMA_DELETE_DIMENSION_INPUT,
         output_schema = _SCHEMA_DELETE_DIMENSION_OUTPUT,
         error_registry = TypeRegistry({
-            ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
-ShapeID("io.superposition#InternalServerError"): InternalServerError,
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+ShapeID("io.superposition#ResourceNotFound"): ResourceNotFound,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#httpBearerAuth")
@@ -6714,8 +6717,8 @@ DELETE_FUNCTION = APIOperation(
         input_schema = _SCHEMA_DELETE_FUNCTION_INPUT,
         output_schema = _SCHEMA_DELETE_FUNCTION_OUTPUT,
         error_registry = TypeRegistry({
-            ShapeID("io.superposition#FunctionNotFound"): FunctionNotFound,
-ShapeID("io.superposition#InternalServerError"): InternalServerError,
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+ShapeID("io.superposition#FunctionNotFound"): FunctionNotFound,
         }),
         effective_auth_schemes = [
             ShapeID("smithy.api#httpBearerAuth")
@@ -12214,6 +12217,178 @@ LIST_WORKSPACE = APIOperation(
         ]
 )
 
+@dataclass(kw_only=True)
+class MigrateWorkspaceSchemaInput:
+
+    org_id: str = "juspay"
+    workspace_name: str | None = None
+
+    def serialize(self, serializer: ShapeSerializer):
+        serializer.write_struct(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT, self)
+
+    def serialize_members(self, serializer: ShapeSerializer):
+        pass
+
+    @classmethod
+    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
+        return cls(**cls.deserialize_kwargs(deserializer))
+
+    @classmethod
+    def deserialize_kwargs(cls, deserializer: ShapeDeserializer) -> dict[str, Any]:
+        kwargs: dict[str, Any] = {}
+
+        def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
+            match schema.expect_member_index():
+                case 0:
+                    kwargs["org_id"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT.members["org_id"])
+
+                case 1:
+                    kwargs["workspace_name"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT.members["workspace_name"])
+
+                case _:
+                    logger.debug("Unexpected member schema: %s", schema)
+
+        deserializer.read_struct(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT, consumer=_consumer)
+        return kwargs
+
+@dataclass(kw_only=True)
+class MigrateWorkspaceSchemaOutput:
+
+    workspace_name: str
+
+    organisation_id: str
+
+    organisation_name: str
+
+    workspace_schema_name: str
+
+    workspace_status: str
+
+    workspace_admin_email: str
+
+    created_by: str
+
+    last_modified_by: str
+
+    last_modified_at: datetime
+
+    created_at: datetime
+
+    strict_mode: bool
+
+    allow_experiment_self_approval: bool
+
+    auto_populate_control: bool
+
+    config_version: str | None = None
+    mandatory_dimensions: list[str] | None = None
+    metrics: Document | None = None
+
+    def serialize(self, serializer: ShapeSerializer):
+        serializer.write_struct(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT, self)
+
+    def serialize_members(self, serializer: ShapeSerializer):
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_name"], self.workspace_name)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["organisation_id"], self.organisation_id)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["organisation_name"], self.organisation_name)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_schema_name"], self.workspace_schema_name)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_status"], self.workspace_status)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_admin_email"], self.workspace_admin_email)
+        if self.config_version is not None:
+            serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["config_version"], self.config_version)
+
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["created_by"], self.created_by)
+        serializer.write_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["last_modified_by"], self.last_modified_by)
+        serializer.write_timestamp(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["last_modified_at"], self.last_modified_at)
+        serializer.write_timestamp(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["created_at"], self.created_at)
+        if self.mandatory_dimensions is not None:
+            _serialize_list_mandatory_dimensions(serializer, _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["mandatory_dimensions"], self.mandatory_dimensions)
+
+        serializer.write_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["strict_mode"], self.strict_mode)
+        if self.metrics is not None:
+            serializer.write_document(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["metrics"], self.metrics)
+
+        serializer.write_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["allow_experiment_self_approval"], self.allow_experiment_self_approval)
+        serializer.write_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["auto_populate_control"], self.auto_populate_control)
+
+    @classmethod
+    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
+        return cls(**cls.deserialize_kwargs(deserializer))
+
+    @classmethod
+    def deserialize_kwargs(cls, deserializer: ShapeDeserializer) -> dict[str, Any]:
+        kwargs: dict[str, Any] = {}
+
+        def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
+            match schema.expect_member_index():
+                case 0:
+                    kwargs["workspace_name"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_name"])
+
+                case 1:
+                    kwargs["organisation_id"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["organisation_id"])
+
+                case 2:
+                    kwargs["organisation_name"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["organisation_name"])
+
+                case 3:
+                    kwargs["workspace_schema_name"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_schema_name"])
+
+                case 4:
+                    kwargs["workspace_status"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_status"])
+
+                case 5:
+                    kwargs["workspace_admin_email"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["workspace_admin_email"])
+
+                case 6:
+                    kwargs["config_version"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["config_version"])
+
+                case 7:
+                    kwargs["created_by"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["created_by"])
+
+                case 8:
+                    kwargs["last_modified_by"] = de.read_string(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["last_modified_by"])
+
+                case 9:
+                    kwargs["last_modified_at"] = de.read_timestamp(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["last_modified_at"])
+
+                case 10:
+                    kwargs["created_at"] = de.read_timestamp(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["created_at"])
+
+                case 11:
+                    kwargs["mandatory_dimensions"] = _deserialize_list_mandatory_dimensions(de, _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["mandatory_dimensions"])
+
+                case 12:
+                    kwargs["strict_mode"] = de.read_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["strict_mode"])
+
+                case 13:
+                    kwargs["metrics"] = de.read_document(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["metrics"])
+
+                case 14:
+                    kwargs["allow_experiment_self_approval"] = de.read_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["allow_experiment_self_approval"])
+
+                case 15:
+                    kwargs["auto_populate_control"] = de.read_boolean(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT.members["auto_populate_control"])
+
+                case _:
+                    logger.debug("Unexpected member schema: %s", schema)
+
+        deserializer.read_struct(_SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT, consumer=_consumer)
+        return kwargs
+
+MIGRATE_WORKSPACE_SCHEMA = APIOperation(
+        input = MigrateWorkspaceSchemaInput,
+        output = MigrateWorkspaceSchemaOutput,
+        schema = _SCHEMA_MIGRATE_WORKSPACE_SCHEMA,
+        input_schema = _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_INPUT,
+        output_schema = _SCHEMA_MIGRATE_WORKSPACE_SCHEMA_OUTPUT,
+        error_registry = TypeRegistry({
+            ShapeID("io.superposition#InternalServerError"): InternalServerError,
+        }),
+        effective_auth_schemes = [
+            ShapeID("smithy.api#httpBearerAuth")
+        ]
+)
+
 @dataclass(kw_only=True)
 class UpdateOrganisationInput:
 

superposition_sdk/serialize.py

@@ -59,6 +59,7 @@ from .models import (
     ListVersionsInput,
     ListWebhookInput,
     ListWorkspaceInput,
+    MigrateWorkspaceSchemaInput,
     MoveContextInput,
     PauseExperimentInput,
     PublishInput,
@@ -1695,6 +1696,36 @@ async def _serialize_list_workspace(input: ListWorkspaceInput, config: Config) -
         body=body,
     )
 
+async def _serialize_migrate_workspace_schema(input: MigrateWorkspaceSchemaInput, config: Config) -> HTTPRequest:
+    if not input.workspace_name:
+        raise ServiceError("workspace_name must not be empty.")
+
+    path = "/workspaces/{workspace_name}/db/migrate".format(
+        workspace_name=urlquote(input.workspace_name, safe=''),
+    )
+    query: str = f''
+
+    body: AsyncIterable[bytes] = AsyncBytesReader(b'')
+    headers = Fields(
+        [
+
+        ]
+    )
+
+    if input.org_id:
+        headers.extend(Fields([Field(name="x-org-id", values=[input.org_id])]))
+    return _HTTPRequest(
+        destination=_URI(
+            host="",
+            path=path,
+            scheme="https",
+            query=query,
+        ),
+        method="POST",
+        fields=headers,
+        body=body,
+    )
+
 async def _serialize_move_context(input: MoveContextInput, config: Config) -> HTTPRequest:
     if not input.id:
         raise ServiceError("id must not be empty.")

{superposition_sdk-0.86.0.dist-info → superposition_sdk-0.87.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superposition_sdk
-Version: 0.86.0
+Version: 0.87.0
 Summary: superposition_sdk client
 License: Apache-2.0
 Keywords: smithy,superposition_sdk

superposition_sdk-0.87.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+superposition_sdk/__init__.py,sha256=Inc_oe2U1CzZ_ZxsNVbU96RGaAlD-6M5xt1ZcW5hGUo,83
+superposition_sdk/auth.py,sha256=U9D3FmOH8TX0cjibGy_CRnAkTnFEavAek0Jw1YUhN2g,404
+superposition_sdk/client.py,sha256=mas4nI_bk32FXVUQMlOSa-S1Y4ewY8GIoDCIsy6kCCM,88710
+superposition_sdk/config.py,sha256=elPULtcHP7am9Wk7VqIpzPMNlcQIdH1Do9gNeAkqdHM,11425
+superposition_sdk/deserialize.py,sha256=fVUcHo_BtWwtINcbBWGXDVo0GmSPT0pEyDwEB6Jf82Y,77499
+superposition_sdk/models.py,sha256=JNV6ts9ouu_brSMfifEmOwBiIBLlawXFmjQfRv-UNeI,545296
+superposition_sdk/serialize.py,sha256=5naLDkPY1SBrHIuU_SGj05TCLcx7jR59BGy65tHYk58,72728
+superposition_sdk/_private/__init__.py,sha256=DxsJq42a0KfQv9zzLrqT0JVpoXrR-IDIYFGuJe4g1Gc,55
+superposition_sdk/_private/schemas.py,sha256=9UZVCgCoo4t7SnD0paDSSZXw3kLSNpt0fq0l8GrJ83Y,303088
+superposition_sdk-0.87.0.dist-info/METADATA,sha256=eadwxEYdC1mIMrQmsceBeYr_VYn_fT2cA-JiW6NHiTQ,2280
+superposition_sdk-0.87.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+superposition_sdk-0.87.0.dist-info/RECORD,,

superposition_sdk-0.86.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-superposition_sdk/__init__.py,sha256=Inc_oe2U1CzZ_ZxsNVbU96RGaAlD-6M5xt1ZcW5hGUo,83
-superposition_sdk/auth.py,sha256=U9D3FmOH8TX0cjibGy_CRnAkTnFEavAek0Jw1YUhN2g,404
-superposition_sdk/client.py,sha256=9kj6k0uX6UaP-tAxJoZNhMJVFpC-kWhRXNAqWA4-OO0,87557
-superposition_sdk/config.py,sha256=WfJkW9bPCkNXyxcWet47cUY2OdU--9rs5zuWlssuAag,11276
-superposition_sdk/deserialize.py,sha256=VlPxbY8hwD5Sbgn1IvvMFmm9-wMOVjmYrUkPvDE_WtU,76337
-superposition_sdk/models.py,sha256=hMdju90O_KJ68Il2fCfVOAKhPNGXQxV2M6x_iA-BXiA,537165
-superposition_sdk/serialize.py,sha256=__hMORn6Mx61sy0t3Y5Hs7O5iO1GOyRHYg0kehjbZYA,71892
-superposition_sdk/_private/__init__.py,sha256=DxsJq42a0KfQv9zzLrqT0JVpoXrR-IDIYFGuJe4g1Gc,55
-superposition_sdk/_private/schemas.py,sha256=VkNDp8dOeaLU615pGiKU0OCKiMM_wFmakcDywQjItHY,298510
-superposition_sdk-0.86.0.dist-info/METADATA,sha256=ZxBfZG2kEAjsEevMATCyX5_a2hKBE5eVnWxOw3Wa8oc,2280
-superposition_sdk-0.86.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-superposition_sdk-0.86.0.dist-info/RECORD,,