orca-sdk 0.0.94__tar.gz → 0.0.95__tar.gz

{orca_sdk-0.0.94 → orca_sdk-0.0.95}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: orca_sdk
-Version: 0.0.94
+Version: 0.0.95
 Summary: SDK for interacting with Orca Services
 License: Apache-2.0
 Author: Orca DB Inc.
@@ -15,14 +15,12 @@ Requires-Dist: attrs (>=22.2.0)
 Requires-Dist: datasets (>=3.1.0,<4.0.0)
 Requires-Dist: gradio (==5.13.0)
 Requires-Dist: httpx (>=0.20.0,<0.29.0)
-Requires-Dist: networkx (>=3.4.2,<4.0.0)
 Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pyarrow (>=18.0.0,<19.0.0)
 Requires-Dist: python-dateutil (>=2.8.0,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.0,<2.0.0)
 Requires-Dist: scikit-learn (>=1.6.1,<2.0.0)
 Requires-Dist: torch (>=2.5.1,<3.0.0)
-Requires-Dist: transformers (>=4.51.3,<5.0.0)
 Description-Content-Type: text/markdown
 
 <!--
@@ -75,7 +73,7 @@ model = ClassificationModel("my_model", memoryset)
 prediction = model.predict("my input")
 ```
 
-For a more detailed walkthrough, check out our [Quick Start Guide](https://docs.orcadb.ai/quickstart/).
+For a more detailed walkthrough, check out our [Quick Start Guide](https://docs.orcadb.ai/quickstart-sdk/).
 
 ## Support
 

{orca_sdk-0.0.94 → orca_sdk-0.0.95}/README.md

@@ -48,7 +48,7 @@ model = ClassificationModel("my_model", memoryset)
 prediction = model.predict("my input")
 ```
 
-For a more detailed walkthrough, check out our [Quick Start Guide](https://docs.orcadb.ai/quickstart/).
+For a more detailed walkthrough, check out our [Quick Start Guide](https://docs.orcadb.ai/quickstart-sdk/).
 
 ## Support
 

{orca_sdk-0.0.94 → orca_sdk-0.0.95}/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-0.0.94 → orca_sdk-0.0.95}/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_classification_model_post import (
+    sync as create_classification_model,
 )
-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_regression_model_post import (
+    sync as create_regression_model,
+)
+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",
     "create_datasource",
     "create_embedding_evaluation",
-    "create_evaluation",
     "create_finetuned_embedding_model",
     "create_memoryset",
-    "create_model",
+    "create_regression_model",
     "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-0.0.95/orca_sdk/_generated_api_client/api/classification_model/create_classification_model_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": "/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-0.0.95/orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py

@@ -0,0 +1,156 @@
+"""
+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
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...types import Response
+
+
+def _get_kwargs(
+    name_or_id: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "delete",
+        "url": f"/classification_model/{name_or_id}",
+    }
+
+    return _kwargs
+
+
+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:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    if response.status_code == 404:
+        raise get_error_for_response(response)
+    if response.status_code == 409:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[None]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    name_or_id: str,
+) -> Response[None]:
+    """Delete Classification Model
+
+    Args:
+        name_or_id (str):
+
+    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[None]
+    """
+
+    kwargs = _get_kwargs(
+        name_or_id=name_or_id,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    name_or_id: str,
+) -> None:
+    """Delete Classification Model
+
+    Args:
+        name_or_id (str):
+
+    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:
+        None
+    """
+
+    return sync_detailed(
+        name_or_id=name_or_id,
+    ).parsed
+
+
+async def asyncio_detailed(
+    name_or_id: str,
+) -> Response[None]:
+    """Delete Classification Model
+
+    Args:
+        name_or_id (str):
+
+    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[None]
+    """
+
+    kwargs = _get_kwargs(
+        name_or_id=name_or_id,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    name_or_id: str,
+) -> None:
+    """Delete Classification Model
+
+    Args:
+        name_or_id (str):
+
+    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:
+        None
+    """
+
+    return (
+        await asyncio_detailed(
+            name_or_id=name_or_id,
+        )
+    ).parsed

orca_sdk-0.0.94/orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py → orca_sdk-0.0.95/orca_sdk/_generated_api_client/api/classification_model/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):