orca-sdk 0.0.96__py3-none-any.whl → 0.0.97__py3-none-any.whl

orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py

@@ -0,0 +1,239 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
+
+The main changes are:
+- Update the API call responses to either return the successful response type or raise an error by:
+    - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
+- Inject a client into every method via a context manager
+"""
+
+from http import HTTPStatus
+from typing import Any, Dict, Union
+from uuid import UUID
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.add_memory_recommendations import AddMemoryRecommendations
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    prediction_id: UUID,
+    *,
+    num_memories: Union[Unset, int] = 3,
+    refresh: Union[Unset, bool] = False,
+) -> dict[str, Any]:
+    params: Dict[str, Any] = {}
+
+    params["num_memories"] = num_memories
+
+    params["refresh"] = refresh
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": f"/telemetry/prediction/{prediction_id}/memory-suggestions",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> AddMemoryRecommendations:
+    if response.status_code == 200:
+        response_200 = AddMemoryRecommendations.from_dict(response.json())
+
+        return response_200
+    if response.status_code == 401:
+        raise get_error_for_response(response)
+    if response.status_code == 500:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    if response.status_code == 404:
+        raise get_error_for_response(response)
+    if response.status_code == 409:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[AddMemoryRecommendations]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    prediction_id: UUID,
+    *,
+    num_memories: Union[Unset, int] = 3,
+    refresh: Union[Unset, bool] = False,
+) -> Response[AddMemoryRecommendations]:
+    """Generate Memory Suggestions
+
+     Generate synthetic memory suggestions to improve a specific prediction.
+
+    The returned suggestions have labels as string representations of integer indices
+    corresponding to the memoryset's label_names.
+
+    Params:
+        prediction_id: ID of the prediction to generate suggestions for
+        num_memories: Number of memory suggestions to generate (default: 3)
+        refresh: Force the explanation agent to re-run even if a cached explanation exists
+
+    Args:
+        prediction_id (UUID):
+        num_memories (Union[Unset, int]):  Default: 3.
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[AddMemoryRecommendations]
+    """
+
+    kwargs = _get_kwargs(
+        prediction_id=prediction_id,
+        num_memories=num_memories,
+        refresh=refresh,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    prediction_id: UUID,
+    *,
+    num_memories: Union[Unset, int] = 3,
+    refresh: Union[Unset, bool] = False,
+) -> AddMemoryRecommendations:
+    """Generate Memory Suggestions
+
+     Generate synthetic memory suggestions to improve a specific prediction.
+
+    The returned suggestions have labels as string representations of integer indices
+    corresponding to the memoryset's label_names.
+
+    Params:
+        prediction_id: ID of the prediction to generate suggestions for
+        num_memories: Number of memory suggestions to generate (default: 3)
+        refresh: Force the explanation agent to re-run even if a cached explanation exists
+
+    Args:
+        prediction_id (UUID):
+        num_memories (Union[Unset, int]):  Default: 3.
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        AddMemoryRecommendations
+    """
+
+    return sync_detailed(
+        prediction_id=prediction_id,
+        num_memories=num_memories,
+        refresh=refresh,
+    ).parsed
+
+
+async def asyncio_detailed(
+    prediction_id: UUID,
+    *,
+    num_memories: Union[Unset, int] = 3,
+    refresh: Union[Unset, bool] = False,
+) -> Response[AddMemoryRecommendations]:
+    """Generate Memory Suggestions
+
+     Generate synthetic memory suggestions to improve a specific prediction.
+
+    The returned suggestions have labels as string representations of integer indices
+    corresponding to the memoryset's label_names.
+
+    Params:
+        prediction_id: ID of the prediction to generate suggestions for
+        num_memories: Number of memory suggestions to generate (default: 3)
+        refresh: Force the explanation agent to re-run even if a cached explanation exists
+
+    Args:
+        prediction_id (UUID):
+        num_memories (Union[Unset, int]):  Default: 3.
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[AddMemoryRecommendations]
+    """
+
+    kwargs = _get_kwargs(
+        prediction_id=prediction_id,
+        num_memories=num_memories,
+        refresh=refresh,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    prediction_id: UUID,
+    *,
+    num_memories: Union[Unset, int] = 3,
+    refresh: Union[Unset, bool] = False,
+) -> AddMemoryRecommendations:
+    """Generate Memory Suggestions
+
+     Generate synthetic memory suggestions to improve a specific prediction.
+
+    The returned suggestions have labels as string representations of integer indices
+    corresponding to the memoryset's label_names.
+
+    Params:
+        prediction_id: ID of the prediction to generate suggestions for
+        num_memories: Number of memory suggestions to generate (default: 3)
+        refresh: Force the explanation agent to re-run even if a cached explanation exists
+
+    Args:
+        prediction_id (UUID):
+        num_memories (Union[Unset, int]):  Default: 3.
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        AddMemoryRecommendations
+    """
+
+    return (
+        await asyncio_detailed(
+            prediction_id=prediction_id,
+            num_memories=num_memories,
+            refresh=refresh,
+        )
+    ).parsed

orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py

@@ -0,0 +1,192 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
+
+The main changes are:
+- Update the API call responses to either return the successful response type or raise an error by:
+    - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
+- Inject a client into every method via a context manager
+"""
+
+from http import HTTPStatus
+from typing import Any, Dict, Union
+from uuid import UUID
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.action_recommendation import ActionRecommendation
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    prediction_id: UUID,
+    *,
+    refresh: Union[Unset, bool] = False,
+) -> dict[str, Any]:
+    params: Dict[str, Any] = {}
+
+    params["refresh"] = refresh
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/telemetry/prediction/{prediction_id}/action",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> ActionRecommendation:
+    if response.status_code == 200:
+        response_200 = ActionRecommendation.from_dict(response.json())
+
+        return response_200
+    if response.status_code == 401:
+        raise get_error_for_response(response)
+    if response.status_code == 500:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    if response.status_code == 404:
+        raise get_error_for_response(response)
+    if response.status_code == 409:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[ActionRecommendation]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    prediction_id: UUID,
+    *,
+    refresh: Union[Unset, bool] = False,
+) -> Response[ActionRecommendation]:
+    """Get Action Recommendation
+
+     Get action recommendation for improving a specific prediction.
+
+    Args:
+        prediction_id (UUID):
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[ActionRecommendation]
+    """
+
+    kwargs = _get_kwargs(
+        prediction_id=prediction_id,
+        refresh=refresh,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    prediction_id: UUID,
+    *,
+    refresh: Union[Unset, bool] = False,
+) -> ActionRecommendation:
+    """Get Action Recommendation
+
+     Get action recommendation for improving a specific prediction.
+
+    Args:
+        prediction_id (UUID):
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        ActionRecommendation
+    """
+
+    return sync_detailed(
+        prediction_id=prediction_id,
+        refresh=refresh,
+    ).parsed
+
+
+async def asyncio_detailed(
+    prediction_id: UUID,
+    *,
+    refresh: Union[Unset, bool] = False,
+) -> Response[ActionRecommendation]:
+    """Get Action Recommendation
+
+     Get action recommendation for improving a specific prediction.
+
+    Args:
+        prediction_id (UUID):
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[ActionRecommendation]
+    """
+
+    kwargs = _get_kwargs(
+        prediction_id=prediction_id,
+        refresh=refresh,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    prediction_id: UUID,
+    *,
+    refresh: Union[Unset, bool] = False,
+) -> ActionRecommendation:
+    """Get Action Recommendation
+
+     Get action recommendation for improving a specific prediction.
+
+    Args:
+        prediction_id (UUID):
+        refresh (Union[Unset, bool]):  Default: False.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        ActionRecommendation
+    """
+
+    return (
+        await asyncio_detailed(
+            prediction_id=prediction_id,
+            refresh=refresh,
+        )
+    ).parsed

