stackgen-sdk 0.4.0rc29__tar.gz → 0.4.0rc30__tar.gz

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: stackgen-sdk
-Version: 0.4.0rc29
+Version: 0.4.0rc30
 Summary: StackGen SDK — Aiden, SRE, and Vault from one client
 Project-URL: Homepage, https://appcd-dev.github.io/stackgen-sdk/
 Project-URL: Documentation, https://appcd-dev.github.io/stackgen-sdk/

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/_http.py

@@ -228,6 +228,45 @@ def request(
             raise HttpError(f"{method} {url} -> {err}") from err
 
 
+def build_multipart_body(
+    fields: dict[str, str],
+    files: list[dict[str, Any]] | None = None,
+) -> tuple[bytes, str]:
+    """Build a ``multipart/form-data`` body by hand (stdlib has no encoder).
+
+    ``files`` entries look like ``{"field_name", "filename", "content",
+    "content_type"}`` (``content_type`` optional, defaults to
+    ``application/octet-stream``). Returns ``(body_bytes, content_type)``.
+    """
+    boundary = f"----stackgenSdk{time.time_ns():x}{os.getpid():x}"
+    crlf = b"\r\n"
+    parts: list[bytes] = []
+    for name, value in fields.items():
+        if value is None:
+            continue
+        parts.append(
+            f'--{boundary}\r\nContent-Disposition: form-data; name="{name}"\r\n\r\n{value}\r\n'.encode(
+                "utf-8"
+            )
+        )
+    for file in files or []:
+        content_type = file.get("content_type") or "application/octet-stream"
+        header = (
+            f"--{boundary}\r\n"
+            f'Content-Disposition: form-data; name="{file["field_name"]}"; '
+            f'filename="{file["filename"]}"\r\n'
+            f"Content-Type: {content_type}\r\n\r\n"
+        ).encode("utf-8")
+        content = file["content"]
+        if isinstance(content, str):
+            content = content.encode("utf-8")
+        parts.append(header)
+        parts.append(content)
+        parts.append(crlf)
+    parts.append(f"--{boundary}--\r\n".encode("utf-8"))
+    return b"".join(parts), f"multipart/form-data; boundary={boundary}"
+
+
 def as_list(payload: Any) -> list[dict[str, Any]]:
     if isinstance(payload, list):
         return [item for item in payload if isinstance(item, dict)]

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.4.0rc29'
-__version_tuple__ = version_tuple = (0, 4, 0, 'rc29')
+__version__ = version = '0.4.0rc30'
+__version_tuple__ = version_tuple = (0, 4, 0, 'rc30')
 
 __commit_id__ = commit_id = None

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/aiden/models.py

@@ -203,6 +203,65 @@ class ModelRegistry:
         )
         return {"provider": chosen, "model": model}
 
+    def test_model(
+        self,
+        *,
+        provider_name: str,
+        model_id: str,
+        variant: str | None = None,
+        max_tokens: int | None = None,
+        temperature: float | None = None,
+        reasoning_effort: str | None = None,
+    ) -> dict[str, Any]:
+        """POST /guild/api/v1/models/test — probe a config without registering it.
+
+        Runs a small generation call against the provider and returns
+        ``{"success", "message"}``, or raises ``HttpError(400)`` with the
+        provider's error on an invalid configuration.
+        """
+        body: dict[str, Any] = {"provider_name": provider_name, "model_id": model_id}
+        if variant is not None:
+            body["variant"] = variant
+        if max_tokens is not None:
+            body["max_tokens"] = max_tokens
+        if temperature is not None:
+            body["temperature"] = temperature
+        if reasoning_effort is not None:
+            body["reasoning_effort"] = reasoning_effort
+        out = self._json("POST", "/models/test", body)
+        if not isinstance(out, dict):
+            raise HttpError(f"unexpected test model payload: {out!r}")
+        return out
+
+    def delete_model(self, name: str, *, scope: str | None = None) -> None:
+        """DELETE /guild/api/v1/models/{name}.
+
+        Pass ``scope`` (``TENANT`` / ``PROJECT``) to target a non-default
+        owner level. Raises ``HttpError(404)`` when the model is missing.
+        """
+        enc = urllib.parse.quote(name, safe="")
+        params: dict[str, Any] = {"orgId": self._org()}
+        if scope:
+            params["scope"] = scope
+        url = f"{self._config.aiden_url()}/api/v1/models/{enc}?{urllib.parse.urlencode(params)}"
+        _http.request(url, self._config.api_token, method="DELETE", expect_json=False)
+
+    def delete_provider(self, name: str, *, scope: str | None = None) -> None:
+        """DELETE /guild/api/v1/model-providers/{name}.
+
+        Delete every model under the provider first — Guild 404s on an
+        unknown provider but does not cascade-delete its models.
+        """
+        enc = urllib.parse.quote(name, safe="")
+        params: dict[str, Any] = {"orgId": self._org()}
+        if scope:
+            params["scope"] = scope
+        url = (
+            f"{self._config.aiden_url()}/api/v1/model-providers/{enc}"
+            f"?{urllib.parse.urlencode(params)}"
+        )
+        _http.request(url, self._config.api_token, method="DELETE", expect_json=False)
+
     def get_model(self, name: str) -> dict[str, Any]:
         """GET /guild/api/v1/models/{name}."""
         enc = urllib.parse.quote(name, safe="")

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/aiden/namespace.py

