orca-sdk 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

orca_sdk/telemetry_test.py

@@ -2,60 +2,79 @@ import pytest
 
 from .classification_model import ClassificationModel
 from .memoryset import LabeledMemoryLookup
-from .telemetry import FeedbackCategory, LabelPrediction
+from .telemetry import ClassificationPrediction, FeedbackCategory
 
 
-def test_get_prediction(model: ClassificationModel):
-    predictions = model.predict(["Do you love soup?", "Are cats cute?"])
-    prediction_with_telemetry = LabelPrediction.get(predictions[0].prediction_id)
+def test_get_prediction(classification_model: ClassificationModel):
+    predictions = classification_model.predict(["Do you love soup?", "Are cats cute?"])
+    assert len(predictions) == 2
+    assert predictions[0].prediction_id is not None
+    assert predictions[1].prediction_id is not None
+    prediction_with_telemetry = ClassificationPrediction.get(predictions[0].prediction_id)
     assert prediction_with_telemetry is not None
     assert prediction_with_telemetry.label == 0
     assert prediction_with_telemetry.input_value == "Do you love soup?"
 
 
-def test_get_predictions(model: ClassificationModel):
-    predictions = model.predict(["Do you love soup?", "Are cats cute?"])
-    prediction_with_telemetry = LabelPrediction.get([predictions[0].prediction_id, predictions[1].prediction_id])
+def test_get_predictions(classification_model: ClassificationModel):
+    predictions = classification_model.predict(["Do you love soup?", "Are cats cute?"])
+    assert len(predictions) == 2
+    assert predictions[0].prediction_id is not None
+    assert predictions[1].prediction_id is not None
+    prediction_with_telemetry = ClassificationPrediction.get(
+        [predictions[0].prediction_id, predictions[1].prediction_id]
+    )
     assert len(prediction_with_telemetry) == 2
     assert prediction_with_telemetry[0].label == 0
     assert prediction_with_telemetry[0].input_value == "Do you love soup?"
