mixpeek 0.21.8__py3-none-any.whl → 0.22.0__py3-none-any.whl

mixpeek/models/objectresponse.py

@@ -6,18 +6,10 @@ from .taskstatus import TaskStatus
 from datetime import datetime
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List
+from typing import Any, Dict, List
 from typing_extensions import NotRequired, TypedDict
 
 
-class ObjectResponseMetadataTypedDict(TypedDict):
-    r"""Additional metadata for the object, appended to downstream documents of connected collections"""
-
-
-class ObjectResponseMetadata(BaseModel):
-    r"""Additional metadata for the object, appended to downstream documents of connected collections"""
-
-
 class ObjectResponseTypedDict(TypedDict):
     r"""Response model for bucket objects"""
 
@@ -28,7 +20,7 @@ class ObjectResponseTypedDict(TypedDict):
     blobs: List[BlobModelTypedDict]
     r"""List of blobs contained in this object"""
     status: TaskStatus
-    metadata: ObjectResponseMetadataTypedDict
+    metadata: Dict[str, Any]
     r"""Additional metadata for the object, appended to downstream documents of connected collections"""
     task_id: NotRequired[Nullable[str]]
     r"""ID of the task that created this object, used to track the object creation progress and status"""
@@ -58,7 +50,7 @@ class ObjectResponse(BaseModel):
 
     status: TaskStatus
 
-    metadata: ObjectResponseMetadata
+    metadata: Dict[str, Any]
     r"""Additional metadata for the object, appended to downstream documents of connected collections"""
 
     task_id: OptionalNullable[str] = UNSET

mixpeek/models/organizationmodel.py

@@ -4,18 +4,10 @@ from __future__ import annotations
 from .usermodel_output import UserModelOutput, UserModelOutputTypedDict
 from datetime import datetime
 from mixpeek.types import BaseModel
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class OrganizationModelMetadataTypedDict(TypedDict):
-    pass
-
-
-class OrganizationModelMetadata(BaseModel):
-    pass
-
-
 class OrganizationModelTypedDict(TypedDict):
     organization_name: str
     organization_id: NotRequired[str]
@@ -23,7 +15,7 @@ class OrganizationModelTypedDict(TypedDict):
     created_at: NotRequired[datetime]
     account_type: NotRequired[str]
     credit_count: NotRequired[int]
-    metadata: NotRequired[OrganizationModelMetadataTypedDict]
+    metadata: NotRequired[Dict[str, Any]]
     users: NotRequired[List[UserModelOutputTypedDict]]
 
 
@@ -40,6 +32,6 @@ class OrganizationModel(BaseModel):
 
     credit_count: Optional[int] = 1000
 
-    metadata: Optional[OrganizationModelMetadata] = None
+    metadata: Optional[Dict[str, Any]] = None
 
     users: Optional[List[UserModelOutput]] = None

mixpeek/models/retrieverbinding.py

@@ -3,23 +3,16 @@
 from __future__ import annotations
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
+from typing import Any, Dict
 from typing_extensions import NotRequired, TypedDict
 
 
-class InputsTypedDict(TypedDict):
-    pass
-
-
-class Inputs(BaseModel):
-    pass
-
-
 class RetrieverBindingTypedDict(TypedDict):
     r"""How a retriever should be used in a taxonomy"""
 
     retriever_id: str
     r"""ID of the retriever to use"""
-    inputs: NotRequired[Nullable[InputsTypedDict]]
+    inputs: NotRequired[Nullable[Dict[str, Any]]]
     r"""Optional inputs to the retriever. If not provided, will use source collection schema"""
 
 
@@ -29,7 +22,7 @@ class RetrieverBinding(BaseModel):
     retriever_id: str
     r"""ID of the retriever to use"""
 