@@ -154,6 +154,46 @@ class AidenNamespace:
         """POST /guild/api/v1/sessions/{sessionId}/terminate."""
         return self._webhook_ops.terminate_session(session_id, reason=reason)
 
+    def update_session(
+        self,
+        session_id: str,
+        *,
+        title: str | None = None,
+        starred: bool | None = None,
+        archived: bool | None = None,
+    ) -> dict[str, Any]:
+        """PATCH /guild/api/v1/sessions/{sessionId}/metadata — rename, star/unstar,
+        or archive/unarchive. Only the fields you pass change."""
+        return self._webhook_ops.update_session(
+            session_id, title=title, starred=starred, archived=archived
+        )
+
+    def download_session_attachment(
+        self, session_id: str, attachment_id: str, output_path: str
+    ) -> Path:
+        """Download a chat attachment by its stable ``attachment_id`` (the UUID
+        half of an ``attachment://<session_id>/<attachment_id>`` handle)."""
+        return self._webhook_ops.download_session_attachment(
+            session_id, attachment_id, output_path
+        )
+
+    def delete_session_attachments(self, session_id: str) -> None:
+        """Delete every stored chat attachment for a session."""
+        self._webhook_ops.delete_session_attachments(session_id)
+
+    def start_agent_chat_with_files(
+        self,
+        agent_name: str,
+        *,
+        message: str = "",
+        session_id: str | None = None,
+        files: list[dict[str, Any]],
+    ) -> dict[str, Any]:
+        """Start a chat turn with one or more file attachments (multipart)."""
+        return self._resources.start_agent_chat_with_files(
+            agent_name, message=message, session_id=session_id, files=files
+        )
+
     def run_ask_and_download_artifact(
         self,
         message: str,

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/aiden/resources.py

@@ -351,6 +351,64 @@ class AidenResources:
         )
         return record
 
+    def start_agent_chat_with_files(
+        self,
+        agent_name: str,
+        *,
+        message: str = "",
+        session_id: str | None = None,
+        files: list[dict[str, Any]],
+    ) -> dict[str, Any]:
+        """POST /agents/{name}/chat/start as ``multipart/form-data``.
+
+        ``files`` entries are ``{"filename", "content", "content_type"}``
+        (``content`` is ``bytes``; ``content_type`` optional). Send multiple
+        files by adding multiple entries — Guild accepts many ``file`` parts
+        per request (cap 10, 3 MiB each for inline agents, 10 MiB for
+        reference agents, 40 MiB total).
+
+        There is no list-attachments endpoint: the returned
+        ``attachment://<session_id>/<attachment_id>`` handle for each file
+        surfaces later in the turn's content (tool-call args, or ask the model
+        to repeat it) or the execution debug ZIP
+        (:meth:`AidenNamespace.download_execution_debug_zip`). Once you have
+        an ``attachment_id``, use :meth:`AidenNamespace.download_session_attachment`
+        / :meth:`AidenNamespace.delete_session_attachments`.
+        """
+        if not files:
+            raise ValueError("start_agent_chat_with_files requires at least one file")
+        fields: dict[str, str] = {"message": message}
+        if session_id:
+            fields["session_id"] = session_id
+        multipart_files = [
+            {
+                "field_name": "file",
+                "filename": f["filename"],
+                "content": f["content"],
+                "content_type": f.get("content_type"),
+            }
+            for f in files
+        ]
+        body, content_type = _http.build_multipart_body(fields, multipart_files)
+        data = _http.request(
+            self._url(f"/agents/{agent_name}/chat/start"),
+            self._config.api_token,
+            method="POST",
+            body=body,
+            content_type=content_type,
+            transient_retry_budget_ms=self._config.transient_retry_budget_ms,
+        )
+        record = _as_object(data, "startAgentChatWithFiles")
+        self._config.emit_session_started(
+            {
+                "kind": "chat",
+                "session_id": record.get("session_id") or record.get("sessionId"),
+                "trace_id": record.get("trace_id") or record.get("traceId"),
+                "agent": agent_name,
+            }
+        )
+        return record
+
     def list_chat_subscriptions(self, **params: Any) -> list[dict[str, Any]]:
         return _http.as_list(self._get("/platforms/chat/subscriptions", params))
 

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/aiden/webhook_ops.py

