arbi 0.1.0__py3-none-any.whl

arbi/types/__init__.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .chunk import Chunk as Chunk
+from .chunk_param import ChunkParam as ChunkParam
+from .chunk_metadata import ChunkMetadata as ChunkMetadata
+from .chunk_metadata_param import ChunkMetadataParam as ChunkMetadataParam

arbi/types/api/__init__.py

@@ -0,0 +1,84 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .token import Token as Token
+from .doc_response import DocResponse as DocResponse
+from .user_response import UserResponse as UserResponse
+from .embedder_config import EmbedderConfig as EmbedderConfig
+from .reranker_config import RerankerConfig as RerankerConfig
+from .query_llm_config import QueryLlmConfig as QueryLlmConfig
+from .retriever_config import RetrieverConfig as RetrieverConfig
+from .sso_login_params import SSOLoginParams as SSOLoginParams
+from .title_llm_config import TitleLlmConfig as TitleLlmConfig
+from .sso_invite_params import SSOInviteParams as SSOInviteParams
+from .tag_create_params import TagCreateParams as TagCreateParams
+from .tag_update_params import TagUpdateParams as TagUpdateParams
+from .user_login_params import UserLoginParams as UserLoginParams
+from .sso_login_response import SSOLoginResponse as SSOLoginResponse
+from .workspace_response import WorkspaceResponse as WorkspaceResponse
+from .parser_config_param import ParserConfigParam as ParserConfigParam
+from .sso_invite_response import SSOInviteResponse as SSOInviteResponse
+from .tag_create_response import TagCreateResponse as TagCreateResponse
+from .tag_delete_response import TagDeleteResponse as TagDeleteResponse
+from .tag_update_response import TagUpdateResponse as TagUpdateResponse
+from .chunker_config_param import ChunkerConfigParam as ChunkerConfigParam
+from .config_create_params import ConfigCreateParams as ConfigCreateParams
+from .document_view_params import DocumentViewParams as DocumentViewParams
+from .user_logout_response import UserLogoutResponse as UserLogoutResponse
+from .user_register_params import UserRegisterParams as UserRegisterParams
+from .embedder_config_param import EmbedderConfigParam as EmbedderConfigParam
+from .model_citation_config import ModelCitationConfig as ModelCitationConfig
+from .reranker_config_param import RerankerConfigParam as RerankerConfigParam
+from .tag_get_docs_response import TagGetDocsResponse as TagGetDocsResponse
+from .assistant_query_params import AssistantQueryParams as AssistantQueryParams
+from .config_create_response import ConfigCreateResponse as ConfigCreateResponse
+from .config_delete_response import ConfigDeleteResponse as ConfigDeleteResponse
+from .document_update_params import DocumentUpdateParams as DocumentUpdateParams
+from .document_upload_params import DocumentUploadParams as DocumentUploadParams
+from .query_llm_config_param import QueryLlmConfigParam as QueryLlmConfigParam
+from .retriever_config_param import RetrieverConfigParam as RetrieverConfigParam
+from .title_llm_config_param import TitleLlmConfigParam as TitleLlmConfigParam
+from .workspace_share_params import WorkspaceShareParams as WorkspaceShareParams
+from .workspace_update_params import WorkspaceUpdateParams as WorkspaceUpdateParams
+from .config_retrieve_response import ConfigRetrieveResponse as ConfigRetrieveResponse
+from .document_delete_response import DocumentDeleteResponse as DocumentDeleteResponse
+from .document_update_response import DocumentUpdateResponse as DocumentUpdateResponse
+from .tag_apply_to_docs_params import TagApplyToDocsParams as TagApplyToDocsParams
+from .workspace_share_response import WorkspaceShareResponse as WorkspaceShareResponse
+from .assistant_retrieve_params import AssistantRetrieveParams as AssistantRetrieveParams
+from .health_check_app_response import HealthCheckAppResponse as HealthCheckAppResponse
+from .workspace_delete_response import WorkspaceDeleteResponse as WorkspaceDeleteResponse
+from .document_get_tags_response import DocumentGetTagsResponse as DocumentGetTagsResponse
+from .health_get_models_response import HealthGetModelsResponse as HealthGetModelsResponse
+from .tag_apply_to_docs_response import TagApplyToDocsResponse as TagApplyToDocsResponse
+from .conversation_share_response import ConversationShareResponse as ConversationShareResponse
+from .model_citation_config_param import ModelCitationConfigParam as ModelCitationConfigParam
+from .tag_remove_from_docs_params import TagRemoveFromDocsParams as TagRemoveFromDocsParams
+from .workspace_get_tags_response import WorkspaceGetTagsResponse as WorkspaceGetTagsResponse
+from .config_get_versions_response import ConfigGetVersionsResponse as ConfigGetVersionsResponse
+from .conversation_delete_response import ConversationDeleteResponse as ConversationDeleteResponse
+from .document_get_parsed_response import DocumentGetParsedResponse as DocumentGetParsedResponse
+from .health_check_models_response import HealthCheckModelsResponse as HealthCheckModelsResponse
+from .sso_rotate_passcode_response import SSORotatePasscodeResponse as SSORotatePasscodeResponse
+from .workspace_get_stats_response import WorkspaceGetStatsResponse as WorkspaceGetStatsResponse
+from .workspace_get_users_response import WorkspaceGetUsersResponse as WorkspaceGetUsersResponse
+from .workspace_remove_user_params import WorkspaceRemoveUserParams as WorkspaceRemoveUserParams
+from .tag_remove_from_docs_response import TagRemoveFromDocsResponse as TagRemoveFromDocsResponse
+from .user_list_workspaces_response import UserListWorkspacesResponse as UserListWorkspacesResponse
+from .health_check_services_response import HealthCheckServicesResponse as HealthCheckServicesResponse
+from .workspace_get_doctags_response import WorkspaceGetDoctagsResponse as WorkspaceGetDoctagsResponse
+from .workspace_remove_user_response import WorkspaceRemoveUserResponse as WorkspaceRemoveUserResponse
+from .conversation_update_title_params import ConversationUpdateTitleParams as ConversationUpdateTitleParams
+from .workspace_get_documents_response import WorkspaceGetDocumentsResponse as WorkspaceGetDocumentsResponse
+from .workspace_create_protected_params import WorkspaceCreateProtectedParams as WorkspaceCreateProtectedParams
+from .conversation_update_title_response import ConversationUpdateTitleResponse as ConversationUpdateTitleResponse
+from .document_date_extractor_llm_config import DocumentDateExtractorLlmConfig as DocumentDateExtractorLlmConfig
+from .conversation_delete_message_response import ConversationDeleteMessageResponse as ConversationDeleteMessageResponse
+from .workspace_get_conversations_response import WorkspaceGetConversationsResponse as WorkspaceGetConversationsResponse
+from .conversation_retrieve_threads_response import (
+    ConversationRetrieveThreadsResponse as ConversationRetrieveThreadsResponse,
+)
+from .document_date_extractor_llm_config_param import (
+    DocumentDateExtractorLlmConfigParam as DocumentDateExtractorLlmConfigParam,
+)

