arbi 0.1.0__py3-none-any.whl

arbi/types/api/tag_create_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Required, TypedDict
+
+__all__ = ["TagCreateParams"]
+
+
+class TagCreateParams(TypedDict, total=False):
+    name: Required[str]
+
+    workspace_ext_id: Required[str]
+
+    parent_ext_id: Optional[str]
+
+    shared: Optional[bool]

arbi/types/api/tag_create_response.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["TagCreateResponse"]
+
+
+class TagCreateResponse(BaseModel):
+    detail: str
+
+    tag_id: str

arbi/types/api/tag_delete_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["TagDeleteResponse"]
+
+
+class TagDeleteResponse(BaseModel):
+    detail: str

arbi/types/api/tag_get_docs_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .document.doc_tag_response import DocTagResponse
+
+__all__ = ["TagGetDocsResponse"]
+
+TagGetDocsResponse: TypeAlias = List[DocTagResponse]

arbi/types/api/tag_remove_from_docs_params.py

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

arbi/types/api/tag_remove_from_docs_response.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from ..._models import BaseModel
+
+__all__ = ["TagRemoveFromDocsResponse"]
+
+
+class TagRemoveFromDocsResponse(BaseModel):
+    detail: str
+
+    removed_doc_tag_ids: List[str]

arbi/types/api/tag_update_params.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import TypedDict
+
+__all__ = ["TagUpdateParams"]
+
+
+class TagUpdateParams(TypedDict, total=False):
+    name: Optional[str]
+
+    shared: Optional[bool]

arbi/types/api/tag_update_response.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["TagUpdateResponse"]
+
+
+class TagUpdateResponse(BaseModel):
+    detail: str
+
+    tag_id: str
+
+    name: Optional[str] = None
+
+    shared: Optional[bool] = None

arbi/types/api/title_llm_config.py

@@ -0,0 +1,29 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from ..._models import BaseModel
+
+__all__ = ["TitleLlmConfig"]
+
+
+class TitleLlmConfig(BaseModel):
+    api_type: Optional[Literal["local", "remote"]] = FieldInfo(alias="API_TYPE", default=None)
+    """The inference type (local or remote)."""
+
+    max_char_size_to_answer: Optional[int] = FieldInfo(alias="MAX_CHAR_SIZE_TO_ANSWER", default=None)
+    """Maximum character size to answer."""
+
+    max_tokens: Optional[int] = FieldInfo(alias="MAX_TOKENS", default=None)
+    """Maximum number of tokens allowed."""
+
+    api_model_name: Optional[str] = FieldInfo(alias="MODEL_NAME", default=None)
+    """The name of the model to be used."""
+
+    system_instruction: Optional[str] = FieldInfo(alias="SYSTEM_INSTRUCTION", default=None)
+
+    temperature: Optional[float] = FieldInfo(alias="TEMPERATURE", default=None)
+    """Temperature value for randomness."""

arbi/types/api/title_llm_config_param.py

@@ -0,0 +1,28 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+
+__all__ = ["TitleLlmConfigParam"]
+
+
+class TitleLlmConfigParam(TypedDict, total=False):
+    api_type: Annotated[Literal["local", "remote"], PropertyInfo(alias="API_TYPE")]
+    """The inference type (local or remote)."""
+
+    max_char_size_to_answer: Annotated[int, PropertyInfo(alias="MAX_CHAR_SIZE_TO_ANSWER")]
+    """Maximum character size to answer."""
+
+    max_tokens: Annotated[int, PropertyInfo(alias="MAX_TOKENS")]
+    """Maximum number of tokens allowed."""
+
+    model_name: Annotated[str, PropertyInfo(alias="MODEL_NAME")]
+    """The name of the model to be used."""
+
+    system_instruction: Annotated[str, PropertyInfo(alias="SYSTEM_INSTRUCTION")]
+
+    temperature: Annotated[float, PropertyInfo(alias="TEMPERATURE")]
+    """Temperature value for randomness."""

arbi/types/api/token.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["Token"]
+
+
+class Token(BaseModel):
+    access_token: str
+
+    token_type: str
+
+    user_ext_id: str

arbi/types/api/user/__init__.py

@@ -0,0 +1,6 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .setting_update_params import SettingUpdateParams as SettingUpdateParams
+from .setting_retrieve_response import SettingRetrieveResponse as SettingRetrieveResponse

arbi/types/api/user/setting_retrieve_response.py

