arbi 0.2.1__tar.gz → 0.3.1__tar.gz

arbi-0.3.1/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.3.1"
+}

{arbi-0.2.1 → arbi-0.3.1}/CHANGELOG.md

@@ -1,5 +1,27 @@
 # Changelog
 
+## 0.3.1 (2025-10-31)
+
+Full Changelog: [v0.3.0...v0.3.1](https://github.com/arbitrationcity/arbi-python/compare/v0.3.0...v0.3.1)
+
+### Chores
+
+* **internal/tests:** avoid race condition with implicit client cleanup ([8bd34d5](https://github.com/arbitrationcity/arbi-python/commit/8bd34d51fcfa6975cf8034617b13dd3024a13f1f))
+
+## 0.3.0 (2025-10-30)
+
+Full Changelog: [v0.2.1...v0.3.0](https://github.com/arbitrationcity/arbi-python/compare/v0.2.1...v0.3.0)
+
+### Features
+
+* Add workspace document copy endpoint ([1c9fd95](https://github.com/arbitrationcity/arbi-python/commit/1c9fd95ebd51302eb2d39a526d1cfdf8d29c3feb))
+* DRAFT: Client-side key generation with JWE encryption ([8ab2fa9](https://github.com/arbitrationcity/arbi-python/commit/8ab2fa931309cec57317aa6894bce47025da447f))
+
+
+### Bug Fixes
+
+* **client:** close streams without requiring full consumption ([9b32ba6](https://github.com/arbitrationcity/arbi-python/commit/9b32ba6c3335273feaed012dbf0d8d7b147e5342))
+
 ## 0.2.1 (2025-10-18)
 
 Full Changelog: [v0.2.0...v0.2.1](https://github.com/arbitrationcity/arbi-python/compare/v0.2.0...v0.2.1)

{arbi-0.2.1 → arbi-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arbi
-Version: 0.2.1
+Version: 0.3.1
 Summary: The official Python library for the arbi API
 Project-URL: Homepage, https://github.com/arbitrationcity/arbi-python
 Project-URL: Repository, https://github.com/arbitrationcity/arbi-python

{arbi-0.2.1 → arbi-0.3.1}/api.md

@@ -16,9 +16,11 @@ Types:
 
 ```python
 from arbi.types.api import (
-    Token,
-    UserResponse,
+    UserChangePasswordResponse,
+    UserCheckSSOStatusResponse,
+    UserInviteResponse,
     UserListWorkspacesResponse,
+    UserLoginResponse,
     UserLogoutResponse,
     UserVerifyEmailResponse,
 )
@@ -26,12 +28,13 @@ from arbi.types.api import (
 
 Methods:
 
+- <code title="post /api/user/change_password">client.api.user.<a href="./src/arbi/resources/api/user/user.py">change_password</a>(\*\*<a href="src/arbi/types/api/user_change_password_params.py">params</a>) -> <a href="./src/arbi/types/api/user_change_password_response.py">UserChangePasswordResponse</a></code>
+- <code title="post /api/user/sso-status">client.api.user.<a href="./src/arbi/resources/api/user/user.py">check_sso_status</a>(\*\*<a href="src/arbi/types/api/user_check_sso_status_params.py">params</a>) -> <a href="./src/arbi/types/api/user_check_sso_status_response.py">UserCheckSSOStatusResponse</a></code>
+- <code title="post /api/user/invite">client.api.user.<a href="./src/arbi/resources/api/user/user.py">invite</a>(\*\*<a href="src/arbi/types/api/user_invite_params.py">params</a>) -> <a href="./src/arbi/types/api/user_invite_response.py">UserInviteResponse</a></code>
 - <code title="get /api/user/workspaces">client.api.user.<a href="./src/arbi/resources/api/user/user.py">list_workspaces</a>() -> <a href="./src/arbi/types/api/user_list_workspaces_response.py">UserListWorkspacesResponse</a></code>
-- <code title="post /api/user/login">client.api.user.<a href="./src/arbi/resources/api/user/user.py">login</a>(\*\*<a href="src/arbi/types/api/user_login_params.py">params</a>) -> <a href="./src/arbi/types/api/token.py">Token</a></code>
+- <code title="post /api/user/login">client.api.user.<a href="./src/arbi/resources/api/user/user.py">login</a>(\*\*<a href="src/arbi/types/api/user_login_params.py">params</a>) -> <a href="./src/arbi/types/api/user_login_response.py">UserLoginResponse</a></code>
 - <code title="post /api/user/logout">client.api.user.<a href="./src/arbi/resources/api/user/user.py">logout</a>() -> <a href="./src/arbi/types/api/user_logout_response.py">UserLogoutResponse</a></code>
-- <code title="post /api/user/token_refresh">client.api.user.<a href="./src/arbi/resources/api/user/user.py">refresh_token</a>() -> <a href="./src/arbi/types/api/token.py">Token</a></code>
-- <code title="post /api/user/register">client.api.user.<a href="./src/arbi/resources/api/user/user.py">register</a>(\*\*<a href="src/arbi/types/api/user_register_params.py">params</a>) -> <a href="./src/arbi/types/api/user_response.py">UserResponse</a></code>
-- <code title="get /api/user/me">client.api.user.<a href="./src/arbi/resources/api/user/user.py">retrieve_me</a>() -> <a href="./src/arbi/types/api/user_response.py">UserResponse</a></code>
+- <code title="post /api/user/register">client.api.user.<a href="./src/arbi/resources/api/user/user.py">register</a>(\*\*<a href="src/arbi/types/api/user_register_params.py">params</a>) -> object</code>
 - <code title="post /api/user/verify-email">client.api.user.<a href="./src/arbi/resources/api/user/user.py">verify_email</a>(\*\*<a href="src/arbi/types/api/user_verify_email_params.py">params</a>) -> <a href="./src/arbi/types/api/user_verify_email_response.py">UserVerifyEmailResponse</a></code>
 
 ### Settings
@@ -47,20 +50,6 @@ Methods:
 - <code title="get /api/user/settings">client.api.user.settings.<a href="./src/arbi/resources/api/user/settings.py">retrieve</a>() -> <a href="./src/arbi/types/api/user/setting_retrieve_response.py">SettingRetrieveResponse</a></code>
 - <code title="patch /api/user/settings">client.api.user.settings.<a href="./src/arbi/resources/api/user/settings.py">update</a>(\*\*<a href="src/arbi/types/api/user/setting_update_params.py">params</a>) -> None</code>
 
-## SSO
-
-Types:
-
-```python
-from arbi.types.api import SSOInviteResponse, SSOLoginResponse, SSORotatePasscodeResponse
-```
-
-Methods:
-
-- <code title="post /api/sso/invite">client.api.sso.<a href="./src/arbi/resources/api/sso.py">invite</a>(\*\*<a href="src/arbi/types/api/sso_invite_params.py">params</a>) -> <a href="./src/arbi/types/api/sso_invite_response.py">SSOInviteResponse</a></code>
-- <code title="post /api/sso/login">client.api.sso.<a href="./src/arbi/resources/api/sso.py">login</a>(\*\*<a href="src/arbi/types/api/sso_login_params.py">params</a>) -> <a href="./src/arbi/types/api/sso_login_response.py">SSOLoginResponse</a></code>
-- <code title="post /api/sso/rotate_passcode">client.api.sso.<a href="./src/arbi/resources/api/sso.py">rotate_passcode</a>() -> <a href="./src/arbi/types/api/sso_rotate_passcode_response.py">SSORotatePasscodeResponse</a></code>
-
 ## Workspace
 
 Types:
@@ -69,6 +58,7 @@ Types:
 from arbi.types.api import (
     WorkspaceResponse,
     WorkspaceDeleteResponse,
+    WorkspaceCopyResponse,
     WorkspaceGetConversationsResponse,
     WorkspaceGetDoctagsResponse,
     WorkspaceGetDocumentsResponse,
@@ -84,6 +74,7 @@ Methods:
 
 - <code title="patch /api/workspace/{workspace_ext_id}">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">update</a>(workspace_ext_id, \*\*<a href="src/arbi/types/api/workspace_update_params.py">params</a>) -> <a href="./src/arbi/types/api/workspace_response.py">WorkspaceResponse</a></code>
 - <code title="delete /api/workspace/{workspace_ext_id}">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">delete</a>(workspace_ext_id) -> <a href="./src/arbi/types/api/workspace_delete_response.py">WorkspaceDeleteResponse</a></code>
+- <code title="post /api/workspace/{workspace_ext_id}/copy">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">copy</a>(workspace_ext_id, \*\*<a href="src/arbi/types/api/workspace_copy_params.py">params</a>) -> <a href="./src/arbi/types/api/workspace_copy_response.py">WorkspaceCopyResponse</a></code>
 - <code title="post /api/workspace/create_protected">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">create_protected</a>(\*\*<a href="src/arbi/types/api/workspace_create_protected_params.py">params</a>) -> <a href="./src/arbi/types/api/workspace_response.py">WorkspaceResponse</a></code>
 - <code title="get /api/workspace/{workspace_ext_id}/conversations">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">get_conversations</a>(workspace_ext_id) -> <a href="./src/arbi/types/api/workspace_get_conversations_response.py">WorkspaceGetConversationsResponse</a></code>
 - <code title="get /api/workspace/{workspace_ext_id}/doctags">client.api.workspace.<a href="./src/arbi/resources/api/workspace.py">get_doctags</a>(workspace_ext_id) -> <a href="./src/arbi/types/api/workspace_get_doctags_response.py">WorkspaceGetDoctagsResponse</a></code>
@@ -142,6 +133,7 @@ Types:
 from arbi.types.api import (
     ConversationDeleteResponse,
     ConversationDeleteMessageResponse,
+    ConversationRetrieveMessageResponse,
     ConversationRetrieveThreadsResponse,
     ConversationShareResponse,
     ConversationUpdateTitleResponse,
@@ -152,6 +144,7 @@ Methods:
 
 - <code title="delete /api/conversation/{conversation_ext_id}">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">delete</a>(conversation_ext_id) -> <a href="./src/arbi/types/api/conversation_delete_response.py">ConversationDeleteResponse</a></code>
 - <code title="delete /api/conversation/message/{message_ext_id}">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">delete_message</a>(message_ext_id) -> <a href="./src/arbi/types/api/conversation_delete_message_response.py">ConversationDeleteMessageResponse</a></code>
+- <code title="get /api/conversation/message/{message_ext_id}">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">retrieve_message</a>(message_ext_id) -> <a href="./src/arbi/types/api/conversation_retrieve_message_response.py">ConversationRetrieveMessageResponse</a></code>
 - <code title="get /api/conversation/{conversation_ext_id}/threads">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">retrieve_threads</a>(conversation_ext_id) -> <a href="./src/arbi/types/api/conversation_retrieve_threads_response.py">ConversationRetrieveThreadsResponse</a></code>
 - <code title="post /api/conversation/{conversation_ext_id}/share">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">share</a>(conversation_ext_id) -> <a href="./src/arbi/types/api/conversation_share_response.py">ConversationShareResponse</a></code>
 - <code title="patch /api/conversation/{conversation_ext_id}/title">client.api.conversation.<a href="./src/arbi/resources/api/conversation/conversation.py">update_title</a>(conversation_ext_id, \*\*<a href="src/arbi/types/api/conversation_update_title_params.py">params</a>) -> <a href="./src/arbi/types/api/conversation_update_title_response.py">ConversationUpdateTitleResponse</a></code>

{arbi-0.2.1 → arbi-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arbi"
-version = "0.2.1"
+version = "0.3.1"
 description = "The official Python library for the arbi API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/_streaming.py

@@ -57,9 +57,8 @@ class Stream(Generic[_T]):
         for sse in iterator:
             yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -121,9 +120,8 @@ class AsyncStream(Generic[_T]):
         async for sse in iterator:
             yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        async for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "arbi"
-__version__ = "0.2.1"  # x-release-please-version
+__version__ = "0.3.1"  # x-release-please-version

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/resources/api/__init__.py

@@ -8,14 +8,6 @@ from .api import (
     APIResourceWithStreamingResponse,
     AsyncAPIResourceWithStreamingResponse,
 )
-from .sso import (
-    SSOResource,
-    AsyncSSOResource,
-    SSOResourceWithRawResponse,
-    AsyncSSOResourceWithRawResponse,
-    SSOResourceWithStreamingResponse,
-    AsyncSSOResourceWithStreamingResponse,
-)
 from .tag import (
     TagResource,
     AsyncTagResource,
@@ -88,12 +80,6 @@ __all__ = [
     "AsyncUserResourceWithRawResponse",
     "UserResourceWithStreamingResponse",
     "AsyncUserResourceWithStreamingResponse",
-    "SSOResource",
-    "AsyncSSOResource",
-    "SSOResourceWithRawResponse",
-    "AsyncSSOResourceWithRawResponse",
-    "SSOResourceWithStreamingResponse",
-    "AsyncSSOResourceWithStreamingResponse",
     "WorkspaceResource",
     "AsyncWorkspaceResource",
     "WorkspaceResourceWithRawResponse",

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/resources/api/api.py

@@ -5,14 +5,6 @@ from __future__ import annotations
 import httpx
 
 from ... import _resource
-from .sso import (
-    SSOResource,
-    AsyncSSOResource,
-    SSOResourceWithRawResponse,
-    AsyncSSOResourceWithRawResponse,
-    SSOResourceWithStreamingResponse,
-    AsyncSSOResourceWithStreamingResponse,
-)
 from .tag import (
     TagResource,
     AsyncTagResource,
@@ -95,10 +87,6 @@ class APIResource(_resource.SyncAPIResource):
     def user(self) -> UserResource:
         return UserResource(self._client)
 
-    @cached_property
-    def sso(self) -> SSOResource:
-        return SSOResource(self._client)
-
     @cached_property
     def workspace(self) -> WorkspaceResource:
         return WorkspaceResource(self._client)
@@ -171,10 +159,6 @@ class AsyncAPIResource(_resource.AsyncAPIResource):
     def user(self) -> AsyncUserResource:
         return AsyncUserResource(self._client)
 
-    @cached_property
-    def sso(self) -> AsyncSSOResource:
-        return AsyncSSOResource(self._client)
-
     @cached_property
     def workspace(self) -> AsyncWorkspaceResource:
         return AsyncWorkspaceResource(self._client)
@@ -254,10 +238,6 @@ class APIResourceWithRawResponse:
     def user(self) -> UserResourceWithRawResponse:
         return UserResourceWithRawResponse(self._api.user)
 
-    @cached_property
-    def sso(self) -> SSOResourceWithRawResponse:
-        return SSOResourceWithRawResponse(self._api.sso)
-
     @cached_property
     def workspace(self) -> WorkspaceResourceWithRawResponse:
         return WorkspaceResourceWithRawResponse(self._api.workspace)
@@ -299,10 +279,6 @@ class AsyncAPIResourceWithRawResponse:
     def user(self) -> AsyncUserResourceWithRawResponse:
         return AsyncUserResourceWithRawResponse(self._api.user)
 
-    @cached_property
-    def sso(self) -> AsyncSSOResourceWithRawResponse:
-        return AsyncSSOResourceWithRawResponse(self._api.sso)
-
     @cached_property
     def workspace(self) -> AsyncWorkspaceResourceWithRawResponse:
         return AsyncWorkspaceResourceWithRawResponse(self._api.workspace)
@@ -344,10 +320,6 @@ class APIResourceWithStreamingResponse:
     def user(self) -> UserResourceWithStreamingResponse:
         return UserResourceWithStreamingResponse(self._api.user)
 
-    @cached_property
-    def sso(self) -> SSOResourceWithStreamingResponse:
-        return SSOResourceWithStreamingResponse(self._api.sso)
-
     @cached_property
     def workspace(self) -> WorkspaceResourceWithStreamingResponse:
         return WorkspaceResourceWithStreamingResponse(self._api.workspace)
@@ -389,10 +361,6 @@ class AsyncAPIResourceWithStreamingResponse:
     def user(self) -> AsyncUserResourceWithStreamingResponse:
         return AsyncUserResourceWithStreamingResponse(self._api.user)
 
-    @cached_property
-    def sso(self) -> AsyncSSOResourceWithStreamingResponse:
-        return AsyncSSOResourceWithStreamingResponse(self._api.sso)
-
     @cached_property
     def workspace(self) -> AsyncWorkspaceResourceWithStreamingResponse:
         return AsyncWorkspaceResourceWithStreamingResponse(self._api.workspace)

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/resources/api/assistant.py

@@ -7,7 +7,7 @@ from typing import Dict, Optional
 import httpx
 
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ..._utils import maybe_transform, async_maybe_transform
+from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -50,6 +50,7 @@ class AssistantResource(SyncAPIResource):
         config_ext_id: Optional[str] | Omit = omit,
         parent_message_ext_id: Optional[str] | Omit = omit,
         tools: Dict[str, assistant_retrieve_params.Tools] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -71,6 +72,7 @@ class AssistantResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return self._post(
             "/api/assistant/retrieve",
             body=maybe_transform(
@@ -97,6 +99,7 @@ class AssistantResource(SyncAPIResource):
         config_ext_id: Optional[str] | Omit = omit,
         parent_message_ext_id: Optional[str] | Omit = omit,
         tools: Dict[str, assistant_query_params.Tools] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -118,6 +121,7 @@ class AssistantResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return self._post(
             "/api/assistant/query",
             body=maybe_transform(
@@ -165,6 +169,7 @@ class AsyncAssistantResource(AsyncAPIResource):
         config_ext_id: Optional[str] | Omit = omit,
         parent_message_ext_id: Optional[str] | Omit = omit,
         tools: Dict[str, assistant_retrieve_params.Tools] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -186,6 +191,7 @@ class AsyncAssistantResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return await self._post(
             "/api/assistant/retrieve",
             body=await async_maybe_transform(
@@ -212,6 +218,7 @@ class AsyncAssistantResource(AsyncAPIResource):
         config_ext_id: Optional[str] | Omit = omit,
         parent_message_ext_id: Optional[str] | Omit = omit,
         tools: Dict[str, assistant_query_params.Tools] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -233,6 +240,7 @@ class AsyncAssistantResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return await self._post(
             "/api/assistant/query",
             body=await async_maybe_transform(

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/resources/api/conversation/conversation.py

@@ -12,8 +12,8 @@ from .user import (
     UserResourceWithStreamingResponse,
     AsyncUserResourceWithStreamingResponse,
 )
-from ...._types import Body, Query, Headers, NotGiven, not_given
-from ...._utils import maybe_transform, async_maybe_transform
+from ...._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ...._utils import maybe_transform, strip_not_given, async_maybe_transform
 from ...._compat import cached_property
 from ...._resource import SyncAPIResource, AsyncAPIResource
 from ...._response import (
@@ -28,6 +28,7 @@ from ....types.api.conversation_share_response import ConversationShareResponse
 from ....types.api.conversation_delete_response import ConversationDeleteResponse
 from ....types.api.conversation_update_title_response import ConversationUpdateTitleResponse
 from ....types.api.conversation_delete_message_response import ConversationDeleteMessageResponse
+from ....types.api.conversation_retrieve_message_response import ConversationRetrieveMessageResponse
 from ....types.api.conversation_retrieve_threads_response import ConversationRetrieveThreadsResponse
 
 __all__ = ["ConversationResource", "AsyncConversationResource"]
@@ -128,6 +129,42 @@ class ConversationResource(SyncAPIResource):
             cast_to=ConversationDeleteMessageResponse,
         )
 
+    def retrieve_message(
+        self,
+        message_ext_id: str,
+        *,
+        workspace_key: str | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConversationRetrieveMessageResponse:
+        """
+        Get a single message with full details including decrypted execution trace.
+        Always includes the trace with decrypted sensitive fields.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not message_ext_id:
+            raise ValueError(f"Expected a non-empty value for `message_ext_id` but received {message_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
+        return self._get(
+            f"/api/conversation/message/{message_ext_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ConversationRetrieveMessageResponse,
+        )
+
     def retrieve_threads(
         self,
         conversation_ext_id: str,
@@ -338,6 +375,42 @@ class AsyncConversationResource(AsyncAPIResource):
             cast_to=ConversationDeleteMessageResponse,
         )
 
+    async def retrieve_message(
+        self,
+        message_ext_id: str,
+        *,
+        workspace_key: str | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConversationRetrieveMessageResponse:
+        """
+        Get a single message with full details including decrypted execution trace.
+        Always includes the trace with decrypted sensitive fields.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not message_ext_id:
+            raise ValueError(f"Expected a non-empty value for `message_ext_id` but received {message_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
+        return await self._get(
+            f"/api/conversation/message/{message_ext_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ConversationRetrieveMessageResponse,
+        )
+
     async def retrieve_threads(
         self,
         conversation_ext_id: str,
@@ -465,6 +538,9 @@ class ConversationResourceWithRawResponse:
         self.delete_message = to_raw_response_wrapper(
             conversation.delete_message,
         )
+        self.retrieve_message = to_raw_response_wrapper(
+            conversation.retrieve_message,
+        )
         self.retrieve_threads = to_raw_response_wrapper(
             conversation.retrieve_threads,
         )
@@ -490,6 +566,9 @@ class AsyncConversationResourceWithRawResponse:
         self.delete_message = async_to_raw_response_wrapper(
             conversation.delete_message,
         )
+        self.retrieve_message = async_to_raw_response_wrapper(
+            conversation.retrieve_message,
+        )
         self.retrieve_threads = async_to_raw_response_wrapper(
             conversation.retrieve_threads,
         )
@@ -515,6 +594,9 @@ class ConversationResourceWithStreamingResponse:
         self.delete_message = to_streamed_response_wrapper(
             conversation.delete_message,
         )
+        self.retrieve_message = to_streamed_response_wrapper(
+            conversation.retrieve_message,
+        )
         self.retrieve_threads = to_streamed_response_wrapper(
             conversation.retrieve_threads,
         )
@@ -540,6 +622,9 @@ class AsyncConversationResourceWithStreamingResponse:
         self.delete_message = async_to_streamed_response_wrapper(
             conversation.delete_message,
         )
+        self.retrieve_message = async_to_streamed_response_wrapper(
+            conversation.retrieve_message,
+        )
         self.retrieve_threads = async_to_streamed_response_wrapper(
             conversation.retrieve_threads,
         )

{arbi-0.2.1 → arbi-0.3.1}/src/arbi/resources/api/document/annotation.py

@@ -7,7 +7,7 @@ from typing import Optional
 import httpx
 
 from ...._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ...._utils import maybe_transform, async_maybe_transform
+from ...._utils import maybe_transform, strip_not_given, async_maybe_transform
 from ...._compat import cached_property
 from ...._resource import SyncAPIResource, AsyncAPIResource
 from ...._response import (
@@ -51,6 +51,7 @@ class AnnotationResource(SyncAPIResource):
         note: Optional[str] | Omit = omit,
         page_ref: Optional[int] | Omit = omit,
         tag_name: Optional[str] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -76,6 +77,7 @@ class AnnotationResource(SyncAPIResource):
         """
         if not doc_ext_id:
             raise ValueError(f"Expected a non-empty value for `doc_ext_id` but received {doc_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return self._post(
             f"/api/document/{doc_ext_id}/annotation",
             body=maybe_transform(
@@ -99,6 +101,7 @@ class AnnotationResource(SyncAPIResource):
         doc_ext_id: str,
         note: Optional[str] | Omit = omit,
         page_ref: Optional[int] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -122,6 +125,7 @@ class AnnotationResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `doc_ext_id` but received {doc_ext_id!r}")
         if not doctag_ext_id:
             raise ValueError(f"Expected a non-empty value for `doctag_ext_id` but received {doctag_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return self._patch(
             f"/api/document/{doc_ext_id}/annotation/{doctag_ext_id}",
             body=maybe_transform(
@@ -201,6 +205,7 @@ class AsyncAnnotationResource(AsyncAPIResource):
         note: Optional[str] | Omit = omit,
         page_ref: Optional[int] | Omit = omit,
         tag_name: Optional[str] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -226,6 +231,7 @@ class AsyncAnnotationResource(AsyncAPIResource):
         """
         if not doc_ext_id:
             raise ValueError(f"Expected a non-empty value for `doc_ext_id` but received {doc_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return await self._post(
             f"/api/document/{doc_ext_id}/annotation",
             body=await async_maybe_transform(
@@ -249,6 +255,7 @@ class AsyncAnnotationResource(AsyncAPIResource):
         doc_ext_id: str,
         note: Optional[str] | Omit = omit,
         page_ref: Optional[int] | Omit = omit,
+        workspace_key: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -272,6 +279,7 @@ class AsyncAnnotationResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `doc_ext_id` but received {doc_ext_id!r}")
         if not doctag_ext_id:
             raise ValueError(f"Expected a non-empty value for `doctag_ext_id` but received {doctag_ext_id!r}")
+        extra_headers = {**strip_not_given({"workspace-key": workspace_key}), **(extra_headers or {})}
         return await self._patch(
             f"/api/document/{doc_ext_id}/annotation/{doctag_ext_id}",
             body=await async_maybe_transform(