arbi/types/api/assistant_query_params.py

@@ -0,0 +1,59 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, List, Union, Iterable, Optional
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
+
+from ..chunk_param import ChunkParam
+
+__all__ = [
+    "AssistantQueryParams",
+    "Tools",
+    "ToolsModelCitationTool",
+    "ToolsRetrievalChunkToolInput",
+    "ToolsRetrievalFullContextToolInput",
+]
+
+
+class AssistantQueryParams(TypedDict, total=False):
+    content: Required[str]
+
+    workspace_ext_id: Required[str]
+
+    config_ext_id: Optional[str]
+
+    parent_message_ext_id: Optional[str]
+
+    tools: Dict[str, Tools]
+
+
+class ToolsModelCitationTool(TypedDict, total=False):
+    description: str
+
+    name: Literal["model_citation"]
+
+    tool_responses: Dict[str, List[str]]
+
+
+class ToolsRetrievalChunkToolInput(TypedDict, total=False):
+    description: str
+
+    name: Literal["retrieval_chunk"]
+
+    tool_args: Dict[str, List[str]]
+
+    tool_responses: Dict[str, Iterable[ChunkParam]]
+
+
+class ToolsRetrievalFullContextToolInput(TypedDict, total=False):
+    description: str
+
+    name: Literal["retrieval_full_context"]
+
+    tool_args: Dict[str, List[str]]
+
+    tool_responses: Dict[str, Iterable[ChunkParam]]
+
+
+Tools: TypeAlias = Union[ToolsModelCitationTool, ToolsRetrievalChunkToolInput, ToolsRetrievalFullContextToolInput]

