arbi 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

arbi/types/api/conversation_retrieve_message_response.py

@@ -0,0 +1,99 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Union, Optional
+from datetime import datetime
+from typing_extensions import Literal, Annotated, TypeAlias
+
+from ..chunk import Chunk
+from ..._utils import PropertyInfo
+from ..._models import BaseModel
+
+__all__ = [
+    "ConversationRetrieveMessageResponse",
+    "Tools",
+    "ToolsModelCitationTool",
+    "ToolsModelCitationToolToolResponses",
+    "ToolsRetrievalChunkToolOutput",
+    "ToolsRetrievalFullContextToolOutput",
+    "ToolsTraceTool",
+]
+
+
+class ToolsModelCitationToolToolResponses(BaseModel):
+    chunk_ids: List[str]
+
+    offset_end: int
+
+    offset_start: int
+
+    statement: str
+
+
+class ToolsModelCitationTool(BaseModel):
+    description: Optional[str] = None
+
+    name: Optional[Literal["model_citation"]] = None
+
+    tool_responses: Optional[Dict[str, ToolsModelCitationToolToolResponses]] = None
+
+
+class ToolsRetrievalChunkToolOutput(BaseModel):
+    description: Optional[str] = None
+
+    name: Optional[Literal["retrieval_chunk"]] = None
+
+    tool_args: Optional[Dict[str, List[str]]] = None
+
+    tool_responses: Optional[Dict[str, List[Chunk]]] = None
+
+
+class ToolsRetrievalFullContextToolOutput(BaseModel):
+    description: Optional[str] = None
+
+    name: Optional[Literal["retrieval_full_context"]] = None
+
+    tool_args: Optional[Dict[str, object]] = None
+
+    tool_responses: Optional[Dict[str, List[Chunk]]] = None
+
+
+class ToolsTraceTool(BaseModel):
+    description: Optional[str] = None
+
+    duration_seconds: Optional[float] = None
+
+    name: Optional[Literal["trace"]] = None
+
+    start_time: Optional[float] = None
+
+    steps: Optional[List[Dict[str, object]]] = None
+
+    trace_id: Optional[str] = None
+
+
+Tools: TypeAlias = Annotated[
+    Union[ToolsModelCitationTool, ToolsRetrievalChunkToolOutput, ToolsRetrievalFullContextToolOutput, ToolsTraceTool],
+    PropertyInfo(discriminator="name"),
+]
+
+
+class ConversationRetrieveMessageResponse(BaseModel):
+    content: str
+
+    conversation_ext_id: str
+
+    created_at: datetime
+
+    created_by_ext_id: str
+
+    external_id: str
+
+    role: Literal["user", "assistant", "system"]
+
+    config_ext_id: Optional[str] = None
+
+    parent_message_ext_id: Optional[str] = None
+
+    shared: Optional[bool] = None
+
+    tools: Optional[Dict[str, Tools]] = None

arbi/types/api/conversation_retrieve_threads_response.py

@@ -17,6 +17,7 @@ __all__ = [
     "ThreadHistoryToolsModelCitationToolToolResponses",
     "ThreadHistoryToolsRetrievalChunkToolOutput",
     "ThreadHistoryToolsRetrievalFullContextToolOutput",
+    "ThreadHistoryToolsTraceTool",
 ]
 
 
@@ -58,11 +59,26 @@ class ThreadHistoryToolsRetrievalFullContextToolOutput(BaseModel):
     tool_responses: Optional[Dict[str, List[Chunk]]] = None
 
 
+class ThreadHistoryToolsTraceTool(BaseModel):
+    description: Optional[str] = None
+
+    duration_seconds: Optional[float] = None
+
+    name: Optional[Literal["trace"]] = None
+
+    start_time: Optional[float] = None
+
+    steps: Optional[List[Dict[str, object]]] = None
+
+    trace_id: Optional[str] = None
+
+
 ThreadHistoryTools: TypeAlias = Annotated[
     Union[
         ThreadHistoryToolsModelCitationTool,
         ThreadHistoryToolsRetrievalChunkToolOutput,
         ThreadHistoryToolsRetrievalFullContextToolOutput,
+        ThreadHistoryToolsTraceTool,
     ],
     PropertyInfo(discriminator="name"),
 ]

arbi/types/api/document/annotation_create_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import TypedDict
+from typing_extensions import Annotated, TypedDict
+
+from ...._utils import PropertyInfo
 
 __all__ = ["AnnotationCreateParams"]
 
@@ -14,3 +16,5 @@ class AnnotationCreateParams(TypedDict, total=False):
     page_ref: Optional[int]
 
     tag_name: Optional[str]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/document/annotation_update_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
+
+from ...._utils import PropertyInfo
 
 __all__ = ["AnnotationUpdateParams"]
 
@@ -14,3 +16,5 @@ class AnnotationUpdateParams(TypedDict, total=False):
     note: Optional[str]
 
     page_ref: Optional[int]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/document_update_params.py

