mixpeek 0.20.20__py3-none-any.whl → 0.21.1__py3-none-any.whl

mixpeek/models/collectionmodel.py

@@ -1,9 +1,24 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .bucketschema_output import BucketSchemaOutput, BucketSchemaOutputTypedDict
+from .collectioncacheconfig_output import (
+    CollectionCacheConfigOutput,
+    CollectionCacheConfigOutputTypedDict,
+)
+from .featureextractorconfig import (
+    FeatureExtractorConfig,
+    FeatureExtractorConfigTypedDict,
+)
+from .singlelineageentry import SingleLineageEntry, SingleLineageEntryTypedDict
+from .taskstatus import TaskStatus
+from .taxonomyapplicationconfig import (
+    TaxonomyApplicationConfig,
+    TaxonomyApplicationConfigTypedDict,
+)
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional
+from typing import List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
@@ -16,40 +31,96 @@ class CollectionModelMetadata(BaseModel):
 
 
 class CollectionModelTypedDict(TypedDict):
+    r"""Collection model defining the high-level structure and behavior of a collection.
+    Collections are created and modified through feature extractors, which define
+    their own document handling behavior (create/enrich/replace) through their
+    document_output_handling settings.
+    """
+
     collection_name: str
     r"""Name for the collection"""
+    feature_extractors: List[FeatureExtractorConfigTypedDict]
+    r"""List of feature extractor configurations used to create the collection"""
+    output_schema: BucketSchemaOutputTypedDict
+    r"""Schema definition for bucket objects"""
     collection_id: NotRequired[str]
     r"""Unique identifier for the collection"""
-    namespace_id: NotRequired[Nullable[str]]
-    r"""Namespace for the collection"""
+    description: NotRequired[Nullable[str]]
+    r"""Description for the collection"""
     metadata: NotRequired[Nullable[CollectionModelMetadataTypedDict]]
     r"""Optional metadata for the collection"""
+    enabled: NotRequired[bool]
+    r"""Enable or disable processing of this collection"""
+    source_lineage: NotRequired[List[SingleLineageEntryTypedDict]]
+    r"""Ordered list of lineage entries, from source to this collection"""
+    status: NotRequired[Nullable[TaskStatus]]
+    r"""Status of the feature extractors in this collection"""
+    taxonomy_applications: NotRequired[List[TaxonomyApplicationConfigTypedDict]]
+    r"""List of taxonomy application configurations"""
+    cache_config: NotRequired[Nullable[CollectionCacheConfigOutputTypedDict]]
+    r"""Configuration for collection-level caching"""
 
 
 class CollectionModel(BaseModel):
+    r"""Collection model defining the high-level structure and behavior of a collection.
+    Collections are created and modified through feature extractors, which define
+    their own document handling behavior (create/enrich/replace) through their
+    document_output_handling settings.
+    """
+
     collection_name: str
     r"""Name for the collection"""
 
+    feature_extractors: List[FeatureExtractorConfig]
+    r"""List of feature extractor configurations used to create the collection"""
+
+    output_schema: BucketSchemaOutput
+    r"""Schema definition for bucket objects"""
+
     collection_id: Optional[str] = None
     r"""Unique identifier for the collection"""
 
-    namespace_id: OptionalNullable[str] = UNSET
-    r"""Namespace for the collection"""
+    description: OptionalNullable[str] = UNSET
+    r"""Description for the collection"""
 
     metadata: OptionalNullable[CollectionModelMetadata] = UNSET
     r"""Optional metadata for the collection"""
 