arbi/types/api/assistant_retrieve_params.py

@@ -0,0 +1,59 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, List, Union, Iterable, Optional
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
+
+from ..chunk_param import ChunkParam
+
+__all__ = [
+    "AssistantRetrieveParams",
+    "Tools",
+    "ToolsModelCitationTool",
+    "ToolsRetrievalChunkToolInput",
+    "ToolsRetrievalFullContextToolInput",
+]
+
+
+class AssistantRetrieveParams(TypedDict, total=False):
+    content: Required[str]
+
+    workspace_ext_id: Required[str]
+
+    config_ext_id: Optional[str]
+
+    parent_message_ext_id: Optional[str]
+
+    tools: Dict[str, Tools]
+
+
+class ToolsModelCitationTool(TypedDict, total=False):
+    description: str
+
+    name: Literal["model_citation"]
+
+    tool_responses: Dict[str, List[str]]
+
+
+class ToolsRetrievalChunkToolInput(TypedDict, total=False):
+    description: str
+
+    name: Literal["retrieval_chunk"]
+
+    tool_args: Dict[str, List[str]]
+
+    tool_responses: Dict[str, Iterable[ChunkParam]]
+
+
+class ToolsRetrievalFullContextToolInput(TypedDict, total=False):
+    description: str
+
+    name: Literal["retrieval_full_context"]
+
+    tool_args: Dict[str, List[str]]
+
+    tool_responses: Dict[str, Iterable[ChunkParam]]
+
+
+Tools: TypeAlias = Union[ToolsModelCitationTool, ToolsRetrievalChunkToolInput, ToolsRetrievalFullContextToolInput]

arbi/types/api/chunker_config_param.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypeAlias
+
+__all__ = ["ChunkerConfigParam"]
+
+ChunkerConfigParam: TypeAlias = object

arbi/types/api/config_create_params.py

@@ -0,0 +1,45 @@
+# 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 Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+from .parser_config_param import ParserConfigParam
+from .chunker_config_param import ChunkerConfigParam
+from .embedder_config_param import EmbedderConfigParam
+from .reranker_config_param import RerankerConfigParam
+from .query_llm_config_param import QueryLlmConfigParam
+from .retriever_config_param import RetrieverConfigParam
+from .title_llm_config_param import TitleLlmConfigParam
+from .model_citation_config_param import ModelCitationConfigParam
+from .document_date_extractor_llm_config_param import DocumentDateExtractorLlmConfigParam
+
+__all__ = ["ConfigCreateParams"]
+
+
+class ConfigCreateParams(TypedDict, total=False):
+    chunker: Annotated[Optional[ChunkerConfigParam], PropertyInfo(alias="Chunker")]
+
+    document_date_extractor_llm: Annotated[
+        Optional[DocumentDateExtractorLlmConfigParam], PropertyInfo(alias="DocumentDateExtractorLLM")
+    ]
+
+    embedder: Annotated[Optional[EmbedderConfigParam], PropertyInfo(alias="Embedder")]
+
+    model_citation: Annotated[Optional[ModelCitationConfigParam], PropertyInfo(alias="ModelCitation")]
+
+    parent_message_ext_id: Optional[str]
+
+    parser: Annotated[Optional[ParserConfigParam], PropertyInfo(alias="Parser")]
+
+    query_llm: Annotated[Optional[QueryLlmConfigParam], PropertyInfo(alias="QueryLLM")]
+
+    reranker: Annotated[Optional[RerankerConfigParam], PropertyInfo(alias="Reranker")]
+
+    retriever: Annotated[Optional[RetrieverConfigParam], PropertyInfo(alias="Retriever")]
+
+    title: str
+
+    title_llm: Annotated[Optional[TitleLlmConfigParam], PropertyInfo(alias="TitleLLM")]