+    assert prediction_with_telemetry[1].label == 1
 
 
-def test_get_prediction_memory_lookups(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_get_predictions_with_expected_label_match(classification_model: ClassificationModel):
+    classification_model.predict(
+        ["Do you love soup?", "Are cats cute?"], expected_labels=[0, 0], tags={"expected_label_match"}
+    )
+    classification_model.predict("no expectations", tags={"expected_label_match"})
+    assert len(classification_model.predictions(tag="expected_label_match")) == 3
+    assert len(classification_model.predictions(expected_label_match=True, tag="expected_label_match")) == 1
+    assert len(classification_model.predictions(expected_label_match=False, tag="expected_label_match")) == 1
+
+
+def test_get_prediction_memory_lookups(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     assert isinstance(prediction.memory_lookups, list)
     assert len(prediction.memory_lookups) > 0
     assert all(isinstance(lookup, LabeledMemoryLookup) for lookup in prediction.memory_lookups)
 
 
-def test_record_feedback(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_record_feedback(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     assert "correct" not in prediction.feedback
     prediction.record_feedback(category="correct", value=prediction.label == 0)
     assert prediction.feedback["correct"] is True
 
 
-def test_record_feedback_with_invalid_value(model: ClassificationModel):
+def test_record_feedback_with_invalid_value(classification_model: ClassificationModel):
     with pytest.raises(ValueError, match=r"Invalid input.*"):
-        model.predict("Do you love soup?").record_feedback(category="correct", value="not a bool")  # type: ignore
+        classification_model.predict("Do you love soup?").record_feedback(category="correct", value="not a bool")  # type: ignore
 
 
-def test_record_feedback_with_inconsistent_value_for_category(model: ClassificationModel):
-    model.predict("Do you love soup?").record_feedback(category="correct", value=True)
+def test_record_feedback_with_inconsistent_value_for_category(classification_model: ClassificationModel):
+    classification_model.predict("Do you love soup?").record_feedback(category="correct", value=True)
     with pytest.raises(ValueError, match=r"Invalid input.*"):
-        model.predict("Do you love soup?").record_feedback(category="correct", value=-1.0)
+        classification_model.predict("Do you love soup?").record_feedback(category="correct", value=-1.0)
 
 
-def test_delete_feedback(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_delete_feedback(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     prediction.record_feedback(category="test_delete", value=True)
     assert "test_delete" in prediction.feedback
     prediction.delete_feedback("test_delete")
     assert "test_delete" not in prediction.feedback
 
 
-def test_list_feedback_categories(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_list_feedback_categories(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     prediction.record_feedback(category="correct", value=True)
     prediction.record_feedback(category="confidence", value=0.8)
     categories = FeedbackCategory.all()
@@ -64,8 +83,8 @@ def test_list_feedback_categories(model: ClassificationModel):
     assert any(c.name == "confidence" and c.value_type == float for c in categories)
 
 
-def test_drop_feedback_category(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_drop_feedback_category(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     prediction.record_feedback(category="test_category", value=True)
     assert any(c.name == "test_category" for c in FeedbackCategory.all())
     FeedbackCategory.drop("test_category")
@@ -74,8 +93,8 @@ def test_drop_feedback_category(model: ClassificationModel):
     assert "test_category" not in prediction.feedback
 
 
-def test_update_prediction(model: ClassificationModel):
-    prediction = model.predict("Do you love soup?")
+def test_update_prediction(classification_model: ClassificationModel):
+    prediction = classification_model.predict("Do you love soup?")
     assert prediction.expected_label is None
     assert prediction.tags == set()
     # update expected label

{orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA

@@ -1,24 +1,25 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: orca_sdk
-Version: 0.1.1
+Version: 0.1.3
 Summary: SDK for interacting with Orca Services
-License: Apache-2.0
+License-Expression: Apache-2.0
 Author: Orca DB Inc.
 Author-email: dev-rel@orcadb.ai
-Requires-Python: >=3.10,<3.13
-Classifier: License :: OSI Approved :: Apache Software License
+Requires-Python: >=3.11,<3.14
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.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: 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: torch (>=2.5.1,<3.0.0)
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: datasets (>=3.1.0,<4)
+Requires-Dist: gradio (>=5.44.1,<6)
+Requires-Dist: httpx (>=0.28.1)
+Requires-Dist: httpx-retries (>=0.4.3,<0.5.0)
+Requires-Dist: numpy (>=2.1.0,<3)
+Requires-Dist: pandas (>=2.2.3,<3)
+Requires-Dist: pyarrow (>=18.0.0,<19)
+Requires-Dist: python-dotenv (>=1.1.0)
+Requires-Dist: scikit-learn (>=1.6.1,<2)
+Requires-Dist: torch (>=2.8.0,<3)
 Description-Content-Type: text/markdown
 
 <!--
@@ -45,7 +46,7 @@ You can find the documentation for all things Orca at [docs.orcadb.ai](https://d
 
 ## Installation
 
-Orca SDK is compatible with Python 3.10 or higher and is available on [PyPI](https://pypi.org/project/orca_sdk/). You can install it with your favorite python package manager:
+OrcaSDK is compatible with Python 3.10 or higher and is available on [PyPI](https://pypi.org/project/orca_sdk/). You can install it with your favorite python package manager:
 
 - Pip: `pip install orca_sdk`
 - Conda: `conda install orca_sdk`
@@ -69,9 +70,26 @@ model = ClassificationModel("my_model", memoryset)
 
 # Make predictions
 prediction = model.predict("my input")
+
+# Get Action Recommendation
+action, rationale = prediction.recommend_action()
+print(f"Recommended action: {action}")
+print(f"Rationale: {rationale}")
+
+# Generate and add synthetic memory suggestions
+if action == "add_memories":
+    suggestions = prediction.generate_memory_suggestions(num_memories=3)
+
+    # Review suggestions
+    for suggestion in suggestions:
+        print(f"Suggested: '{suggestion['value']}' -> {suggestion['label']}")
+
+    # Add suggestions to memoryset
+    model.memoryset.insert(suggestions)
+    print(f"Added {len(suggestions)} new memories to improve model performance!")
 ```
 
-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.1.3.dist-info/RECORD

@@ -0,0 +1,41 @@
+orca_sdk/__init__.py,sha256=xyjNwkLQXaX8A-UYgGwYDjv2btOXArT_yiMTfmW7KA8,1003
+orca_sdk/_shared/__init__.py,sha256=3Kt0Hu3QLI5FEp9nqGTxqAm3hAoBJKcagfaGQZ-lbJQ,223
+orca_sdk/_shared/metrics.py,sha256=LEZfAUWUtUWv_WWy9F_yjGLlUQHQpmR9WxG2fbKxa7U,14419
+orca_sdk/_shared/metrics_test.py,sha256=Rw1MaH37FppNsMnW8Ir9vMd8xxnZt3eo2Iypx1igtBI,9440
+orca_sdk/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+orca_sdk/_utils/analysis_ui.py,sha256=nT-M_YcNRCVPQzvuqYNFKnNHhYkADYBvq1GlIUePrWw,9232
+orca_sdk/_utils/analysis_ui_style.css,sha256=q_ba_-_KtgztepHg829zLzypaxKayl7ySC1-oYDzV3k,836
+orca_sdk/_utils/auth.py,sha256=nC252O171_3_wn4KBAN7kg8GNvoZFiQ5Xtzkrm5dWDo,2645
+orca_sdk/_utils/auth_test.py,sha256=ygVWv1Ex53LaxIP7p2hzPHl8l9qYyBD5IGmEFJMps6s,1056
+orca_sdk/_utils/common.py,sha256=wUm2pNDWytEecC5WiDWd02-yCZw3Akx0bIutG4lHsFA,805
+orca_sdk/_utils/data_parsing.py,sha256=gkAwWEC8qRt3vRUObe7n7Pr0azOayNwc2yFY04WFp7E,5220
+orca_sdk/_utils/data_parsing_test.py,sha256=fNEYzPzE1jt3KWE2Kj91KqIeuv-L5REHFAa98zkNGSQ,8962
+orca_sdk/_utils/pagination.py,sha256=986z0QPZixrZeurJWorF6eMgnTRdDF84AagEA6qNbMw,4245
+orca_sdk/_utils/pagination_test.py,sha256=BUylCrcHnwoKEBmMUzVr0lwLpA35ivcCwdBK4rMw9y8,4887
+orca_sdk/_utils/prediction_result_ui.css,sha256=sqBlkRLnovb5X5EcUDdB6iGpH63nVRlTW4uAmXuD0WM,258
+orca_sdk/_utils/prediction_result_ui.py,sha256=Ur_FY7dz3oWNmtPiP3Wl3yRlEMgK8q9UfT-SDu9UPxA,4805
+orca_sdk/_utils/tqdm_file_reader.py,sha256=Lw7Cg1UgNuRUoN6jjqZb-IlV00H-kbRcrZLdudr1GxE,324
+orca_sdk/_utils/value_parser.py,sha256=c3qMABCCDQcIjn9N1orYYnlRwDW9JWdGwW_2TDZPLdI,1286
+orca_sdk/_utils/value_parser_test.py,sha256=OybsiC-Obi32RRi9NIuwrVBRAnlyPMV1xVAaevSrb7M,1079
+orca_sdk/async_client.py,sha256=HK52VxltotpDdq-aTgsCHQPsDAYzOSZDxdlbOnal99c,125459
+orca_sdk/classification_model.py,sha256=WJM6oLBuGrxleTWakc-ZgSRfNyiZxb6-GIMH-S7k12w,39700
+orca_sdk/classification_model_test.py,sha256=_gaDg8QB0h0ByN4UwTk2fIIDXE4UzahuJBjz7NSPK28,23605
+orca_sdk/client.py,sha256=dcGBnzIwaU74CMzUh1ObKJbVmZekF5n57gQY6YcQwHE,124550
+orca_sdk/conftest.py,sha256=RtINF1xea2iMycMkpMXIOOqRbfWeIZsceSAemhBmgNE,9761
+orca_sdk/credentials.py,sha256=80_1r8n5jruEvN_E629SaRrRhKvF_NhWUEZyZzPXkqQ,6620
+orca_sdk/credentials_test.py,sha256=TLbXJMz3IlThvtSrHeLM7jRsKnrncA_ahOTpHg15Ei4,4089
+orca_sdk/datasource.py,sha256=DJt1Hr8iwaTFbtFD1aqbUPytpmjPr39qISeqSumoraM,20668
+orca_sdk/datasource_test.py,sha256=yBR0NbsAzChV97pSOU0IvlfF5_WbMe49wZeWNXxwNl4,12128
+orca_sdk/embedding_model.py,sha256=IQCpGUUlKHtz33Ld1-Ag8eLMk72qT7K-cHDjBJGJqhQ,27689
+orca_sdk/embedding_model_test.py,sha256=-NItbNb3tTVj5jAvSi3WjV3FP448q08lmT5iObg9vwA,8133
+orca_sdk/job.py,sha256=BOHg9ksVcN26VtAmuA2cNjGed_Gsx2zbdCO6FBZjuqI,13119
+orca_sdk/job_test.py,sha256=nRSWxd_1UIfrj9oMVvrXjt6OBkBpddYAjb2y6P-DTUg,4327
+orca_sdk/memoryset.py,sha256=angGB6OUJRDoBa2xzl4WsYhNsRRj99dEKEaAKUqgxO8,100113
+orca_sdk/memoryset_test.py,sha256=wqoHXP60CBvtsReCunQNUxj6_ZDT67TTTluguQapigs,25368
+orca_sdk/regression_model.py,sha256=kIT2i4XMrTBZXXVqDENoYILLF7Zqa8o2ndhraXHUPbY,26437
+orca_sdk/regression_model_test.py,sha256=slwxbty_vL9d24OCn5xN61eKyri5GS7Jv2YmpEOMTrM,15856
+orca_sdk/telemetry.py,sha256=C0rTudfAV3_t_uADATrl06d7vk-Sgop24FiSSqYhqmc,26209
+orca_sdk/telemetry_test.py,sha256=eT66C5lFdNg-pQdo2I__BP7Tn5fTc9aTkVo9ZhWwhU0,5519
+orca_sdk-0.1.3.dist-info/METADATA,sha256=tbUzJDcZGUOkwPeToA74JXtkJErdpmtx7UtL1mhzm_M,3659
+orca_sdk-0.1.3.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+orca_sdk-0.1.3.dist-info/RECORD,,

{orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

orca_sdk/_generated_api_client/__init__.py

@@ -1,3 +0,0 @@
-import logging
-
-logging.getLogger("httpx").setLevel(logging.WARNING)

orca_sdk/_generated_api_client/api/__init__.py

@@ -1,193 +0,0 @@
-"""Contains methods for accessing the API"""
-
-from .auth.check_authentication_auth_get import sync as check_authentication
-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_model_classification_model_post import (
-    sync as create_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_model_classification_model_name_or_id_delete import (
-    sync as delete_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_model_classification_model_name_or_id_get import (
-    sync as get_model,
-)
-from .classification_model.list_evaluations_classification_model_model_name_or_id_evaluation_get import (
-    sync as list_evaluations,
-)
-from .classification_model.list_models_classification_model_get import (
-    sync as list_models,
-)
-from .classification_model.predict_gpu_classification_model_name_or_id_prediction_post import (
-    sync as predict_gpu,
-)
-from .datasource.create_datasource_datasource_post import sync as create_datasource
-from .datasource.delete_datasource_datasource_name_or_id_delete import (
-    sync as delete_datasource,
-)
-from .datasource.get_datasource_datasource_name_or_id_get import sync as get_datasource
-from .datasource.list_datasources_datasource_get import sync as list_datasources
-from .default.healthcheck_get import sync as healthcheck
-from .default.healthcheck_gpu_get import sync as healthcheck_gpu
-from .finetuned_embedding_model.create_finetuned_embedding_model_finetuned_embedding_model_post import (
-    sync as create_finetuned_embedding_model,
-)
-from .finetuned_embedding_model.delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete import (
-    sync as delete_finetuned_embedding_model,
-)
-from .finetuned_embedding_model.embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post import (
-    sync as embed_with_finetuned_model_gpu,
-)
-from .finetuned_embedding_model.get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get import (
-    sync as get_finetuned_embedding_model,
-)
-from .finetuned_embedding_model.list_finetuned_embedding_models_finetuned_embedding_model_get import (
-    sync as list_finetuned_embedding_models,
-)
-from .memoryset.clone_memoryset_memoryset_name_or_id_clone_post import (
-    sync as clone_memoryset,
-)
-from .memoryset.create_analysis_memoryset_name_or_id_analysis_post import (
-    sync as create_analysis,
-)
-from .memoryset.create_memoryset_memoryset_post import sync as create_memoryset
-from .memoryset.delete_memories_memoryset_name_or_id_memories_delete_post import (
-    sync as delete_memories,
-)
-from .memoryset.delete_memory_memoryset_name_or_id_memory_memory_id_delete import (
-    sync as delete_memory,
-)
-from .memoryset.delete_memoryset_memoryset_name_or_id_delete import (
-    sync as delete_memoryset,
-)
-from .memoryset.get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get import (
-    sync as get_analysis,
-)
-from .memoryset.get_memories_memoryset_name_or_id_memories_get_post import (
-    sync as get_memories,
-)
-from .memoryset.get_memory_memoryset_name_or_id_memory_memory_id_get import (
-    sync as get_memory,
-)
-from .memoryset.get_memoryset_memoryset_name_or_id_get import sync as get_memoryset
-from .memoryset.insert_memories_gpu_memoryset_name_or_id_memory_post import (
-    sync as insert_memories_gpu,
-)
-from .memoryset.list_analyses_memoryset_name_or_id_analysis_get import (
-    sync as list_analyses,
-)
-from .memoryset.list_memorysets_memoryset_get import sync as list_memorysets
-from .memoryset.memoryset_lookup_gpu_memoryset_name_or_id_lookup_post import (
-    sync as memoryset_lookup_gpu,
-)
-from .memoryset.query_memoryset_memoryset_name_or_id_memories_post import (
-    sync as query_memoryset,
-)
-from .memoryset.update_memories_gpu_memoryset_name_or_id_memories_patch import (
-    sync as update_memories_gpu,
-)
-from .memoryset.update_memory_gpu_memoryset_name_or_id_memory_patch import (
-    sync as update_memory_gpu,
-)
-from .pretrained_embedding_model.embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post import (
-    sync as embed_with_pretrained_model_gpu,
-)
-from .pretrained_embedding_model.get_pretrained_embedding_model_pretrained_embedding_model_model_name_get import (
-    sync as get_pretrained_embedding_model,
-)
-from .pretrained_embedding_model.list_pretrained_embedding_models_pretrained_embedding_model_get import (
-    sync as list_pretrained_embedding_models,
-)
-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.list_tasks_task_get import sync as list_tasks
-from .telemetry.drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete import (
-    sync as drop_feedback_category_with_data,
-)
-from .telemetry.get_feedback_category_telemetry_feedback_category_name_or_id_get import (
-    sync as get_feedback_category,
-)
-from .telemetry.get_prediction_telemetry_prediction_prediction_id_get import (
-    sync as get_prediction,
-)
-from .telemetry.list_feedback_categories_telemetry_feedback_category_get import (
-    sync as list_feedback_categories,
-)
-from .telemetry.list_predictions_telemetry_prediction_post import (
-    sync as list_predictions,
-)
-from .telemetry.record_prediction_feedback_telemetry_prediction_feedback_put import (
-    sync as record_prediction_feedback,
-)
-from .telemetry.update_prediction_telemetry_prediction_prediction_id_patch import (
-    sync as update_prediction,
-)
-
-__all__ = [
-    "list_datasources",
-    "create_datasource",
-    "delete_datasource",
-    "get_datasource",
-    "check_authentication",
-    "create_api_key",
-    "list_api_keys",
-    "delete_api_key",
-    "delete_org",
-    "get_finetuned_embedding_model",
-    "create_finetuned_embedding_model",
-    "list_finetuned_embedding_models",
-    "delete_finetuned_embedding_model",
-    "embed_with_finetuned_model_gpu",
-    "list_pretrained_embedding_models",
-    "get_pretrained_embedding_model",
-    "embed_with_pretrained_model_gpu",
-    "healthcheck",
-    "healthcheck_gpu",
-    "create_analysis",
-    "update_memory_gpu",
-    "get_memoryset",
-    "get_memories",
-    "delete_memoryset",
-    "list_analyses",
-    "delete_memories",
-    "clone_memoryset",
-    "query_memoryset",
-    "list_memorysets",
-    "create_memoryset",
-    "get_memory",
-    "insert_memories_gpu",
-    "memoryset_lookup_gpu",
-    "update_memories_gpu",
-    "delete_memory",
-    "get_analysis",
-    "get_task_status_task",
-    "abort_task",
-    "list_tasks",
-    "list_predictions",
-    "update_prediction",
-    "list_feedback_categories",
-    "get_prediction",
-    "drop_feedback_category_with_data",
-    "record_prediction_feedback",
-    "get_feedback_category",
-    "create_model",
-    "delete_model",
-    "predict_gpu",
-    "get_model",
-    "delete_evaluation",
-    "create_evaluation",
-    "list_evaluations",
-    "list_models",
-    "get_evaluation",
-]

orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py

@@ -1,128 +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/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, cast
-
-import httpx
-
-from ...client import _client_context
-from ...errors import get_error_for_response
-from ...types import Response
-
-
-def _get_kwargs() -> dict[str, Any]:
-    _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": "/auth/",
-    }
-
-    return _kwargs
-
-
-def _parse_response(*, response: httpx.Response) -> bool:
-    if response.status_code == 200:
-        response_200 = cast(bool, response.json())
-        return response_200
-    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 == 422:
-        raise get_error_for_response(response)
-
-
-def _build_response(*, response: httpx.Response) -> Response[bool]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(response=response),
-    )
-
-
-def sync_detailed() -> Response[bool]:
-    """Check Authentication
-
-     Returns true if the api key header is valid for the org (will be false for admin api key)
-
-    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[bool]
-    """
-
-    kwargs = _get_kwargs()
-
-    with _client_context() as client:
-        response = client.get_httpx_client().request(
-            **kwargs,
-        )
-
-        return _build_response(response=response)
-
-
-def sync() -> bool:
-    """Check Authentication
-
-     Returns true if the api key header is valid for the org (will be false for admin api key)
-
-    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:
-        bool
-    """
-
-    return sync_detailed().parsed
-
-
-async def asyncio_detailed() -> Response[bool]:
-    """Check Authentication
-
-     Returns true if the api key header is valid for the org (will be false for admin api key)
-
-    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[bool]
-    """
-
-    kwargs = _get_kwargs()
-
-    with _client_context() as client:
-        response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(response=response)
-
-
-async def asyncio() -> bool:
-    """Check Authentication
-
-     Returns true if the api key header is valid for the org (will be false for admin api key)
-
-    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:
-        bool
-    """
-
-    return (await asyncio_detailed()).parsed