-    inputs: OptionalNullable[Inputs] = UNSET
+    inputs: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Optional inputs to the retriever. If not provided, will use source collection schema"""
 
     @model_serializer(mode="wrap")

mixpeek/models/retrievermodel.py

@@ -9,18 +9,10 @@ from .retrieverschema_output import (
 from .stageconfig_output import StageConfigOutput, StageConfigOutputTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class RetrieverModelMetadataTypedDict(TypedDict):
-    pass
-
-
-class RetrieverModelMetadata(BaseModel):
-    pass
-
-
 class RetrieverModelTypedDict(TypedDict):
     r"""Definition of a retriever"""
 
@@ -36,7 +28,7 @@ class RetrieverModelTypedDict(TypedDict):
     r"""Unique identifier for the retriever"""
     description: NotRequired[Nullable[str]]
     r"""Description of the retriever"""
-    metadata: NotRequired[RetrieverModelMetadataTypedDict]
+    metadata: NotRequired[Dict[str, Any]]
     cache_config: NotRequired[Nullable[RetrieverCacheConfigTypedDict]]
     r"""Configuration for retriever-level caching"""
 
@@ -62,7 +54,7 @@ class RetrieverModel(BaseModel):
     description: OptionalNullable[str] = UNSET
     r"""Description of the retriever"""
 
-    metadata: Optional[RetrieverModelMetadata] = None
+    metadata: Optional[Dict[str, Any]] = None
 
     cache_config: OptionalNullable[RetrieverCacheConfig] = UNSET
     r"""Configuration for retriever-level caching"""

mixpeek/models/retrieverqueryrequest.py

@@ -5,18 +5,10 @@ from .logicaloperator_input import LogicalOperatorInput, LogicalOperatorInputTyp
 from .sortoption import SortOption, SortOptionTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class RetrieverQueryRequestInputsTypedDict(TypedDict):
-    r"""Input values for the retriever query. These map to the required inputs defined in the retriever's first stage."""
-
-
-class RetrieverQueryRequestInputs(BaseModel):
-    r"""Input values for the retriever query. These map to the required inputs defined in the retriever's first stage."""
-
-
 class RetrieverQueryRequestTypedDict(TypedDict):
     r"""Query parameters for executing a retriever pipeline.
 
@@ -25,7 +17,7 @@ class RetrieverQueryRequestTypedDict(TypedDict):
     query inputs, filtering, sorting, pagination, and result formatting.
     """
 
-    inputs: RetrieverQueryRequestInputsTypedDict
+    inputs: Dict[str, Any]
     r"""Input values for the retriever query. These map to the required inputs defined in the retriever's first stage."""
     filters: NotRequired[Nullable[LogicalOperatorInputTypedDict]]
     r"""Logical operations for filtering results. Can include AND, OR, NOT conditions with field comparisons."""
@@ -51,7 +43,7 @@ class RetrieverQueryRequest(BaseModel):
     query inputs, filtering, sorting, pagination, and result formatting.
     """
 
-    inputs: RetrieverQueryRequestInputs
+    inputs: Dict[str, Any]
     r"""Input values for the retriever query. These map to the required inputs defined in the retriever's first stage."""
 
     filters: OptionalNullable[LogicalOperatorInput] = UNSET

mixpeek/models/retrieverschemafield_input.py

@@ -1,14 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .bucketschemafield_input import (
-    BucketSchemaFieldInput,
-    BucketSchemaFieldInputTypedDict,
-)
+from .bucketschemafieldbase import BucketSchemaFieldBase, BucketSchemaFieldBaseTypedDict
 from .bucketschemafieldtype import BucketSchemaFieldType
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-import pydantic
-from pydantic import ConfigDict, model_serializer
+from pydantic import model_serializer
 from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
@@ -19,8 +15,8 @@ class RetrieverSchemaFieldInputTypedDict(TypedDict):
     type: BucketSchemaFieldType
     r"""Enum for field types in bucket schemas"""
     default: NotRequired[Nullable[Any]]
-    items: NotRequired[Nullable[BucketSchemaFieldInputTypedDict]]
-    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldInputTypedDict]]]
+    items: NotRequired[Nullable[BucketSchemaFieldBaseTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldBaseTypedDict]]]
     example: NotRequired[Nullable[Any]]
     description: NotRequired[Nullable[str]]
     enum: NotRequired[Nullable[List[Any]]]
