subako 0.1.0__py3-none-any.whl

subako/__init__.py

@@ -0,0 +1,468 @@
+from subako._client import SubakoClient
+from subako._core import Archive, RequestOptions, ResourceId, TokenSource
+from subako._errors import (
+    ClientOutdatedError,
+    ConflictError,
+    ForbiddenError,
+    InternalError,
+    InvalidRequestError,
+    NotFoundError,
+    PreconditionFailedError,
+    SubakoConnectionError,
+    SubakoError,
+    SubakoTimeoutError,
+    TooManyRequestsError,
+    UnauthorizedError,
+)
+from subako._events import (
+    ApprovalState,
+    ToolCallResult,
+    TranscriptFold,
+    TranscriptMessage,
+    TranscriptToolCall,
+    is_running,
+    pending_approvals,
+    transcript,
+)
+from subako._pagination import Page
+from subako._session import ConnectionStatus, SessionConnection
+from subako._tool_client import Lifetime, Tool, ToolAnswer, ToolCall, ToolClient, ToolClientStatus
+from subako._version import VERSION
+
+# generated-exports: begin
+from subako.generated.models import (
+    ActorBody,
+    AddCredentialBody,
+    AddWorkspaceMemberBody,
+    AgentBody,
+    AgentConfigBody,
+    AgentSecurityBody,
+    AgentVersionBody,
+    AllowApprovalDecisionForm,
+    AnthropicModelBody,
+    AnthropicModelConfig,
+    AnthropicModelUsage,
+    AnthropicThinkingBody,
+    AnthropicUsageBody,
+    ApiKeyBody,
+    ApiKeyReceiptBody,
+    ApiKeySessionCreator,
+    ApprovalDecisionBody,
+    ApprovalDecisionForm,
+    ApprovalRequestedEvent,
+    ApprovalResolvedEvent,
+    AuthSchemeBody,
+    ByokModelProvider,
+    CancelOutcomeBody,
+    ChangeMemberRoleBody,
+    ChangeWorkspaceMemberRoleBody,
+    ClientCallFailureBody,
+    ClientCallResultBody,
+    ClientLeaveReasonBody,
+    ClientLeftEvent,
+    ClientLifetime,
+    ClientLifetimeBody,
+    ClientLifetimeForm,
+    ClientPingBody,
+    ClientRegisteredEvent,
+    ClientToolAckedEvent,
+    ClientToolBody,
+    ClientToolDispatchedEvent,
+    ClientToolFailedEvent,
+    ClientToolsUpdatedEvent,
+    ContentBlock,
+    ContentBlockBody,
+    CreateAgentBody,
+    CreatedSessionBody,
+    CreatedSessionReceiptBody,
+    CreateInvitationBody,
+    CreateSessionBody,
+    CreateVaultBody,
+    CreateWorkspaceBody,
+    CreateWorkspaceModelProviderBody,
+    CredentialBody,
+    CredentialCreated,
+    CredentialPayload,
+    CredentialPayloadBody,
+    CreditLotBody,
+    CreditsBody,
+    CreditSourceBody,
+    CustomEvent,
+    DenyApprovalDecisionForm,
+    ErrorBody,
+    ErrorCode,
+    ErrorDetail,
+    ErrorEvent,
+    HarnessRuntimeUsageResource,
+    InputSourceBody,
+    InvitationBody,
+    KeyPermissionSchema,
+    LatestVersionPin,
+    ListAgentsQuery,
+    ListAgentVersionsQuery,
+    ListApiKeysQuery,
+    ListCredentialsQuery,
+    ListInvitationsQuery,
+    ListMembersQuery,
+    ListModelProvidersQuery,
+    ListSessionEventsQuery,
+    ListSessionsQuery,
+    ListSkillsQuery,
+    ListSkillVersionsQuery,
+    ListUsageQuery,
+    ListVaultsQuery,
+    ListWorkspaceMembersQuery,
+    ListWorkspacesQuery,
+    ListWorkspaceUsageQuery,
+    LlmTokensUsageResource,
+    McpGrantBody,
+    MeBody,
+    MemberRemovedBody,
+    MeOrgBody,
+    MessageAssistantEvent,
+    MessageBody,
+    MessageUserEvent,
+    MeUserBody,
+    MintApiKeyBody,
+    MintedApiKeyBody,
+    ModelConfig,
+    ModelConfigBody,
+    ModelProvider,
+    ModelProviderBody,
+    ModelProviderDetailsBody,
+    ModelProviderFormatBody,
+    ModelProviderOwnerBody,
+    ModelUsage,
+    ModelUsageBody,
+    NeverClientLifetime,
+    NeverClientLifetimeForm,
+    OauthCredentialPayload,
+    OpenaiResponsesModelBody,
+    OpenaiResponsesModelConfig,
+    OpenaiResponsesModelUsage,
+    OpenaiResponsesUsageBody,
+    OrderParam,
+    OrganizationBody,
+    OrgMemberBody,
+    OrgRoleBody,
+    OrphanedApiKeyBody,
+    PageOutAgentBody,
+    PageOutAgentVersionBody,
+    PageOutApiKeyBody,
+    PageOutCredentialBody,
+    PageOutInvitationBody,
+    PageOutModelProviderBody,
+    PageOutOrgMemberBody,
+    PageOutSessionBody,
+    PageOutSkillBody,
+    PageOutSkillVersionBody,
+    PageOutUsageEventBody,
+    PageOutVaultBody,
+    PageOutWorkspaceBody,
+    PageOutWorkspaceMemberBody,
+    PinnedVersionPin,
+    PlatformModelBody,
+    PlatformModelProvider,
+    PolicyActionBody,
+    PostSessionEventBody,
+    PublishAgentVersionBody,
+    PutAgentSecurityBody,
+    ReasoningEffortBody,
+    RefreshBlockBody,
+    RegisterClientBody,
+    RegisteredClientBody,
+    RenameAgentBody,
+    RenameOrganizationBody,
+    RenameSkillBody,
+    RenameWorkspaceBody,
+    RoleBody,
+    RunCancelledEvent,
+    RunCompletedEvent,
+    RunFailedEvent,
+    RunQueuedEvent,
+    RunStartedEvent,
+    SessionBody,
+    SessionCreatedEvent,
+    SessionCreator,
+    SessionCreatorBody,
+    SessionEvent,
+    SessionEventBody,
+    SessionEventPageOut,
+    SessionStatusBody,
+    SessionTokenBody,
+    SessionTokenReceiptBody,
+    SkillBody,
+    SkillGrantBody,
+    SkillVersionBody,
+    StaticBearerCredentialPayload,
+    StreamSessionEventsQuery,
+    SubmitKindBody,
+    SubmitOutcomeBody,
+    TextContentBlock,
+    TextContentBlockData,
+    ThinkingContentBlock,
+    ThinkingContentBlockData,
+    ToolCallContentBlock,
+    ToolCallContentBlockData,
+    ToolCallEvent,
+    ToolDescriptorBody,
+    ToolResultEvent,
+    ToolRuleBody,
+    TtlClientLifetime,
+    TtlClientLifetimeData,
+    TtlClientLifetimeForm,
+    TtlClientLifetimeFormData,
+    UpdateClientToolsBody,
+    UpdateModelProviderBody,
+    UpdateVaultBody,
+    UsageEventBody,
+    UsageResource,
+    UsageResourceBody,
+    UserSessionCreator,
+    VaultBody,
+    VaultCreated,
+    VersionPin,
+    VersionPinBody,
+    WorkspaceBody,
+    WorkspaceMemberBody,
+    WorkspaceRoleBody,
+)
+
+# generated-exports: end
+
+__all__ = [
+    "VERSION",
+    "ApprovalState",
+    "Archive",
+    "ClientOutdatedError",
+    "ConflictError",
+    "ConnectionStatus",
+    "ForbiddenError",
+    "InternalError",
+    "InvalidRequestError",
+    "Lifetime",
+    "NotFoundError",
+    "Page",
+    "PreconditionFailedError",
+    "RequestOptions",
+    "ResourceId",
+    "SessionConnection",
+    "SubakoClient",
+    "SubakoConnectionError",
+    "SubakoError",
+    "SubakoTimeoutError",
+    "TokenSource",
+    "TooManyRequestsError",
+    "Tool",
+    "ToolAnswer",
+    "ToolCall",
+    "ToolCallResult",
+    "ToolClient",
+    "ToolClientStatus",
+    "TranscriptFold",
+    "TranscriptMessage",
+    "TranscriptToolCall",
+    "UnauthorizedError",
+    "is_running",
+    "pending_approvals",
+    "transcript",
+]
+# generated-exports-all: begin
+__all__ += [
+    "ActorBody",
+    "AddCredentialBody",
+    "AddWorkspaceMemberBody",
+    "AgentBody",
+    "AgentConfigBody",
+    "AgentSecurityBody",
+    "AgentVersionBody",
+    "AllowApprovalDecisionForm",
+    "AnthropicModelBody",
+    "AnthropicModelConfig",
+    "AnthropicModelUsage",
+    "AnthropicThinkingBody",
+    "AnthropicUsageBody",
+    "ApiKeyBody",
+    "ApiKeyReceiptBody",
+    "ApiKeySessionCreator",
+    "ApprovalDecisionBody",
+    "ApprovalDecisionForm",
+    "ApprovalRequestedEvent",
+    "ApprovalResolvedEvent",
+    "AuthSchemeBody",
+    "ByokModelProvider",
+    "CancelOutcomeBody",
+    "ChangeMemberRoleBody",
+    "ChangeWorkspaceMemberRoleBody",
+    "ClientCallFailureBody",
+    "ClientCallResultBody",
+    "ClientLeaveReasonBody",
+    "ClientLeftEvent",
+    "ClientLifetime",
+    "ClientLifetimeBody",
+    "ClientLifetimeForm",
+    "ClientPingBody",
+    "ClientRegisteredEvent",
+    "ClientToolAckedEvent",
+    "ClientToolBody",
+    "ClientToolDispatchedEvent",
+    "ClientToolFailedEvent",
+    "ClientToolsUpdatedEvent",
+    "ContentBlock",
+    "ContentBlockBody",
+    "CreateAgentBody",
+    "CreateInvitationBody",
+    "CreateSessionBody",
+    "CreateVaultBody",
+    "CreateWorkspaceBody",
+    "CreateWorkspaceModelProviderBody",
+    "CreatedSessionBody",
+    "CreatedSessionReceiptBody",
+    "CredentialBody",
+    "CredentialCreated",
+    "CredentialPayload",
+    "CredentialPayloadBody",
+    "CreditLotBody",
+    "CreditSourceBody",
+    "CreditsBody",
+    "CustomEvent",
+    "DenyApprovalDecisionForm",
+    "ErrorBody",
+    "ErrorCode",
+    "ErrorDetail",
+    "ErrorEvent",
+    "HarnessRuntimeUsageResource",
+    "InputSourceBody",
+    "InvitationBody",
+    "KeyPermissionSchema",
+    "LatestVersionPin",
+    "ListAgentVersionsQuery",
+    "ListAgentsQuery",
+    "ListApiKeysQuery",
+    "ListCredentialsQuery",
+    "ListInvitationsQuery",
+    "ListMembersQuery",
+    "ListModelProvidersQuery",
+    "ListSessionEventsQuery",
+    "ListSessionsQuery",
+    "ListSkillVersionsQuery",
+    "ListSkillsQuery",
+    "ListUsageQuery",
+    "ListVaultsQuery",
+    "ListWorkspaceMembersQuery",
+    "ListWorkspaceUsageQuery",
+    "ListWorkspacesQuery",
+    "LlmTokensUsageResource",
+    "McpGrantBody",
+    "MeBody",
+    "MeOrgBody",
+    "MeUserBody",
+    "MemberRemovedBody",
+    "MessageAssistantEvent",
+    "MessageBody",
+    "MessageUserEvent",
+    "MintApiKeyBody",
+    "MintedApiKeyBody",
+    "ModelConfig",
+    "ModelConfigBody",
+    "ModelProvider",
+    "ModelProviderBody",
+    "ModelProviderDetailsBody",
+    "ModelProviderFormatBody",
+    "ModelProviderOwnerBody",
+    "ModelUsage",
+    "ModelUsageBody",
+    "NeverClientLifetime",
+    "NeverClientLifetimeForm",
+    "OauthCredentialPayload",
+    "OpenaiResponsesModelBody",
+    "OpenaiResponsesModelConfig",
+    "OpenaiResponsesModelUsage",
+    "OpenaiResponsesUsageBody",
+    "OrderParam",
+    "OrgMemberBody",
+    "OrgRoleBody",
+    "OrganizationBody",
+    "OrphanedApiKeyBody",
+    "PageOutAgentBody",
+    "PageOutAgentVersionBody",
+    "PageOutApiKeyBody",
+    "PageOutCredentialBody",
+    "PageOutInvitationBody",
+    "PageOutModelProviderBody",
+    "PageOutOrgMemberBody",
+    "PageOutSessionBody",
+    "PageOutSkillBody",
+    "PageOutSkillVersionBody",
+    "PageOutUsageEventBody",
+    "PageOutVaultBody",
+    "PageOutWorkspaceBody",
+    "PageOutWorkspaceMemberBody",
+    "PinnedVersionPin",
+    "PlatformModelBody",
+    "PlatformModelProvider",
+    "PolicyActionBody",
+    "PostSessionEventBody",
+    "PublishAgentVersionBody",
+    "PutAgentSecurityBody",
+    "ReasoningEffortBody",
+    "RefreshBlockBody",
+    "RegisterClientBody",
+    "RegisteredClientBody",
+    "RenameAgentBody",
+    "RenameOrganizationBody",
+    "RenameSkillBody",
+    "RenameWorkspaceBody",
+    "RoleBody",
+    "RunCancelledEvent",
+    "RunCompletedEvent",
+    "RunFailedEvent",
+    "RunQueuedEvent",
+    "RunStartedEvent",
+    "SessionBody",
+    "SessionCreatedEvent",
+    "SessionCreator",
+    "SessionCreatorBody",
+    "SessionEvent",
+    "SessionEventBody",
+    "SessionEventPageOut",
+    "SessionStatusBody",
+    "SessionTokenBody",
+    "SessionTokenReceiptBody",
+    "SkillBody",
+    "SkillGrantBody",
+    "SkillVersionBody",
+    "StaticBearerCredentialPayload",
+    "StreamSessionEventsQuery",
+    "SubmitKindBody",
+    "SubmitOutcomeBody",
+    "TextContentBlock",
+    "TextContentBlockData",
+    "ThinkingContentBlock",
+    "ThinkingContentBlockData",
+    "ToolCallContentBlock",
+    "ToolCallContentBlockData",
+    "ToolCallEvent",
+    "ToolDescriptorBody",
+    "ToolResultEvent",
+    "ToolRuleBody",
+    "TtlClientLifetime",
+    "TtlClientLifetimeData",
+    "TtlClientLifetimeForm",
+    "TtlClientLifetimeFormData",
+    "UpdateClientToolsBody",
+    "UpdateModelProviderBody",
+    "UpdateVaultBody",
+    "UsageEventBody",
+    "UsageResource",
+    "UsageResourceBody",
+    "UserSessionCreator",
+    "VaultBody",
+    "VaultCreated",
+    "VersionPin",
+    "VersionPinBody",
+    "WorkspaceBody",
+    "WorkspaceMemberBody",
+    "WorkspaceRoleBody",
+]
+# generated-exports-all: end