orca_sdk/_generated_api_client/models/__init__.py

@@ -1,14 +1,19 @@
 """Contains all the data models used in inputs/outputs"""
 
+from .action_recommendation import ActionRecommendation
+from .action_recommendation_action import ActionRecommendationAction
+from .add_memory_recommendations import AddMemoryRecommendations
+from .add_memory_suggestion import AddMemorySuggestion
 from .analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
 from .api_key_metadata import ApiKeyMetadata
 from .api_key_metadata_scope_item import ApiKeyMetadataScopeItem
 from .base_label_prediction_result import BaseLabelPredictionResult
 from .base_model import BaseModel
 from .base_score_prediction_result import BaseScorePredictionResult
-from .body_create_datasource_datasource_post import BodyCreateDatasourceDatasourcePost
+from .body_create_datasource_from_files_datasource_upload_post import BodyCreateDatasourceFromFilesDatasourceUploadPost
 from .cascade_edit_suggestions_request import CascadeEditSuggestionsRequest
 from .cascading_edit_suggestion import CascadingEditSuggestion
+from .class_representatives import ClassRepresentatives
 from .classification_evaluation_request import ClassificationEvaluationRequest
 from .classification_metrics import ClassificationMetrics
 from .classification_model_metadata import ClassificationModelMetadata
