stacklink-sdk 0.3.13__tar.gz → 0.3.15__tar.gz

{stacklink_sdk-0.3.13/stacklink_sdk.egg-info → stacklink_sdk-0.3.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stacklink-sdk
-Version: 0.3.13
+Version: 0.3.15
 Summary: Python SDK for Stacklink sessions, tools, MCP, auth, workbench, logs, knowledge, and billing.
 Author: Stacklink
 License-Expression: LicenseRef-Proprietary

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stacklink-sdk"
-version = "0.3.13"
+version = "0.3.15"
 description = "Python SDK for Stacklink sessions, tools, MCP, auth, workbench, logs, knowledge, and billing."
 readme = "README.md"
 license = "LicenseRef-Proprietary"

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/stacklink/billing.py

@@ -1,6 +1,45 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, TypedDict, cast
+
+
+class ResourceAllowanceAdjustment(TypedDict, total=False):
+    credited: int
+
+
+class ResourceUsage(ResourceAllowanceAdjustment):
+    key: str
+    label: str
+    unit: str
+    included: int
+    used: int
+    reserved: int
+    remaining: int
+    additional: int
+    additionalCents: int
+    priceMicros: str
+    unitQuantity: int
+    rateVersion: str
+
+
+class AdditionalSpending(TypedDict):
+    approvedCents: int
+    usedCents: int
+    reservedCents: int
+    availableCents: int
+    currency: str
+
+
+class UsageSummary(TypedDict, total=False):
+    agreementVersion: str
+    resources: list[ResourceUsage]
+    spending: dict[str, AdditionalSpending]
+    workspace: dict[str, Any]
+    subscription: dict[str, Any]
+    plan: dict[str, Any]
+    period: dict[str, str]
+    usage: dict[str, Any]
+    overagePending: bool
 
 
 class BillingResource:
@@ -16,8 +55,8 @@ class BillingResource:
     def subscription(self):
         return self.client.request("GET", "/v1/subscription")
 
-    def usage(self, **query):
-        return self.client.request("GET", "/v1/usage", query=query)
+    def usage(self, **query) -> UsageSummary:
+        return cast(UsageSummary, self.client.request("GET", "/v1/usage", query=query))
 
     def usage_events(self, **query):
         return self.client.request("GET", "/v1/usage/events", query=query)

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/stacklink/bots.py

@@ -54,7 +54,52 @@ class BotRequirement(TypedDict):
     nextAction: Literal["configure_bot", "repair_connection", "use_live_data", "contact_administrator", "unavailable"]
 
 
+class BotAttachment(TypedDict, total=False):
+    id: str
+    name: str
+    mediaType: str
+    sizeBytes: int
+    sha256: str | None
+    status: Literal["uploading", "uploaded", "preparing", "ready", "failed", "gone"]
+    facts: dict[str, Any]
+    warnings: list[dict[str, str]]
+    error: dict[str, str] | None
+
+
+class BotFileUpload(TypedDict):
+    fileId: str
+    uploadUrl: str
+    requiredHeaders: dict[str, str]
+    expiresAt: str
+
+
+class BotFiles:
+    """Sync and async clients share this contract; upload bytes with the returned headers."""
+    def __init__(self, client):
+        self.client = client
+
+    def _path(self, bot_id: str, file_id: str | None = None):
+        return "/v1/bots/" + quote(bot_id, safe="") + "/files" + ("/" + quote(file_id, safe="") if file_id else "")
+
+    def create_upload(self, bot_id: str, *, session_id: str, name: str, media_type: str, size_bytes: int):
+        return self.client.request("POST", self._path(bot_id), json_body={"sessionId": session_id, "name": name, "mediaType": media_type, "sizeBytes": size_bytes})
+
+    def finalize(self, bot_id: str, file_id: str, *, session_id: str):
+        return self.client.request("POST", self._path(bot_id, file_id) + "/finalize", json_body={"sessionId": session_id})
+
+    def get(self, bot_id: str, file_id: str, *, session_id: str):
+        return self.client.request("GET", self._path(bot_id, file_id), query={"sessionId": session_id})
+
+    def link(self, bot_id: str, file_id: str, *, session_id: str):
+        return self.client.request("POST", self._path(bot_id, file_id) + "/link", json_body={"sessionId": session_id})
+
+
 class BotRun(TypedDict, total=False):
+    attachments: list[BotAttachment]
+    citations: list[dict[str, Any]]
+    outputs: list[dict[str, Any]]
+    coverage: list[dict[str, Any]]
+    updates: list[dict[str, str]]
     id: str
     botId: str
     conversationId: str
@@ -115,6 +160,7 @@ def verify_bot_api_request(*, method: str, url: str, body: str, headers: dict[st
 class BotsResource:
     def __init__(self, client, *, async_mode: bool = False):
         self.client = client
+        self.files = BotFiles(client)
         self.async_mode = async_mode
 
     def _path(self, bot_id: str, run_id: str | None = None) -> str:
@@ -187,8 +233,10 @@ class BotsResource:
     def publish(self, bot_id: str, revision: int, channels: dict[str, bool]):
         return self.client.request("POST", self._path(bot_id) + "/publish", json_body={"revision": revision, "channels": channels})
 
-    def run(self, bot_id: str, *, session_id: str, message: str, variables: dict[str, Any] | None = None, conversation_id: str | None = None, run_id: str | None = None, connection_return_url: str | None = None):
+    def run(self, bot_id: str, *, session_id: str, message: str, variables: dict[str, Any] | None = None, conversation_id: str | None = None, run_id: str | None = None, connection_return_url: str | None = None, attachments: list[str] | None = None):
         body: dict[str, Any] = {"sessionId": session_id, "message": message, "variables": variables or {}, "runId": run_id or str(uuid4())}
+        if attachments is not None:
+            body["attachments"] = attachments
         if conversation_id:
             body["conversationId"] = conversation_id
         if connection_return_url is not None:

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/stacklink/client.py

@@ -37,7 +37,7 @@ from .types import Json, Transport
 from .webhooks import WebhooksResource
 from .workbench import WorkbenchResource
 
-STACKLINK_PYTHON_SDK_VERSION = "0.3.13"
+STACKLINK_PYTHON_SDK_VERSION = "0.3.15"
 STACKLINK_PYTHON_USER_AGENT = f"stacklink-python/{STACKLINK_PYTHON_SDK_VERSION}"
 
 

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/stacklink/workbench.py

@@ -68,6 +68,28 @@ class WorkbenchSession(TypedDict, total=False):
     updatedAt: str
 
 
+class WorkbenchComputeState(TypedDict, total=False):
+    status: str
+    stopped: bool
+    fundedThrough: int | None
+    activeExecutionCount: int
+    generation: int | None
+
+
+class WorkbenchRuntimeCapacity(TypedDict):
+    measurement: Literal["customer_active_ms", "legacy_operations"]
+    settlementStatus: Literal["pending", "acknowledgement_pending", "settled"]
+    reservedMilliseconds: int
+    workspace: dict[str, Any] | None
+    intervals: list[dict[str, Any]]
+
+
+class WorkbenchComputeStatus(TypedDict):
+    workbenchSession: WorkbenchSession
+    compute: WorkbenchComputeState
+    billing: WorkbenchRuntimeCapacity
+
+
 class WorkbenchSessionUseResponse(TypedDict):
     workbenchSession: WorkbenchSession
     reused: bool
@@ -1084,6 +1106,11 @@ class WorkbenchResource:
             return _wait_for_workbench_response(self.client, body, response)
         return response
 
+    def compute_status(self, session_id: str, workbench_session_id: str) -> WorkbenchComputeStatus:
+        return cast(WorkbenchComputeStatus, self.client.request("POST", "/v1/workbench/run", json_body={
+            "sessionId": session_id, "workbenchSessionId": workbench_session_id, "operation": "compute_status",
+        }))
+
     def destroy(self, session_id: str, workbench_session_id: str):
         return self.run({
             "sessionId": session_id,

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15/stacklink_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stacklink-sdk
-Version: 0.3.13
+Version: 0.3.15
 Summary: Python SDK for Stacklink sessions, tools, MCP, auth, workbench, logs, knowledge, and billing.
 Author: Stacklink
 License-Expression: LicenseRef-Proprietary

{stacklink_sdk-0.3.13 → stacklink_sdk-0.3.15}/tests/test_client.py

@@ -159,6 +159,37 @@ class StacklinkClientTests(unittest.TestCase):
             ("POST", "/v1/bots/bot-one/apis/api-one/signing-key", {"action": "activate", "keyId": "key-two"}),
         ])
 
+    def test_bot_files_preserve_customer_identity_for_sync_and_async_clients(self):
+        expected = [
+            ("POST", "/v1/bots/bot-one/files", {"sessionId": "customer-a", "name": "notes.sql", "mediaType": "text/plain", "sizeBytes": 42}, {}),
+            ("POST", "/v1/bots/bot-one/files/file-one/finalize", {"sessionId": "customer-a"}, {}),
+            ("GET", "/v1/bots/bot-one/files/file-one", None, {"sessionId": "customer-a"}),
+            ("POST", "/v1/bots/bot-one/files/file-one/link", {"sessionId": "customer-a"}, {}),
+        ]
+        sync_calls = []
+        def transport(method, path, headers, body, query, timeout):
+            sync_calls.append((method, path, body, query))
+            return 200, {}, {"ok": True}
+        client = Stacklink(api_key="test", base_url="https://stacklink.test", transport=transport)
+        client.bots.files.create_upload("bot-one", session_id="customer-a", name="notes.sql", media_type="text/plain", size_bytes=42)
+        client.bots.files.finalize("bot-one", "file-one", session_id="customer-a")
+        client.bots.files.get("bot-one", "file-one", session_id="customer-a")
+        client.bots.files.link("bot-one", "file-one", session_id="customer-a")
+        self.assertEqual(sync_calls, expected)
+        async_calls = []
+        async def handler(request):
+            async_calls.append((request.method, request.url.path, json.loads(request.content) if request.content else None, dict(request.url.params)))
+            return httpx.Response(200, json={"ok": True})
+        async def run():
+            async with httpx.AsyncClient(transport=httpx.MockTransport(handler)) as http:
+                client = AsyncStacklink(api_key="test", base_url="https://stacklink.test", http_client=http)
+                await client.bots.files.create_upload("bot-one", session_id="customer-a", name="notes.sql", media_type="text/plain", size_bytes=42)
+                await client.bots.files.finalize("bot-one", "file-one", session_id="customer-a")
+                await client.bots.files.get("bot-one", "file-one", session_id="customer-a")
+                await client.bots.files.link("bot-one", "file-one", session_id="customer-a")
+        asyncio.run(run())
+        self.assertEqual(async_calls, expected)
+
     def test_callback_signature_uses_original_body_and_expiry(self):
         import base64, hashlib, hmac
         from stacklink import verify_webhook_signature