mixpeek 0.21.6__tar.gz → 0.21.8__tar.gz

{mixpeek-0.21.6 → mixpeek-0.21.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mixpeek
-Version: 0.21.6
+Version: 0.21.8
 Summary: Python Client SDK Generated by Speakeasy.
 Author: Speakeasy
 Requires-Python: >=3.9

{mixpeek-0.21.6 → mixpeek-0.21.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mixpeek"
-version = "0.21.6"
+version = "0.21.8"
 description = "Python Client SDK Generated by Speakeasy."
 authors = [{ name = "Speakeasy" },]
 readme = "README-PYPI.md"

{mixpeek-0.21.6 → mixpeek-0.21.8}/src/mixpeek/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mixpeek"
-__version__: str = "0.21.6"
+__version__: str = "0.21.8"
 __openapi_doc_version__: str = "0.81"
 __gen_version__: str = "2.585.2"
-__user_agent__: str = "speakeasy-sdk/python 0.21.6 2.585.2 0.81 mixpeek"
+__user_agent__: str = "speakeasy-sdk/python 0.21.8 2.585.2 0.81 mixpeek"
 
 try:
     if __package__ is not None:

{mixpeek-0.21.6 → mixpeek-0.21.8}/src/mixpeek/models/__init__.py

@@ -389,6 +389,19 @@ from .retrieverqueryrequest import (
     RetrieverQueryRequestTypedDict,
 )
 from .retrieverresponse import RetrieverResponse, RetrieverResponseTypedDict
+from .retrieverschema_input import RetrieverSchemaInput, RetrieverSchemaInputTypedDict
+from .retrieverschema_output import (
+    RetrieverSchemaOutput,
+    RetrieverSchemaOutputTypedDict,
+)
+from .retrieverschemafield_input import (
+    RetrieverSchemaFieldInput,
+    RetrieverSchemaFieldInputTypedDict,
+)
+from .retrieverschemafield_output import (
+    RetrieverSchemaFieldOutput,
+    RetrieverSchemaFieldOutputTypedDict,
+)
 from .searchinteraction import (
     SearchInteraction,
     SearchInteractionMetadata,
@@ -804,6 +817,14 @@ __all__ = [
     "RetrieverQueryRequestTypedDict",
     "RetrieverResponse",
     "RetrieverResponseTypedDict",
+    "RetrieverSchemaFieldInput",
+    "RetrieverSchemaFieldInputTypedDict",
+    "RetrieverSchemaFieldOutput",
+    "RetrieverSchemaFieldOutputTypedDict",
+    "RetrieverSchemaInput",
+    "RetrieverSchemaInputTypedDict",
+    "RetrieverSchemaOutput",
+    "RetrieverSchemaOutputTypedDict",
     "SearchInteraction",
     "SearchInteractionMetadata",
     "SearchInteractionMetadataTypedDict",

{mixpeek-0.21.6 → mixpeek-0.21.8}/src/mixpeek/models/createretrieverrequest.py

@@ -1,8 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .bucketschema_input import BucketSchemaInput, BucketSchemaInputTypedDict
 from .retrievercacheconfig import RetrieverCacheConfig, RetrieverCacheConfigTypedDict
+from .retrieverschema_input import RetrieverSchemaInput, RetrieverSchemaInputTypedDict
 from .stageconfig_input import StageConfigInput, StageConfigInputTypedDict
 from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
@@ -23,8 +23,8 @@ class CreateRetrieverRequestTypedDict(TypedDict):
 
     retriever_name: str
     r"""Name of the retriever"""
-    input_schema: BucketSchemaInputTypedDict
-    r"""Schema definition for bucket objects"""
+    input_schema: RetrieverSchemaInputTypedDict
+    r"""Schema definition for retriever inputs"""
     collection_ids: List[str]
     r"""List of collection IDs to search in"""
     stages: List[StageConfigInputTypedDict]
@@ -42,8 +42,8 @@ class CreateRetrieverRequest(BaseModel):
     retriever_name: str
     r"""Name of the retriever"""
 
-    input_schema: BucketSchemaInput
-    r"""Schema definition for bucket objects"""
+    input_schema: RetrieverSchemaInput
+    r"""Schema definition for retriever inputs"""
 
     collection_ids: List[str]
     r"""List of collection IDs to search in"""

{mixpeek-0.21.6 → mixpeek-0.21.8}/src/mixpeek/models/retrievermodel.py

@@ -1,8 +1,11 @@
 """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
@@ -23,8 +26,8 @@ class RetrieverModelTypedDict(TypedDict):
 
     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]
@@ -44,8 +47,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"""

mixpeek-0.21.8/src/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-0.21.8/src/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-0.21.8/src/mixpeek/models/retrieverschemafield_input.py

@@ -0,0 +1,111 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .bucketschemafield_input import (
+    BucketSchemaFieldInput,
+    BucketSchemaFieldInputTypedDict,
+)
+from .bucketschemafieldtype import BucketSchemaFieldType
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+import pydantic
+from pydantic import ConfigDict, 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[BucketSchemaFieldInputTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldInputTypedDict]]]
+    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"""
+
+    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
+
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldInput]] = 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"""
+
+    @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 = [
+            "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
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

mixpeek-0.21.8/src/mixpeek/models/retrieverschemafield_output.py

@@ -0,0 +1,111 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .bucketschemafield_output import (
+    BucketSchemaFieldOutput,
+    BucketSchemaFieldOutputTypedDict,
+)
+from .bucketschemafieldtype import BucketSchemaFieldType
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+import pydantic
+from pydantic import ConfigDict, 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[BucketSchemaFieldOutputTypedDict]]
+    properties: NotRequired[Nullable[Dict[str, BucketSchemaFieldOutputTypedDict]]]
+    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"""
+
+    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
+
+    properties: OptionalNullable[Dict[str, BucketSchemaFieldOutput]] = 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"""
+
+    @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 = [
+            "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
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

{mixpeek-0.21.6 → mixpeek-0.21.8}/src/mixpeek/retrievers.py

@@ -14,7 +14,7 @@ class Retrievers(BaseSDK):
         *,
         retriever_name: str,
         input_schema: Union[
-            models.BucketSchemaInput, models.BucketSchemaInputTypedDict
+            models.RetrieverSchemaInput, models.RetrieverSchemaInputTypedDict
         ],
         collection_ids: List[str],
         stages: Union[
@@ -39,7 +39,7 @@ class Retrievers(BaseSDK):
         r"""Create Retriever
 
         :param retriever_name: Name of the retriever
-        :param input_schema: Schema definition for bucket objects
+        :param input_schema: Schema definition for retriever inputs
         :param collection_ids: List of collection IDs to search in
         :param stages: List of stages to execute in order
         :param x_namespace: 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.
@@ -67,7 +67,7 @@ class Retrievers(BaseSDK):
                 retriever_name=retriever_name,
                 description=description,
                 input_schema=utils.get_pydantic_model(
-                    input_schema, models.BucketSchemaInput
+                    input_schema, models.RetrieverSchemaInput
                 ),
                 collection_ids=collection_ids,
                 stages=utils.get_pydantic_model(stages, List[models.StageConfigInput]),
@@ -170,7 +170,7 @@ class Retrievers(BaseSDK):
         *,
         retriever_name: str,
         input_schema: Union[
-            models.BucketSchemaInput, models.BucketSchemaInputTypedDict
+            models.RetrieverSchemaInput, models.RetrieverSchemaInputTypedDict
         ],
         collection_ids: List[str],
         stages: Union[
@@ -195,7 +195,7 @@ class Retrievers(BaseSDK):
         r"""Create Retriever
 
         :param retriever_name: Name of the retriever
-        :param input_schema: Schema definition for bucket objects
+        :param input_schema: Schema definition for retriever inputs
         :param collection_ids: List of collection IDs to search in
         :param stages: List of stages to execute in order
         :param x_namespace: 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.
@@ -223,7 +223,7 @@ class Retrievers(BaseSDK):
                 retriever_name=retriever_name,
                 description=description,
                 input_schema=utils.get_pydantic_model(
-                    input_schema, models.BucketSchemaInput
+                    input_schema, models.RetrieverSchemaInput
                 ),
                 collection_ids=collection_ids,
                 stages=utils.get_pydantic_model(stages, List[models.StageConfigInput]),