@@ -31,19 +27,14 @@ class RetrieverSchemaFieldInputTypedDict(TypedDict):
 class RetrieverSchemaFieldInput(BaseModel):
     r"""Schema field definition for retriever inputs"""
 
-    model_config = ConfigDict(
-        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
-    )
-    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
-
     type: BucketSchemaFieldType
     r"""Enum for field types in bucket schemas"""
 
     default: OptionalNullable[Any] = UNSET
 
-    items: OptionalNullable[BucketSchemaFieldInput] = UNSET
+    items: OptionalNullable[BucketSchemaFieldBase] = UNSET
 
-    properties: OptionalNullable[Dict[str, BucketSchemaFieldInput]] = UNSET
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldBase]] = UNSET
 
     example: OptionalNullable[Any] = UNSET
 
@@ -54,14 +45,6 @@ class RetrieverSchemaFieldInput(BaseModel):
     required: Optional[bool] = True
     r"""Whether this field is required for the retriever"""
 
-    @property
-    def additional_properties(self):
-        return self.__pydantic_extra__
-
-    @additional_properties.setter
-    def additional_properties(self, value):
-        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
-
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -105,7 +88,4 @@ class RetrieverSchemaFieldInput(BaseModel):
             ):
                 m[k] = val
 
-        for k, v in serialized.items():
-            m[k] = v
-
         return m

mixpeek/models/retrieverschemafield_output.py

@@ -1,14 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .bucketschemafield_output import (
-    BucketSchemaFieldOutput,
-    BucketSchemaFieldOutputTypedDict,
-)
+from .bucketschemafieldbase import BucketSchemaFieldBase, BucketSchemaFieldBaseTypedDict
 from .bucketschemafieldtype import BucketSchemaFieldType
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-import pydantic
-from pydantic import ConfigDict, model_serializer
+from pydantic import model_serializer
 from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
@@ -19,8 +15,8 @@ class RetrieverSchemaFieldOutputTypedDict(TypedDict):
     type: BucketSchemaFieldType
     r"""Enum for field types in bucket schemas"""
     default: NotRequired[Nullable[Any]]
-    items: NotRequired[Nullable[BucketSchemaFieldOutputTypedDict]]
-    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldOutputTypedDict]]]
+    items: NotRequired[Nullable[BucketSchemaFieldBaseTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldBaseTypedDict]]]
     example: NotRequired[Nullable[Any]]
     description: NotRequired[Nullable[str]]
     enum: NotRequired[Nullable[List[Any]]]
@@ -31,19 +27,14 @@ class RetrieverSchemaFieldOutputTypedDict(TypedDict):
 class RetrieverSchemaFieldOutput(BaseModel):
     r"""Schema field definition for retriever inputs"""
 
-    model_config = ConfigDict(
-        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
-    )
-    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
-
     type: BucketSchemaFieldType
     r"""Enum for field types in bucket schemas"""
 
     default: OptionalNullable[Any] = UNSET
 
-    items: OptionalNullable[BucketSchemaFieldOutput] = UNSET
+    items: OptionalNullable[BucketSchemaFieldBase] = UNSET
 
-    properties: OptionalNullable[Dict[str, BucketSchemaFieldOutput]] = UNSET
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldBase]] = UNSET
 
     example: OptionalNullable[Any] = UNSET
 
@@ -54,14 +45,6 @@ class RetrieverSchemaFieldOutput(BaseModel):
     required: Optional[bool] = True
     r"""Whether this field is required for the retriever"""
 
-    @property
-    def additional_properties(self):
-        return self.__pydantic_extra__
-
-    @additional_properties.setter
-    def additional_properties(self, value):
-        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
-
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -105,7 +88,4 @@ class RetrieverSchemaFieldOutput(BaseModel):
             ):
                 m[k] = val
 
-        for k, v in serialized.items():
-            m[k] = v
-
         return m

mixpeek/models/searchinteraction.py