@@ -18,15 +23,19 @@ from .cluster_metrics import ClusterMetrics
 from .column_info import ColumnInfo
 from .column_type import ColumnType
 from .constraint_violation_error_response import ConstraintViolationErrorResponse
+from .constraint_violation_error_response_status_code import ConstraintViolationErrorResponseStatusCode
 from .count_predictions_request import CountPredictionsRequest
 from .create_api_key_request import CreateApiKeyRequest
 from .create_api_key_request_scope_item import CreateApiKeyRequestScopeItem
 from .create_api_key_response import CreateApiKeyResponse
 from .create_api_key_response_scope_item import CreateApiKeyResponseScopeItem
 from .create_classification_model_request import CreateClassificationModelRequest
+from .create_datasource_from_content_request import CreateDatasourceFromContentRequest
 from .create_memoryset_request import CreateMemorysetRequest
 from .create_memoryset_request_index_params import CreateMemorysetRequestIndexParams
 from .create_memoryset_request_index_type import CreateMemorysetRequestIndexType
+from .create_org_plan_request import CreateOrgPlanRequest
+from .create_org_plan_request_tier import CreateOrgPlanRequestTier
 from .create_regression_model_request import CreateRegressionModelRequest
 from .datasource_metadata import DatasourceMetadata
 from .delete_memories_request import DeleteMemoriesRequest
@@ -45,6 +54,8 @@ from .feedback_metrics import FeedbackMetrics
 from .feedback_type import FeedbackType
 from .filter_item import FilterItem
 from .filter_item_field_type_0_item import FilterItemFieldType0Item
+from .filter_item_field_type_1_item_type_0 import FilterItemFieldType1ItemType0
+from .filter_item_field_type_2_item_type_0 import FilterItemFieldType2ItemType0
 from .filter_item_field_type_2_item_type_1 import FilterItemFieldType2ItemType1
 from .filter_item_op import FilterItemOp
 from .finetune_embedding_model_request import FinetuneEmbeddingModelRequest
@@ -53,6 +64,7 @@ from .finetuned_embedding_model_metadata import FinetunedEmbeddingModelMetadata
 from .get_memories_request import GetMemoriesRequest
 from .http_validation_error import HTTPValidationError
 from .internal_server_error_response import InternalServerErrorResponse
+from .internal_server_error_response_status_code import InternalServerErrorResponseStatusCode
 from .label_class_metrics import LabelClassMetrics
 from .label_prediction_memory_lookup import LabelPredictionMemoryLookup
 from .label_prediction_memory_lookup_metadata import LabelPredictionMemoryLookupMetadata
@@ -63,7 +75,6 @@ from .labeled_memory_insert_metadata import LabeledMemoryInsertMetadata
 from .labeled_memory_lookup import LabeledMemoryLookup
 from .labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
 from .labeled_memory_metadata import LabeledMemoryMetadata
-from .labeled_memory_metrics import LabeledMemoryMetrics
 from .labeled_memory_update import LabeledMemoryUpdate
 from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
 from .labeled_memory_with_feedback_metrics import LabeledMemoryWithFeedbackMetrics
@@ -78,6 +89,8 @@ from .memory_type import MemoryType
 from .memoryset_analysis_configs import MemorysetAnalysisConfigs
 from .memoryset_analysis_request import MemorysetAnalysisRequest
 from .memoryset_analysis_response import MemorysetAnalysisResponse
+from .memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
+from .memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
 from .memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
 from .memoryset_cluster_analysis_config_clustering_method import MemorysetClusterAnalysisConfigClusteringMethod
 from .memoryset_cluster_analysis_config_partitioning_method import MemorysetClusterAnalysisConfigPartitioningMethod
@@ -98,6 +111,10 @@ from .memoryset_projection_metrics import MemorysetProjectionMetrics
 from .memoryset_update import MemorysetUpdate
 from .not_found_error_response import NotFoundErrorResponse
 from .not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
+from .not_found_error_response_status_code import NotFoundErrorResponseStatusCode
+from .org_plan import OrgPlan
+from .org_plan_tier import OrgPlanTier
+from .paginated_task import PaginatedTask
 from .paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics import (
     PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics,
 )
@@ -133,48 +150,65 @@ from .scored_memory_with_feedback_metrics import ScoredMemoryWithFeedbackMetrics
 from .scored_memory_with_feedback_metrics_feedback_metrics import ScoredMemoryWithFeedbackMetricsFeedbackMetrics
 from .scored_memory_with_feedback_metrics_metadata import ScoredMemoryWithFeedbackMetricsMetadata
 from .service_unavailable_error_response import ServiceUnavailableErrorResponse
+from .service_unavailable_error_response_status_code import ServiceUnavailableErrorResponseStatusCode
 from .task import Task
 from .task_status import TaskStatus
 from .task_status_info import TaskStatusInfo
+from .telemetry_field_type_0_item_type_0 import TelemetryFieldType0ItemType0
 from .telemetry_field_type_0_item_type_2 import TelemetryFieldType0ItemType2
+from .telemetry_field_type_1_item_type_0 import TelemetryFieldType1ItemType0
+from .telemetry_field_type_1_item_type_1 import TelemetryFieldType1ItemType1
 from .telemetry_filter_item import TelemetryFilterItem
 from .telemetry_filter_item_op import TelemetryFilterItemOp
 from .telemetry_memories_request import TelemetryMemoriesRequest
 from .telemetry_sort_options import TelemetrySortOptions
 from .telemetry_sort_options_direction import TelemetrySortOptionsDirection
 from .unauthenticated_error_response import UnauthenticatedErrorResponse
+from .unauthenticated_error_response_status_code import UnauthenticatedErrorResponseStatusCode
 from .unauthorized_error_response import UnauthorizedErrorResponse
