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

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 (

orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py

@@ -11,13 +11,14 @@ The main changes are:
 """
 
 from http import HTTPStatus
-from typing import Any, List
+from typing import Any, List, 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
 
 
@@ -32,19 +33,45 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List[List["LabeledMemory"]]:
+def _parse_response(*, response: httpx.Response) -> Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]:
     if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = []
-            _response_200_item = response_200_item_data
-            for response_200_item_item_data in _response_200_item:
-                response_200_item_item = LabeledMemory.from_dict(response_200_item_item_data)
 
-                response_200_item.append(response_200_item_item)
+        def _parse_response_200(data: object) -> Union[List[List["LabeledMemory"]], List[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 = []
+                    _response_200_type_0_item = response_200_type_0_item_data
+                    for response_200_type_0_item_item_data in _response_200_type_0_item:
+                        response_200_type_0_item_item = LabeledMemory.from_dict(response_200_type_0_item_item_data)
 
-            response_200.append(response_200_item)
+                        response_200_type_0_item.append(response_200_type_0_item_item)
+
+                    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 = []
+                _response_200_type_1_item = response_200_type_1_item_data
+                for response_200_type_1_item_item_data in _response_200_type_1_item:
+                    response_200_type_1_item_item = ScoredMemory.from_dict(response_200_type_1_item_item_data)
+
+                    response_200_type_1_item.append(response_200_type_1_item_item)
+
+                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:
@@ -59,7 +86,9 @@ def _parse_response(*, response: httpx.Response) -> List[List["LabeledMemory"]]:
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List[List["LabeledMemory"]]]:
+def _build_response(
+    *, response: httpx.Response
+) -> Response[Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -70,7 +99,7 @@ def _build_response(*, response: httpx.Response) -> Response[List[List["LabeledM
 
 def sync_detailed(
     name_or_id: str,
-) -> Response[List[List["LabeledMemory"]]]:
+) -> Response[Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]]:
     """Potential Duplicate Groups
 
     Args:
@@ -81,7 +110,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List[List['LabeledMemory']]]
+        Response[Union[List[List['LabeledMemory']], List[List['ScoredMemory']]]]
     """
 
     kwargs = _get_kwargs(
@@ -98,7 +127,7 @@ def sync_detailed(
 
 def sync(
     name_or_id: str,
-) -> List[List["LabeledMemory"]]:
+) -> Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]:
     """Potential Duplicate Groups
 
     Args:
@@ -109,7 +138,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List[List['LabeledMemory']]
+        Union[List[List['LabeledMemory']], List[List['ScoredMemory']]]
     """
 
     return sync_detailed(
@@ -119,7 +148,7 @@ def sync(
 
 async def asyncio_detailed(
     name_or_id: str,
-) -> Response[List[List["LabeledMemory"]]]:
+) -> Response[Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]]:
     """Potential Duplicate Groups
 
     Args:
@@ -130,7 +159,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List[List['LabeledMemory']]]
+        Response[Union[List[List['LabeledMemory']], List[List['ScoredMemory']]]]
     """
 
     kwargs = _get_kwargs(
@@ -145,7 +174,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     name_or_id: str,
-) -> List[List["LabeledMemory"]]:
+) -> Union[List[List["LabeledMemory"]], List[List["ScoredMemory"]]]:
     """Potential Duplicate Groups
 
     Args:
@@ -156,7 +185,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List[List['LabeledMemory']]
+        Union[List[List['LabeledMemory']], List[List['ScoredMemory']]]
     """
 
     return (

orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_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 import LabeledMemory
 from ...models.list_memories_request import ListMemoriesRequest
+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: ListMemoriesRequest,
-) -> Response[List["LabeledMemory"]]:
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Query Memoryset
 
     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: ListMemoriesRequest,
-) -> List["LabeledMemory"]:
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Query Memoryset
 
     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: ListMemoriesRequest,
-) -> Response[List["LabeledMemory"]]:
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Query Memoryset
 
     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: ListMemoriesRequest,
-) -> List["LabeledMemory"]:
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Query Memoryset
 
     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/update_memories_gpu_memoryset_name_or_id_memories_patch.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,13 +19,15 @@ from ...client import _client_context
 from ...errors import get_error_for_response
 from ...models.labeled_memory import LabeledMemory
 from ...models.labeled_memory_update import LabeledMemoryUpdate
+from ...models.scored_memory import ScoredMemory
+from ...models.scored_memory_update import ScoredMemoryUpdate
 from ...types import Response
 
 
 def _get_kwargs(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryUpdate"],
+    body: Union[List["LabeledMemoryUpdate"], List["ScoredMemoryUpdate"]],
 ) -> dict[str, Any]:
     headers: Dict[str, Any] = {}
 
@@ -34,10 +36,12 @@ def _get_kwargs(
         "url": f"/gpu/memoryset/{name_or_id}/memories",
     }
 
-    _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"
@@ -46,14 +50,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:
@@ -68,7 +93,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,
@@ -80,20 +105,20 @@ def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemory
 def sync_detailed(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryUpdate"],
-) -> Response[List["LabeledMemory"]]:
+    body: Union[List["LabeledMemoryUpdate"], List["ScoredMemoryUpdate"]],
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Update Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryUpdate']):
+        body (Union[List['LabeledMemoryUpdate'], List['ScoredMemoryUpdate']]):
 
     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['LabeledMemory']]
+        Response[Union[List['LabeledMemory'], List['ScoredMemory']]]
     """
 
     kwargs = _get_kwargs(
@@ -112,20 +137,20 @@ def sync_detailed(
 def sync(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryUpdate"],
-) -> List["LabeledMemory"]:
+    body: Union[List["LabeledMemoryUpdate"], List["ScoredMemoryUpdate"]],
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Update Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryUpdate']):
+        body (Union[List['LabeledMemoryUpdate'], List['ScoredMemoryUpdate']]):
 
     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['LabeledMemory']
+        Union[List['LabeledMemory'], List['ScoredMemory']]
     """
 
     return sync_detailed(
@@ -137,20 +162,20 @@ def sync(
 async def asyncio_detailed(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryUpdate"],
-) -> Response[List["LabeledMemory"]]:
+    body: Union[List["LabeledMemoryUpdate"], List["ScoredMemoryUpdate"]],
+) -> Response[Union[List["LabeledMemory"], List["ScoredMemory"]]]:
     """Update Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryUpdate']):
+        body (Union[List['LabeledMemoryUpdate'], List['ScoredMemoryUpdate']]):
 
     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['LabeledMemory']]
+        Response[Union[List['LabeledMemory'], List['ScoredMemory']]]
     """
 
     kwargs = _get_kwargs(
@@ -167,20 +192,20 @@ async def asyncio_detailed(
 async def asyncio(
     name_or_id: str,
     *,
-    body: List["LabeledMemoryUpdate"],
-) -> List["LabeledMemory"]:
+    body: Union[List["LabeledMemoryUpdate"], List["ScoredMemoryUpdate"]],
+) -> Union[List["LabeledMemory"], List["ScoredMemory"]]:
     """Update Memories
 
     Args:
         name_or_id (str):
-        body (List['LabeledMemoryUpdate']):
+        body (Union[List['LabeledMemoryUpdate'], List['ScoredMemoryUpdate']]):
 
     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['LabeledMemory']
+        Union[List['LabeledMemory'], List['ScoredMemory']]
     """
 
     return (