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

orca_sdk/__init__.py

@@ -3,8 +3,7 @@ OrcaSDK is a Python library for building and using retrieval augmented models in
 """
 
 from ._utils.common import UNSET, CreateMode, DropMode
-from ._utils.task import TaskStatus
-from .classification_model import ClassificationModel
+from .classification_model import ClassificationMetrics, ClassificationModel
 from .credentials import OrcaCredentials
 from .datasource import Datasource
 from .embedding_model import (
@@ -12,13 +11,23 @@ from .embedding_model import (
     PretrainedEmbeddingModel,
     PretrainedEmbeddingModelName,
 )
+from .job import Job, Status
 from .memoryset import (
+    CascadingEditSuggestion,
     FilterItemTuple,
     LabeledMemory,
     LabeledMemoryLookup,
     LabeledMemoryset,
+    ScoredMemory,
+    ScoredMemoryLookup,
+    ScoredMemoryset,
+)
+from .regression_model import RegressionModel
+from .telemetry import (
+    ClassificationPrediction,
+    FeedbackCategory,
+    RegressionPrediction,
 )
-from .telemetry import LabelPrediction
 
 # only specify things that should show up on the root page of the reference docs because they are in private modules
-__all__ = ["TaskStatus", "UNSET", "CreateMode", "DropMode"]
+__all__ = ["UNSET", "CreateMode", "DropMode"]

orca_sdk/_generated_api_client/api/__init__.py

@@ -5,35 +5,35 @@ from .auth.create_api_key_auth_api_key_post import sync as create_api_key
 from .auth.delete_api_key_auth_api_key_name_or_id_delete import sync as delete_api_key
 from .auth.delete_org_auth_org_delete import sync as delete_org
 from .auth.list_api_keys_auth_api_key_get import sync as list_api_keys
-from .classification_model.create_evaluation_classification_model_model_name_or_id_evaluation_post import (
-    sync as create_evaluation,
+from .classification_model.create_classification_model_gpu_classification_model_post import (
+    sync as create_classification_model_gpu,
 )
-from .classification_model.create_model_classification_model_post import (
-    sync as create_model,
+from .classification_model.delete_classification_model_classification_model_name_or_id_delete import (
+    sync as delete_classification_model,
 )
-from .classification_model.delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete import (
-    sync as delete_evaluation,
+from .classification_model.delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete import (
+    sync as delete_classification_model_evaluation,
 )
-from .classification_model.delete_model_classification_model_name_or_id_delete import (
-    sync as delete_model,
+from .classification_model.evaluate_classification_model_classification_model_model_name_or_id_evaluation_post import (
+    sync as evaluate_classification_model,
 )
-from .classification_model.get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get import (
-    sync as get_evaluation,
+from .classification_model.get_classification_model_classification_model_name_or_id_get import (
+    sync as get_classification_model,
 )
-from .classification_model.get_model_classification_model_name_or_id_get import (
-    sync as get_model,
+from .classification_model.get_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_get import (
+    sync as get_classification_model_evaluation,
 )
-from .classification_model.list_evaluations_classification_model_model_name_or_id_evaluation_get import (
-    sync as list_evaluations,
+from .classification_model.list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get import (
+    sync as list_classification_model_evaluations,
 )
-from .classification_model.list_models_classification_model_get import (
-    sync as list_models,
+from .classification_model.list_classification_models_classification_model_get import (
+    sync as list_classification_models,
 )
-from .classification_model.predict_gpu_classification_model_name_or_id_prediction_post import (
-    sync as predict_gpu,
+from .classification_model.predict_label_gpu_classification_model_name_or_id_prediction_post import (
+    sync as predict_label_gpu,
 )
-from .classification_model.update_model_classification_model_name_or_id_patch import (
-    sync as update_model,
+from .classification_model.update_classification_model_classification_model_name_or_id_patch import (
+    sync as update_classification_model,
 )
 from .datasource.create_datasource_datasource_post import sync as create_datasource
 from .datasource.create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post import (
@@ -74,7 +74,7 @@ from .memoryset.analyze_memoryset_memoryset_name_or_id_analysis_post import (
     sync as analyze_memoryset,
 )
 from .memoryset.batch_delete_memoryset_batch_delete_memoryset_post import (
-    sync as batch_delete,
+    sync as batch_delete_memoryset_batch_delete,
 )
 from .memoryset.clone_memoryset_memoryset_name_or_id_clone_post import (
     sync as clone_memoryset,
@@ -127,6 +127,9 @@ from .memoryset.update_memory_gpu_memoryset_name_or_id_memory_patch import (
 from .memoryset.update_memoryset_memoryset_name_or_id_patch import (
     sync as update_memoryset,
 )
+from .predictive_model.list_predictive_models_predictive_model_get import (
+    sync as list_predictive_models,
+)
 from .pretrained_embedding_model.embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post import (
     sync as embed_with_pretrained_model_gpu,
 )
@@ -136,8 +139,39 @@ from .pretrained_embedding_model.get_pretrained_embedding_model_pretrained_embed
 from .pretrained_embedding_model.list_pretrained_embedding_models_pretrained_embedding_model_get import (
     sync as list_pretrained_embedding_models,
 )
+from .regression_model.create_regression_model_gpu_regression_model_post import (
+    sync as create_regression_model_gpu,
+)
+from .regression_model.delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete import (
+    sync as delete_regression_model_evaluation,
+)
+from .regression_model.delete_regression_model_regression_model_name_or_id_delete import (
+    sync as delete_regression_model,
+)
+from .regression_model.evaluate_regression_model_regression_model_model_name_or_id_evaluation_post import (
+    sync as evaluate_regression_model,
+)
+from .regression_model.get_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_get import (
+    sync as get_regression_model_evaluation,
+)
+from .regression_model.get_regression_model_regression_model_name_or_id_get import (
+    sync as get_regression_model,
+)
+from .regression_model.list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get import (
+    sync as list_regression_model_evaluations,
+)
+from .regression_model.list_regression_models_regression_model_get import (
+    sync as list_regression_models,
+)
+from .regression_model.predict_score_gpu_regression_model_name_or_id_prediction_post import (
+    sync as predict_score_gpu,
+)
+from .regression_model.update_regression_model_regression_model_name_or_id_patch import (
+    sync as update_regression_model,
+)
 from .task.abort_task_task_task_id_abort_delete import sync as abort_task
-from .task.get_task_status_task_task_id_status_get import sync as get_task_status_task
+from .task.get_task_status_task_task_id_status_get import sync as get_task_status
+from .task.get_task_task_task_id_get import sync as get_task
 from .task.list_tasks_task_get import sync as list_tasks
 from .telemetry.count_predictions_telemetry_prediction_count_post import (
     sync as count_predictions,
@@ -173,69 +207,81 @@ from .telemetry.update_prediction_telemetry_prediction_prediction_id_patch impor
 __all__ = [
     "abort_task",
     "analyze_memoryset",
-    "batch_delete",
+    "batch_delete_memoryset_batch_delete",
     "check_authentication",
     "clone_memoryset",
     "count_predictions",
     "create_api_key",
+    "create_classification_model_gpu",
     "create_datasource",
     "create_embedding_evaluation",
-    "create_evaluation",
     "create_finetuned_embedding_model",
     "create_memoryset",
-    "create_model",
+    "create_regression_model_gpu",
     "delete_api_key",
+    "delete_classification_model",
+    "delete_classification_model_evaluation",
     "delete_datasource",
-    "delete_evaluation",
     "delete_finetuned_embedding_model",
     "delete_memories",
     "delete_memory",
     "delete_memoryset",
-    "delete_model",
     "delete_org",
+    "delete_regression_model",
+    "delete_regression_model_evaluation",
     "download_datasource",
     "drop_feedback_category_with_data",
     "embed_with_finetuned_model_gpu",
     "embed_with_pretrained_model_gpu",
+    "evaluate_classification_model",
+    "evaluate_regression_model",
     "explain_prediction",
     "get_analysis",
+    "get_classification_model",
+    "get_classification_model_evaluation",
     "get_datasource",
     "get_embedding_evaluation",
-    "get_evaluation",
     "get_feedback_category",
     "get_finetuned_embedding_model",
     "get_memories",
     "get_memory",
     "get_memoryset",
-    "get_model",
     "get_prediction",
     "get_pretrained_embedding_model",
-    "get_task_status_task",
+    "get_regression_model",
+    "get_regression_model_evaluation",
+    "get_task",
+    "get_task_status",
     "healthcheck",
     "healthcheck_gpu",
     "insert_memories_gpu",
     "list_analyses",
     "list_api_keys",
+    "list_classification_model_evaluations",
+    "list_classification_models",
     "list_datasources",
     "list_embedding_evaluations",
-    "list_evaluations",
     "list_feedback_categories",
     "list_finetuned_embedding_models",
     "list_memories_with_feedback",
     "list_memorysets",
-    "list_models",
     "list_predictions",
+    "list_predictive_models",
     "list_pretrained_embedding_models",
+    "list_regression_model_evaluations",
+    "list_regression_models",
     "list_tasks",
     "memoryset_lookup_gpu",
     "potential_duplicate_groups",
-    "predict_gpu",
+    "predict_label_gpu",
+    "predict_score_gpu",
     "query_memoryset",
     "record_prediction_feedback",
     "suggest_cascading_edits",
+    "update_classification_model",
     "update_memories_gpu",
     "update_memory_gpu",
     "update_memoryset",
-    "update_model",
     "update_prediction",
+    "update_regression_model",
 ]

orca_sdk/_generated_api_client/api/classification_model/create_classification_model_gpu_classification_model_post.py

@@ -0,0 +1,170 @@
+"""
+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
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.classification_model_metadata import ClassificationModelMetadata
+from ...models.create_classification_model_request import CreateClassificationModelRequest
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: CreateClassificationModelRequest,
+) -> dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/gpu/classification_model/",
+    }
+
+    _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> ClassificationModelMetadata:
+    if response.status_code == 201:
+        response_201 = ClassificationModelMetadata.from_dict(response.json())
+
+        return response_201
+    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 == 409:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[ClassificationModelMetadata]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    body: CreateClassificationModelRequest,
+) -> Response[ClassificationModelMetadata]:
+    """Create Classification Model
+
+    Args:
+        body (CreateClassificationModelRequest):
+
+    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[ClassificationModelMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    *,
+    body: CreateClassificationModelRequest,
+) -> ClassificationModelMetadata:
+    """Create Classification Model
+
+    Args:
+        body (CreateClassificationModelRequest):
+
+    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:
+        ClassificationModelMetadata
+    """
+
+    return sync_detailed(
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    body: CreateClassificationModelRequest,
+) -> Response[ClassificationModelMetadata]:
+    """Create Classification Model
+
+    Args:
+        body (CreateClassificationModelRequest):
+
+    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[ClassificationModelMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    *,
+    body: CreateClassificationModelRequest,
+) -> ClassificationModelMetadata:
+    """Create Classification Model
+
+    Args:
+        body (CreateClassificationModelRequest):
+
+    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:
+        ClassificationModelMetadata
+    """
+
+    return (
+        await asyncio_detailed(
+            body=body,
+        )
+    ).parsed

