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

mixpeek/models/namespaceresponse.py

@@ -1,17 +1,22 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mixpeek.types import BaseModel, Nullable, UNSET_SENTINEL
+from .basicfeatureextractor import BasicFeatureExtractor, BasicFeatureExtractorTypedDict
+from .namespacestatus import NamespaceStatus
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import Any, List
-from typing_extensions import TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class NamespaceResponseTypedDict(TypedDict):
     namespace_id: str
     namespace_name: str
-    embedding_models: List[str]
+    description: Nullable[str]
+    feature_extractors: List[BasicFeatureExtractorTypedDict]
     payload_indexes: Nullable[List[Any]]
+    status: NamespaceStatus
+    provisioning_task_id: NotRequired[Nullable[str]]
 
 
 class NamespaceResponse(BaseModel):
@@ -19,14 +24,20 @@ class NamespaceResponse(BaseModel):
 
     namespace_name: str
 
-    embedding_models: List[str]
+    description: Nullable[str]
+
+    feature_extractors: List[BasicFeatureExtractor]
 
     payload_indexes: Nullable[List[Any]]
 
+    status: NamespaceStatus
+
+    provisioning_task_id: OptionalNullable[str] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = []
-        nullable_fields = ["payload_indexes"]
+        optional_fields = ["provisioning_task_id"]
+        nullable_fields = ["description", "payload_indexes", "provisioning_task_id"]
         null_default_fields = []
 
         serialized = handler(self)

mixpeek/models/namespacestatus.py

@@ -0,0 +1,10 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from enum import Enum
+
+
+class NamespaceStatus(str, Enum):
+    PENDING = "pending"
+    READY = "ready"
+    FAILED = "failed"

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

@@ -1,30 +1,25 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .bucketschema_output import BucketSchemaOutput, BucketSchemaOutputTypedDict
 from .retrievercacheconfig import RetrieverCacheConfig, RetrieverCacheConfigTypedDict
+from .retrieverschema_output import (
+    RetrieverSchemaOutput,
+    RetrieverSchemaOutputTypedDict,
+)
 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"""
 
     retriever_name: str
     r"""Name of the retriever"""
-    input_schema: BucketSchemaOutputTypedDict
-    r"""Schema definition for bucket objects"""
+    input_schema: RetrieverSchemaOutputTypedDict
+    r"""Schema definition for retriever inputs"""
     collection_ids: List[str]
     r"""List of collection IDs to search in"""
     stages: List[StageConfigOutputTypedDict]
@@ -33,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"""
 
@@ -44,8 +39,8 @@ class RetrieverModel(BaseModel):
     retriever_name: str
     r"""Name of the retriever"""
 
-    input_schema: BucketSchemaOutput
-    r"""Schema definition for bucket objects"""
+    input_schema: RetrieverSchemaOutput
+    r"""Schema definition for retriever inputs"""
 
     collection_ids: List[str]
     r"""List of collection IDs to search in"""
@@ -59,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/retrieverschema_input.py

@@ -0,0 +1,37 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .retrieverschemafield_input import (
+    RetrieverSchemaFieldInput,
+    RetrieverSchemaFieldInputTypedDict,
+)
+from mixpeek.types import BaseModel
+import pydantic
+from pydantic import ConfigDict
+from typing import Any, Dict
+from typing_extensions import TypedDict
+
+
+class RetrieverSchemaInputTypedDict(TypedDict):
+    r"""Schema definition for retriever inputs"""
+
+    properties: Dict[str, RetrieverSchemaFieldInputTypedDict]
+
+
+class RetrieverSchemaInput(BaseModel):
+    r"""Schema 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)
+
+    properties: Dict[str, RetrieverSchemaFieldInput]
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]

mixpeek/models/retrieverschema_output.py

@@ -0,0 +1,37 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .retrieverschemafield_output import (
+    RetrieverSchemaFieldOutput,
+    RetrieverSchemaFieldOutputTypedDict,
+)
+from mixpeek.types import BaseModel
+import pydantic
+from pydantic import ConfigDict
+from typing import Any, Dict
+from typing_extensions import TypedDict
+
+
+class RetrieverSchemaOutputTypedDict(TypedDict):
+    r"""Schema definition for retriever inputs"""
+
+    properties: Dict[str, RetrieverSchemaFieldOutputTypedDict]
+
+
+class RetrieverSchemaOutput(BaseModel):
+    r"""Schema 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)
+
+    properties: Dict[str, RetrieverSchemaFieldOutput]
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]