@@ -254,6 +254,80 @@ class WebhookOps:
             out.append({"name": name, "path": str(dest), "meta": meta})
         return out
 
+    def update_session(
+        self,
+        session_id: str,
+        *,
+        title: str | None = None,
+        starred: bool | None = None,
+        archived: bool | None = None,
+    ) -> dict[str, Any]:
+        """PATCH /guild/api/v1/sessions/{sessionId}/metadata.
+
+        Only the keyword args you pass are written; omit ``title`` /
+        ``starred`` / ``archived`` to leave them unchanged. Pass
+        ``title=""`` to clear a previously set title and revert to the
+        original prompt snippet. Returns the updated ``SessionSummary``.
+        """
+        patch: dict[str, Any] = {}
+        if title is not None:
+            patch["title"] = title
+        if starred is not None:
+            patch["starred"] = starred
+        if archived is not None:
+            patch["archived"] = archived
+        url = (
+            f"{self._config.aiden_url()}/api/v1/sessions/{session_id}/metadata"
+            f"{_http.query(self._config.org_id)}"
+        )
+        data = _http.request(
+            url,
+            self._config.api_token,
+            method="PATCH",
+            body=json.dumps(patch).encode("utf-8"),
+            content_type="application/json",
+        )
+        if not isinstance(data, dict):
+            raise StackgenError(f"unexpected update session payload: {data!r}")
+        return data
+
+    def download_session_attachment(
+        self,
+        session_id: str,
+        attachment_id: str,
+        output_path: str,
+    ) -> Path:
+        """GET .../sessions/{sessionId}/attachments/{attachmentId}/download.
+
+        ``attachment_id`` is the UUID half of an
+        ``attachment://<session_id>/<attachment_id>`` handle — see
+        :meth:`AidenResources.start_agent_chat_with_files`. Writes the
+        stored bytes (original content type) to ``output_path``.
+        """
+        encoded = urllib.parse.quote(attachment_id, safe="")
+        url = (
+            f"{self._config.aiden_url()}/api/v1/sessions/{session_id}/attachments/"
+            f"{encoded}/download"
+        )
+        raw = _http.request(url, self._config.api_token, expect_json=False)
+        dest = Path(output_path)
+        dest.parent.mkdir(parents=True, exist_ok=True)
+        dest.write_bytes(raw)
+        return dest
+
+    def delete_session_attachments(self, session_id: str) -> None:
+        """DELETE .../sessions/{sessionId}/attachments.
+
+        Removes every stored chat attachment (and best-effort RAG vector
+        rows) for the session. Raises ``HttpError(404)`` when no attachment
+        session exists for ``session_id``.
+        """
+        url = (
+            f"{self._config.aiden_url()}/api/v1/sessions/{session_id}/attachments"
+            f"{_http.query(self._config.org_id)}"
+        )
+        _http.request(url, self._config.api_token, method="DELETE", expect_json=False)
+
     def export_session(
         self,
         session_id: str,

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/generated/.openapi-generator/FILES

@@ -213,6 +213,8 @@ stackgen/generated/docs/SyncResponse.md
 stackgen/generated/docs/TaskErrorCategory.md
 stackgen/generated/docs/TaskStatus.md
 stackgen/generated/docs/TerminateSessionResponse.md
+stackgen/generated/docs/TestModelRequest.md
+stackgen/generated/docs/TestModelResponse.md
 stackgen/generated/docs/ToggleChatSubscriptionRequest.md
 stackgen/generated/docs/TraceDAGResponse.md
 stackgen/generated/docs/TriggerSchedule202Response.md
@@ -220,6 +222,7 @@ stackgen/generated/docs/TriggerWebhook202Response.md
 stackgen/generated/docs/UpdateAgentRequest.md
 stackgen/generated/docs/UpdateChatSubscriptionRequest.md
 stackgen/generated/docs/UpdateSecretByIdRequest.md
+stackgen/generated/docs/UpdateSessionMetadataRequest.md
 stackgen/generated/docs/UpsertIntegrationRequest.md
 stackgen/generated/docs/VaultApi.md
 stackgen/generated/docs/VersionStatus.md
@@ -442,6 +445,8 @@ stackgen/generated/models/sync_response.py
 stackgen/generated/models/task_error_category.py
 stackgen/generated/models/task_status.py
 stackgen/generated/models/terminate_session_response.py
+stackgen/generated/models/test_model_request.py
+stackgen/generated/models/test_model_response.py
 stackgen/generated/models/toggle_chat_subscription_request.py
 stackgen/generated/models/trace_dag_response.py
 stackgen/generated/models/trigger_schedule202_response.py
@@ -449,6 +454,7 @@ stackgen/generated/models/trigger_webhook202_response.py
 stackgen/generated/models/update_agent_request.py
 stackgen/generated/models/update_chat_subscription_request.py
 stackgen/generated/models/update_secret_by_id_request.py
+stackgen/generated/models/update_session_metadata_request.py
 stackgen/generated/models/upsert_integration_request.py
 stackgen/generated/models/version_status.py
 stackgen/generated/models/webhook.py
@@ -672,6 +678,8 @@ stackgen/generated/test/test_sync_response.py
 stackgen/generated/test/test_task_error_category.py
 stackgen/generated/test/test_task_status.py
 stackgen/generated/test/test_terminate_session_response.py
+stackgen/generated/test/test_test_model_request.py
+stackgen/generated/test/test_test_model_response.py
 stackgen/generated/test/test_toggle_chat_subscription_request.py
 stackgen/generated/test/test_trace_dag_response.py
 stackgen/generated/test/test_trigger_schedule202_response.py
@@ -679,6 +687,7 @@ stackgen/generated/test/test_trigger_webhook202_response.py
 stackgen/generated/test/test_update_agent_request.py
 stackgen/generated/test/test_update_chat_subscription_request.py
 stackgen/generated/test/test_update_secret_by_id_request.py
+stackgen/generated/test/test_update_session_metadata_request.py
 stackgen/generated/test/test_upsert_integration_request.py
 stackgen/generated/test/test_vault_api.py
 stackgen/generated/test/test_version_status.py

{stackgen_sdk-0.4.0rc29 → stackgen_sdk-0.4.0rc30}/src/stackgen/generated/__init__.py

@@ -233,6 +233,8 @@ __all__ = [
     "TaskErrorCategory",
     "TaskStatus",
     "TerminateSessionResponse",
+    "TestModelRequest",
+    "TestModelResponse",
     "ToggleChatSubscriptionRequest",
     "TraceDAGResponse",
     "TriggerSchedule202Response",
@@ -240,6 +242,7 @@ __all__ = [
     "UpdateAgentRequest",
     "UpdateChatSubscriptionRequest",
     "UpdateSecretByIdRequest",
+    "UpdateSessionMetadataRequest",
     "UpsertIntegrationRequest",
     "VersionStatus",
     "Webhook",
@@ -478,6 +481,8 @@ from stackgen.generated.models.sync_response import SyncResponse as SyncResponse
 from stackgen.generated.models.task_error_category import TaskErrorCategory as TaskErrorCategory
 from stackgen.generated.models.task_status import TaskStatus as TaskStatus
 from stackgen.generated.models.terminate_session_response import TerminateSessionResponse as TerminateSessionResponse
+from stackgen.generated.models.test_model_request import TestModelRequest as TestModelRequest
+from stackgen.generated.models.test_model_response import TestModelResponse as TestModelResponse
 from stackgen.generated.models.toggle_chat_subscription_request import ToggleChatSubscriptionRequest as ToggleChatSubscriptionRequest
 from stackgen.generated.models.trace_dag_response import TraceDAGResponse as TraceDAGResponse
 from stackgen.generated.models.trigger_schedule202_response import TriggerSchedule202Response as TriggerSchedule202Response
@@ -485,6 +490,7 @@ from stackgen.generated.models.trigger_webhook202_response import TriggerWebhook
 from stackgen.generated.models.update_agent_request import UpdateAgentRequest as UpdateAgentRequest
 from stackgen.generated.models.update_chat_subscription_request import UpdateChatSubscriptionRequest as UpdateChatSubscriptionRequest
 from stackgen.generated.models.update_secret_by_id_request import UpdateSecretByIdRequest as UpdateSecretByIdRequest
+from stackgen.generated.models.update_session_metadata_request import UpdateSessionMetadataRequest as UpdateSessionMetadataRequest
 from stackgen.generated.models.upsert_integration_request import UpsertIntegrationRequest as UpsertIntegrationRequest
 from stackgen.generated.models.version_status import VersionStatus as VersionStatus
 from stackgen.generated.models.webhook import Webhook as Webhook