+    enabled: Optional[bool] = True
+    r"""Enable or disable processing of this collection"""
+
+    source_lineage: Optional[List[SingleLineageEntry]] = None
+    r"""Ordered list of lineage entries, from source to this collection"""
+
+    status: OptionalNullable[TaskStatus] = UNSET
+    r"""Status of the feature extractors in this collection"""
+
+    taxonomy_applications: Optional[List[TaxonomyApplicationConfig]] = None
+    r"""List of taxonomy application configurations"""
+
+    cache_config: OptionalNullable[CollectionCacheConfigOutput] = UNSET
+    r"""Configuration for collection-level caching"""
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["collection_id", "namespace_id", "metadata"]
-        nullable_fields = ["namespace_id", "metadata"]
+        optional_fields = [
+            "collection_id",
+            "description",
+            "metadata",
+            "enabled",
+            "source_lineage",
+            "status",
+            "taxonomy_applications",
+            "cache_config",
+        ]
+        nullable_fields = ["description", "metadata", "status", "cache_config"]
         null_default_fields = []
 
         serialized = handler(self)
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{create_taxonomy_v1_entities_taxonomies_postop.py → create_bucket_v1_buckets_create_postop.py}

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .taxonomycreate import TaxonomyCreate, TaxonomyCreateTypedDict
+from .bucketcreaterequest import BucketCreateRequest, BucketCreateRequestTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import FieldMetadata, HeaderMetadata, RequestMetadata
 import pydantic
