orca-sdk 0.0.93__py3-none-any.whl → 0.0.95__py3-none-any.whl

orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_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.get_memories_request import GetMemoriesRequest
 from ...models.labeled_memory import LabeledMemory
+from ...models.scored_memory import ScoredMemory
 from ...types import Response
 
 
@@ -43,14 +44,35 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List["LabeledMemory"]:
+def _parse_response(*, response: httpx.Response) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = LabeledMemory.from_dict(response_200_item_data)
 
-            response_200.append(response_200_item)
+        def _parse_response_200(data: object) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                response_200_type_0 = []
+                _response_200_type_0 = data
+                for response_200_type_0_item_data in _response_200_type_0:
+                    response_200_type_0_item = LabeledMemory.from_dict(response_200_type_0_item_data)
+
+                    response_200_type_0.append(response_200_type_0_item)
+
+                return response_200_type_0
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, list):
+                raise TypeError()
+            response_200_type_1 = []
+            _response_200_type_1 = data
+            for response_200_type_1_item_data in _response_200_type_1:
+                response_200_type_1_item = ScoredMemory.from_dict(response_200_type_1_item_data)
+
+                response_200_type_1.append(response_200_type_1_item)
+
+            return response_200_type_1
+
+        response_200 = _parse_response_200(response.json())
 
         return response_200
     if response.status_code == 401:
@@ -65,7 +87,7 @@ def _parse_response(*, response: httpx.Response) -> List["LabeledMemory"]:
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemory"]]:
+def _build_response(*, response: httpx.Response) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -78,7 +100,7 @@ def sync_detailed(
     name_or_id: str,
     *,
     body: GetMemoriesRequest,
-) -> Response[List["LabeledMemory"]]:
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Get Memories
 
     Args:
@@ -90,7 +112,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['LabeledMemory']]
+        Response[Union[List['LabeledMemory'], List['ScoredMemory']]]
     """
 
     kwargs = _get_kwargs(
@@ -110,7 +132,7 @@ def sync(
     name_or_id: str,
     *,
     body: GetMemoriesRequest,
-) -> List["LabeledMemory"]:
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Get Memories
 
     Args:
@@ -122,7 +144,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['LabeledMemory']
+        Union[List['LabeledMemory'], List['ScoredMemory']]
     """
 
     return sync_detailed(
@@ -135,7 +157,7 @@ async def asyncio_detailed(
     name_or_id: str,
     *,
     body: GetMemoriesRequest,
-) -> Response[List["LabeledMemory"]]:
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Get Memories
 
     Args:
@@ -147,7 +169,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['LabeledMemory']]
+        Response[Union[List['LabeledMemory'], List['ScoredMemory']]]
     """
 
     kwargs = _get_kwargs(
@@ -165,7 +187,7 @@ async def asyncio(
     name_or_id: str,
     *,
     body: GetMemoriesRequest,
-) -> List["LabeledMemory"]:
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Get Memories
 
     Args:
@@ -177,7 +199,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['LabeledMemory']
+        Union[List['LabeledMemory'], List['ScoredMemory']]
     """
 
     return (

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