subako/_client.py

@@ -0,0 +1,82 @@
+import asyncio
+from collections.abc import Mapping
+from types import TracebackType
+from typing import Self
+
+import httpx
+
+from subako._core import DEFAULT_MAX_RETRIES, DEFAULT_TIMEOUT, Core, Sleep, TokenSource
+from subako.resources.agents import Agents
+from subako.resources.api_keys import ApiKeys
+from subako.resources.me import Me
+from subako.resources.model_providers import ModelProviders
+from subako.resources.organization import Organization
+from subako.resources.sessions import Sessions
+from subako.resources.skills import Skills
+from subako.resources.vaults import Vaults
+from subako.resources.workspaces import Workspaces
+
+
+def _credential(api_key: TokenSource | None, access_token: TokenSource | None) -> TokenSource:
+    match [source for source in (api_key, access_token) if source is not None]:
+        case [""]:
+            raise TypeError("the credential given to SubakoClient is empty")
+        case [source]:
+            return source
+        case _:
+            raise TypeError("SubakoClient takes exactly one of api_key or access_token")
+
+
+class SubakoClient:
+    """One instance per credential and workspace; the resources on it share one token cache and one HTTP client."""
+
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        api_key: TokenSource | None = None,
+        access_token: TokenSource | None = None,
+        workspace_id: str | None = None,
+        timeout: float = DEFAULT_TIMEOUT,
+        max_retries: int = DEFAULT_MAX_RETRIES,
+        headers: Mapping[str, str] | None = None,
+        http_client: httpx.AsyncClient | None = None,
+        sleep: Sleep = asyncio.sleep,
+    ) -> None:
+        if base_url == "":
+            raise TypeError("SubakoClient takes a base_url, e.g. https://api.kikuvi.com")
+        credential = _credential(api_key, access_token)
+        self._owns_http = http_client is None
+        self._http = httpx.AsyncClient() if http_client is None else http_client
+        core = Core(
+            http=self._http,
+            base_url=base_url,
+            credential=credential,
+            workspace_id=workspace_id,
+            timeout=timeout,
+            max_retries=max_retries,
+            headers=headers,
+            sleep=sleep,
+        )
+        self.agents = Agents(core)
+        self.api_keys = ApiKeys(core)
+        self.me = Me(core)
+        self.model_providers = ModelProviders(core)
+        self.organization = Organization(core)
+        self.sessions = Sessions(core)
+        self.skills = Skills(core)
+        self.vaults = Vaults(core)
+        self.workspaces = Workspaces(core)
+
+    async def aclose(self) -> None:
+        """Close the HTTP client this instance created; one that was passed in stays open."""
+        if self._owns_http:
+            await self._http.aclose()
+
+    async def __aenter__(self) -> Self:
+        return self
+
+    async def __aexit__(
+        self, exc_type: type[BaseException] | None, exc: BaseException | None, traceback: TracebackType | None
+    ) -> None:
+        await self.aclose()