@@ -9,15 +9,15 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class CreateTaxonomyV1EntitiesTaxonomiesPostRequestTypedDict(TypedDict):
-    taxonomy_create: TaxonomyCreateTypedDict
+class CreateBucketV1BucketsCreatePostRequestTypedDict(TypedDict):
+    bucket_create_request: BucketCreateRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class CreateTaxonomyV1EntitiesTaxonomiesPostRequest(BaseModel):
-    taxonomy_create: Annotated[
-        TaxonomyCreate,
+class CreateBucketV1BucketsCreatePostRequest(BaseModel):
+    bucket_create_request: Annotated[
+        BucketCreateRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -38,7 +38,7 @@ class CreateTaxonomyV1EntitiesTaxonomiesPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{search_assets_v1_assets_search_postop.py → create_cluster_v1_clusters_postop.py}

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .searchassetsrequest import SearchAssetsRequest, SearchAssetsRequestTypedDict
+from .createclusterrequest import CreateClusterRequest, CreateClusterRequestTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import FieldMetadata, HeaderMetadata, RequestMetadata
 import pydantic
@@ -9,15 +9,15 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class SearchAssetsV1AssetsSearchPostRequestTypedDict(TypedDict):
-    search_assets_request: SearchAssetsRequestTypedDict
+class CreateClusterV1ClustersPostRequestTypedDict(TypedDict):
+    create_cluster_request: CreateClusterRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class SearchAssetsV1AssetsSearchPostRequest(BaseModel):
-    search_assets_request: Annotated[
-        SearchAssetsRequest,
+class CreateClusterV1ClustersPostRequest(BaseModel):
+    create_cluster_request: Annotated[
+        CreateClusterRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -38,7 +38,7 @@ class SearchAssetsV1AssetsSearchPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{create_collection_v1_collections_postop.py → create_collection_v1_collections_create_postop.py}

@@ -12,13 +12,13 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class CreateCollectionV1CollectionsPostRequestTypedDict(TypedDict):
+class CreateCollectionV1CollectionsCreatePostRequestTypedDict(TypedDict):
     create_collection_request: CreateCollectionRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class CreateCollectionV1CollectionsPostRequest(BaseModel):
+class CreateCollectionV1CollectionsCreatePostRequest(BaseModel):
     create_collection_request: Annotated[
         CreateCollectionRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
@@ -41,7 +41,7 @@ class CreateCollectionV1CollectionsPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{ingest_video_url_v1_ingest_videos_url_postop.py → create_interaction_v1_retrievers_interactions_postop.py}

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .processvideourlinput import ProcessVideoURLInput, ProcessVideoURLInputTypedDict
+from .searchinteraction import SearchInteraction, SearchInteractionTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import FieldMetadata, HeaderMetadata, RequestMetadata
 import pydantic
@@ -9,15 +9,15 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class IngestVideoURLV1IngestVideosURLPostRequestTypedDict(TypedDict):
-    process_video_url_input: ProcessVideoURLInputTypedDict
+class CreateInteractionV1RetrieversInteractionsPostRequestTypedDict(TypedDict):
+    search_interaction: SearchInteractionTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class IngestVideoURLV1IngestVideosURLPostRequest(BaseModel):
-    process_video_url_input: Annotated[
-        ProcessVideoURLInput,
+class CreateInteractionV1RetrieversInteractionsPostRequest(BaseModel):
+    search_interaction: Annotated[
+        SearchInteraction,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -38,7 +38,7 @@ class IngestVideoURLV1IngestVideosURLPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{partial_asset_update_v1_assets_asset_id_patchop.py → create_object_v1_buckets_bucket_identifier_objects_create_postop.py}

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .updateassetrequest import UpdateAssetRequest, UpdateAssetRequestTypedDict
+from .createobjectrequest import CreateObjectRequest, CreateObjectRequestTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import (
     FieldMetadata,
@@ -14,20 +14,22 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class PartialAssetUpdateV1AssetsAssetIDPatchRequestTypedDict(TypedDict):
-    asset_id: str
-    update_asset_request: UpdateAssetRequestTypedDict
+class CreateObjectV1BucketsBucketIdentifierObjectsCreatePostRequestTypedDict(TypedDict):
+    bucket_identifier: str
+    r"""Identifier of the bucket"""
+    create_object_request: CreateObjectRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class PartialAssetUpdateV1AssetsAssetIDPatchRequest(BaseModel):
-    asset_id: Annotated[
+class CreateObjectV1BucketsBucketIdentifierObjectsCreatePostRequest(BaseModel):
+    bucket_identifier: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""Identifier of the bucket"""
 
-    update_asset_request: Annotated[
-        UpdateAssetRequest,
+    create_object_request: Annotated[
+        CreateObjectRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -48,7 +50,7 @@ class PartialAssetUpdateV1AssetsAssetIDPatchRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{ingest_text_v1_ingest_text_postop.py → create_retriever_v1_retrievers_retrievers_postop.py}

@@ -1,7 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .processtextinput import ProcessTextInput, ProcessTextInputTypedDict
+from .createretrieverrequest import (
+    CreateRetrieverRequest,
+    CreateRetrieverRequestTypedDict,
+)
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import FieldMetadata, HeaderMetadata, RequestMetadata
 import pydantic
@@ -9,15 +12,15 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class IngestTextV1IngestTextPostRequestTypedDict(TypedDict):
-    process_text_input: ProcessTextInputTypedDict
+class CreateRetrieverV1RetrieversRetrieversPostRequestTypedDict(TypedDict):
+    create_retriever_request: CreateRetrieverRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class IngestTextV1IngestTextPostRequest(BaseModel):
-    process_text_input: Annotated[
-        ProcessTextInput,
+class CreateRetrieverV1RetrieversRetrieversPostRequest(BaseModel):
+    create_retriever_request: Annotated[
+        CreateRetrieverRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -38,7 +41,7 @@ class IngestTextV1IngestTextPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/{ingest_image_url_v1_ingest_images_url_postop.py → create_taxonomy_v1_taxonomies_postop.py}

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .processimageurlinput import ProcessImageURLInput, ProcessImageURLInputTypedDict
+from .createtaxonomyrequest import CreateTaxonomyRequest, CreateTaxonomyRequestTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mixpeek.utils import FieldMetadata, HeaderMetadata, RequestMetadata
 import pydantic
@@ -9,15 +9,15 @@ from pydantic import model_serializer
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class IngestImageURLV1IngestImagesURLPostRequestTypedDict(TypedDict):
-    process_image_url_input: ProcessImageURLInputTypedDict
+class CreateTaxonomyV1TaxonomiesPostRequestTypedDict(TypedDict):
+    create_taxonomy_request: CreateTaxonomyRequestTypedDict
     x_namespace: NotRequired[Nullable[str]]
     r"""Optional namespace for data isolation. This can be a namespace name or namespace ID. Example: 'netflix_prod' or 'ns_1234567890'. To create a namespace, use the /namespaces endpoint."""
 
 
-class IngestImageURLV1IngestImagesURLPostRequest(BaseModel):
-    process_image_url_input: Annotated[
-        ProcessImageURLInput,
+class CreateTaxonomyV1TaxonomiesPostRequest(BaseModel):
+    create_taxonomy_request: Annotated[
+        CreateTaxonomyRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
 
@@ -38,7 +38,7 @@ class IngestImageURLV1IngestImagesURLPostRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mixpeek/models/createblobrequest.py

@@ -0,0 +1,80 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .bucketschemafieldtype import BucketSchemaFieldType
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import Any
+from typing_extensions import NotRequired, TypedDict
+
+
+class CreateBlobRequestMetadataTypedDict(TypedDict):
+    pass
+
+
+class CreateBlobRequestMetadata(BaseModel):
+    pass
+
+
+class CreateBlobRequestTypedDict(TypedDict):
+    r"""Request model for creating a new blob"""
+
+    property: str
+    r"""Property name in the schema that this blob belongs to"""
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+    data: Any
+    r"""Data for the blob, this will only be applied to the documents that use this blob"""
+    key_prefix: NotRequired[Nullable[str]]
+    r"""Optional prefix for the blob key"""
+    metadata: NotRequired[Nullable[CreateBlobRequestMetadataTypedDict]]
+    r"""Metadata for the blob, this will only be applied to the documents that use this blob"""
+
+
+class CreateBlobRequest(BaseModel):
+    r"""Request model for creating a new blob"""
+
+    property: str
+    r"""Property name in the schema that this blob belongs to"""
+
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+
+    data: Any
+    r"""Data for the blob, this will only be applied to the documents that use this blob"""
+
+    key_prefix: OptionalNullable[str] = UNSET
+    r"""Optional prefix for the blob key"""
+
+    metadata: OptionalNullable[CreateBlobRequestMetadata] = UNSET
+    r"""Metadata for the blob, this will only be applied to the documents that use this blob"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["key_prefix", "metadata"]
+        nullable_fields = ["key_prefix", "metadata"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

mixpeek/models/createclusterrequest.py

@@ -0,0 +1,74 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .attributebasedconfig import AttributeBasedConfig, AttributeBasedConfigTypedDict
+from .automaticnaming import AutomaticNaming, AutomaticNamingTypedDict
+from .clustertype import ClusterType
+from .vectorbasedconfig import VectorBasedConfig, VectorBasedConfigTypedDict
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class CreateClusterRequestTypedDict(TypedDict):
+    collection_id: str
+    r"""ID of the collection to cluster"""
+    cluster_name: NotRequired[Nullable[str]]
+    r"""Name for the cluster (auto-generated if empty)"""
+    cluster_type: NotRequired[ClusterType]
+    vector_config: NotRequired[Nullable[VectorBasedConfigTypedDict]]
+    attribute_config: NotRequired[Nullable[AttributeBasedConfigTypedDict]]
+    automatic_naming: NotRequired[AutomaticNamingTypedDict]
+
+
+class CreateClusterRequest(BaseModel):
+    collection_id: str
+    r"""ID of the collection to cluster"""
+
+    cluster_name: OptionalNullable[str] = UNSET
+    r"""Name for the cluster (auto-generated if empty)"""
+
+    cluster_type: Optional[ClusterType] = None
+
+    vector_config: OptionalNullable[VectorBasedConfig] = UNSET
+
+    attribute_config: OptionalNullable[AttributeBasedConfig] = UNSET
+
+    automatic_naming: Optional[AutomaticNaming] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "cluster_name",
+            "cluster_type",
+            "vector_config",
+            "attribute_config",
+            "automatic_naming",
+        ]
+        nullable_fields = ["cluster_name", "vector_config", "attribute_config"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

mixpeek/models/createcollectionrequest.py

@@ -1,8 +1,26 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .collectioncacheconfig_input import (
+    CollectionCacheConfigInput,
+    CollectionCacheConfigInputTypedDict,
+)
+from .documenthandlingconfig import (
+    DocumentHandlingConfig,
+    DocumentHandlingConfigTypedDict,
+)
+from .featureextractorconfig import (
+    FeatureExtractorConfig,
+    FeatureExtractorConfigTypedDict,
+)
+from .sourceconfig_input import SourceConfigInput, SourceConfigInputTypedDict
+from .taxonomyapplicationconfig import (
+    TaxonomyApplicationConfig,
+    TaxonomyApplicationConfigTypedDict,
+)
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
+from typing import List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
@@ -15,30 +33,81 @@ class CreateCollectionRequestMetadata(BaseModel):
 
 
 class CreateCollectionRequestTypedDict(TypedDict):
+    r"""Request model for creating a new collection"""
+
     collection_name: str
     r"""Name for the collection"""
+    source: SourceConfigInputTypedDict
+    r"""Configuration for a collection source"""
+    feature_extractors: List[FeatureExtractorConfigTypedDict]
+    r"""List of feature extractor configurations to use"""
+    description: NotRequired[Nullable[str]]
+    r"""Description for the collection"""
+    taxonomy_applications: NotRequired[List[TaxonomyApplicationConfigTypedDict]]
+    r"""List of taxonomy application configurations. there are two options: on ingestion store the taxonomy application results to this collection, or on demand compute the taxonomy application results at query time"""
+    enabled: NotRequired[bool]
+    r"""Enable or disable processing of this collection"""
     metadata: NotRequired[Nullable[CreateCollectionRequestMetadataTypedDict]]
     r"""Optional metadata for the collection"""
+    document_handling: NotRequired[Nullable[DocumentHandlingConfigTypedDict]]
+    r"""Configuration for how documents are handled by this extractor"""
+    cache_config: NotRequired[Nullable[CollectionCacheConfigInputTypedDict]]
+    r"""Configuration for collection-level caching"""
 
 
 class CreateCollectionRequest(BaseModel):
+    r"""Request model for creating a new collection"""
+
     collection_name: str
     r"""Name for the collection"""
 
+    source: SourceConfigInput
+    r"""Configuration for a collection source"""
+
+    feature_extractors: List[FeatureExtractorConfig]
+    r"""List of feature extractor configurations to use"""
+
+    description: OptionalNullable[str] = UNSET
+    r"""Description for the collection"""
+
+    taxonomy_applications: Optional[List[TaxonomyApplicationConfig]] = None
+    r"""List of taxonomy application configurations. there are two options: on ingestion store the taxonomy application results to this collection, or on demand compute the taxonomy application results at query time"""
+
+    enabled: Optional[bool] = True
+    r"""Enable or disable processing of this collection"""
+
     metadata: OptionalNullable[CreateCollectionRequestMetadata] = UNSET
     r"""Optional metadata for the collection"""
 
+    document_handling: OptionalNullable[DocumentHandlingConfig] = UNSET
+    r"""Configuration for how documents are handled by this extractor"""
+
+    cache_config: OptionalNullable[CollectionCacheConfigInput] = UNSET
+    r"""Configuration for collection-level caching"""
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["metadata"]
-        nullable_fields = ["metadata"]
+        optional_fields = [
+            "description",
+            "taxonomy_applications",
+            "enabled",
+            "metadata",
+            "document_handling",
+            "cache_config",
+        ]
+        nullable_fields = [
+            "description",
+            "metadata",
+            "document_handling",
+            "cache_config",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)