orca_sdk/_generated_api_client/api/classification_model/{get_model_classification_model_name_or_id_get.py → delete_classification_model_classification_model_name_or_id_delete.py}

@@ -17,7 +17,6 @@ import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.rac_model_metadata import RACModelMetadata
 from ...types import Response
 
 
@@ -25,18 +24,17 @@ def _get_kwargs(
     name_or_id: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
-        "method": "get",
+        "method": "delete",
         "url": f"/classification_model/{name_or_id}",
     }
 
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> RACModelMetadata:
-    if response.status_code == 200:
-        response_200 = RACModelMetadata.from_dict(response.json())
-
-        return response_200
+def _parse_response(*, response: httpx.Response) -> None:
+    if response.status_code == 204:
+        response_204 = None
+        return response_204
     if response.status_code == 401:
         raise get_error_for_response(response)
     if response.status_code == 500:
@@ -45,11 +43,13 @@ def _parse_response(*, response: httpx.Response) -> RACModelMetadata:
         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[RACModelMetadata]:
+def _build_response(*, response: httpx.Response) -> Response[None]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -60,8 +60,8 @@ def _build_response(*, response: httpx.Response) -> Response[RACModelMetadata]:
 
 def sync_detailed(
     name_or_id: str,
-) -> Response[RACModelMetadata]:
-    """Get Model
+) -> Response[None]:
+    """Delete Classification Model
 
     Args:
         name_or_id (str):
