orca-sdk 0.0.102__py3-none-any.whl → 0.0.103__py3-none-any.whl

orca_sdk/_utils/auth.py

@@ -53,7 +53,7 @@ def _delete_org(org_id: str) -> None:
 def _authenticate_local_api(org_id: str = _DEFAULT_ORG_ID, api_key_name: str = "local") -> None:
     """Connect to the local API at http://localhost:1584/ and authenticate with a new API key"""
     _delete_api_key(org_id, api_key_name, if_not_exists="ignore")
-    OrcaCredentials.set_base_url("http://localhost:1584")
+    OrcaCredentials.set_api_url("http://localhost:1584")
     OrcaCredentials.set_api_key(_create_api_key(org_id, api_key_name))
     logging.info(f"Authenticated against local API at 'http://localhost:1584' with '{api_key_name}' API key")
 

orca_sdk/classification_model_test.py

@@ -3,7 +3,7 @@ from uuid import uuid4
 
 import numpy as np
 import pytest
-from datasets.arrow_dataset import Dataset
+from datasets import Dataset
 
 from .classification_model import ClassificationMetrics, ClassificationModel
 from .conftest import skip_in_ci

orca_sdk/client.py

@@ -36,6 +36,10 @@ class AddMemorySuggestion(TypedDict):
     label_name: str
 
 
+class AliveResponse(TypedDict):
+    ok: bool
+
+
 class ApiKeyMetadata(TypedDict):
     id: str
     org_id: str
@@ -201,6 +205,14 @@ class GetMemoriesRequest(TypedDict):
     memory_ids: list[str]
 
 
+class HealthyResponse(TypedDict):
+    ok: bool
+    checks: dict[str, bool]
+    durations: dict[str, int]
+    draining: bool
+    config: dict[str, str | float | int | bool | None]
+
+
 class InternalServerErrorResponse(TypedDict):
     status_code: NotRequired[int]
     message: str
@@ -282,6 +294,7 @@ class MemorysetClassPatternsMetrics(TypedDict):
     variance_spread: float
     mean_uniformity: float
     variance_uniformity: float
+    updated_at: str
 
 
 class MemorysetClusterAnalysisConfig(TypedDict):
@@ -299,6 +312,7 @@ class MemorysetClusterMetrics(TypedDict):
     cluster_metrics: list[ClusterMetrics]
     num_outliers: int
     num_clusters: int
+    updated_at: str
 
 
 class MemorysetConceptAnalysisConfig(TypedDict):
@@ -321,6 +335,7 @@ class MemorysetDuplicateAnalysisConfig(TypedDict):
 class MemorysetDuplicateMetrics(TypedDict):
     num_duplicates: int
     num_potential_duplicates: int
+    updated_at: str
 
 
 class MemorysetLabelAnalysisConfig(TypedDict):
@@ -334,6 +349,7 @@ class MemorysetLabelMetrics(TypedDict):
     mean_neighbor_label_entropy: float
     mean_neighbor_predicted_label_ambiguity: float
     num_potential_mislabels: int
+    updated_at: str
 
 
 class MemorysetNeighborAnalysisConfig(TypedDict):
@@ -343,6 +359,7 @@ class MemorysetNeighborAnalysisConfig(TypedDict):
 
 class MemorysetNeighborMetrics(TypedDict):
     lookup_score_metrics: dict[str, LookupScoreMetrics]
+    updated_at: str
 
 
 class MemorysetProjectionAnalysisConfig(TypedDict):
@@ -351,7 +368,7 @@ class MemorysetProjectionAnalysisConfig(TypedDict):
 
 
 class MemorysetProjectionMetrics(TypedDict):
-    pass
+    updated_at: str
 
 
 class MemorysetUpdate(TypedDict):
@@ -359,6 +376,7 @@ class MemorysetUpdate(TypedDict):
     description: NotRequired[str | None]
     name: NotRequired[str]
     notes: NotRequired[str | None]