arbi/types/api/config_create_response.py

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

arbi/types/api/config_delete_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__ = ["ConfigDeleteResponse"]
+
+
+class ConfigDeleteResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/config_get_versions_response.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ..._models import BaseModel
+
+__all__ = ["ConfigGetVersionsResponse", "Version"]
+
+
+class Version(BaseModel):
+    created_at: str
+
+    external_id: str
+
+    title: Optional[str] = None
+
+
+class ConfigGetVersionsResponse(BaseModel):
+    versions: List[Version]

arbi/types/api/config_retrieve_response.py

@@ -0,0 +1,46 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Union, Optional
+from typing_extensions import TypeAlias
+
+from pydantic import Field as FieldInfo
+
+from ..._models import BaseModel
+from .embedder_config import EmbedderConfig
+from .reranker_config import RerankerConfig
+from .query_llm_config import QueryLlmConfig
+from .retriever_config import RetrieverConfig
+from .title_llm_config import TitleLlmConfig
+from .model_citation_config import ModelCitationConfig
+from .document_date_extractor_llm_config import DocumentDateExtractorLlmConfig
+
+__all__ = ["ConfigRetrieveResponse", "AllConfigs", "NonDeveloperConfig"]
+
+
+class AllConfigs(BaseModel):
+    chunker: Optional[object] = FieldInfo(alias="Chunker", default=None)
+
+    document_date_extractor_llm: Optional[DocumentDateExtractorLlmConfig] = FieldInfo(
+        alias="DocumentDateExtractorLLM", default=None
+    )
+
+    embedder: Optional[EmbedderConfig] = FieldInfo(alias="Embedder", default=None)
+
+    api_model_citation: Optional[ModelCitationConfig] = FieldInfo(alias="ModelCitation", default=None)
+
+    parser: Optional[object] = FieldInfo(alias="Parser", default=None)
+
+    query_llm: Optional[QueryLlmConfig] = FieldInfo(alias="QueryLLM", default=None)
+
+    reranker: Optional[RerankerConfig] = FieldInfo(alias="Reranker", default=None)
+
+    retriever: Optional[RetrieverConfig] = FieldInfo(alias="Retriever", default=None)
+
+    title_llm: Optional[TitleLlmConfig] = FieldInfo(alias="TitleLLM", default=None)
+
+
+class NonDeveloperConfig(BaseModel):
+    query_llm: Dict[str, str] = FieldInfo(alias="QueryLLM")
+
+
+ConfigRetrieveResponse: TypeAlias = Union[AllConfigs, NonDeveloperConfig]

arbi/types/api/conversation/__init__.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .user_add_params import UserAddParams as UserAddParams
+from .user_add_response import UserAddResponse as UserAddResponse
+from .user_remove_params import UserRemoveParams as UserRemoveParams
+from .user_remove_response import UserRemoveResponse as UserRemoveResponse

arbi/types/api/conversation/user_add_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__ = ["UserAddParams"]
+
+
+class UserAddParams(TypedDict, total=False):
+    user_ext_id: Required[str]

arbi/types/api/conversation/user_add_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__ = ["UserAddResponse"]
+
+
+class UserAddResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/conversation/user_remove_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__ = ["UserRemoveParams"]
+
+
+class UserRemoveParams(TypedDict, total=False):
+    user_ext_id: Required[str]

arbi/types/api/conversation/user_remove_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__ = ["UserRemoveResponse"]
+
+
+class UserRemoveResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/conversation_delete_message_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__ = ["ConversationDeleteMessageResponse"]
+
+
+class ConversationDeleteMessageResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/conversation_delete_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__ = ["ConversationDeleteResponse"]
+
+
+class ConversationDeleteResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/conversation_retrieve_threads_response.py