@@ -71,7 +71,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[RACModelMetadata]
+        Response[None]
     """
 
     kwargs = _get_kwargs(
@@ -88,8 +88,8 @@ def sync_detailed(
 
 def sync(
     name_or_id: str,
-) -> RACModelMetadata:
-    """Get Model
+) -> None:
+    """Delete Classification Model
 
     Args:
         name_or_id (str):
@@ -99,7 +99,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        RACModelMetadata
+        None
     """
 
     return sync_detailed(
@@ -109,8 +109,8 @@ def sync(
 
 async def asyncio_detailed(
     name_or_id: str,
-) -> Response[RACModelMetadata]:
-    """Get Model
+) -> Response[None]:
+    """Delete Classification Model
 
     Args:
         name_or_id (str):
@@ -120,7 +120,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[RACModelMetadata]
+        Response[None]
     """
 
     kwargs = _get_kwargs(
@@ -135,8 +135,8 @@ async def asyncio_detailed(
 
 async def asyncio(
     name_or_id: str,
-) -> RACModelMetadata:
-    """Get Model
+) -> None:
+    """Delete Classification Model
 
     Args:
         name_or_id (str):
@@ -146,7 +146,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        RACModelMetadata
+        None
     """
 
     return (

orca_sdk/_generated_api_client/api/classification_model/{delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py → delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py}

@@ -62,7 +62,7 @@ def sync_detailed(
     model_name_or_id: str,
     task_id: UUID,
 ) -> Response[None]:
-    """Delete Evaluation
+    """Delete Classification Model Evaluation
 
     Args:
         model_name_or_id (str):
@@ -93,7 +93,7 @@ def sync(
     model_name_or_id: str,
     task_id: UUID,
 ) -> None:
-    """Delete Evaluation
+    """Delete Classification Model Evaluation
 
     Args:
         model_name_or_id (str):
@@ -117,7 +117,7 @@ async def asyncio_detailed(
     model_name_or_id: str,
     task_id: UUID,
 ) -> Response[None]:
-    """Delete Evaluation
+    """Delete Classification Model Evaluation
 
     Args:
         model_name_or_id (str):
@@ -146,7 +146,7 @@ async def asyncio(
     model_name_or_id: str,
     task_id: UUID,
 ) -> None:
-    """Delete Evaluation
+    """Delete Classification Model Evaluation
 
     Args:
         model_name_or_id (str):

orca_sdk/_generated_api_client/api/classification_model/{create_evaluation_classification_model_model_name_or_id_evaluation_post.py → evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py}

@@ -17,7 +17,7 @@ import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.evaluation_request import EvaluationRequest
+from ...models.classification_evaluation_request import ClassificationEvaluationRequest
 from ...models.evaluation_response import EvaluationResponse
 from ...types import Response
 
@@ -25,7 +25,7 @@ from ...types import Response
 def _get_kwargs(
     model_name_or_id: str,
     *,
-    body: EvaluationRequest,
+    body: ClassificationEvaluationRequest,
 ) -> dict[str, Any]:
     headers: Dict[str, Any] = {}
 
@@ -72,13 +72,13 @@ def _build_response(*, response: httpx.Response) -> Response[EvaluationResponse]
 def sync_detailed(
     model_name_or_id: str,
     *,
-    body: EvaluationRequest,
+    body: ClassificationEvaluationRequest,
 ) -> Response[EvaluationResponse]:
-    """Create Evaluation
+    """Evaluate Classification Model
 
     Args:
         model_name_or_id (str):
-        body (EvaluationRequest):
+        body (ClassificationEvaluationRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -104,13 +104,13 @@ def sync_detailed(
 def sync(
     model_name_or_id: str,
     *,
-    body: EvaluationRequest,
+    body: ClassificationEvaluationRequest,
 ) -> EvaluationResponse:
-    """Create Evaluation
+    """Evaluate Classification Model
 
     Args:
         model_name_or_id (str):
-        body (EvaluationRequest):
+        body (ClassificationEvaluationRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -129,13 +129,13 @@ def sync(
 async def asyncio_detailed(
     model_name_or_id: str,
     *,
-    body: EvaluationRequest,
+    body: ClassificationEvaluationRequest,
 ) -> Response[EvaluationResponse]:
-    """Create Evaluation
+    """Evaluate Classification Model
 
     Args:
         model_name_or_id (str):
-        body (EvaluationRequest):
+        body (ClassificationEvaluationRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -159,13 +159,13 @@ async def asyncio_detailed(
 async def asyncio(
     model_name_or_id: str,
     *,
-    body: EvaluationRequest,
+    body: ClassificationEvaluationRequest,
 ) -> EvaluationResponse:
-    """Create Evaluation
+    """Evaluate Classification Model
 
     Args:
         model_name_or_id (str):
-        body (EvaluationRequest):
+        body (ClassificationEvaluationRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.