@@ -0,0 +1,36 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import TYPE_CHECKING, Dict, List, Optional
+
+from pydantic import Field as FieldInfo
+
+from ...._models import BaseModel
+
+__all__ = ["SettingRetrieveResponse"]
+
+
+class SettingRetrieveResponse(BaseModel):
+    developer: Optional[bool] = None
+
+    last_workspace: Optional[str] = None
+
+    pinned_workspaces: Optional[List[str]] = None
+
+    show_document_navigator: Optional[bool] = None
+
+    show_help_page: Optional[bool] = None
+
+    show_invite_tab: Optional[bool] = None
+
+    show_security_settings: Optional[bool] = None
+
+    show_smart_search: Optional[bool] = None
+
+    show_thread_visualization: Optional[bool] = None
+
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+    if TYPE_CHECKING:
+        # Stub to indicate that arbitrary properties are accepted.
+        # To access properties that are not valid identifiers you can use `getattr`, e.g.
+        # `getattr(obj, '$type')`
+        def __getattr__(self, attr: str) -> object: ...

arbi/types/api/user/setting_update_params.py

@@ -0,0 +1,24 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from typing_extensions import TypedDict
+
+__all__ = ["SettingUpdateParams"]
+
+
+class SettingUpdateParams(TypedDict, total=False):
+    pinned_workspaces: Optional[List[str]]
+
+    show_document_navigator: Optional[bool]
+
+    show_help_page: Optional[bool]
+
+    show_invite_tab: Optional[bool]
+
+    show_security_settings: Optional[bool]
+
+    show_smart_search: Optional[bool]
+
+    show_thread_visualization: Optional[bool]

arbi/types/api/user_list_workspaces_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .workspace_response import WorkspaceResponse
+
+__all__ = ["UserListWorkspacesResponse"]
+
+UserListWorkspacesResponse: TypeAlias = List[WorkspaceResponse]

arbi/types/api/user_login_params.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["UserLoginParams"]
+
+
+class UserLoginParams(TypedDict, total=False):
+    email: Required[str]
+
+    password: Required[str]

arbi/types/api/user_logout_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["UserLogoutResponse"]
+
+
+class UserLogoutResponse(BaseModel):
+    detail: str

arbi/types/api/user_register_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["UserRegisterParams"]
+
+
+class UserRegisterParams(TypedDict, total=False):
+    email: Required[str]
+
+    last_name: Required[str]
+
+    name: Required[str]
+
+    password: Required[str]

arbi/types/api/user_response.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["UserResponse"]
+
+
+class UserResponse(BaseModel):
+    email: Optional[str] = None
+
+    external_id: Optional[str] = None
+
+    last_name: Optional[str] = None
+
+    name: Optional[str] = None
+
+    public_key: Optional[str] = None

arbi/types/api/workspace_create_protected_params.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Required, TypedDict
+
+__all__ = ["WorkspaceCreateProtectedParams"]
+
+
+class WorkspaceCreateProtectedParams(TypedDict, total=False):
+    name: Required[str]
+
+    description: Optional[str]

arbi/types/api/workspace_delete_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceDeleteResponse"]
+
+
+class WorkspaceDeleteResponse(BaseModel):
+    detail: str

arbi/types/api/workspace_get_conversations_response.py

@@ -0,0 +1,28 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+from typing_extensions import TypeAlias
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceGetConversationsResponse", "WorkspaceGetConversationsResponseItem"]
+
+
+class WorkspaceGetConversationsResponseItem(BaseModel):
+    created_at: datetime
+
+    external_id: str
+
+    is_creator: bool
+
+    message_count: int
+
+    title: Optional[str] = None
+
+    updated_at: datetime
+
+    is_shared: Optional[bool] = None
+
+
+WorkspaceGetConversationsResponse: TypeAlias = List[WorkspaceGetConversationsResponseItem]

arbi/types/api/workspace_get_doctags_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .document.doc_tag_response import DocTagResponse
+
+__all__ = ["WorkspaceGetDoctagsResponse"]
+
+WorkspaceGetDoctagsResponse: TypeAlias = List[DocTagResponse]

arbi/types/api/workspace_get_documents_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .doc_response import DocResponse
+
+__all__ = ["WorkspaceGetDocumentsResponse"]
+
+WorkspaceGetDocumentsResponse: TypeAlias = List[DocResponse]