@@ -4,18 +4,10 @@ from __future__ import annotations
 from .interactiontype import InteractionType
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List
+from typing import Any, Dict, List
 from typing_extensions import NotRequired, TypedDict
 
 
-class SearchInteractionMetadataTypedDict(TypedDict):
-    pass
-
-
-class SearchInteractionMetadata(BaseModel):
-    pass
-
-
 class SearchInteractionTypedDict(TypedDict):
     feature_id: str
     r"""ID of the item that was interacted with"""
@@ -23,7 +15,7 @@ class SearchInteractionTypedDict(TypedDict):
     r"""Type of interaction or feedback"""
     position: NotRequired[Nullable[int]]
     r"""Position in search results where interaction occurred"""
-    metadata: NotRequired[Nullable[SearchInteractionMetadataTypedDict]]
+    metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Additional context about the interaction"""
     user_id: NotRequired[Nullable[str]]
     r"""Customer's authenticated user identifier - persists across sessions"""
@@ -41,7 +33,7 @@ class SearchInteraction(BaseModel):
     position: OptionalNullable[int] = UNSET
     r"""Position in search results where interaction occurred"""
 
-    metadata: OptionalNullable[SearchInteractionMetadata] = UNSET
+    metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Additional context about the interaction"""
 
     user_id: OptionalNullable[str] = UNSET

mixpeek/models/stageconfig_input.py

@@ -4,18 +4,10 @@ from __future__ import annotations
 from .logicaloperator_input import LogicalOperatorInput, LogicalOperatorInputTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional
+from typing import Any, Dict, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class StageConfigInputParametersTypedDict(TypedDict):
-    r"""Parameters for the stage"""
-
-
-class StageConfigInputParameters(BaseModel):
-    r"""Parameters for the stage"""
-
-
 class StageConfigInputTypedDict(TypedDict):
     r"""Configuration for how a retriever stage processes queries and handles results.
 
@@ -24,13 +16,13 @@ class StageConfigInputTypedDict(TypedDict):
     with consistent options.
     """
 
+    stage_id: str
+    r"""Unique identifier for the stage"""
     stage_name: str
     r"""Name of the stage"""
     version: str
     r"""Version of the stage"""
-    module_path: str
-    r"""Python module path for the stage"""
-    parameters: NotRequired[StageConfigInputParametersTypedDict]
+    parameters: NotRequired[Dict[str, Any]]
     r"""Parameters for the stage"""
     pre_filters: NotRequired[Nullable[LogicalOperatorInputTypedDict]]
     r"""Filters to apply before the main search is executed"""
@@ -50,16 +42,16 @@ class StageConfigInput(BaseModel):
     with consistent options.
     """
 
+    stage_id: str
+    r"""Unique identifier for the stage"""
+
     stage_name: str
     r"""Name of the stage"""
 
     version: str
     r"""Version of the stage"""
 
-    module_path: str
-    r"""Python module path for the stage"""
-
-    parameters: Optional[StageConfigInputParameters] = None
+    parameters: Optional[Dict[str, Any]] = None
     r"""Parameters for the stage"""
 
     pre_filters: OptionalNullable[LogicalOperatorInput] = UNSET

mixpeek/models/stageconfig_output.py

@@ -7,18 +7,10 @@ from .logicaloperator_output import (
 )
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional
+from typing import Any, Dict, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class StageConfigOutputParametersTypedDict(TypedDict):
-    r"""Parameters for the stage"""
-
-
-class StageConfigOutputParameters(BaseModel):
-    r"""Parameters for the stage"""
-
-
 class StageConfigOutputTypedDict(TypedDict):
     r"""Configuration for how a retriever stage processes queries and handles results.
 
@@ -27,13 +19,13 @@ class StageConfigOutputTypedDict(TypedDict):
     with consistent options.
     """
 
+    stage_id: str
+    r"""Unique identifier for the stage"""
     stage_name: str
     r"""Name of the stage"""
     version: str
     r"""Version of the stage"""
