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

orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py

@@ -11,13 +11,14 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.labeled_memory import LabeledMemory
+from ...models.scored_memory import ScoredMemory
 from ...types import Response
 
 
@@ -33,9 +34,25 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> LabeledMemory:
+def _parse_response(*, response: httpx.Response) -> Union["LabeledMemory", "ScoredMemory"]:
     if response.status_code == 200:
-        response_200 = LabeledMemory.from_dict(response.json())
+
+        def _parse_response_200(data: object) -> Union["LabeledMemory", "ScoredMemory"]:
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_200_type_0 = LabeledMemory.from_dict(data)
+
+                return response_200_type_0
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            response_200_type_1 = ScoredMemory.from_dict(data)
+
+            return response_200_type_1
+
+        response_200 = _parse_response_200(response.json())
 
         return response_200
     if response.status_code == 401:
@@ -50,7 +67,7 @@ def _parse_response(*, response: httpx.Response) -> LabeledMemory:
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[LabeledMemory]:
+def _build_response(*, response: httpx.Response) -> Response[Union["LabeledMemory", "ScoredMemory"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -62,7 +79,7 @@ def _build_response(*, response: httpx.Response) -> Response[LabeledMemory]:
 def sync_detailed(
     name_or_id: str,
     memory_id: str,
-) -> Response[LabeledMemory]:
+) -> Response[Union["LabeledMemory", "ScoredMemory"]]:
     """Get Memory
 
     Args:
@@ -74,7 +91,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabeledMemory]
+        Response[Union['LabeledMemory', 'ScoredMemory']]
     """
 
     kwargs = _get_kwargs(
@@ -93,7 +110,7 @@ def sync_detailed(
 def sync(
     name_or_id: str,
     memory_id: str,
-) -> LabeledMemory:
+) -> Union["LabeledMemory", "ScoredMemory"]:
     """Get Memory
 
     Args:
@@ -105,7 +122,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabeledMemory
+        Union['LabeledMemory', 'ScoredMemory']
     """
 
     return sync_detailed(
@@ -117,7 +134,7 @@ def sync(
 async def asyncio_detailed(
     name_or_id: str,
     memory_id: str,
-) -> Response[LabeledMemory]:
+) -> Response[Union["LabeledMemory", "ScoredMemory"]]:
     """Get Memory
 
     Args:
@@ -129,7 +146,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabeledMemory]
+        Response[Union['LabeledMemory', 'ScoredMemory']]
     """
 
     kwargs = _get_kwargs(
@@ -146,7 +163,7 @@ async def asyncio_detailed(
 async def asyncio(
     name_or_id: str,
     memory_id: str,
-) -> LabeledMemory:
+) -> Union["LabeledMemory", "ScoredMemory"]:
     """Get Memory
 
     Args:
@@ -158,7 +175,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabeledMemory
+        Union['LabeledMemory', 'ScoredMemory']
     """
 
     return (

orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py

@@ -17,7 +17,7 @@ import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.labeled_memoryset_metadata import LabeledMemorysetMetadata
+from ...models.memoryset_metadata import MemorysetMetadata
 from ...types import Response
 
 
@@ -32,9 +32,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
+def _parse_response(*, response: httpx.Response) -> MemorysetMetadata:
     if response.status_code == 200:
-        response_200 = LabeledMemorysetMetadata.from_dict(response.json())
+        response_200 = MemorysetMetadata.from_dict(response.json())
 
         return response_200
     if response.status_code == 401:
@@ -49,7 +49,7 @@ def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMetadata]:
+def _build_response(*, response: httpx.Response) -> Response[MemorysetMetadata]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -60,7 +60,7 @@ def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMet
 
 def sync_detailed(
     name_or_id: str,
-) -> Response[LabeledMemorysetMetadata]:
+) -> Response[MemorysetMetadata]:
     """Get Memoryset
 
     Args:
@@ -71,7 +71,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabeledMemorysetMetadata]
+        Response[MemorysetMetadata]
     """
 
     kwargs = _get_kwargs(
@@ -88,7 +88,7 @@ def sync_detailed(
 
 def sync(
     name_or_id: str,
-) -> LabeledMemorysetMetadata:
+) -> MemorysetMetadata:
     """Get Memoryset
 
     Args:
@@ -99,7 +99,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabeledMemorysetMetadata
+        MemorysetMetadata
     """
 
     return sync_detailed(
@@ -109,7 +109,7 @@ def sync(
 
 async def asyncio_detailed(
     name_or_id: str,
-) -> Response[LabeledMemorysetMetadata]:
+) -> Response[MemorysetMetadata]:
     """Get Memoryset
 
     Args:
@@ -120,7 +120,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[LabeledMemorysetMetadata]
+        Response[MemorysetMetadata]
     """
 
     kwargs = _get_kwargs(
@@ -135,7 +135,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     name_or_id: str,
-) -> LabeledMemorysetMetadata:
+) -> MemorysetMetadata:
     """Get Memoryset
 
     Args:
@@ -146,7 +146,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        LabeledMemorysetMetadata
+        MemorysetMetadata
     """
 
     return (

orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py

@@ -11,20 +11,21 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any, Dict, List, cast
+from typing import Any, Dict, List, Union, cast
 
 import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.labeled_memory_insert import LabeledMemoryInsert
+from ...models.scored_memory_insert import ScoredMemoryInsert
 from ...types import Response
 
 
 def _get_kwargs(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryInsert"],
+    body: Union[List["LabeledMemoryInsert"], List["ScoredMemoryInsert"]],
 ) -> dict[str, Any]:
     headers: Dict[str, Any] = {}
 
@@ -33,10 +34,12 @@ def _get_kwargs(
         "url": f"/gpu/memoryset/{name_or_id}/memory",
     }
 
-    _body = []
-    for body_item_data in body:
-        body_item = body_item_data.to_dict()
-        _body.append(body_item)
+    _body: List[Dict[str, Any]]
+    if isinstance(body, list):
+        _body = []
+        for body_type_0_item_data in body:
+            body_type_0_item = body_type_0_item_data.to_dict()
+            _body.append(body_type_0_item)
 
     _kwargs["json"] = _body
     headers["Content-Type"] = "application/json"
@@ -74,13 +77,13 @@ def _build_response(*, response: httpx.Response) -> Response[List[str]]:
 def sync_detailed(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryInsert"],
+    body: Union[List["LabeledMemoryInsert"], List["ScoredMemoryInsert"]],
 ) -> Response[List[str]]:
     """Insert Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryInsert']):
+        body (Union[List['LabeledMemoryInsert'], List['ScoredMemoryInsert']]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -106,13 +109,13 @@ def sync_detailed(
 def sync(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryInsert"],
+    body: Union[List["LabeledMemoryInsert"], List["ScoredMemoryInsert"]],
 ) -> List[str]:
     """Insert Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryInsert']):
+        body (Union[List['LabeledMemoryInsert'], List['ScoredMemoryInsert']]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -131,13 +134,13 @@ def sync(
 async def asyncio_detailed(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryInsert"],
+    body: Union[List["LabeledMemoryInsert"], List["ScoredMemoryInsert"]],
 ) -> Response[List[str]]:
     """Insert Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryInsert']):
+        body (Union[List['LabeledMemoryInsert'], List['ScoredMemoryInsert']]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -161,13 +164,13 @@ async def asyncio_detailed(
 async def asyncio(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryInsert"],
+    body: Union[List["LabeledMemoryInsert"], List["ScoredMemoryInsert"]],
 ) -> List[str]:
     """Insert Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryInsert']):
+        body (Union[List['LabeledMemoryInsert'], List['ScoredMemoryInsert']]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.

orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py

@@ -10,32 +10,51 @@ The main changes are:
 - Inject a client into every method via a context manager
 """
 
+from enum import Enum
 from http import HTTPStatus
-from typing import Any, List
+from typing import Any, Dict, List, Union
 
 import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.labeled_memoryset_metadata import LabeledMemorysetMetadata
-from ...types import Response
+from ...models.memory_type import MemoryType
+from ...models.memoryset_metadata import MemorysetMetadata
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    type: Union[MemoryType, None, Unset] = UNSET,
+) -> dict[str, Any]:
+    params: Dict[str, Any] = {}
+
+    json_type: Union[None, Unset, str]
+    if isinstance(type, Unset):
+        json_type = UNSET
+    elif isinstance(type, MemoryType):
+        json_type = type.value if isinstance(type, Enum) else type
+    else:
+        json_type = type
+    params["type"] = json_type
 
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 
-def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
         "url": "/memoryset/",
+        "params": params,
     }
 
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List["LabeledMemorysetMetadata"]:
+def _parse_response(*, response: httpx.Response) -> List["MemorysetMetadata"]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = LabeledMemorysetMetadata.from_dict(response_200_item_data)
+            response_200_item = MemorysetMetadata.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)
 
@@ -50,7 +69,7 @@ def _parse_response(*, response: httpx.Response) -> List["LabeledMemorysetMetada
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemorysetMetadata"]]:
+def _build_response(*, response: httpx.Response) -> Response[List["MemorysetMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -59,18 +78,26 @@ def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemory
     )
 
 
-def sync_detailed() -> Response[List["LabeledMemorysetMetadata"]]:
+def sync_detailed(
+    *,
+    type: Union[MemoryType, None, Unset] = UNSET,
+) -> Response[List["MemorysetMetadata"]]:
     """List Memorysets
 
+    Args:
+        type (Union[MemoryType, None, Unset]):
+
     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[List['LabeledMemorysetMetadata']]
+        Response[List['MemorysetMetadata']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        type=type,
+    )
 
     with _client_context() as client:
         response = client.get_httpx_client().request(
@@ -80,32 +107,48 @@ def sync_detailed() -> Response[List["LabeledMemorysetMetadata"]]:
         return _build_response(response=response)
 
 
-def sync() -> List["LabeledMemorysetMetadata"]:
+def sync(
+    *,
+    type: Union[MemoryType, None, Unset] = UNSET,
+) -> List["MemorysetMetadata"]:
     """List Memorysets
 
+    Args:
+        type (Union[MemoryType, None, Unset]):
+
     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:
-        List['LabeledMemorysetMetadata']
+        List['MemorysetMetadata']
     """
 
-    return sync_detailed().parsed
+    return sync_detailed(
+        type=type,
+    ).parsed
 
 
-async def asyncio_detailed() -> Response[List["LabeledMemorysetMetadata"]]:
+async def asyncio_detailed(
+    *,
+    type: Union[MemoryType, None, Unset] = UNSET,
+) -> Response[List["MemorysetMetadata"]]:
     """List Memorysets
 
+    Args:
+        type (Union[MemoryType, None, Unset]):
+
     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[List['LabeledMemorysetMetadata']]
+        Response[List['MemorysetMetadata']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        type=type,
+    )
 
     with _client_context() as client:
         response = await client.get_async_httpx_client().request(**kwargs)
@@ -113,15 +156,25 @@ async def asyncio_detailed() -> Response[List["LabeledMemorysetMetadata"]]:
     return _build_response(response=response)
 
 
-async def asyncio() -> List["LabeledMemorysetMetadata"]:
+async def asyncio(
+    *,
+    type: Union[MemoryType, None, Unset] = UNSET,
+) -> List["MemorysetMetadata"]:
     """List Memorysets
 
+    Args:
+        type (Union[MemoryType, None, Unset]):
+
     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:
-        List['LabeledMemorysetMetadata']
+        List['MemorysetMetadata']
     """
 
-    return (await asyncio_detailed()).parsed
+    return (
+        await asyncio_detailed(
+            type=type,
+        )
+    ).parsed

orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py

@@ -11,7 +11,7 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Union
 
 import httpx
 
@@ -19,6 +19,7 @@ from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.labeled_memory_lookup import LabeledMemoryLookup
 from ...models.lookup_request import LookupRequest
+from ...models.scored_memory_lookup import ScoredMemoryLookup
 from ...types import Response
 
 
@@ -43,7 +44,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List[List["LabeledMemoryLookup"]]:
+def _parse_response(*, response: httpx.Response) -> List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -51,7 +52,23 @@ def _parse_response(*, response: httpx.Response) -> List[List["LabeledMemoryLook
             response_200_item = []
             _response_200_item = response_200_item_data
             for response_200_item_item_data in _response_200_item:
-                response_200_item_item = LabeledMemoryLookup.from_dict(response_200_item_item_data)
+
+                def _parse_response_200_item_item(data: object) -> Union["LabeledMemoryLookup", "ScoredMemoryLookup"]:
+                    try:
+                        if not isinstance(data, dict):
+                            raise TypeError()
+                        response_200_item_item_type_0 = LabeledMemoryLookup.from_dict(data)
+
+                        return response_200_item_item_type_0
+                    except:  # noqa: E722
+                        pass
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    response_200_item_item_type_1 = ScoredMemoryLookup.from_dict(data)
+
+                    return response_200_item_item_type_1
+
+                response_200_item_item = _parse_response_200_item_item(response_200_item_item_data)
 
                 response_200_item.append(response_200_item_item)
 
@@ -70,7 +87,9 @@ def _parse_response(*, response: httpx.Response) -> List[List["LabeledMemoryLook
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List[List["LabeledMemoryLookup"]]]:
+def _build_response(
+    *, response: httpx.Response
+) -> Response[List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -83,7 +102,7 @@ def sync_detailed(
     name_or_id: str,
     *,
     body: LookupRequest,
-) -> Response[List[List["LabeledMemoryLookup"]]]:
+) -> Response[List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]]:
     """Memoryset Lookup
 
     Args:
@@ -95,7 +114,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List[List['LabeledMemoryLookup']]]
+        Response[List[List[Union['LabeledMemoryLookup', 'ScoredMemoryLookup']]]]
     """
 
     kwargs = _get_kwargs(
@@ -115,7 +134,7 @@ def sync(
     name_or_id: str,
     *,
     body: LookupRequest,
-) -> List[List["LabeledMemoryLookup"]]:
+) -> List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]:
     """Memoryset Lookup
 
     Args:
@@ -127,7 +146,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List[List['LabeledMemoryLookup']]
+        List[List[Union['LabeledMemoryLookup', 'ScoredMemoryLookup']]]
     """
 
     return sync_detailed(
@@ -140,7 +159,7 @@ async def asyncio_detailed(
     name_or_id: str,
     *,
     body: LookupRequest,
-) -> Response[List[List["LabeledMemoryLookup"]]]:
+) -> Response[List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]]:
     """Memoryset Lookup
 
     Args:
@@ -152,7 +171,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List[List['LabeledMemoryLookup']]]
+        Response[List[List[Union['LabeledMemoryLookup', 'ScoredMemoryLookup']]]]
     """
 
     kwargs = _get_kwargs(
@@ -170,7 +189,7 @@ async def asyncio(
     name_or_id: str,
     *,
     body: LookupRequest,
-) -> List[List["LabeledMemoryLookup"]]:
+) -> List[List[Union["LabeledMemoryLookup", "ScoredMemoryLookup"]]]:
     """Memoryset Lookup
 
     Args:
@@ -182,7 +201,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List[List['LabeledMemoryLookup']]
+        List[List[Union['LabeledMemoryLookup', 'ScoredMemoryLookup']]]
     """
 
     return (