mixpeek 0.17.5__tar.gz → 0.17.7__tar.gz

{mixpeek-0.17.5 → mixpeek-0.17.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mixpeek
-Version: 0.17.5
+Version: 0.17.7
 Summary: Python Client SDK Generated by Speakeasy.
 Author: Speakeasy
 Requires-Python: >=3.9
@@ -226,6 +226,7 @@ with Mixpeek(
 
 * [kill_task_v1_tasks_task_id_delete](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#kill_task_v1_tasks_task_id_delete) - Kill Task
 * [get_task_v1_tasks_task_id_get](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#get_task_v1_tasks_task_id_get) - Get Task Information
+* [list_active_tasks_v1_tasks_get](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#list_active_tasks_v1_tasks_get) - List Active Tasks
 
 ### [taxonomy_entities](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/taxonomyentities/README.md)
 

{mixpeek-0.17.5 → mixpeek-0.17.7}/README-PYPI.md

@@ -205,6 +205,7 @@ with Mixpeek(
 
 * [kill_task_v1_tasks_task_id_delete](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#kill_task_v1_tasks_task_id_delete) - Kill Task
 * [get_task_v1_tasks_task_id_get](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#get_task_v1_tasks_task_id_get) - Get Task Information
+* [list_active_tasks_v1_tasks_get](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/tasks/README.md#list_active_tasks_v1_tasks_get) - List Active Tasks
 
 ### [taxonomy_entities](https://github.com/mixpeek/python-sdk/blob/master/docs/sdks/taxonomyentities/README.md)
 

{mixpeek-0.17.5 → mixpeek-0.17.7}/pyproject.toml

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

{mixpeek-0.17.5 → mixpeek-0.17.7}/src/mixpeek/_version.py

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

{mixpeek-0.17.5 → mixpeek-0.17.7}/src/mixpeek/models/__init__.py

@@ -19,6 +19,7 @@ from .assets_model_searchquery import (
 )
 from .assetupdate import AssetUpdate, AssetUpdateTypedDict, Mode
 from .assignmentconfig import AssignmentConfig, AssignmentConfigTypedDict
+from .assignmentmode import AssignmentMode
 from .availablemodels import AvailableModels
 from .availablemodelsresponse import (
     AvailableModelsResponse,
@@ -246,6 +247,10 @@ from .kill_task_v1_tasks_task_id_deleteop import (
     KillTaskV1TasksTaskIDDeleteRequest,
     KillTaskV1TasksTaskIDDeleteRequestTypedDict,
 )
+from .list_active_tasks_v1_tasks_getop import (
+    ListActiveTasksV1TasksGetRequest,
+    ListActiveTasksV1TasksGetRequestTypedDict,
+)
 from .list_assets_v1_assets_postop import (
     ListAssetsV1AssetsPostRequest,
     ListAssetsV1AssetsPostRequestTypedDict,
@@ -282,6 +287,7 @@ from .listcollectionsresponse import (
 )
 from .listfeaturesrequest import ListFeaturesRequest, ListFeaturesRequestTypedDict
 from .listfeaturesresponse import ListFeaturesResponse, ListFeaturesResponseTypedDict
+from .listtasksresponse import ListTasksResponse, ListTasksResponseTypedDict
 from .listtaxonomiesresponse import (
     ListTaxonomiesResponse,
     ListTaxonomiesResponseTypedDict,
@@ -469,6 +475,7 @@ __all__ = [
     "AssetsModelSearchQueryTypedDict",
     "AssignmentConfig",
     "AssignmentConfigTypedDict",
+    "AssignmentMode",
     "AvailableModels",
     "AvailableModelsResponse",
     "AvailableModelsResponseTypedDict",
@@ -637,6 +644,8 @@ __all__ = [
     "KeywordIndexParamsTypedDict",
     "KillTaskV1TasksTaskIDDeleteRequest",
     "KillTaskV1TasksTaskIDDeleteRequestTypedDict",
+    "ListActiveTasksV1TasksGetRequest",
+    "ListActiveTasksV1TasksGetRequestTypedDict",
     "ListAssetsRequest",
     "ListAssetsRequestTypedDict",
     "ListAssetsResponse",
@@ -659,6 +668,8 @@ __all__ = [
     "ListFeaturesResponseTypedDict",
     "ListFeaturesV1FeaturesPostRequest",
     "ListFeaturesV1FeaturesPostRequestTypedDict",
+    "ListTasksResponse",
+    "ListTasksResponseTypedDict",
     "ListTaxonomiesResponse",
     "ListTaxonomiesResponseTypedDict",
     "ListTaxonomiesV1EntitiesTaxonomiesGetRequest",

mixpeek-0.17.7/src/mixpeek/models/assignmentconfig.py

@@ -0,0 +1,67 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .assignmentmode import AssignmentMode
+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 AssignmentConfigTypedDict(TypedDict):
+    r"""Configuration for how classifications should be assigned to features"""
+
+    enabled: NotRequired[bool]
+    r"""Whether to assign the taxonomy to the feature"""
+    mode: NotRequired[AssignmentMode]
+    r"""Mode for how classifications should be assigned"""
+    append: NotRequired[bool]
+    r"""Whether to append the classification to the feature, if false, replaces any existing classification"""
+    confidence_threshold: NotRequired[Nullable[float]]
+    r"""Minimum confidence score required for classification (only used in threshold mode)"""
+
+
+class AssignmentConfig(BaseModel):
+    r"""Configuration for how classifications should be assigned to features"""
+
+    enabled: Optional[bool] = False
+    r"""Whether to assign the taxonomy to the feature"""
+
+    mode: Optional[AssignmentMode] = None
+    r"""Mode for how classifications should be assigned"""
+
+    append: Optional[bool] = False
+    r"""Whether to append the classification to the feature, if false, replaces any existing classification"""
+
+    confidence_threshold: OptionalNullable[float] = UNSET
+    r"""Minimum confidence score required for classification (only used in threshold mode)"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["enabled", "mode", "append", "confidence_threshold"]
+        nullable_fields = ["confidence_threshold"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in 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-0.17.7/src/mixpeek/models/assignmentmode.py

@@ -0,0 +1,11 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from enum import Enum
+
+
+class AssignmentMode(str, Enum):
+    r"""Mode for how classifications should be assigned"""
+
+    THRESHOLD = "threshold"
+    NEAREST = "nearest"

{mixpeek-0.17.5 → mixpeek-0.17.7}/src/mixpeek/models/discoverrequest.py

@@ -14,12 +14,10 @@ class DiscoverRequestTypedDict(TypedDict):
     r"""List of collection names or ids to search for features"""
     filters: NotRequired[Nullable[LogicalOperatorTypedDict]]
     r"""Filters to apply to the discovery task"""
-    confidence_threshold: NotRequired[float]
-    r"""Minimum confidence score required for classification"""
     assignment: NotRequired[AssignmentConfigTypedDict]
     r"""Configuration for how classifications should be assigned to features"""
-    sample_size: NotRequired[Nullable[int]]
-    r"""Number of feature samples to process"""
+    limit: NotRequired[Nullable[int]]
+    r"""Number of feature samples to process, if None, all features that match the filters are processed"""
 
 
 class DiscoverRequest(BaseModel):
@@ -29,24 +27,16 @@ class DiscoverRequest(BaseModel):
     filters: OptionalNullable[LogicalOperator] = UNSET
     r"""Filters to apply to the discovery task"""
 
-    confidence_threshold: Optional[float] = 0.8
-    r"""Minimum confidence score required for classification"""
-
     assignment: Optional[AssignmentConfig] = None
     r"""Configuration for how classifications should be assigned to features"""
 
-    sample_size: OptionalNullable[int] = UNSET
-    r"""Number of feature samples to process"""
+    limit: OptionalNullable[int] = UNSET
+    r"""Number of feature samples to process, if None, all features that match the filters are processed"""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "filters",
-            "confidence_threshold",
-            "assignment",
-            "sample_size",
-        ]
-        nullable_fields = ["filters", "sample_size"]
+        optional_fields = ["filters", "assignment", "limit"]
+        nullable_fields = ["filters", "limit"]
         null_default_fields = []
 
         serialized = handler(self)

mixpeek-0.17.7/src/mixpeek/models/list_active_tasks_v1_tasks_getop.py

@@ -0,0 +1,65 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mixpeek.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mixpeek.utils import FieldMetadata, HeaderMetadata, QueryParamMetadata
+import pydantic
+from pydantic import model_serializer
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class ListActiveTasksV1TasksGetRequestTypedDict(TypedDict):
+    page: NotRequired[Nullable[int]]
+    page_size: NotRequired[int]
+    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 ListActiveTasksV1TasksGetRequest(BaseModel):
+    page: Annotated[
+        OptionalNullable[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = UNSET
+
+    page_size: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 10
+
+    x_namespace: Annotated[
+        OptionalNullable[str],
+        pydantic.Field(alias="X-Namespace"),
+        FieldMetadata(header=HeaderMetadata(style="simple", explode=False)),
+    ] = UNSET
+    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."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["page", "page_size", "X-Namespace"]
+        nullable_fields = ["page", "X-Namespace"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in 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-0.17.7/src/mixpeek/models/listtasksresponse.py

@@ -0,0 +1,22 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .db_model_paginationresponse import (
+    DbModelPaginationResponse,
+    DbModelPaginationResponseTypedDict,
+)
+from .taskresponse import TaskResponse, TaskResponseTypedDict
+from mixpeek.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListTasksResponseTypedDict(TypedDict):
+    results: List[TaskResponseTypedDict]
+    pagination: DbModelPaginationResponseTypedDict
+
+
+class ListTasksResponse(BaseModel):
+    results: List[TaskResponse]
+
+    pagination: DbModelPaginationResponse

{mixpeek-0.17.5 → mixpeek-0.17.7}/src/mixpeek/tasks.py

@@ -436,3 +436,219 @@ class Tasks(BaseSDK):
             http_res_text,
             http_res,
         )
+
+    def list_active_tasks_v1_tasks_get(
+        self,
+        *,
+        page: OptionalNullable[int] = UNSET,
+        page_size: Optional[int] = 10,
+        x_namespace: OptionalNullable[str] = UNSET,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.ListTasksResponse:
+        r"""List Active Tasks
+
+        Retrieve all tasks that are not in a complete state (DONE, FAILED, SKIPPED, or CANCELLED)
+
+        :param page:
+        :param page_size:
+        :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.
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.ListActiveTasksV1TasksGetRequest(
+            page=page,
+            page_size=page_size,
+            x_namespace=x_namespace,
+        )
+
+        req = self._build_request(
+            method="GET",
+            path="/v1/tasks",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                operation_id="list_active_tasks_v1_tasks_get",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
+            retry_config=retry_config,
+        )
+
+        data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.ListTasksResponse)
+        if utils.match_response(
+            http_res, ["400", "401", "403", "404"], "application/json"
+        ):
+            data = utils.unmarshal_json(http_res.text, models.ErrorResponseData)
+            raise models.ErrorResponse(data=data)
+        if utils.match_response(http_res, "422", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
+            raise models.HTTPValidationError(data=data)
+        if utils.match_response(http_res, "500", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.ErrorResponseData)
+            raise models.ErrorResponse(data=data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.APIError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.APIError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.APIError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def list_active_tasks_v1_tasks_get_async(
+        self,
+        *,
+        page: OptionalNullable[int] = UNSET,
+        page_size: Optional[int] = 10,
+        x_namespace: OptionalNullable[str] = UNSET,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.ListTasksResponse:
+        r"""List Active Tasks
+
+        Retrieve all tasks that are not in a complete state (DONE, FAILED, SKIPPED, or CANCELLED)
+
+        :param page:
+        :param page_size:
+        :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.
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.ListActiveTasksV1TasksGetRequest(
+            page=page,
+            page_size=page_size,
+            x_namespace=x_namespace,
+        )
+
+        req = self._build_request_async(
+            method="GET",
+            path="/v1/tasks",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                operation_id="list_active_tasks_v1_tasks_get",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
+            retry_config=retry_config,
+        )
+
+        data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.ListTasksResponse)
+        if utils.match_response(
+            http_res, ["400", "401", "403", "404"], "application/json"
+        ):
+            data = utils.unmarshal_json(http_res.text, models.ErrorResponseData)
+            raise models.ErrorResponse(data=data)
+        if utils.match_response(http_res, "422", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
+            raise models.HTTPValidationError(data=data)
+        if utils.match_response(http_res, "500", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.ErrorResponseData)
+            raise models.ErrorResponse(data=data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.APIError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.APIError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.APIError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )

{mixpeek-0.17.5 → mixpeek-0.17.7}/src/mixpeek/taxonomy_entities.py

@@ -1594,11 +1594,10 @@ class TaxonomyEntities(BaseSDK):
         filters: OptionalNullable[
             Union[models.LogicalOperator, models.LogicalOperatorTypedDict]
         ] = UNSET,
-        confidence_threshold: Optional[float] = 0.8,
         assignment: Optional[
             Union[models.AssignmentConfig, models.AssignmentConfigTypedDict]
         ] = None,
-        sample_size: OptionalNullable[int] = UNSET,
+        limit: OptionalNullable[int] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1612,9 +1611,8 @@ class TaxonomyEntities(BaseSDK):
         :param collections: List of collection names or ids to search for features
         :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.
         :param filters: Filters to apply to the discovery task
-        :param confidence_threshold: Minimum confidence score required for classification
         :param assignment: Configuration for how classifications should be assigned to features
-        :param sample_size: Number of feature samples to process
+        :param limit: Number of feature samples to process, if None, all features that match the filters are processed
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -1637,11 +1635,10 @@ class TaxonomyEntities(BaseSDK):
                     filters=utils.get_pydantic_model(
                         filters, OptionalNullable[models.LogicalOperator]
                     ),
-                    confidence_threshold=confidence_threshold,
                     assignment=utils.get_pydantic_model(
                         assignment, Optional[models.AssignmentConfig]
                     ),
-                    sample_size=sample_size,
+                    limit=limit,
                 ),
             )
         )
@@ -1729,11 +1726,10 @@ class TaxonomyEntities(BaseSDK):
         filters: OptionalNullable[
             Union[models.LogicalOperator, models.LogicalOperatorTypedDict]
         ] = UNSET,
-        confidence_threshold: Optional[float] = 0.8,
         assignment: Optional[
             Union[models.AssignmentConfig, models.AssignmentConfigTypedDict]
         ] = None,
-        sample_size: OptionalNullable[int] = UNSET,
+        limit: OptionalNullable[int] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1747,9 +1743,8 @@ class TaxonomyEntities(BaseSDK):
         :param collections: List of collection names or ids to search for features
         :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.
         :param filters: Filters to apply to the discovery task
-        :param confidence_threshold: Minimum confidence score required for classification
         :param assignment: Configuration for how classifications should be assigned to features
-        :param sample_size: Number of feature samples to process
+        :param limit: Number of feature samples to process, if None, all features that match the filters are processed
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -1772,11 +1767,10 @@ class TaxonomyEntities(BaseSDK):
                     filters=utils.get_pydantic_model(
                         filters, OptionalNullable[models.LogicalOperator]
                     ),
-                    confidence_threshold=confidence_threshold,
                     assignment=utils.get_pydantic_model(
                         assignment, Optional[models.AssignmentConfig]
                     ),
-                    sample_size=sample_size,
+                    limit=limit,
                 ),
             )
         )

mixpeek-0.17.5/src/mixpeek/models/assignmentconfig.py

@@ -1,25 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-from __future__ import annotations
-from mixpeek.types import BaseModel
-from typing import Optional
-from typing_extensions import NotRequired, TypedDict
-
-
-class AssignmentConfigTypedDict(TypedDict):
-    r"""Configuration for how classifications should be assigned to features"""
-
-    enabled: NotRequired[bool]
-    r"""Whether to assign the taxonomy to the feature"""
-    append: NotRequired[bool]
-    r"""Whether to append the classification to the feature, if false, replaces any existing classification"""
-
-
-class AssignmentConfig(BaseModel):
-    r"""Configuration for how classifications should be assigned to features"""
-
-    enabled: Optional[bool] = False
-    r"""Whether to assign the taxonomy to the feature"""
-
-    append: Optional[bool] = False
-    r"""Whether to append the classification to the feature, if false, replaces any existing classification"""