mixpeek/models/retrieverschemafield_input.py

@@ -0,0 +1,91 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .bucketschemafieldbase import BucketSchemaFieldBase, BucketSchemaFieldBaseTypedDict
+from .bucketschemafieldtype import BucketSchemaFieldType
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import Any, Dict, List, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class RetrieverSchemaFieldInputTypedDict(TypedDict):
+    r"""Schema field definition for retriever inputs"""
+
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+    default: NotRequired[Nullable[Any]]
+    items: NotRequired[Nullable[BucketSchemaFieldBaseTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldBaseTypedDict]]]
+    example: NotRequired[Nullable[Any]]
+    description: NotRequired[Nullable[str]]
+    enum: NotRequired[Nullable[List[Any]]]
+    required: NotRequired[bool]
+    r"""Whether this field is required for the retriever"""
+
+
+class RetrieverSchemaFieldInput(BaseModel):
+    r"""Schema field definition for retriever inputs"""
+
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+
+    default: OptionalNullable[Any] = UNSET
+
+    items: OptionalNullable[BucketSchemaFieldBase] = UNSET
+
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldBase]] = UNSET
+
+    example: OptionalNullable[Any] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    enum: OptionalNullable[List[Any]] = UNSET
+
+    required: Optional[bool] = True
+    r"""Whether this field is required for the retriever"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "default",
+            "items",
+            "properties",
+            "example",
+            "description",
+            "enum",
+            "required",
+        ]
+        nullable_fields = [
+            "default",
+            "items",
+            "properties",
+            "example",
+            "description",
+            "enum",
+        ]
+        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/retrieverschemafield_output.py

@@ -0,0 +1,91 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .bucketschemafieldbase import BucketSchemaFieldBase, BucketSchemaFieldBaseTypedDict
+from .bucketschemafieldtype import BucketSchemaFieldType
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import Any, Dict, List, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class RetrieverSchemaFieldOutputTypedDict(TypedDict):
+    r"""Schema field definition for retriever inputs"""
+
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+    default: NotRequired[Nullable[Any]]
+    items: NotRequired[Nullable[BucketSchemaFieldBaseTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldBaseTypedDict]]]
+    example: NotRequired[Nullable[Any]]
+    description: NotRequired[Nullable[str]]
+    enum: NotRequired[Nullable[List[Any]]]
+    required: NotRequired[bool]
+    r"""Whether this field is required for the retriever"""
+
+
+class RetrieverSchemaFieldOutput(BaseModel):
+    r"""Schema field definition for retriever inputs"""
+
+    type: BucketSchemaFieldType
+    r"""Enum for field types in bucket schemas"""
+
+    default: OptionalNullable[Any] = UNSET
+
+    items: OptionalNullable[BucketSchemaFieldBase] = UNSET
+
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldBase]] = UNSET
+
+    example: OptionalNullable[Any] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    enum: OptionalNullable[List[Any]] = UNSET
+
+    required: Optional[bool] = True
+    r"""Whether this field is required for the retriever"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "default",
+            "items",
+            "properties",
+            "example",
+            "description",
+            "enum",
+            "required",
+        ]
+        nullable_fields = [
+            "default",
+            "items",
+            "properties",
+            "example",
+            "description",
+            "enum",
+        ]
+        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/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