@@ -17,3 +17,5 @@ class DocumentUpdateParams(TypedDict, total=False):
     shared: Optional[bool]
 
     title: Optional[str]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/document_upload_from_url_params.py

@@ -3,9 +3,10 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
 
 from ..._types import SequenceNotStr
+from ..._utils import PropertyInfo
 
 __all__ = ["DocumentUploadFromURLParams"]
 
@@ -21,3 +22,5 @@ class DocumentUploadFromURLParams(TypedDict, total=False):
 
     shared: bool
     """Whether the document should be shared with workspace members"""
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/document_upload_params.py

@@ -3,9 +3,10 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
 
 from ..._types import FileTypes, SequenceNotStr
+from ..._utils import PropertyInfo
 
 __all__ = ["DocumentUploadParams"]
 
@@ -21,3 +22,5 @@ class DocumentUploadParams(TypedDict, total=False):
 
     shared: bool
     """Whether the document should be shared with workspace members"""
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/document_view_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import TypedDict
+from typing_extensions import Annotated, TypedDict
+
+from ..._utils import PropertyInfo
 
 __all__ = ["DocumentViewParams"]
 
@@ -11,3 +13,5 @@ __all__ = ["DocumentViewParams"]
 class DocumentViewParams(TypedDict, total=False):
     page: Optional[int]
     """Optional page to open on load"""
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/tag_create_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
 
 __all__ = ["TagCreateParams"]
 
@@ -16,3 +18,5 @@ class TagCreateParams(TypedDict, total=False):
     parent_ext_id: Optional[str]
 
     shared: Optional[bool]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/tag_update_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import TypedDict
+from typing_extensions import Annotated, TypedDict
+
+from ..._utils import PropertyInfo
 
 __all__ = ["TagUpdateParams"]
 
@@ -12,3 +14,5 @@ class TagUpdateParams(TypedDict, total=False):
     name: Optional[str]
 
     shared: Optional[bool]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/user_change_password_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict
+from typing_extensions import Required, TypedDict
+
+__all__ = ["UserChangePasswordParams"]
+
+
+class UserChangePasswordParams(TypedDict, total=False):
+    current_public_key: Required[str]
+
+    new_public_key: Required[str]
+
+    rewrapped_workspace_keys: Required[Dict[str, str]]

arbi/types/api/{sso_rotate_passcode_response.py → user_change_password_response.py}

@@ -2,10 +2,10 @@
 
 from ..._models import BaseModel
 
-__all__ = ["SSORotatePasscodeResponse"]
+__all__ = ["UserChangePasswordResponse"]
 
 
-class SSORotatePasscodeResponse(BaseModel):
+class UserChangePasswordResponse(BaseModel):
     detail: str
 
-    new_passcode: str
+    workspaces_updated: int

arbi/types/api/{sso_login_params.py → user_check_sso_status_params.py}

@@ -2,15 +2,12 @@
 
 from __future__ import annotations
 
-from typing import Optional
 from typing_extensions import Required, TypedDict
 
-__all__ = ["SSOLoginParams"]
+__all__ = ["UserCheckSSOStatusParams"]
 
 
-class SSOLoginParams(TypedDict, total=False):
-    token: Required[str]
-
+class UserCheckSSOStatusParams(TypedDict, total=False):
     email: Required[str]
 
-    passcode: Optional[str]
+    sso_token: Required[str]

arbi/types/api/{user_response.py → user_check_sso_status_response.py}

@@ -4,16 +4,14 @@ from typing import Optional
 
 from ..._models import BaseModel
 
-__all__ = ["UserResponse"]
+__all__ = ["UserCheckSSOStatusResponse"]
 
 
-class UserResponse(BaseModel):
-    email: Optional[str] = None
+class UserCheckSSOStatusResponse(BaseModel):
+    email: str
 
-    external_id: Optional[str] = None
+    status: str
 
     last_name: Optional[str] = None
 
     name: Optional[str] = None
-
-    public_key: Optional[str] = None

arbi/types/api/{sso_invite_params.py → user_invite_params.py}

@@ -4,8 +4,8 @@ from __future__ import annotations
 
 from typing_extensions import Required, TypedDict
 
-__all__ = ["SSOInviteParams"]
+__all__ = ["UserInviteParams"]
 
 
-class SSOInviteParams(TypedDict, total=False):
+class UserInviteParams(TypedDict, total=False):
     email: Required[str]

arbi/types/api/{sso_invite_response.py → user_invite_response.py}

@@ -2,10 +2,8 @@
 
 from ..._models import BaseModel
 
-__all__ = ["SSOInviteResponse"]
+__all__ = ["UserInviteResponse"]
 
 
-class SSOInviteResponse(BaseModel):
+class UserInviteResponse(BaseModel):
     detail: str
-
-    user_ext_id: str

arbi/types/api/user_login_params.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Optional
 from typing_extensions import Required, TypedDict
 
 __all__ = ["UserLoginParams"]