+    hidden: NotRequired[bool]
 
 
 class NotFoundErrorResponse(TypedDict):
@@ -477,6 +495,11 @@ class ROCCurve(TypedDict):
     true_positive_rates: list[float]
 
 
+class ReadyResponse(TypedDict):
+    ok: bool
+    draining: bool
+
+
 class RegressionEvaluationRequest(TypedDict):
     datasource_name_or_id: str
     memoryset_override_name_or_id: NotRequired[str | None]
@@ -639,7 +662,7 @@ class SubConceptMetrics(TypedDict):
     memory_count: int
 
 
-TaskStatus = Literal["INITIALIZED", "DISPATCHED", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
+TaskStatus = Literal["INITIALIZED", "DISPATCHED", "WAITING", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
 
 
 class TaskStatusInfo(TypedDict):
@@ -697,6 +720,7 @@ class DeleteAuthApiKeyByNameOrIdParams(TypedDict):
 
 class GetMemorysetParams(TypedDict):
     type: NotRequired[MemoryType | None]
+    show_hidden: NotRequired[bool | None]
 
 
 class PostMemorysetByNameOrIdCloneParams(TypedDict):
@@ -1149,6 +1173,7 @@ class CreateMemorysetRequest(TypedDict):
     index_type: NotRequired[Literal["FLAT", "IVF_FLAT", "IVF_SQ8", "IVF_PQ", "HNSW", "DISKANN"]]
     index_params: NotRequired[dict[str, int | float | str]]
     prompt: NotRequired[str]
+    hidden: NotRequired[bool]
 
 
 class CreateRegressionModelRequest(TypedDict):
@@ -1410,6 +1435,7 @@ class MemorysetAnalysisRequest(TypedDict):
 class MemorysetConceptMetrics(TypedDict):
     concepts: list[ConceptMetrics]
     num_outliers: int
+    updated_at: str
 
 
 class MemorysetMetrics(TypedDict):
@@ -1514,6 +1540,7 @@ class MemorysetMetadata(TypedDict):
     database_uri: str | None
     document_prompt_override: str | None
     query_prompt_override: str | None
+    hidden: bool
 
 
 class PaginatedTask(TypedDict):
@@ -1565,7 +1592,7 @@ class OrcaClient(Client):
     @overload
     def GET(
         self,
-        path: Literal["/"],
+        path: Literal["/check/alive"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1575,13 +1602,13 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> AliveResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/check/alive"],
+        path: Literal["/check/ready"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1591,13 +1618,13 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> ReadyResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/check/ready"],
+        path: Literal["/gpu/check/healthy"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1607,7 +1634,87 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> HealthyResponse:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/check/healthy"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> HealthyResponse:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/gpu/config"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> dict[str, str | float | int | bool | None]:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/config"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> dict[str, str | float | int | bool | None]:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/gpu/"],
+        *,
+        params: None = None,
+        parse_as: Literal["text"],
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> str:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/"],
+        *,
+        params: None = None,
+        parse_as: Literal["text"],
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> str:
         pass
 
     @overload
@@ -2345,54 +2452,6 @@ class OrcaClient(Client):
         """Get the status of a bootstrap classification model task"""
         pass
 
-    @overload
-    def GET(
-        self,
-        path: Literal["/gpu/"],
-        *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> Any:
-        pass
-
-    @overload
-    def GET(
-        self,
-        path: Literal["/gpu/check/alive"],
-        *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> Any:
-        pass
-
-    @overload
-    def GET(
-        self,
-        path: Literal["/gpu/check/ready"],
-        *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> Any:
-        pass
-
     def GET(
         self,
         path: str,

orca_sdk/conftest.py

@@ -24,7 +24,7 @@ os.environ["ORCA_SAVE_TELEMETRY_SYNCHRONOUSLY"] = "true"
 
 def skip_in_prod(reason: str):
     """Custom decorator to skip tests when running against production API"""
-    PROD_API_URLs = ["https://api.orcadb.ai", "https://api.dev.orcadb.ai"]
+    PROD_API_URLs = ["https://api.orcadb.ai", "https://api.staging.orcadb.ai"]
     return pytest.mark.skipif(
         os.environ["ORCA_API_URL"] in PROD_API_URLs,
         reason=reason,
@@ -45,7 +45,7 @@ def _create_org_id():
 
 
 @pytest.fixture()
-def base_url_reset():
+def api_url_reset():
     original_base_url = orca_api.base_url
     yield
     orca_api.base_url = original_base_url

orca_sdk/credentials.py

@@ -35,11 +35,33 @@ class OrcaCredentials:
     """
 
     @staticmethod
-    def get_api_url() -> str:
+    def is_authenticated() -> bool:
         """
-        Get the Orca API base URL that is currently being used
+        Check if you are authenticated to interact with the Orca API
+
+        Returns:
+            True if you are authenticated, False otherwise
         """
-        return str(orca_api.base_url)
+        try:
+            return orca_api.GET("/auth")
+        except ValueError as e:
+            if "Invalid API key" in str(e):
+                return False
+            raise e
+
+    @staticmethod
+    def is_healthy() -> bool:
+        """
+        Check whether the API is healthy
+
+        Returns:
+            True if the API is healthy, False otherwise
+        """
+        try:
+            orca_api.GET("/check/healthy")
+        except Exception:
+            return False
+        return True
 
     @staticmethod
     def list_api_keys() -> list[ApiKeyInfo]:
@@ -58,21 +80,6 @@ class OrcaCredentials:
             for api_key in orca_api.GET("/auth/api_key")
         ]
 
-    @staticmethod
-    def is_authenticated() -> bool:
-        """
-        Check if you are authenticated to interact with the Orca API
-
-        Returns:
-            True if you are authenticated, False otherwise
-        """
-        try:
-            return orca_api.GET("/auth")
-        except ValueError as e:
-            if "Invalid API key" in str(e):
-                return False
-            raise e
-
     @staticmethod
     def create_api_key(name: str, scopes: set[Scope] = {"ADMINISTER"}) -> str:
         """
@@ -104,20 +111,6 @@ class OrcaCredentials:
         """
         orca_api.DELETE("/auth/api_key/{name_or_id}", params={"name_or_id": name})
 
-    @staticmethod
-    def set_headers(headers: dict[str, str]):
-        """
-        Add or override default HTTP headers for all Orca API requests.
-
-        Params:
-            headers: Mapping of header names to their string values
-
-        Notes:
-            New keys are merged into the existing headers, this will overwrite headers with the
-            same name, but leave other headers untouched.
-        """
-        orca_api.headers.update(Headers(headers))
-
     @staticmethod
     def set_api_key(api_key: str, check_validity: bool = True):
         """
@@ -133,17 +126,24 @@ class OrcaCredentials:
         Raises:
             ValueError: if the API key is invalid and `check_validity` is True
         """
-        OrcaCredentials.set_headers({"Api-Key": api_key})
+        OrcaCredentials.set_api_headers({"Api-Key": api_key})
         if check_validity:
             orca_api.GET("/auth")
 
     @staticmethod
-    def set_base_url(base_url: str, check_validity: bool = True):
+    def get_api_url() -> str:
+        """
+        Get the base URL of the Orca API that is currently being used
+        """
+        return str(orca_api.base_url)
+
+    @staticmethod
+    def set_api_url(url: str, check_validity: bool = True):
         """
         Set the base URL for the Orca API
 
         Args:
-            base_url: The base URL to set
+            url: The base URL to set
             check_validity: Whether to check if there is an API running at the given base URL
 
         Raises:
@@ -152,27 +152,26 @@ class OrcaCredentials:
         # check if the base url is reachable before setting it
         if check_validity:
             try:
-                httpx.get(base_url, timeout=1)
+                httpx.get(url, timeout=1)
             except ConnectError as e:
-                raise ValueError(f"No API found at {base_url}") from e
+                raise ValueError(f"No API found at {url}") from e
 
-        orca_api.base_url = base_url
+        orca_api.base_url = url
 
         # check if the api passes the health check
         if check_validity:
-            orca_api.GET("/")
+            OrcaCredentials.is_healthy()
 
     @staticmethod
-    def is_healthy() -> bool:
+    def set_api_headers(headers: dict[str, str]):
         """
-        Check whether the API is healthy
+        Add or override default HTTP headers for all Orca API requests.
 
-        Returns:
-            True if the API is healthy, False otherwise
+        Params:
+            headers: Mapping of header names to their string values
+
+        Notes:
+            New keys are merged into the existing headers, this will overwrite headers with the
+            same name, but leave other headers untouched.
         """
-        try:
-            orca_api.GET("/")
-            orca_api.GET("/gpu/")
-        except Exception:
-            return False
-        return True
+        orca_api.headers.update(Headers(headers))

orca_sdk/credentials_test.py

@@ -38,20 +38,20 @@ def test_set_invalid_api_key(api_key):
     assert not OrcaCredentials.is_authenticated()
 
 
-def test_set_base_url(base_url_reset):
-    OrcaCredentials.set_base_url("http://api.orcadb.ai")
+def test_set_api_url(api_url_reset):
+    OrcaCredentials.set_api_url("http://api.orcadb.ai")
     assert str(orca_api.base_url) == "http://api.orcadb.ai"
 
 
 def test_set_invalid_base_url():
     with pytest.raises(ValueError, match="No API found at http://localhost:1582"):
-        OrcaCredentials.set_base_url("http://localhost:1582")
+        OrcaCredentials.set_api_url("http://localhost:1582")
 
 
 def test_is_healthy():
     assert OrcaCredentials.is_healthy()
 
 
-def test_is_healthy_false(base_url_reset):
-    OrcaCredentials.set_base_url("http://localhost:1582", check_validity=False)
+def test_is_healthy_false(api_url_reset):
+    OrcaCredentials.set_api_url("http://localhost:1582", check_validity=False)
     assert not OrcaCredentials.is_healthy()

orca_sdk/job.py

@@ -7,7 +7,7 @@ from typing import Callable, Generic, TypedDict, TypeVar, cast
 
 from tqdm.auto import tqdm
 
-from .client import TaskStatus, orca_api
+from .client import orca_api
 
 
 class JobConfig(TypedDict):
@@ -26,6 +26,9 @@ class Status(Enum):
     DISPATCHED = "DISPATCHED"
     """The job has been queued and is waiting to be processed"""
 
+    WAITING = "WAITING"
+    """The job is waiting for dependencies to complete"""
+
     PROCESSING = "PROCESSING"
     """The job is being processed"""
 

orca_sdk/memoryset.py

@@ -21,7 +21,9 @@ from .client import (
     FilterItem,
 )
 from .client import LabeledMemory as LabeledMemoryResponse
-from .client import LabeledMemoryInsert
+from .client import (
+    LabeledMemoryInsert,
+)
 from .client import LabeledMemoryLookup as LabeledMemoryLookupResponse
 from .client import (
     LabeledMemoryUpdate,
@@ -35,7 +37,9 @@ from .client import (
     MemoryType,
 )
 from .client import ScoredMemory as ScoredMemoryResponse
-from .client import ScoredMemoryInsert
+from .client import (
+    ScoredMemoryInsert,
+)
 from .client import ScoredMemoryLookup as ScoredMemoryLookupResponse
 from .client import (
     ScoredMemoryUpdate,
@@ -645,6 +649,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
     embedding_model: EmbeddingModelBase
     index_type: IndexType
     index_params: dict[str, Any]
+    hidden: bool
 
     def __init__(self, metadata: MemorysetMetadata):
         # for internal use only, do not document
@@ -665,6 +670,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
         self.index_type = metadata["index_type"]
         self.index_params = metadata["index_params"]
         self.memory_type = metadata["memory_type"]
+        self.hidden = metadata["hidden"]
 
     def __eq__(self, other) -> bool:
         return isinstance(other, MemorysetBase) and self.id == other.id
@@ -699,6 +705,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
         index_params: dict[str, Any] = {},
         if_exists: CreateMode = "error",
         background: Literal[True],
+        hidden: bool = False,
     ) -> Job[Self]:
         pass
 
@@ -723,6 +730,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
         index_params: dict[str, Any] = {},
         if_exists: CreateMode = "error",
         background: Literal[False] = False,
+        hidden: bool = False,
     ) -> Self:
         pass
 
@@ -746,6 +754,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
         index_params: dict[str, Any] = {},
         if_exists: CreateMode = "error",
         background: bool = False,
+        hidden: bool = False,
     ) -> Self | Job[Self]:
         """
         Create a new memoryset in the OrcaCloud
@@ -783,6 +792,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
             if_exists: What to do if a memoryset with the same name already exists, defaults to
                 `"error"`. Other option is `"open"` to open the existing memoryset.
             background: Whether to run the operation none blocking and return a job handle
+            hidden: Whether the memoryset should be hidden
 
         Returns:
             Handle to the new memoryset in the OrcaCloud
@@ -820,6 +830,7 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
             "remove_duplicates": remove_duplicates,
             "index_type": index_type,
             "index_params": index_params,
+            "hidden": hidden,
         }
         if prompt is not None:
             payload["prompt"] = prompt
@@ -1272,14 +1283,20 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
             return False
 
     @classmethod
-    def all(cls) -> list[Self]:
+    def all(cls, show_hidden: bool = False) -> list[Self]:
         """
         Get a list of handles to all memorysets in the OrcaCloud
 
+        Params:
+            show_hidden: Whether to include hidden memorysets in results, defaults to `False`
+
         Returns:
             List of handles to all memorysets in the OrcaCloud
         """
-        return [cls(metadata) for metadata in orca_api.GET("/memoryset", params={"type": cls.memory_type})]
+        return [
+            cls(metadata)
+            for metadata in orca_api.GET("/memoryset", params={"type": cls.memory_type, "show_hidden": show_hidden})
+        ]
 
     @classmethod
     def drop(cls, name_or_id: str, if_not_exists: DropMode = "error"):
@@ -1301,7 +1318,14 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
             if if_not_exists == "error":
                 raise
 
-    def set(self, *, name: str = UNSET, description: str | None = UNSET, label_names: list[str] = UNSET):
+    def set(
+        self,
+        *,
+        name: str = UNSET,
+        description: str | None = UNSET,
+        label_names: list[str] = UNSET,
+        hidden: bool = UNSET,
+    ):
         """
         Update editable attributes of the memoryset
 
@@ -1320,6 +1344,8 @@ class MemorysetBase(Generic[MemoryT, MemoryLookupT], ABC):
             payload["description"] = description
         if label_names is not UNSET:
             payload["label_names"] = label_names
+        if hidden is not UNSET:
+            payload["hidden"] = hidden
 
         orca_api.PATCH("/memoryset/{name_or_id}", params={"name_or_id": self.id}, json=payload)
         self.refresh()

orca_sdk/memoryset_test.py

@@ -122,6 +122,26 @@ def test_all_memorysets(readonly_memoryset: LabeledMemoryset):
     assert any(memoryset.name == readonly_memoryset.name for memoryset in memorysets)
 
 
+def test_all_memorysets_hidden(
+    readonly_memoryset: LabeledMemoryset,
+):
+    # Create a hidden memoryset
+    hidden_memoryset = LabeledMemoryset.clone(readonly_memoryset, "test_hidden_memoryset")
+    hidden_memoryset.set(hidden=True)
+
+    # Test that show_hidden=False excludes hidden memorysets
+    visible_memorysets = LabeledMemoryset.all(show_hidden=False)
+    assert len(visible_memorysets) > 0
+    assert readonly_memoryset in visible_memorysets
+    assert hidden_memoryset not in visible_memorysets
+
+    # Test that show_hidden=True includes hidden memorysets
+    all_memorysets = LabeledMemoryset.all(show_hidden=True)
+    assert len(all_memorysets) == len(visible_memorysets) + 1
+    assert readonly_memoryset in all_memorysets
+    assert hidden_memoryset in all_memorysets
+
+
 def test_all_memorysets_unauthenticated(unauthenticated):
     with pytest.raises(ValueError, match="Invalid API key"):
         LabeledMemoryset.all()
@@ -167,6 +187,9 @@ def test_update_memoryset_attributes(writable_memoryset: LabeledMemoryset):
     writable_memoryset.set(label_names=["New label 1", "New label 2"])
     assert writable_memoryset.label_names == ["New label 1", "New label 2"]
 
+    writable_memoryset.set(hidden=True)
+    assert writable_memoryset.hidden is True
+
 
 def test_search(readonly_memoryset: LabeledMemoryset):
     memory_lookups = readonly_memoryset.search(["i love soup", "cats are cute"])

{orca_sdk-0.0.102.dist-info → orca_sdk-0.0.103.dist-info}/METADATA

@@ -1,24 +1,25 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: orca_sdk
-Version: 0.0.102
+Version: 0.0.103
 Summary: SDK for interacting with Orca Services
 License: Apache-2.0
 Author: Orca DB Inc.
 Author-email: dev-rel@orcadb.ai
-Requires-Python: >=3.11,<4.0
+Requires-Python: >=3.11,<3.14
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 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: gradio (>=5.44.1,<6.0.0)
+Requires-Dist: httpx (>=0.28.1,<0.29.0)
+Requires-Dist: numpy (>=2.1.0,<3.0.0)
 Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pyarrow (>=18.0.0,<19.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: torch (>=2.8.0,<3.0.0)
 Description-Content-Type: text/markdown
 
 <!--

{orca_sdk-0.0.102.dist-info → orca_sdk-0.0.103.dist-info}/RECORD

@@ -5,7 +5,7 @@ orca_sdk/_shared/metrics_test.py,sha256=3wD1e2WE_wGtWBQ0E2gML6J3LCcbQBMezxvhDM94
 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=58Gaa4KA9iy7WP-hMZT4PI6x5lkpctFbbJ6Piwu1ZS4,2484
+orca_sdk/_utils/auth.py,sha256=Yf39jucuAkrM-TbTsLm_iRgLbLEcAIRzlHrCqgy8RiM,2483
 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
@@ -18,23 +18,23 @@ orca_sdk/_utils/tqdm_file_reader.py,sha256=Lw7Cg1UgNuRUoN6jjqZb-IlV00H-kbRcrZLdu
 orca_sdk/_utils/value_parser.py,sha256=c3qMABCCDQcIjn9N1orYYnlRwDW9JWdGwW_2TDZPLdI,1286
 orca_sdk/_utils/value_parser_test.py,sha256=OybsiC-Obi32RRi9NIuwrVBRAnlyPMV1xVAaevSrb7M,1079
 orca_sdk/classification_model.py,sha256=SUiUgv_o3UUngpz3Le_L6DsijhjXVEB3yo84hrD1MX4,31172
-orca_sdk/classification_model_test.py,sha256=WganVoP-0vw1cqiVWJ2vXyGi4lwYp_hbZHultpxvFqk,19536
-orca_sdk/client.py,sha256=Q2tjK-CeNnX0dgK-oZPzElgEK3wkxXC1q5YK85mZcqE,120338
-orca_sdk/conftest.py,sha256=LHA46gDU_D0T_ogS6XOVQvGDDMD01nVZFWVcBYDConc,8885
-orca_sdk/credentials.py,sha256=G_jdeC-1j6oL6VtHnRHL-LxvLXnL-KH2CDCB2iQ9POY,5084
-orca_sdk/credentials_test.py,sha256=whUweSJIEws6C8Bku-5V31Dv9hD1mFDDW7X2cCsB6g0,1629
+orca_sdk/classification_model_test.py,sha256=VlmHJ5CYzyGLDn-6hVqTkqoH7mBIgyzFiNOse0olYl4,19522
+orca_sdk/client.py,sha256=Iw0zHKe9kWvtSS-xv-J1LgDDWwNbrxCSg8LqtBcGmiE,122073
+orca_sdk/conftest.py,sha256=vTw-gkWpTutMU0GdNNwtqAEAEcyPrT5hCogghK4jtTY,8888
+orca_sdk/credentials.py,sha256=YuWipb5r1R_uPRe0nrm8mbbrfrXKFimgFOxny9bzAbI,5063
+orca_sdk/credentials_test.py,sha256=ZIIZtfv507UyIIgLhnECWn6KS5NjbfHMdT31f8k0zJM,1623
 orca_sdk/datasource.py,sha256=6wARRq-eNDJVSBABdVzn41z7s69xasTsqbozaVAsf9U,20263
 orca_sdk/datasource_test.py,sha256=wENPourrJvQN-uJJPaJI9EDuof6wVw3GirOhbrY4sFI,11564
 orca_sdk/embedding_model.py,sha256=FBuMDYKS8x6iyhSHtNO8TI2pCi6yYDycirKw3hqaGAc,26043
 orca_sdk/embedding_model_test.py,sha256=1aELyCuIzxSxUg7Z4tYtNOd8-hV5hFb-gsZTNh712OQ,7765
-orca_sdk/job.py,sha256=wWJPkkQbkNu_ylBtZN4AscU00VwWTfqlSmysRBUlivw,12787
+orca_sdk/job.py,sha256=yHmHgm5vf2DHJlvAGgA2o1mNbKs3yoJ5RMeUpPSwl7E,12858
 orca_sdk/job_test.py,sha256=r4zCyCuJ8eVN-O6_IUSakmSBn64rY7oBmlNtb5RNsi0,3919
-orca_sdk/memoryset.py,sha256=aeceFH6WpIt3hVXnpjLGr7cdj5lPy_PeT3ZxR8fdhKA,85585
-orca_sdk/memoryset_test.py,sha256=14WG6u_adVmm6CSn2dM5lEfQYxybwh3s9Q7RBeTuoPE,20486
+orca_sdk/memoryset.py,sha256=QIwW1kAWh_wtFHvjX2SvNMuLUBoFYK8tgsaaPXgtoQA,86188
+orca_sdk/memoryset_test.py,sha256=0Al--P2jDwhlxFpTgL0piDPCrKtxZNvaYjzPugw4yAk,21361
 orca_sdk/regression_model.py,sha256=je2g1BmoPCuouv5iWqDolja90F2w2vD6TooXA8KjL7c,24552
 orca_sdk/regression_model_test.py,sha256=8LDhtQeh52grZQ2Xd0ju1MQvb_hwosY_ORDDE3wS2LA,14570
 orca_sdk/telemetry.py,sha256=qTEPkOlqjxsPaS-HR5Jh5ZnIvuF58aIy5OpzA-wQkAE,25713
 orca_sdk/telemetry_test.py,sha256=eT66C5lFdNg-pQdo2I__BP7Tn5fTc9aTkVo9ZhWwhU0,5519
-orca_sdk-0.0.102.dist-info/METADATA,sha256=dfe9GFGNjzso6R73Y-uVqyRgvFxvJZRIOsCgVvPpnn8,3665
-orca_sdk-0.0.102.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-orca_sdk-0.0.102.dist-info/RECORD,,
+orca_sdk-0.0.103.dist-info/METADATA,sha256=eXig9FBe1OKfMNVm30rrJIUzhvGLw5AaHj1gMKupqbE,3711
+orca_sdk-0.0.103.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+orca_sdk-0.0.103.dist-info/RECORD,,

{orca_sdk-0.0.102.dist-info → orca_sdk-0.0.103.dist-info}/WHEEL

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