sofias-sdk-lite 0.3.0__tar.gz → 0.4.0__tar.gz

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sofias-sdk-lite
-Version: 0.3.0
+Version: 0.4.0
 Summary: Build agents on the Sofias agent graph and run them over RabbitMQ.
 Keywords: agents,ai,llm,rabbitmq,sofias
 Author: Sofias Tech, S.L.

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sofias-sdk-lite"
-version = "0.3.0"
+version = "0.4.0"
 description = "Build agents on the Sofias agent graph and run them over RabbitMQ."
 readme = "README.md"
 requires-python = ">=3.11"

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "sofias-sdk-lite"
-version = "0.3.0"
+version = "0.4.0"
 description = "Build agents on the Sofias agent graph and run them over RabbitMQ."
 readme = "README.md"
 requires-python = ">=3.11"

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/src/sofias_sdk_lite/__init__.py

@@ -103,6 +103,12 @@ from sofias_sdk_lite.llm import (
     ToolCall,
     ToolSpec,
 )
+from sofias_sdk_lite.files import (
+    Attachment,
+    FileAccess,
+    FileAccessError,
+    ReturnedFile,
+)
 from sofias_sdk_lite.messaging import (
     AgentTaskMessage,
     ApiTaskMessage,
@@ -171,6 +177,11 @@ except PackageNotFoundError:  # pragma: no cover - source checkout without insta
 
 __all__ = [
     "__version__",
+    # Files (0.4.0)
+    "Attachment",
+    "FileAccess",
+    "FileAccessError",
+    "ReturnedFile",
     # Agent
     "Agent",
     "AgentBuilder",

sofias_sdk_lite-0.4.0/src/sofias_sdk_lite/files.py

@@ -0,0 +1,195 @@
+"""Files for third-party agents: the turn's attachments and the files you return.
+
+Everything goes through the platform's intermediary with the turn's short-lived
+grant (the same one your connectors use). Your agent never sees storage keys or
+permanent links, and the grant expires when the turn does.
+
+The task message carries what you need::
+
+    def prepare_input(self, task, history, role):
+        for attachment in task.attachment_details:      # what the user attached
+            data = task.files.download(attachment)        # bytes
+        saved = task.files.upload("report.csv", csv_bytes, "text/csv")
+        answer = f"Here is the report.\\n\\n{saved.artifact()}"
+
+``task.files`` is ``None`` when the platform has no intermediary for your agent
+(for instance, a local run): check it before using it.
+
+Images for a vision model::
+
+    part = task.files.image_part(attachment)   # OpenAI-compatible content part
+    messages = [{"role": "user", "content": [{"type": "text", "text": "What is it?"}, part]}]
+
+Limits: 25 MB per file and 10 returned files per turn.
+"""
+
+from __future__ import annotations
+
+import base64
+import json
+from typing import Any, ClassVar
+from urllib.parse import quote
+
+import httpx
+from pydantic import BaseModel, ConfigDict, Field
+
+__all__ = [
+    "MAX_FILE_BYTES",
+    "Attachment",
+    "FileAccess",
+    "FileAccessError",
+    "ReturnedFile",
+]
+
+#: The intermediary rejects bigger files; checked here too, before sending.
+MAX_FILE_BYTES = 25 * 1024 * 1024
+
+
+class FileAccessError(RuntimeError):
+    """A download or upload the intermediary refused (status in ``status_code``)."""
+
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class Attachment(BaseModel):
+    """A file the user attached, as listed in ``task.attachment_details``."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str | None = None
+    filename: str
+    content_type: str | None = None
+    byte_size: int | None = None
+    url: str
+
+
+class ReturnedFile(BaseModel):
+    """A file you uploaded with ``FileAccess.upload``: return it with ``artifact()``."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str
+    filename: str
+    content_type: str | None = None
+    byte_size: int | None = None
+    url: str | None = None
+
+    def artifact(self) -> str:
+        """The tag that hands this file to the user. Put it in your answer."""
+        return "[ARTIFACT]\n" + json.dumps({"file_id": self.id}) + "\n[/ARTIFACT]"
+
+
+class FileAccess(BaseModel):
+    """Download the turn's attachments and upload the files you return.
+
+    Built by the SDK from the ``files`` entry of the task message; the token is
+    the turn's grant and never appears in ``repr`` or logs.
+    """
+
+    model_config = ConfigDict(extra="ignore")
+
+    url: str
+    token: str = Field(repr=False)
+    timeout: float = 60.0
+
+    #: Tests swap the HTTP transport here (``httpx.MockTransport``).
+    _transport: ClassVar[httpx.BaseTransport | None] = None
+    _async_transport: ClassVar[httpx.AsyncBaseTransport | None] = None
+
+    # -- download -----------------------------------------------------------
+
+    def download(self, attachment: Attachment | str) -> bytes:
+        """The bytes of an attachment (or of an intermediary file URL)."""
+        with httpx.Client(timeout=self.timeout, transport=self._transport) as client:
+            response = client.get(_url_of(attachment), headers=self._headers())
+        return _content(response, "download")
+
+    async def adownload(self, attachment: Attachment | str) -> bytes:
+        """Async ``download``."""
+        async with httpx.AsyncClient(
+            timeout=self.timeout, transport=self._async_transport
+        ) as client:
+            response = await client.get(_url_of(attachment), headers=self._headers())
+        return _content(response, "download")
+
+    def image_part(self, attachment: Attachment) -> dict[str, Any]:
+        """An OpenAI-compatible ``image_url`` content part for a vision model."""
+        return _image_part(attachment, self.download(attachment))
+
+    async def aimage_part(self, attachment: Attachment) -> dict[str, Any]:
+        """Async ``image_part``."""
+        return _image_part(attachment, await self.adownload(attachment))
+
+    # -- upload -------------------------------------------------------------
+
+    def upload(
+        self, filename: str, data: bytes, content_type: str = "application/octet-stream"
+    ) -> ReturnedFile:
+        """Store a file for this turn; return it to the user with ``.artifact()``."""
+        _check_size(data)
+        with httpx.Client(timeout=self.timeout, transport=self._transport) as client:
+            response = client.post(
+                self._upload_url(filename), content=data, headers=self._headers(content_type)
+            )
+        return _returned(response)
+
+    async def aupload(
+        self, filename: str, data: bytes, content_type: str = "application/octet-stream"
+    ) -> ReturnedFile:
+        """Async ``upload``."""
+        _check_size(data)
+        async with httpx.AsyncClient(
+            timeout=self.timeout, transport=self._async_transport
+        ) as client:
+            response = await client.post(
+                self._upload_url(filename), content=data, headers=self._headers(content_type)
+            )
+        return _returned(response)
+
+    # -- internals ----------------------------------------------------------
+
+    def _headers(self, content_type: str | None = None) -> dict[str, str]:
+        headers = {"Authorization": f"Bearer {self.token}"}
+        if content_type:
+            headers["Content-Type"] = content_type
+        return headers
+
+    def _upload_url(self, filename: str) -> str:
+        return f"{self.url.rstrip('/')}?filename={quote(filename)}"
+
+
+def _url_of(attachment: Attachment | str) -> str:
+    return attachment if isinstance(attachment, str) else attachment.url
+
+
+def _check_size(data: bytes) -> None:
+    if not data:
+        raise ValueError("cannot upload an empty file")
+    if len(data) > MAX_FILE_BYTES:
+        raise ValueError(f"file too large: {len(data)} bytes (limit {MAX_FILE_BYTES})")
+
+
+def _content(response: httpx.Response, action: str) -> bytes:
+    if response.status_code != 200:
+        raise FileAccessError(
+            f"file {action} failed (HTTP {response.status_code})", response.status_code
+        )
+    return response.content
+
+
+def _returned(response: httpx.Response) -> ReturnedFile:
+    if response.status_code != 201:
+        raise FileAccessError(
+            f"file upload failed (HTTP {response.status_code})", response.status_code
+        )
+    return ReturnedFile.model_validate(response.json())
+
+
+def _image_part(attachment: Attachment, data: bytes) -> dict[str, Any]:
+    content_type = attachment.content_type or "image/png"
+    if not content_type.startswith("image/"):
+        raise ValueError(f"{attachment.filename} is not an image ({content_type})")
+    encoded = base64.b64encode(data).decode("ascii")
+    return {"type": "image_url", "image_url": {"url": f"data:{content_type};base64,{encoded}"}}

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/src/sofias_sdk_lite/messaging/models.py

@@ -37,6 +37,7 @@ from pydantic import BaseModel, Field, field_validator, model_validator
 import nh3
 
 from sofias_sdk_lite.contracts.strict_contract import StrictContract
+from sofias_sdk_lite.files import Attachment, FileAccess
 from sofias_sdk_lite.observability._log import get_logger
 
 logger = get_logger("messaging.models")
@@ -105,6 +106,15 @@ class _BaseTaskMessage(BaseModel):
     available_agents: dict[str, Any] | list[str] = {}
     agent_internal_capabilities: str | None = None
 
+    # --- Files (0.4.0) ---
+    # What the user attached: download URLs, and their details. For third-party
+    # agents both point at the platform's intermediary, with the turn's grant.
+    attachments: list[str] = []
+    attachment_details: list[Attachment] = []
+    # Where to download from and upload to with the turn's grant (``None`` when
+    # the platform sends no intermediary, e.g. a local run). See ``sofias_sdk_lite.files``.
+    files: FileAccess | None = None
+
     # --- Delegation (optional) ---
     reply_to: str | None = None
     correlation_id: str | None = None

{sofias_sdk_lite-0.3.0 → sofias_sdk_lite-0.4.0}/src/sofias_sdk_lite/nodes/llm_node.py

@@ -48,6 +48,7 @@ from sofias_sdk_lite.llm import (
 )
 from sofias_sdk_lite.nodes.base_node import BaseNode, NodeType
 from sofias_sdk_lite.nodes.prompt_assembler import PromptAssembler
+from sofias_sdk_lite.observability.tool_events import emit_tool_call
 from sofias_sdk_lite.observability._log import get_logger
 
 if TYPE_CHECKING:
@@ -727,7 +728,11 @@ class LLMNode(BaseNode):
         )
 
     async def _execute_tool(self, tool_call: ToolCall) -> ToolResult:
-        """Execute a single tool call.
+        """Execute a single tool call and leave its event line (``emit_tool_call``).
+
+        Both paths -- streaming and not -- and both kinds of tool -- your own
+        and the connectors' -- come through here, so this is the one place the
+        sandbox logs learn which tools ran.
 
         Args:
             tool_call: The tool call to execute.
@@ -735,6 +740,20 @@ class LLMNode(BaseNode):
         Returns:
             ToolResult with success/failure and output/error.
         """
+        started = time.perf_counter()
+        result = await self._run_tool(tool_call)
+        known = tool_call.name in self._tools_by_name or self._tool_registry is not None
+        emit_tool_call(
+            tool_call.name,
+            ok=result.success,
+            duration_ms=(time.perf_counter() - started) * 1000,
+            node_name=self.name,
+            error_type=None if result.success else ("failed" if known else "not_found"),
+        )
+        return result
+
+    async def _run_tool(self, tool_call: ToolCall) -> ToolResult:
+        """Execute a single tool call (see ``_execute_tool``)."""
         tool_start = time.perf_counter()
         tool = self._tools_by_name.get(tool_call.name)
         if tool is None:

sofias_sdk_lite-0.4.0/src/sofias_sdk_lite/observability/tool_events.py

@@ -0,0 +1,80 @@
+"""One line per tool call, for the developer portal's sandbox logs.
+
+Every tool the model calls -- your own or a connector's -- leaves one line on
+standard output, whatever your logging configuration::
+
+    SOFIAS_EVENT {"type":"tool_call","tool":"search","server":"notion","ok":true,"duration_ms":412,...}
+
+The portal reads these lines to show which tools your agent called. They carry
+names, timings and outcome only: never the arguments or the results, which may
+hold your users' data. Set ``SOFIAS_EVENTS=off`` to silence them.
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+import os
+import sys
+import time
+from typing import Any
+
+__all__ = ["EVENT_PREFIX", "emit_tool_call"]
+
+#: Marks the lines the portal parses; everything else in the log is shown raw.
+EVENT_PREFIX = "SOFIAS_EVENT"
+
+_logger = logging.getLogger("sofias_sdk_lite.events")
+
+
+class _StdoutHandler(logging.StreamHandler):  # type: ignore[type-arg]
+    """Writes to whatever ``sys.stdout`` is at emit time (it may be swapped later)."""
+
+    @property  # type: ignore[override]
+    def stream(self) -> Any:
+        return sys.stdout
+
+    @stream.setter
+    def stream(self, _value: Any) -> None:
+        pass
+
+
+def _ensure_handler() -> None:
+    """Own stdout handler, so the line appears even without logging configured."""
+    if _logger.handlers:
+        return
+    handler = _StdoutHandler()
+    handler.setFormatter(logging.Formatter("%(message)s"))
+    _logger.addHandler(handler)
+    _logger.setLevel(logging.INFO)
+    _logger.propagate = False
+
+
+def emit_tool_call(
+    tool_name: str,
+    *,
+    ok: bool,
+    duration_ms: float,
+    node_name: str | None = None,
+    error_type: str | None = None,
+) -> None:
+    """Write the event line for one tool call (no arguments, no results)."""
+    if os.environ.get("SOFIAS_EVENTS", "").lower() in {"off", "0", "false"}:
+        return
+    server, separator, tool = tool_name.partition("__")
+    if not separator:
+        server, tool = "", tool_name
+    event: dict[str, Any] = {
+        "type": "tool_call",
+        "tool": tool,
+        "server": server or None,
+        "ok": ok,
+        "duration_ms": round(duration_ms),
+        "node": node_name,
+        "error_type": error_type,
+        "ts": round(time.time(), 3),
+    }
+    _ensure_handler()
+    _logger.info(
+        "%s %s", EVENT_PREFIX, json.dumps(event, separators=(",", ":"), ensure_ascii=False)
+    )