@@ -10,4 +11,6 @@ __all__ = ["UserLoginParams"]
 class UserLoginParams(TypedDict, total=False):
     email: Required[str]
 
-    password: Required[str]
+    public_key: Required[str]
+
+    sso_token: Optional[str]

arbi/types/api/user_login_response.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict
+from typing_extensions import TypeAlias
+
+__all__ = ["UserLoginResponse"]
+
+UserLoginResponse: TypeAlias = Dict[str, object]

arbi/types/api/user_register_params.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Optional
 from typing_extensions import Required, TypedDict
 
 __all__ = ["UserRegisterParams"]
@@ -10,10 +11,10 @@ __all__ = ["UserRegisterParams"]
 class UserRegisterParams(TypedDict, total=False):
     email: Required[str]
 
-    last_name: Required[str]
+    public_key: Required[str]
 
-    name: Required[str]
+    verification_credential: Required[str]
 
-    password: Required[str]
+    last_name: Optional[str]
 
-    verification_code: Required[str]
+    name: Optional[str]

arbi/types/api/workspace_copy_params.py

@@ -0,0 +1,21 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._types import SequenceNotStr
+from ..._utils import PropertyInfo
+
+__all__ = ["WorkspaceCopyParams"]
+
+
+class WorkspaceCopyParams(TypedDict, total=False):
+    items: Required[SequenceNotStr[str]]
+    """List of document external IDs to copy (e.g., ['doc-a1b2c3d4', 'doc-e5f6g7h8'])"""
+
+    target_workspace_ext_id: Required[str]
+
+    target_workspace_key: Annotated[str, PropertyInfo(alias="target-workspace-key")]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/workspace_copy_response.py

@@ -0,0 +1,25 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceCopyResponse", "Result"]
+
+
+class Result(BaseModel):
+    source_doc_ext_id: str
+
+    success: bool
+
+    error: Optional[str] = None
+
+    new_doc_ext_id: Optional[str] = None
+
+
+class WorkspaceCopyResponse(BaseModel):
+    detail: str
+
+    documents_copied: Optional[int] = None
+
+    results: Optional[List[Result]] = None

arbi/types/api/workspace_get_stats_response.py

@@ -8,6 +8,10 @@ __all__ = ["WorkspaceGetStatsResponse"]
 
 
 class WorkspaceGetStatsResponse(BaseModel):
-    conversation_count: Optional[int] = None
+    private_conversation_count: Optional[int] = None
 
-    document_count: Optional[int] = None
+    private_document_count: Optional[int] = None
+
+    shared_conversation_count: Optional[int] = None
+
+    shared_document_count: Optional[int] = None

arbi/types/api/workspace_get_users_response.py

@@ -9,7 +9,11 @@ __all__ = ["WorkspaceGetUsersResponse", "WorkspaceGetUsersResponseItem"]
 
 
 class WorkspaceGetUsersResponseItem(BaseModel):
-    user_email: str
+    email: str
+
+    last_name: str
+
+    name: str
 
     user_ext_id: str
 

arbi/types/api/workspace_response.py

@@ -1,11 +1,21 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Optional
+from typing import List, Optional
 from datetime import datetime
 
 from ..._models import BaseModel
 
-__all__ = ["WorkspaceResponse"]
+__all__ = ["WorkspaceResponse", "User"]
+
+
+class User(BaseModel):
+    email: str
+
+    last_name: str
+
+    name: str
+
+    user_ext_id: str
 
 
 class WorkspaceResponse(BaseModel):
@@ -24,3 +34,15 @@ class WorkspaceResponse(BaseModel):
     updated_at: datetime
 
     updated_by_ext_id: Optional[str] = None
+
+    private_conversation_count: Optional[int] = None
+
+    private_document_count: Optional[int] = None
+
+    shared_conversation_count: Optional[int] = None
+
+    shared_document_count: Optional[int] = None
+
+    users: Optional[List[User]] = None
+
+    wrapped_key: Optional[str] = None

arbi/types/api/workspace_share_params.py

@@ -2,10 +2,14 @@
 
 from __future__ import annotations
 
-from typing_extensions import Required, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
 
 __all__ = ["WorkspaceShareParams"]
 
 
 class WorkspaceShareParams(TypedDict, total=False):
     recipient_email: Required[str]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

arbi/types/api/workspace_update_params.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import TypedDict
+from typing_extensions import Annotated, TypedDict
+
+from ..._utils import PropertyInfo
 
 __all__ = ["WorkspaceUpdateParams"]
 
@@ -14,3 +16,5 @@ class WorkspaceUpdateParams(TypedDict, total=False):
     is_public: Optional[bool]
 
     name: Optional[str]
+
+    workspace_key: Annotated[str, PropertyInfo(alias="workspace-key")]

{arbi-0.2.1.dist-info → arbi-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arbi
-Version: 0.2.1
+Version: 0.3.0
 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