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

orca_sdk/_generated_api_client/client.py

@@ -1,216 +0,0 @@
-"""
-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/5cfe4e1d594951725844ea470fc9d61f40c08093/openapi_python_client/templates/client.py.jinja
-
-The main changes are:
-- add a lazy initialized global client singleton that is injected into every api method via a context manager
-- add methods for setting headers and base URL on the global client singleton
-"""
-
-import os
-import ssl
-from contextlib import contextmanager
-from typing import Any, Optional, Union
-
-import httpx
-from attrs import define, evolve, field
-
-
-@define
-class Client:
-    """A class for keeping track of data related to the API
-
-    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
-
-        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
-
-        ``cookies``: A dictionary of cookies to be sent with every request
-
-        ``headers``: A dictionary of headers to be sent with every request
-
-        ``timeout``: The maximum amount of a time a request can take. API functions will raise
-        httpx.TimeoutException if this is exceeded.
-
-        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
-        but can be set to False for testing purposes.
-
-        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
-
-        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
-
-
-    Attributes:
-        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
-            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
-            argument to the constructor.
-    """
-
-    raise_on_unexpected_status: bool = field(default=False, kw_only=True)
-    _base_url: str = field(alias="base_url")
-    _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
-    _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
-    _timeout: Optional[httpx.Timeout] = field(default=None, kw_only=True, alias="timeout")
-    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
-    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
-    _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
-    _client: Optional[httpx.Client] = field(default=None, init=False)
-    _async_client: Optional[httpx.AsyncClient] = field(default=None, init=False)
-
-    def with_headers(self, headers: dict[str, str]) -> "Client":
-        """Get a new client matching this one with additional headers"""
-        if self._client is not None:
-            self._client.headers.update(headers)
-        if self._async_client is not None:
-            self._async_client.headers.update(headers)
-        return evolve(self, headers={**self._headers, **headers})
-
-    def with_cookies(self, cookies: dict[str, str]) -> "Client":
-        """Get a new client matching this one with additional cookies"""
-        if self._client is not None:
-            self._client.cookies.update(cookies)
-        if self._async_client is not None:
-            self._async_client.cookies.update(cookies)
-        return evolve(self, cookies={**self._cookies, **cookies})
-
-    def with_timeout(self, timeout: httpx.Timeout) -> "Client":
-        """Get a new client matching this one with a new timeout (in seconds)"""
-        if self._client is not None:
-            self._client.timeout = timeout
-        if self._async_client is not None:
-            self._async_client.timeout = timeout
-        return evolve(self, timeout=timeout)
-
-    def set_httpx_client(self, client: httpx.Client) -> "Client":
-        """Manually set the underlying httpx.Client
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._client = client
-        return self
-
-    def get_httpx_client(self) -> httpx.Client:
-        """Get the underlying httpx.Client, constructing a new one if not previously set"""
-        if self._client is None:
-            self._client = httpx.Client(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._client
-
-    def __enter__(self) -> "Client":
-        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
-        self.get_httpx_client().__enter__()
-        return self
-
-    def __exit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for internal httpx.Client (see httpx docs)"""
-        self.get_httpx_client().__exit__(*args, **kwargs)
-
-    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
-        """Manually the underlying httpx.AsyncClient
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._async_client = async_client
-        return self
-
-    def get_async_httpx_client(self) -> httpx.AsyncClient:
-        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
-        if self._async_client is None:
-            self._async_client = httpx.AsyncClient(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._async_client
-
-    async def __aenter__(self) -> "Client":
-        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
-        await self.get_async_httpx_client().__aenter__()
-        return self
-
-    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
-        await self.get_async_httpx_client().__aexit__(*args, **kwargs)
-
-
-# this must be initialized lazily to ensure that the environment variables are set
-_client: Optional["Client"] = None
-
-
-def get_base_url():
-    """Get the base URL for the global client singleton"""
-    global _client
-    if _client is not None:
-        return _client._base_url
-    return os.environ.get("ORCA_API_URL", "https://api.orcadb.ai/")
-
-
-def get_headers():
-    """Get the headers for the global client singleton"""
-    global _client
-    if _client is not None:
-        return _client._headers
-    headers = {}
-    api_key = os.environ.get("ORCA_API_KEY")
-    if api_key is not None:
-        headers["Api-Key"] = api_key
-    return headers
-
-
-def get_client():
-    """Get the global client singleton"""
-    global _client
-    if _client is None:
-        _client = Client(base_url=get_base_url(), headers=get_headers())
-    return _client
-
-
-def set_headers(headers: dict[str, str]):
-    """Set the headers for the global client singleton"""
-    global _client
-    _client = Client(base_url=get_base_url(), headers=headers)
-    return _client
-
-
-def set_base_url(base_url: str):
-    """Set the base URL for the global client singleton"""
-    global _client
-    _client = Client(base_url=base_url, headers=_client._headers if _client is not None else {})
-    return _client
-
-
-@contextmanager
-def headers_context(headers: dict[str, str]):
-    """Set the headers for the global client singleton within a context manager"""
-    global _client
-    old_client = get_client()
-    set_headers(headers=headers)
-    yield
-    _client = old_client
-
-
-@contextmanager
-def base_url_context(base_url: str):
-    """Set the base URL for the global client singleton within a context manager"""
-    global _client
-    old_client = get_client()
-    set_base_url(base_url=base_url)
-    yield
-    _client = old_client
-
-
-@contextmanager
-def _client_context():
-    yield get_client()

orca_sdk/_generated_api_client/errors.py

@@ -1,38 +0,0 @@
-"""
-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/errors.py.jinja
-
-The main change is:
-- Returning the appropriate python exception for Orca API's error responses
-"""
-
-import httpx
-
-
-def get_error_for_response(response: httpx.Response):
-    """Raise an appropriate exception based on the response status code and content"""
-    if response.status_code == 401:
-        return ValueError("Invalid API key")
-    elif response.status_code == 403:
-        return PermissionError(response.json()["reason"])
-    elif response.status_code == 404:
-        resource = response.json()["resource"]
-        if resource is not None:
-            return LookupError(f"The {resource} you are looking for does not exist")
-        else:
-            return RuntimeError(f"Unknown API route: {response.url}")
-    elif response.status_code == 405:
-        return RuntimeError(f"Unknown method {response.request.method} for API route: {response.url}")
-    elif response.status_code == 409:
-        return RuntimeError(response.json()["constraint"])
-    elif response.status_code == 422:
-        issues = [f"{issue['loc'][-1]}: {issue['msg']}" for issue in response.json()["validation_issues"]]
-        return ValueError("Invalid input:\n\t" + "\n\t".join(issues))
-    elif response.status_code == 500:
-        return RuntimeError(f"Unexpected server error: {response.json()['message']}")
-    elif response.status_code == 503:
-        return RuntimeError("Orca API is currently unavailable, please try again later")
-    else:
-        return RuntimeError(f"Unexpected status code: {response.status_code}")

orca_sdk/_generated_api_client/models/__init__.py

@@ -1,345 +0,0 @@
-"""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_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
-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 .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
-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_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
-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
-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 .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
-from .label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
-from .labeled_memory import LabeledMemory
-from .labeled_memory_insert import LabeledMemoryInsert
-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_update import LabeledMemoryUpdate
-from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
-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 .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
-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
-from .memoryset_cluster_metrics import MemorysetClusterMetrics
-from .memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
-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 .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,
-)
-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_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 .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 .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",
-    "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",
-    "DeleteMemorysetsRequest",
-    "EmbeddingEvaluationPayload",
-    "EmbeddingEvaluationRequest",
-    "EmbeddingEvaluationResponse",
-    "EmbeddingEvaluationResult",
-    "EmbeddingFinetuningMethod",
-    "EmbeddingModelResult",
-    "EmbedRequest",
-    "EvaluationResponse",
-    "EvaluationResponseClassificationMetrics",
-    "EvaluationResponseRegressionMetrics",
-    "FeedbackMetrics",
-    "FeedbackType",
-    "FilterItem",
-    "FilterItemFieldType0Item",
-    "FilterItemFieldType1ItemType0",
-    "FilterItemFieldType2ItemType0",
-    "FilterItemFieldType2ItemType1",
-    "FilterItemOp",
-    "FinetunedEmbeddingModelMetadata",
-    "FinetuneEmbeddingModelRequest",
-    "FinetuneEmbeddingModelRequestTrainingArgs",
-    "GetMemoriesRequest",
-    "HTTPValidationError",
-    "InternalServerErrorResponse",
-    "InternalServerErrorResponseStatusCode",
-    "LabelClassMetrics",
-    "LabeledMemory",
-    "LabeledMemoryInsert",
-    "LabeledMemoryInsertMetadata",
-    "LabeledMemoryLookup",
-    "LabeledMemoryLookupMetadata",
-    "LabeledMemoryMetadata",
-    "LabeledMemoryUpdate",
-    "LabeledMemoryUpdateMetadataType0",
-    "LabeledMemoryWithFeedbackMetrics",
-    "LabeledMemoryWithFeedbackMetricsFeedbackMetrics",
-    "LabeledMemoryWithFeedbackMetricsMetadata",
-    "LabelPredictionMemoryLookup",
-    "LabelPredictionMemoryLookupMetadata",
-    "LabelPredictionWithMemoriesAndFeedback",
-    "ListMemoriesRequest",
-    "ListPredictionsRequest",
-    "LookupRequest",
-    "LookupScoreMetrics",
-    "MemoryMetrics",
-    "MemorysetAnalysisConfigs",
-    "MemorysetAnalysisRequest",
-    "MemorysetAnalysisResponse",
-    "MemorysetClassPatternsAnalysisConfig",
-    "MemorysetClassPatternsMetrics",
-    "MemorysetClusterAnalysisConfig",
-    "MemorysetClusterAnalysisConfigClusteringMethod",
-    "MemorysetClusterAnalysisConfigPartitioningMethod",
-    "MemorysetClusterMetrics",
-    "MemorysetDuplicateAnalysisConfig",
-    "MemorysetDuplicateMetrics",
-    "MemorysetLabelAnalysisConfig",
-    "MemorysetLabelMetrics",
-    "MemorysetMetadata",
-    "MemorysetMetadataIndexParams",
-    "MemorysetMetadataIndexType",
-    "MemorysetMetrics",
-    "MemorysetNeighborAnalysisConfig",
-    "MemorysetNeighborMetrics",
-    "MemorysetNeighborMetricsLookupScoreMetrics",
-    "MemorysetProjectionAnalysisConfig",
-    "MemorysetProjectionMetrics",
-    "MemorysetUpdate",
-    "MemoryType",
-    "NotFoundErrorResponse",
-    "NotFoundErrorResponseResourceType0",
-    "NotFoundErrorResponseStatusCode",
-    "OrgPlan",
-    "OrgPlanTier",
-    "PaginatedTask",
-    "PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics",
-    "PRCurve",
-    "PredictionFeedback",
-    "PredictionFeedbackCategory",
-    "PredictionFeedbackRequest",
-    "PredictionFeedbackResult",
-    "PredictionSortItemItemType0",
-    "PredictionSortItemItemType1",
-    "PredictiveModelUpdate",
-    "PretrainedEmbeddingModelMetadata",
-    "PretrainedEmbeddingModelName",
-    "RACHeadType",
-    "RARHeadType",
-    "RegressionEvaluationRequest",
-    "RegressionMetrics",
-    "RegressionModelMetadata",
-    "RegressionPredictionRequest",
-    "ROCCurve",
-    "ScoredMemory",
-    "ScoredMemoryInsert",
-    "ScoredMemoryInsertMetadata",
-    "ScoredMemoryLookup",
-    "ScoredMemoryLookupMetadata",
-    "ScoredMemoryMetadata",
-    "ScoredMemoryUpdate",
-    "ScoredMemoryUpdateMetadataType0",
-    "ScoredMemoryWithFeedbackMetrics",
-    "ScoredMemoryWithFeedbackMetricsFeedbackMetrics",
-    "ScoredMemoryWithFeedbackMetricsMetadata",
-    "ScorePredictionMemoryLookup",
-    "ScorePredictionMemoryLookupMetadata",
-    "ScorePredictionWithMemoriesAndFeedback",
-    "ServiceUnavailableErrorResponse",
-    "ServiceUnavailableErrorResponseStatusCode",
-    "Task",
-    "TaskStatus",
-    "TaskStatusInfo",
-    "TelemetryFieldType0ItemType0",
-    "TelemetryFieldType0ItemType2",
-    "TelemetryFieldType1ItemType0",
-    "TelemetryFieldType1ItemType1",
-    "TelemetryFilterItem",
-    "TelemetryFilterItemOp",
-    "TelemetryMemoriesRequest",
-    "TelemetrySortOptions",
-    "TelemetrySortOptionsDirection",
-    "UnauthenticatedErrorResponse",
-    "UnauthenticatedErrorResponseStatusCode",
-    "UnauthorizedErrorResponse",
-    "UnauthorizedErrorResponseStatusCode",
-    "UpdateOrgPlanRequest",
-    "UpdateOrgPlanRequestTier",
-    "UpdatePredictionRequest",
-    "ValidationError",
-)

orca_sdk/_generated_api_client/models/action_recommendation.py

@@ -1,82 +0,0 @@
-"""
-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 enum import Enum
-from typing import Any, Type, TypeVar
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..models.action_recommendation_action import ActionRecommendationAction
-
-T = TypeVar("T", bound="ActionRecommendation")
-
-
-@_attrs_define
-class ActionRecommendation:
-    """Recommended actions to take based on the prediction and explanation
-
-    Attributes:
-        action (ActionRecommendationAction): The recommended action to take
-        rationale (str): Explanation for why this action was recommended
-    """
-
-    action: ActionRecommendationAction
-    rationale: str
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        action = self.action.value if isinstance(self.action, Enum) else self.action
-
-        rationale = self.rationale
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "action": action,
-                "rationale": rationale,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        action = ActionRecommendationAction(d.pop("action"))
-
-        rationale = d.pop("rationale")
-
-        action_recommendation = cls(
-            action=action,
-            rationale=rationale,
-        )
-
-        action_recommendation.additional_properties = d
-        return action_recommendation
-
-    @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