orca-sdk 0.0.94__py3-none-any.whl → 0.0.95__py3-none-any.whl

orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py

@@ -11,7 +11,7 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 from uuid import UUID
 
 import httpx
@@ -19,6 +19,7 @@ import httpx
 from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
+from ...models.score_prediction_with_memories_and_feedback import ScorePredictionWithMemoriesAndFeedback
 from ...types import Response
 
 
@@ -33,9 +34,29 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> LabelPredictionWithMemoriesAndFeedback:
+def _parse_response(
+    *, response: httpx.Response
+) -> Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]:
     if response.status_code == 200:
-        response_200 = LabelPredictionWithMemoriesAndFeedback.from_dict(response.json())
+
+        def _parse_response_200(
+            data: object,
+        ) -> Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]:
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_200_type_0 = LabelPredictionWithMemoriesAndFeedback.from_dict(data)
+
+                return response_200_type_0
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            response_200_type_1 = ScorePredictionWithMemoriesAndFeedback.from_dict(data)
+
+            return response_200_type_1
+
+        response_200 = _parse_response_200(response.json())
 
         return response_200
     if response.status_code == 401:
@@ -50,7 +71,9 @@ def _parse_response(*, response: httpx.Response) -> LabelPredictionWithMemoriesA
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[LabelPredictionWithMemoriesAndFeedback]:
+def _build_response(
+    *, response: httpx.Response
+) -> Response[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -61,7 +84,7 @@ def _build_response(*, response: httpx.Response) -> Response[LabelPredictionWith
 
 def sync_detailed(
     prediction_id: UUID,
-) -> Response[LabelPredictionWithMemoriesAndFeedback]:
+) -> Response[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     """Get Prediction
 
     Args:
@@ -72,7 +95,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabelPredictionWithMemoriesAndFeedback]
+        Response[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]
     """
 
     kwargs = _get_kwargs(
@@ -89,7 +112,7 @@ def sync_detailed(
 
 def sync(
     prediction_id: UUID,
-) -> LabelPredictionWithMemoriesAndFeedback:
+) -> Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]:
     """Get Prediction
 
     Args:
@@ -100,7 +123,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabelPredictionWithMemoriesAndFeedback
+        Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']
     """
 
     return sync_detailed(
@@ -110,7 +133,7 @@ def sync(
 
 async def asyncio_detailed(
     prediction_id: UUID,
-) -> Response[LabelPredictionWithMemoriesAndFeedback]:
+) -> Response[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     """Get Prediction
 
     Args:
@@ -121,7 +144,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabelPredictionWithMemoriesAndFeedback]
+        Response[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]
     """
 
     kwargs = _get_kwargs(
@@ -136,7 +159,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     prediction_id: UUID,
-) -> LabelPredictionWithMemoriesAndFeedback:
+) -> Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]:
     """Get Prediction
 
     Args:
@@ -147,7 +170,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabelPredictionWithMemoriesAndFeedback
+        Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']
     """
 
     return (

orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py

@@ -17,7 +17,9 @@ import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.paginated_labeled_memory_with_feedback_metrics import PaginatedLabeledMemoryWithFeedbackMetrics
+from ...models.paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics import (
+    PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics,
+)
 from ...models.telemetry_memories_request import TelemetryMemoriesRequest
 from ...types import Response
 
@@ -42,9 +44,13 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> PaginatedLabeledMemoryWithFeedbackMetrics:
+def _parse_response(
+    *, response: httpx.Response
+) -> PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics:
     if response.status_code == 200:
-        response_200 = PaginatedLabeledMemoryWithFeedbackMetrics.from_dict(response.json())
+        response_200 = PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics.from_dict(
+            response.json()
+        )
 
         return response_200
     if response.status_code == 401:
@@ -59,7 +65,9 @@ def _parse_response(*, response: httpx.Response) -> PaginatedLabeledMemoryWithFe
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[PaginatedLabeledMemoryWithFeedbackMetrics]:
+def _build_response(
+    *, response: httpx.Response
+) -> Response[PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -71,7 +79,7 @@ def _build_response(*, response: httpx.Response) -> Response[PaginatedLabeledMem
 def sync_detailed(
     *,
     body: TelemetryMemoriesRequest,
-) -> Response[PaginatedLabeledMemoryWithFeedbackMetrics]:
+) -> Response[PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics]:
     """List Memories With Feedback
 
      List memories with feedback metrics.
@@ -87,7 +95,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[PaginatedLabeledMemoryWithFeedbackMetrics]
+        Response[PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -105,7 +113,7 @@ def sync_detailed(
 def sync(
     *,
     body: TelemetryMemoriesRequest,
-) -> PaginatedLabeledMemoryWithFeedbackMetrics:
+) -> PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics:
     """List Memories With Feedback
 
      List memories with feedback metrics.
@@ -121,7 +129,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        PaginatedLabeledMemoryWithFeedbackMetrics
+        PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics
     """
 
     return sync_detailed(
@@ -132,7 +140,7 @@ def sync(
 async def asyncio_detailed(
     *,
     body: TelemetryMemoriesRequest,
-) -> Response[PaginatedLabeledMemoryWithFeedbackMetrics]:
+) -> Response[PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics]:
     """List Memories With Feedback
 
      List memories with feedback metrics.
@@ -148,7 +156,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[PaginatedLabeledMemoryWithFeedbackMetrics]
+        Response[PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -164,7 +172,7 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     body: TelemetryMemoriesRequest,
-) -> PaginatedLabeledMemoryWithFeedbackMetrics:
+) -> PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics:
     """List Memories With Feedback
 
      List memories with feedback metrics.
@@ -180,7 +188,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        PaginatedLabeledMemoryWithFeedbackMetrics
+        PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics
     """
 
     return (

orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py

@@ -11,7 +11,7 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Union
 
 import httpx
 
@@ -19,6 +19,7 @@ from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
 from ...models.list_predictions_request import ListPredictionsRequest
+from ...models.score_prediction_with_memories_and_feedback import ScorePredictionWithMemoriesAndFeedback
 from ...types import Response
 
 
@@ -42,12 +43,32 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List["LabelPredictionWithMemoriesAndFeedback"]:
+def _parse_response(
+    *, response: httpx.Response
+) -> List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = LabelPredictionWithMemoriesAndFeedback.from_dict(response_200_item_data)
+
+            def _parse_response_200_item(
+                data: object,
+            ) -> Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    response_200_item_type_0 = LabelPredictionWithMemoriesAndFeedback.from_dict(data)
+
+                    return response_200_item_type_0
+                except:  # noqa: E722
+                    pass
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_200_item_type_1 = ScorePredictionWithMemoriesAndFeedback.from_dict(data)
+
+                return response_200_item_type_1
+
+            response_200_item = _parse_response_200_item(response_200_item_data)
 
             response_200.append(response_200_item)
 
@@ -64,7 +85,9 @@ def _parse_response(*, response: httpx.Response) -> List["LabelPredictionWithMem
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List["LabelPredictionWithMemoriesAndFeedback"]]:
+def _build_response(
+    *, response: httpx.Response
+) -> Response[List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -76,7 +99,7 @@ def _build_response(*, response: httpx.Response) -> Response[List["LabelPredicti
 def sync_detailed(
     *,
     body: ListPredictionsRequest,
-) -> Response[List["LabelPredictionWithMemoriesAndFeedback"]]:
+) -> Response[List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]]:
     """List Predictions
 
     Args:
@@ -87,7 +110,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['LabelPredictionWithMemoriesAndFeedback']]
+        Response[List[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]]
     """
 
     kwargs = _get_kwargs(
@@ -105,7 +128,7 @@ def sync_detailed(
 def sync(
     *,
     body: ListPredictionsRequest,
-) -> List["LabelPredictionWithMemoriesAndFeedback"]:
+) -> List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     """List Predictions
 
     Args:
@@ -116,7 +139,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['LabelPredictionWithMemoriesAndFeedback']
+        List[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]
     """
 
     return sync_detailed(
@@ -127,7 +150,7 @@ def sync(
 async def asyncio_detailed(
     *,
     body: ListPredictionsRequest,
-) -> Response[List["LabelPredictionWithMemoriesAndFeedback"]]:
+) -> Response[List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]]:
     """List Predictions
 
     Args:
@@ -138,7 +161,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['LabelPredictionWithMemoriesAndFeedback']]
+        Response[List[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]]
     """
 
     kwargs = _get_kwargs(
@@ -154,7 +177,7 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     body: ListPredictionsRequest,
-) -> List["LabelPredictionWithMemoriesAndFeedback"]:
+) -> List[Union["LabelPredictionWithMemoriesAndFeedback", "ScorePredictionWithMemoriesAndFeedback"]]:
     """List Predictions
 
     Args:
@@ -165,7 +188,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['LabelPredictionWithMemoriesAndFeedback']
+        List[Union['LabelPredictionWithMemoriesAndFeedback', 'ScorePredictionWithMemoriesAndFeedback']]
     """
 
     return (

orca_sdk/_generated_api_client/models/__init__.py

@@ -5,11 +5,15 @@ 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 .cascade_edit_suggestions_request import CascadeEditSuggestionsRequest
 from .cascading_edit_suggestion import CascadingEditSuggestion
-from .classification_evaluation_result import ClassificationEvaluationResult
-from .clone_labeled_memoryset_request import CloneLabeledMemorysetRequest
+from .classification_evaluation_request import ClassificationEvaluationRequest
+from .classification_metrics import ClassificationMetrics
+from .classification_model_metadata import ClassificationModelMetadata
+from .classification_prediction_request import ClassificationPredictionRequest
+from .clone_memoryset_request import CloneMemorysetRequest
 from .cluster_metrics import ClusterMetrics
 from .column_info import ColumnInfo
 from .column_type import ColumnType
@@ -19,19 +23,24 @@ 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_labeled_memoryset_request import CreateLabeledMemorysetRequest
-from .create_rac_model_request import CreateRACModelRequest
+from .create_classification_model_request import CreateClassificationModelRequest
+from .create_memoryset_request import CreateMemorysetRequest
+from .create_memoryset_request_index_params import CreateMemorysetRequestIndexParams
+from .create_memoryset_request_index_type import CreateMemorysetRequestIndexType
+from .create_regression_model_request import CreateRegressionModelRequest
 from .datasource_metadata import DatasourceMetadata
 from .delete_memories_request import DeleteMemoriesRequest
 from .delete_memorysets_request import DeleteMemorysetsRequest
 from .embed_request import EmbedRequest
+from .embedding_evaluation_payload import EmbeddingEvaluationPayload
 from .embedding_evaluation_request import EmbeddingEvaluationRequest
 from .embedding_evaluation_response import EmbeddingEvaluationResponse
 from .embedding_evaluation_result import EmbeddingEvaluationResult
 from .embedding_finetuning_method import EmbeddingFinetuningMethod
 from .embedding_model_result import EmbeddingModelResult
-from .evaluation_request import EvaluationRequest
 from .evaluation_response import EvaluationResponse
+from .evaluation_response_classification_metrics import EvaluationResponseClassificationMetrics
+from .evaluation_response_regression_metrics import EvaluationResponseRegressionMetrics
 from .feedback_metrics import FeedbackMetrics
 from .feedback_type import FeedbackType
 from .filter_item import FilterItem
@@ -60,13 +69,12 @@ from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataTy
 from .labeled_memory_with_feedback_metrics import LabeledMemoryWithFeedbackMetrics
 from .labeled_memory_with_feedback_metrics_feedback_metrics import LabeledMemoryWithFeedbackMetricsFeedbackMetrics
 from .labeled_memory_with_feedback_metrics_metadata import LabeledMemoryWithFeedbackMetricsMetadata
-from .labeled_memoryset_metadata import LabeledMemorysetMetadata
-from .labeled_memoryset_update import LabeledMemorysetUpdate
 from .list_memories_request import ListMemoriesRequest
 from .list_predictions_request import ListPredictionsRequest
 from .lookup_request import LookupRequest
 from .lookup_score_metrics import LookupScoreMetrics
 from .memory_metrics import MemoryMetrics
+from .memory_type import MemoryType
 from .memoryset_analysis_configs import MemorysetAnalysisConfigs
 from .memoryset_analysis_request import MemorysetAnalysisRequest
 from .memoryset_analysis_response import MemorysetAnalysisResponse
@@ -78,29 +86,52 @@ from .memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfi
 from .memoryset_duplicate_metrics import MemorysetDuplicateMetrics
 from .memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
 from .memoryset_label_metrics import MemorysetLabelMetrics
+from .memoryset_metadata import MemorysetMetadata
+from .memoryset_metadata_index_params import MemorysetMetadataIndexParams
+from .memoryset_metadata_index_type import MemorysetMetadataIndexType
 from .memoryset_metrics import MemorysetMetrics
 from .memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
 from .memoryset_neighbor_metrics import MemorysetNeighborMetrics
 from .memoryset_neighbor_metrics_lookup_score_metrics import MemorysetNeighborMetricsLookupScoreMetrics
 from .memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
 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 .paginated_labeled_memory_with_feedback_metrics import PaginatedLabeledMemoryWithFeedbackMetrics
-from .precision_recall_curve import PrecisionRecallCurve
+from .paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics import (
+    PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics,
+)
+from .pr_curve import PRCurve
 from .prediction_feedback import PredictionFeedback
 from .prediction_feedback_category import PredictionFeedbackCategory
 from .prediction_feedback_request import PredictionFeedbackRequest
 from .prediction_feedback_result import PredictionFeedbackResult
-from .prediction_request import PredictionRequest
 from .prediction_sort_item_item_type_0 import PredictionSortItemItemType0
 from .prediction_sort_item_item_type_1 import PredictionSortItemItemType1
+from .predictive_model_update import PredictiveModelUpdate
 from .pretrained_embedding_model_metadata import PretrainedEmbeddingModelMetadata
 from .pretrained_embedding_model_name import PretrainedEmbeddingModelName
 from .rac_head_type import RACHeadType
-from .rac_model_metadata import RACModelMetadata
-from .rac_model_update import RACModelUpdate
+from .rar_head_type import RARHeadType
+from .regression_evaluation_request import RegressionEvaluationRequest
+from .regression_metrics import RegressionMetrics
+from .regression_model_metadata import RegressionModelMetadata
+from .regression_prediction_request import RegressionPredictionRequest
 from .roc_curve import ROCCurve
+from .score_prediction_memory_lookup import ScorePredictionMemoryLookup
+from .score_prediction_memory_lookup_metadata import ScorePredictionMemoryLookupMetadata
+from .score_prediction_with_memories_and_feedback import ScorePredictionWithMemoriesAndFeedback
+from .scored_memory import ScoredMemory
+from .scored_memory_insert import ScoredMemoryInsert
+from .scored_memory_insert_metadata import ScoredMemoryInsertMetadata
+from .scored_memory_lookup import ScoredMemoryLookup
+from .scored_memory_lookup_metadata import ScoredMemoryLookupMetadata
+from .scored_memory_metadata import ScoredMemoryMetadata
+from .scored_memory_update import ScoredMemoryUpdate
+from .scored_memory_update_metadata_type_0 import ScoredMemoryUpdateMetadataType0
+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 .task import Task
 from .task_status import TaskStatus
@@ -122,11 +153,15 @@ __all__ = (
     "ApiKeyMetadataScopeItem",
     "BaseLabelPredictionResult",
     "BaseModel",
+    "BaseScorePredictionResult",
     "BodyCreateDatasourceDatasourcePost",
     "CascadeEditSuggestionsRequest",
     "CascadingEditSuggestion",
-    "ClassificationEvaluationResult",
-    "CloneLabeledMemorysetRequest",
+    "ClassificationEvaluationRequest",
+    "ClassificationMetrics",
+    "ClassificationModelMetadata",
+    "ClassificationPredictionRequest",
+    "CloneMemorysetRequest",
     "ClusterMetrics",
     "ColumnInfo",
     "ColumnType",
@@ -136,19 +171,24 @@ __all__ = (
     "CreateApiKeyRequestScopeItem",
     "CreateApiKeyResponse",
     "CreateApiKeyResponseScopeItem",
-    "CreateLabeledMemorysetRequest",
-    "CreateRACModelRequest",
+    "CreateClassificationModelRequest",
+    "CreateMemorysetRequest",
+    "CreateMemorysetRequestIndexParams",
+    "CreateMemorysetRequestIndexType",
+    "CreateRegressionModelRequest",
     "DatasourceMetadata",
     "DeleteMemoriesRequest",
     "DeleteMemorysetsRequest",
+    "EmbeddingEvaluationPayload",
     "EmbeddingEvaluationRequest",
     "EmbeddingEvaluationResponse",
     "EmbeddingEvaluationResult",
     "EmbeddingFinetuningMethod",
     "EmbeddingModelResult",
     "EmbedRequest",
-    "EvaluationRequest",
     "EvaluationResponse",
+    "EvaluationResponseClassificationMetrics",
+    "EvaluationResponseRegressionMetrics",
     "FeedbackMetrics",
     "FeedbackType",
     "FilterItem",
@@ -169,8 +209,6 @@ __all__ = (
     "LabeledMemoryLookupMetadata",
     "LabeledMemoryMetadata",
     "LabeledMemoryMetrics",
-    "LabeledMemorysetMetadata",
-    "LabeledMemorysetUpdate",
     "LabeledMemoryUpdate",
     "LabeledMemoryUpdateMetadataType0",
     "LabeledMemoryWithFeedbackMetrics",
@@ -195,29 +233,51 @@ __all__ = (
     "MemorysetDuplicateMetrics",
     "MemorysetLabelAnalysisConfig",
     "MemorysetLabelMetrics",
+    "MemorysetMetadata",
+    "MemorysetMetadataIndexParams",
+    "MemorysetMetadataIndexType",
     "MemorysetMetrics",
     "MemorysetNeighborAnalysisConfig",
     "MemorysetNeighborMetrics",
     "MemorysetNeighborMetricsLookupScoreMetrics",
     "MemorysetProjectionAnalysisConfig",
     "MemorysetProjectionMetrics",
+    "MemorysetUpdate",
+    "MemoryType",
     "NotFoundErrorResponse",
     "NotFoundErrorResponseResourceType0",
-    "PaginatedLabeledMemoryWithFeedbackMetrics",
-    "PrecisionRecallCurve",
+    "PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics",
+    "PRCurve",
     "PredictionFeedback",
     "PredictionFeedbackCategory",
     "PredictionFeedbackRequest",
     "PredictionFeedbackResult",
-    "PredictionRequest",
     "PredictionSortItemItemType0",
     "PredictionSortItemItemType1",
+    "PredictiveModelUpdate",
     "PretrainedEmbeddingModelMetadata",
     "PretrainedEmbeddingModelName",
     "RACHeadType",
-    "RACModelMetadata",
-    "RACModelUpdate",
+    "RARHeadType",
+    "RegressionEvaluationRequest",
+    "RegressionMetrics",
+    "RegressionModelMetadata",
+    "RegressionPredictionRequest",
     "ROCCurve",
+    "ScoredMemory",
+    "ScoredMemoryInsert",
+    "ScoredMemoryInsertMetadata",
+    "ScoredMemoryLookup",
+    "ScoredMemoryLookupMetadata",
+    "ScoredMemoryMetadata",
+    "ScoredMemoryUpdate",
+    "ScoredMemoryUpdateMetadataType0",
+    "ScoredMemoryWithFeedbackMetrics",
+    "ScoredMemoryWithFeedbackMetricsFeedbackMetrics",
+    "ScoredMemoryWithFeedbackMetricsMetadata",
+    "ScorePredictionMemoryLookup",
+    "ScorePredictionMemoryLookupMetadata",
+    "ScorePredictionWithMemoriesAndFeedback",
     "ServiceUnavailableErrorResponse",
     "Task",
     "TaskStatus",