-    module_path: str
-    r"""Python module path for the stage"""
-    parameters: NotRequired[StageConfigOutputParametersTypedDict]
+    parameters: NotRequired[Dict[str, Any]]
     r"""Parameters for the stage"""
     pre_filters: NotRequired[Nullable[LogicalOperatorOutputTypedDict]]
     r"""Filters to apply before the main search is executed"""
@@ -53,16 +45,16 @@ class StageConfigOutput(BaseModel):
     with consistent options.
     """
 
+    stage_id: str
+    r"""Unique identifier for the stage"""
+
     stage_name: str
     r"""Name of the stage"""
 
     version: str
     r"""Version of the stage"""
 
-    module_path: str
-    r"""Python module path for the stage"""
-
-    parameters: Optional[StageConfigOutputParameters] = None
+    parameters: Optional[Dict[str, Any]] = None
     r"""Parameters for the stage"""
 
     pre_filters: OptionalNullable[LogicalOperatorOutput] = UNSET

mixpeek/models/stagedefinition.py

@@ -12,73 +12,84 @@ from typing import Any, Dict, List
 from typing_extensions import NotRequired, TypedDict
 
 
-class StageDefinitionDefaultParametersTypedDict(TypedDict):
-    pass
-
-
-class StageDefinitionDefaultParameters(BaseModel):
-    pass
-
-
 class StageDefinitionTypedDict(TypedDict):
     r"""Model for retriever stage definition that describes how to process and search documents."""
 
     stage_name: str
-    r"""Name of the stage"""
+    r"""Unique name identifying the stage type (e.g., 'knn_search', 'face_search')"""
     description: str
+    r"""Human-readable explanation of what the stage does"""
     version: str
-    module_path: str
+    r"""Version string for the stage definition (e.g., '1.0.0')"""
     input_schema: Dict[str, BucketSchemaFieldOutputTypedDict]
+    r"""Defines the data structure required from the retriever's overall input query"""
     output_schema: Dict[str, BucketSchemaFieldOutputTypedDict]
+    r"""Defines the data structure produced by this stage"""
     parameter_schema: Dict[str, BucketSchemaFieldOutputTypedDict]
+    r"""Defines the configurable parameters accepted by this stage instance (e.g., thresholds, limits)"""
     supported_input_types: List[str]
+    r"""List of general data types the stage can process (e.g., ['text', 'vector'])"""
     max_inputs: Dict[str, int]
-    default_parameters: StageDefinitionDefaultParametersTypedDict
+    r"""Specifies the maximum number of inputs allowed per type (e.g., {'text': 1})"""
+    max_total_inputs: NotRequired[Nullable[int]]
+    r"""Specifies the maximum total number of inputs allowed across all types defined in input_schema (e.g., 1 means only one input field can be provided)"""
     required_vector_indexes: NotRequired[Nullable[List[VectorIndexDefinitionTypedDict]]]
+    r"""List of vector index configurations that must exist in the target collections for this stage to function"""
     required_payload_indexes: NotRequired[Nullable[List[Any]]]
+    r"""List of payload index configurations (e.g., for specific field filtering) required in the target collections"""
     supported_filters: NotRequired[Nullable[List[str]]]
-    r"""List of filter operations supported by this stage"""
+    r"""List of filter operations supported by this stage (e.g., ['=', '>', '<'])"""
 
 
 class StageDefinition(BaseModel):
     r"""Model for retriever stage definition that describes how to process and search documents."""
 
     stage_name: str
-    r"""Name of the stage"""
+    r"""Unique name identifying the stage type (e.g., 'knn_search', 'face_search')"""
 
     description: str
+    r"""Human-readable explanation of what the stage does"""
 
     version: str
-
-    module_path: str
+    r"""Version string for the stage definition (e.g., '1.0.0')"""
 
     input_schema: Dict[str, BucketSchemaFieldOutput]
+    r"""Defines the data structure required from the retriever's overall input query"""
 
     output_schema: Dict[str, BucketSchemaFieldOutput]
+    r"""Defines the data structure produced by this stage"""
 
     parameter_schema: Dict[str, BucketSchemaFieldOutput]