arbi/types/api/workspace_get_stats_response.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceGetStatsResponse"]
+
+
+class WorkspaceGetStatsResponse(BaseModel):
+    conversation_count: Optional[int] = None
+
+    document_count: Optional[int] = None

arbi/types/api/workspace_get_tags_response.py

@@ -0,0 +1,30 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceGetTagsResponse", "Tag"]
+
+
+class Tag(BaseModel):
+    created_at: datetime
+
+    created_by_ext_id: str
+
+    name: str
+
+    tag_ext_id: str
+
+    type: str
+
+    updated_at: datetime
+
+    workspace_ext_id: str
+
+    shared: Optional[bool] = None
+
+
+class WorkspaceGetTagsResponse(BaseModel):
+    tags: List[Tag]

arbi/types/api/workspace_get_users_response.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceGetUsersResponse", "WorkspaceGetUsersResponseItem"]
+
+
+class WorkspaceGetUsersResponseItem(BaseModel):
+    user_email: str
+
+    user_ext_id: str
+
+
+WorkspaceGetUsersResponse: TypeAlias = List[WorkspaceGetUsersResponseItem]

arbi/types/api/workspace_remove_user_params.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["WorkspaceRemoveUserParams"]
+
+
+class WorkspaceRemoveUserParams(TypedDict, total=False):
+    user_ext_id: Required[str]

arbi/types/api/workspace_remove_user_response.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceRemoveUserResponse"]
+
+
+class WorkspaceRemoveUserResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/workspace_response.py

@@ -0,0 +1,24 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceResponse"]
+
+
+class WorkspaceResponse(BaseModel):
+    created_at: datetime
+
+    created_by_ext_id: str
+
+    description: Optional[str] = None
+
+    external_id: str
+
+    name: str
+
+    updated_at: datetime
+
+    updated_by_ext_id: Optional[str] = None

arbi/types/api/workspace_share_params.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["WorkspaceShareParams"]
+
+
+class WorkspaceShareParams(TypedDict, total=False):
+    recipient_email: Required[str]

arbi/types/api/workspace_share_response.py

@@ -0,0 +1,21 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["WorkspaceShareResponse", "Notifications"]
+
+
+class Notifications(BaseModel):
+    recipient: str
+
+    sender: str
+
+
+class WorkspaceShareResponse(BaseModel):
+    detail: str
+
+    notifications: Notifications
+
+    shared_with: str
+
+    workspace_ext_id: str

arbi/types/api/workspace_update_params.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import TypedDict
+
+__all__ = ["WorkspaceUpdateParams"]
+
+
+class WorkspaceUpdateParams(TypedDict, total=False):
+    description: Optional[str]
+
+    name: Optional[str]

arbi/types/chunk.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .._models import BaseModel
+from .chunk_metadata import ChunkMetadata
+
+__all__ = ["Chunk"]
+
+
+class Chunk(BaseModel):
+    content: str
+
+    metadata: ChunkMetadata

arbi/types/chunk_metadata.py

@@ -0,0 +1,31 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from .._models import BaseModel
+
+__all__ = ["ChunkMetadata"]
+
+
+class ChunkMetadata(BaseModel):
+    chunk_doc_idx: int
+
+    chunk_ext_id: str
+
+    chunk_pg_idx: int
+
+    created_at: str
+
+    page_number: int
+
+    chunk_id: Optional[str] = None
+
+    doc_ext_id: Optional[str] = None
+
+    doc_title: Optional[str] = None
+
+    rerank_score: Optional[float] = None
+
+    score: Optional[float] = None
+
+    tokens: Optional[int] = None

arbi/types/chunk_metadata_param.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Required, TypedDict
+
+__all__ = ["ChunkMetadataParam"]
+
+
+class ChunkMetadataParam(TypedDict, total=False):
+    chunk_doc_idx: Required[int]
+
+    chunk_ext_id: Required[str]
+
+    chunk_pg_idx: Required[int]
+
+    created_at: Required[str]
+
+    page_number: Required[int]
+
+    chunk_id: Optional[str]
+
+    doc_ext_id: Optional[str]
+
+    doc_title: Optional[str]
+
+    rerank_score: Optional[float]
+
+    score: Optional[float]
+
+    tokens: Optional[int]

arbi/types/chunk_param.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+from .chunk_metadata_param import ChunkMetadataParam
+
+__all__ = ["ChunkParam"]
+
+
+class ChunkParam(TypedDict, total=False):
+    content: Required[str]
+
+    metadata: Required[ChunkMetadataParam]