@@ -0,0 +1,91 @@
+# 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__ = [
+    "ConversationRetrieveThreadsResponse",
+    "Thread",
+    "ThreadHistory",
+    "ThreadHistoryTools",
+    "ThreadHistoryToolsModelCitationTool",
+    "ThreadHistoryToolsRetrievalChunkToolOutput",
+    "ThreadHistoryToolsRetrievalFullContextToolOutput",
+]
+
+
+class ThreadHistoryToolsModelCitationTool(BaseModel):
+    description: Optional[str] = None
+
+    name: Optional[Literal["model_citation"]] = None
+
+    tool_responses: Optional[Dict[str, List[str]]] = None
+
+
+class ThreadHistoryToolsRetrievalChunkToolOutput(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 ThreadHistoryToolsRetrievalFullContextToolOutput(BaseModel):
+    description: Optional[str] = None
+
+    name: Optional[Literal["retrieval_full_context"]] = None
+
+    tool_args: Optional[Dict[str, List[str]]] = None
+
+    tool_responses: Optional[Dict[str, List[Chunk]]] = None
+
+
+ThreadHistoryTools: TypeAlias = Annotated[
+    Union[
+        ThreadHistoryToolsModelCitationTool,
+        ThreadHistoryToolsRetrievalChunkToolOutput,
+        ThreadHistoryToolsRetrievalFullContextToolOutput,
+    ],
+    PropertyInfo(discriminator="name"),
+]
+
+
+class ThreadHistory(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, ThreadHistoryTools]] = None
+
+
+class Thread(BaseModel):
+    history: List[ThreadHistory]
+
+    leaf_message_ext_id: str
+
+
+class ConversationRetrieveThreadsResponse(BaseModel):
+    conversation_ext_id: str
+
+    threads: List[Thread]

arbi/types/api/conversation_share_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__ = ["ConversationShareResponse"]
+
+
+class ConversationShareResponse(BaseModel):
+    detail: Optional[str] = None

arbi/types/api/conversation_update_title_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_extensions import Required, TypedDict
+
+__all__ = ["ConversationUpdateTitleParams"]
+
+
+class ConversationUpdateTitleParams(TypedDict, total=False):
+    title: Required[str]
+    """New conversation title (1-60 characters)"""

arbi/types/api/conversation_update_title_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__ = ["ConversationUpdateTitleResponse"]
+
+
+class ConversationUpdateTitleResponse(BaseModel):
+    title: str
+
+    detail: Optional[str] = None

arbi/types/api/doc_response.py

@@ -0,0 +1,48 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import date, datetime
+
+from ..._models import BaseModel
+
+__all__ = ["DocResponse"]
+
+
+class DocResponse(BaseModel):
+    created_at: datetime
+
+    external_id: str
+
+    title: str
+
+    updated_at: datetime
+
+    workspace_ext_id: str
+
+    config_ext_id: Optional[str] = None
+
+    created_by_ext_id: Optional[str] = None
+
+    doc_date: Optional[date] = None
+
+    file_name: Optional[str] = None
+
+    file_size: Optional[int] = None
+
+    file_type: Optional[str] = None
+
+    n_chunks: Optional[int] = None
+
+    n_pages: Optional[int] = None
+
+    re_ocred: Optional[bool] = None
+
+    shared: Optional[bool] = None
+
+    status: Optional[str] = None
+
+    storage_type: Optional[str] = None
+
+    storage_uri: Optional[str] = None
+
+    tokens: Optional[int] = None

arbi/types/api/document/__init__.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .doc_tag_response import DocTagResponse as DocTagResponse
+from .annotation_create_params import AnnotationCreateParams as AnnotationCreateParams
+from .annotation_update_params import AnnotationUpdateParams as AnnotationUpdateParams
+from .annotation_delete_response import AnnotationDeleteResponse as AnnotationDeleteResponse

arbi/types/api/document/annotation_create_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 Optional
+from typing_extensions import TypedDict
+
+__all__ = ["AnnotationCreateParams"]
+
+
+class AnnotationCreateParams(TypedDict, total=False):
+    note: Optional[str]
+
+    page_ref: Optional[int]
+
+    tag_name: Optional[str]

arbi/types/api/document/annotation_delete_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__ = ["AnnotationDeleteResponse"]
+
+
+class AnnotationDeleteResponse(BaseModel):
+    message: Optional[str] = None