+    r"""Defines the configurable parameters accepted by this stage instance (e.g., thresholds, limits)"""
 
     supported_input_types: List[str]
+    r"""List of general data types the stage can process (e.g., ['text', 'vector'])"""
 
     max_inputs: Dict[str, int]
+    r"""Specifies the maximum number of inputs allowed per type (e.g., {'text': 1})"""
 
-    default_parameters: StageDefinitionDefaultParameters
+    max_total_inputs: OptionalNullable[int] = UNSET
+    r"""Specifies the maximum total number of inputs allowed across all types defined in input_schema (e.g., 1 means only one input field can be provided)"""
 
     required_vector_indexes: OptionalNullable[List[VectorIndexDefinition]] = UNSET
+    r"""List of vector index configurations that must exist in the target collections for this stage to function"""
 
     required_payload_indexes: OptionalNullable[List[Any]] = UNSET
+    r"""List of payload index configurations (e.g., for specific field filtering) required in the target collections"""
 
     supported_filters: OptionalNullable[List[str]] = UNSET
-    r"""List of filter operations supported by this stage"""
+    r"""List of filter operations supported by this stage (e.g., ['=', '>', '<'])"""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
+            "max_total_inputs",
             "required_vector_indexes",
             "required_payload_indexes",
             "supported_filters",
         ]
         nullable_fields = [
+            "max_total_inputs",
             "required_vector_indexes",
             "required_payload_indexes",
             "supported_filters",

mixpeek/models/stageresponse.py

@@ -3,22 +3,14 @@
 from __future__ import annotations
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List
+from typing import Any, Dict, List
 from typing_extensions import NotRequired, TypedDict
 
 
-class ResultsTypedDict(TypedDict):
-    pass
-
-
-class Results(BaseModel):
-    pass
-
-
 class StageResponseTypedDict(TypedDict):
     r"""Output from a retriever stage"""
 
-    results: List[ResultsTypedDict]
+    results: List[Dict[str, Any]]
     execution_time: float
     total_results: NotRequired[Nullable[int]]
     r"""Total number of results not filtered by pagination"""
@@ -27,7 +19,7 @@ class StageResponseTypedDict(TypedDict):
 class StageResponse(BaseModel):
     r"""Output from a retriever stage"""
 
-    results: List[Results]
+    results: List[Dict[str, Any]]
 
     execution_time: float
 

mixpeek/models/updateobjectrequest.py

@@ -4,18 +4,10 @@ from __future__ import annotations
 from .createblobrequest import CreateBlobRequest, CreateBlobRequestTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import List
+from typing import Any, Dict, List
 from typing_extensions import NotRequired, TypedDict
 
 
-class UpdateObjectRequestMetadataTypedDict(TypedDict):
-    pass
-
-
-class UpdateObjectRequestMetadata(BaseModel):
-    pass
-
-
 class UpdateObjectRequestTypedDict(TypedDict):
     r"""Request model for updating an existing bucket object"""
 
@@ -23,7 +15,7 @@ class UpdateObjectRequestTypedDict(TypedDict):
     r"""Updated storage key/path prefix of the object, this will be used to retrieve the object from the storage. It's at the root of the object."""
     blobs: NotRequired[Nullable[List[CreateBlobRequestTypedDict]]]
     r"""List of new or updated blobs for this object"""
-    metadata: NotRequired[Nullable[UpdateObjectRequestMetadataTypedDict]]
+    metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Updated metadata for the object, this will be merged with existing metadata."""
     skip_duplicates: NotRequired[Nullable[bool]]
     r"""Skip duplicate blobs, if a blob with the same hash already exists, it will be skipped."""
@@ -38,7 +30,7 @@ class UpdateObjectRequest(BaseModel):
     blobs: OptionalNullable[List[CreateBlobRequest]] = UNSET
     r"""List of new or updated blobs for this object"""
 
-    metadata: OptionalNullable[UpdateObjectRequestMetadata] = UNSET
+    metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Updated metadata for the object, this will be merged with existing metadata."""
 
     skip_duplicates: OptionalNullable[bool] = UNSET