+from .unauthorized_error_response_status_code import UnauthorizedErrorResponseStatusCode
+from .update_org_plan_request import UpdateOrgPlanRequest
+from .update_org_plan_request_tier import UpdateOrgPlanRequestTier
 from .update_prediction_request import UpdatePredictionRequest
 from .validation_error import ValidationError
 
 __all__ = (
+    "ActionRecommendation",
+    "ActionRecommendationAction",
+    "AddMemoryRecommendations",
+    "AddMemorySuggestion",
     "AnalyzeNeighborLabelsResult",
     "ApiKeyMetadata",
     "ApiKeyMetadataScopeItem",
     "BaseLabelPredictionResult",
     "BaseModel",
     "BaseScorePredictionResult",
-    "BodyCreateDatasourceDatasourcePost",
+    "BodyCreateDatasourceFromFilesDatasourceUploadPost",
     "CascadeEditSuggestionsRequest",
     "CascadingEditSuggestion",
     "ClassificationEvaluationRequest",
     "ClassificationMetrics",
     "ClassificationModelMetadata",
     "ClassificationPredictionRequest",
+    "ClassRepresentatives",
     "CloneMemorysetRequest",
     "ClusterMetrics",
     "ColumnInfo",
     "ColumnType",
     "ConstraintViolationErrorResponse",
+    "ConstraintViolationErrorResponseStatusCode",
     "CountPredictionsRequest",
     "CreateApiKeyRequest",
     "CreateApiKeyRequestScopeItem",
     "CreateApiKeyResponse",
     "CreateApiKeyResponseScopeItem",
     "CreateClassificationModelRequest",
+    "CreateDatasourceFromContentRequest",
     "CreateMemorysetRequest",
     "CreateMemorysetRequestIndexParams",
     "CreateMemorysetRequestIndexType",
+    "CreateOrgPlanRequest",
+    "CreateOrgPlanRequestTier",
     "CreateRegressionModelRequest",
     "DatasourceMetadata",
     "DeleteMemoriesRequest",
@@ -193,6 +227,8 @@ __all__ = (
     "FeedbackType",
     "FilterItem",
     "FilterItemFieldType0Item",
+    "FilterItemFieldType1ItemType0",
+    "FilterItemFieldType2ItemType0",
     "FilterItemFieldType2ItemType1",
     "FilterItemOp",
     "FinetunedEmbeddingModelMetadata",
@@ -201,6 +237,7 @@ __all__ = (
     "GetMemoriesRequest",
     "HTTPValidationError",
     "InternalServerErrorResponse",
+    "InternalServerErrorResponseStatusCode",
     "LabelClassMetrics",
     "LabeledMemory",
     "LabeledMemoryInsert",
@@ -208,7 +245,6 @@ __all__ = (
     "LabeledMemoryLookup",
     "LabeledMemoryLookupMetadata",
     "LabeledMemoryMetadata",
-    "LabeledMemoryMetrics",
     "LabeledMemoryUpdate",
     "LabeledMemoryUpdateMetadataType0",
     "LabeledMemoryWithFeedbackMetrics",
@@ -225,6 +261,8 @@ __all__ = (
     "MemorysetAnalysisConfigs",
     "MemorysetAnalysisRequest",
     "MemorysetAnalysisResponse",
+    "MemorysetClassPatternsAnalysisConfig",
+    "MemorysetClassPatternsMetrics",
     "MemorysetClusterAnalysisConfig",
     "MemorysetClusterAnalysisConfigClusteringMethod",
     "MemorysetClusterAnalysisConfigPartitioningMethod",
@@ -246,6 +284,10 @@ __all__ = (
     "MemoryType",
     "NotFoundErrorResponse",
     "NotFoundErrorResponseResourceType0",
+    "NotFoundErrorResponseStatusCode",
+    "OrgPlan",
+    "OrgPlanTier",
+    "PaginatedTask",
     "PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics",
     "PRCurve",
     "PredictionFeedback",
@@ -279,17 +321,25 @@ __all__ = (
     "ScorePredictionMemoryLookupMetadata",
     "ScorePredictionWithMemoriesAndFeedback",
     "ServiceUnavailableErrorResponse",
+    "ServiceUnavailableErrorResponseStatusCode",
     "Task",
     "TaskStatus",
     "TaskStatusInfo",
+    "TelemetryFieldType0ItemType0",
     "TelemetryFieldType0ItemType2",
+    "TelemetryFieldType1ItemType0",
+    "TelemetryFieldType1ItemType1",
     "TelemetryFilterItem",
     "TelemetryFilterItemOp",
     "TelemetryMemoriesRequest",
     "TelemetrySortOptions",
     "TelemetrySortOptionsDirection",
     "UnauthenticatedErrorResponse",
+    "UnauthenticatedErrorResponseStatusCode",
     "UnauthorizedErrorResponse",
+    "UnauthorizedErrorResponseStatusCode",
+    "UpdateOrgPlanRequest",
+    "UpdateOrgPlanRequestTier",
     "UpdatePredictionRequest",
     "ValidationError",
 )