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

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,32 +5,42 @@ 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
 from .constraint_violation_error_response import ConstraintViolationErrorResponse
+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_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
@@ -41,6 +51,7 @@ from .finetune_embedding_model_request import FinetuneEmbeddingModelRequest
 from .finetune_embedding_model_request_training_args import FinetuneEmbeddingModelRequestTrainingArgs
 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 .label_class_metrics import LabelClassMetrics
 from .label_prediction_memory_lookup import LabelPredictionMemoryLookup
@@ -58,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
@@ -76,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
@@ -112,6 +145,7 @@ from .telemetry_sort_options_direction import TelemetrySortOptionsDirection
 from .unauthenticated_error_response import UnauthenticatedErrorResponse
 from .unauthorized_error_response import UnauthorizedErrorResponse
 from .update_prediction_request import UpdatePredictionRequest
+from .validation_error import ValidationError
 
 __all__ = (
     "AnalyzeNeighborLabelsResult",
@@ -119,32 +153,42 @@ __all__ = (
     "ApiKeyMetadataScopeItem",
     "BaseLabelPredictionResult",
     "BaseModel",
+    "BaseScorePredictionResult",
     "BodyCreateDatasourceDatasourcePost",
     "CascadeEditSuggestionsRequest",
     "CascadingEditSuggestion",
-    "ClassificationEvaluationResult",
-    "CloneLabeledMemorysetRequest",
+    "ClassificationEvaluationRequest",
+    "ClassificationMetrics",
+    "ClassificationModelMetadata",
+    "ClassificationPredictionRequest",
+    "CloneMemorysetRequest",
     "ClusterMetrics",
     "ColumnInfo",
     "ColumnType",
     "ConstraintViolationErrorResponse",
+    "CountPredictionsRequest",
     "CreateApiKeyRequest",
     "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",
@@ -155,6 +199,7 @@ __all__ = (
     "FinetuneEmbeddingModelRequest",
     "FinetuneEmbeddingModelRequestTrainingArgs",
     "GetMemoriesRequest",
+    "HTTPValidationError",
     "InternalServerErrorResponse",
     "LabelClassMetrics",
     "LabeledMemory",
@@ -164,8 +209,6 @@ __all__ = (
     "LabeledMemoryLookupMetadata",
     "LabeledMemoryMetadata",
     "LabeledMemoryMetrics",
-    "LabeledMemorysetMetadata",
-    "LabeledMemorysetUpdate",
     "LabeledMemoryUpdate",
     "LabeledMemoryUpdateMetadataType0",
     "LabeledMemoryWithFeedbackMetrics",
@@ -190,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",
@@ -226,4 +291,5 @@ __all__ = (
     "UnauthenticatedErrorResponse",
     "UnauthorizedErrorResponse",
     "UpdatePredictionRequest",
+    "ValidationError",
 )

orca_sdk/_generated_api_client/models/base_score_prediction_result.py

@@ -0,0 +1,108 @@
+"""
+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/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="BaseScorePredictionResult")
+
+
+@_attrs_define
+class BaseScorePredictionResult:
+    """Predicted score and confidence for a single input.
+
+    Attributes:
+        prediction_id (Union[None, str]):
+        confidence (float):
+        anomaly_score (Union[None, float]):
+        score (float):
+    """
+
+    prediction_id: Union[None, str]
+    confidence: float
+    anomaly_score: Union[None, float]
+    score: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        prediction_id: Union[None, str]
+        prediction_id = self.prediction_id
+
+        confidence = self.confidence
+
+        anomaly_score: Union[None, float]
+        anomaly_score = self.anomaly_score
+
+        score = self.score
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "prediction_id": prediction_id,
+                "confidence": confidence,
+                "anomaly_score": anomaly_score,
+                "score": score,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+
+        def _parse_prediction_id(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        prediction_id = _parse_prediction_id(d.pop("prediction_id"))
+
+        confidence = d.pop("confidence")
+
+        def _parse_anomaly_score(data: object) -> Union[None, float]:
+            if data is None:
+                return data
+            return cast(Union[None, float], data)
+
+        anomaly_score = _parse_anomaly_score(d.pop("anomaly_score"))
+
+        score = d.pop("score")
+
+        base_score_prediction_result = cls(
+            prediction_id=prediction_id,
+            confidence=confidence,
+            anomaly_score=anomaly_score,
+            score=score,
+        )
+
+        base_score_prediction_result.additional_properties = d
+        return base_score_prediction_result
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties