acontext 0.0.1.dev1__tar.gz → 0.0.1.dev2__tar.gz

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/PKG-INFO

@@ -1,9 +1,11 @@
 Metadata-Version: 2.3
 Name: acontext
-Version: 0.0.1.dev1
+Version: 0.0.1.dev2
 Summary: Python SDK for the Acontext API
 Keywords: acontext,sdk,client,api
 Requires-Dist: httpx>=0.28.1
+Requires-Dist: openai>=2.6.1
+Requires-Dist: anthropic>=0.72.0
 Requires-Python: >=3.10
 Project-URL: Homepage, https://github.com/memodb-io/Acontext
 Project-URL: Issues, https://github.com/memodb-io/Acontext/issues

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/pyproject.toml

@@ -1,10 +1,14 @@
 [project]
 name = "acontext"
-version = "0.0.1.dev1"
+version = "0.0.1.dev2"
 description = "Python SDK for the Acontext API"
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = ["httpx>=0.28.1"]
+dependencies = [
+    "httpx>=0.28.1",
+    "openai>=2.6.1",
+    "anthropic>=0.72.0"
+]
 keywords = ["acontext", "sdk", "client", "api"]
 
 [project.urls]

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/__init__.py

@@ -5,12 +5,14 @@ Python SDK for the Acontext API.
 from importlib import metadata as _metadata
 
 from .client import AcontextClient, FileUpload, MessagePart
+from .messages import AcontextMessage
 from .resources import BlocksAPI, DiskArtifactsAPI, DisksAPI, SessionsAPI, SpacesAPI
 
 __all__ = [
     "AcontextClient",
     "FileUpload",
     "MessagePart",
+    "AcontextMessage",
     "DisksAPI",
     "DiskArtifactsAPI",
     "BlocksAPI",

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/client.py

@@ -2,7 +2,8 @@
 High-level synchronous client for the Acontext API.
 """
 
-from typing import Any, BinaryIO, Mapping, MutableMapping
+from collections.abc import Mapping, MutableMapping
+from typing import Any, BinaryIO
 
 import httpx
 

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/client_types.py

@@ -2,7 +2,8 @@
 Common typing helpers used by resource modules to avoid circular imports.
 """
 
-from typing import Any, BinaryIO, Mapping, MutableMapping, Protocol
+from collections.abc import Mapping, MutableMapping
+from typing import Any, BinaryIO, Protocol
 
 
 class RequesterProtocol(Protocol):

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/errors.py

@@ -2,7 +2,8 @@
 Custom exceptions raised by the acontext Python client.
 """
 
-from typing import Any, Mapping, MutableMapping
+from collections.abc import Mapping, MutableMapping
+from typing import Any
 
 
 class AcontextError(Exception):

acontext-0.0.1.dev2/src/acontext/messages.py

@@ -0,0 +1,83 @@
+"""
+Support for constructing session messages.
+"""
+
+from collections.abc import Mapping, MutableMapping, Sequence
+from dataclasses import dataclass
+from typing import Any, Literal
+
+@dataclass(slots=True)
+class MessagePart:
+    """
+    Represents a single message part for ``/session/{id}/messages``.
+
+    Args:
+        type: One of ``text``, ``image``, ``audio``, ``video``, ``file``, ``tool-call``,
+            ``tool-result`` or ``data``.
+        text: Optional textual payload for ``text`` parts.
+        meta: Optional metadata dictionary accepted by the API.
+        file_field: Optional field name to use in the multipart body. When omitted the
+            client will auto-generate deterministic field names.
+    """
+
+    type: str
+    text: str | None = None
+    meta: Mapping[str, Any] | None = None
+    file_field: str | None = None
+
+    @classmethod
+    def text_part(cls, text: str, *, meta: Mapping[str, Any] | None = None) -> "MessagePart":
+        return cls(type="text", text=text, meta=meta)
+    
+    @classmethod
+    def file_field_part(cls, file_field: str, *, meta: Mapping[str, Any] | None = None) -> "MessagePart":
+        return cls(type="file", file_field=file_field, meta=meta)
+
+@dataclass(slots=True)
+class AcontextMessage:
+    """
+    Represents an Acontext-format message payload.
+    """
+
+    role: Literal["user", "assistant", "system"]
+    parts: list[MessagePart]
+    meta: MutableMapping[str, Any] | None = None
+
+
+def build_acontext_message(
+    *,
+    role: Literal["user", "assistant", "system"],
+    parts: Sequence[MessagePart | str | Mapping[str, Any]],
+    meta: Mapping[str, Any] | None = None,
+) -> AcontextMessage:
+    """
+    Construct an Acontext-format message blob and associated multipart files.
+    """
+    if role not in {"user", "assistant", "system"}:
+        raise ValueError("role must be one of {'user', 'assistant', 'system'}")
+
+    normalized_parts = [normalize_message_part(part) for part in parts]
+
+    message = AcontextMessage(
+        role=role,
+        parts=normalized_parts,
+        meta=dict(meta) if meta is not None else None,
+    )
+    return message
+
+
+def normalize_message_part(part: MessagePart | str | Mapping[str, Any]) -> MessagePart:
+    if isinstance(part, MessagePart):
+        return part
+    if isinstance(part, str):
+        return MessagePart(type="text", text=part)
+    if isinstance(part, Mapping):
+        if "type" not in part:
+            raise ValueError("mapping message parts must include a 'type'")
+        return MessagePart(
+            type=str(part["type"]),
+            text=part.get("text"),
+            meta=part.get("meta"),
+            file_field=part.get("file_field"),
+        )
+    raise TypeError("unsupported message part type")

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/resources/blocks.py

@@ -2,7 +2,8 @@
 Block endpoints.
 """
 
-from typing import Any, Mapping, MutableMapping
+from collections.abc import Mapping, MutableMapping
+from typing import Any
 
 from ..client_types import RequesterProtocol
 

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/resources/disks.py

@@ -3,7 +3,8 @@ Disk and artifact endpoints.
 """
 
 import json
-from typing import Any, BinaryIO, Mapping, MutableMapping, cast
+from collections.abc import Mapping, MutableMapping
+from typing import Any, BinaryIO, cast
 
 from ..client_types import RequesterProtocol
 from ..uploads import FileUpload, normalize_file_upload

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/resources/sessions.py

@@ -1,13 +1,18 @@
-"""
-Sessions endpoints.
-"""
+"""Sessions endpoints."""
 
 import json
-from typing import Any, Mapping, MutableMapping, Sequence
+from collections.abc import Mapping, MutableMapping
+from dataclasses import asdict
+from typing import Any, BinaryIO, Literal
 
-from .._constants import SUPPORTED_ROLES
-from ..messages import MessagePart, build_message_payload
 from ..client_types import RequesterProtocol
+from ..messages import AcontextMessage
+from ..uploads import FileUpload
+from openai.types.chat import ChatCompletionMessageParam
+from anthropic.types import MessageParam
+
+UploadPayload = FileUpload | tuple[str, BinaryIO | bytes] | tuple[str, BinaryIO | bytes, str | None]
+MessageBlob = AcontextMessage | ChatCompletionMessageParam | MessageParam
 
 
 class SessionsAPI:
@@ -59,33 +64,60 @@ class SessionsAPI:
         payload = {"space_id": space_id}
         self._requester.request("POST", f"/session/{session_id}/connect_to_space", json_data=payload)
 
+    def get_tasks(
+        self,
+        session_id: str,
+        *,
+        limit: int | None = None,
+        cursor: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {}
+        if limit is not None:
+            params["limit"] = limit
+        if cursor is not None:
+            params["cursor"] = cursor
+        return self._requester.request(
+            "GET",
+            f"/session/{session_id}/task",
+            params=params or None,
+        )
+
     def send_message(
         self,
         session_id: str,
         *,
-        role: str,
-        parts: Sequence[MessagePart | str | Mapping[str, Any]],
-        format: str | None = None,
+        blob: MessageBlob,
+        format: Literal["acontext", "openai", "anthropic"] = "acontext",
+        file_field: str | None = "",
+        file: FileUpload | None = None
     ) -> Any:
-        if role not in SUPPORTED_ROLES:
-            raise ValueError(f"role must be one of {SUPPORTED_ROLES!r}")
-        if not parts:
-            raise ValueError("parts must contain at least one entry")
+        if format not in {"acontext", "openai", "anthropic"}:
+            raise ValueError("format must be one of {'acontext', 'openai', 'anthropic'}")
+
+        payload = {
+            "format": format,
+        }
+        if format == "acontext":
+           payload["blob"] = asdict(blob)
+        else:
+           payload["blob"] = blob
 
-        payload_parts, files = build_message_payload(parts)
-        payload = {"role": role, "parts": payload_parts}
-        if format is not None:
-            payload["format"] = format
 
-        if files:
+        file_payload: dict[str, tuple[str, BinaryIO, str | None]] | None = None
+        if file:
+            # only support upload one file now
+            file_payload = {
+                file_field: file.as_httpx()
+            }
+
+        if file_payload:
             form_data = {"payload": json.dumps(payload)}
             return self._requester.request(
                 "POST",
                 f"/session/{session_id}/messages",
                 data=form_data,
-                files=files,
+                files=file_payload,
             )
-
         return self._requester.request(
             "POST",
             f"/session/{session_id}/messages",
@@ -99,7 +131,8 @@ class SessionsAPI:
         limit: int | None = None,
         cursor: str | None = None,
         with_asset_public_url: bool | None = None,
-        format: str | None = None,
+        format: Literal["acontext", "openai", "anthropic"] = "acontext",
+        time_desc: bool | None = None,
     ) -> Any:
         params: dict[str, Any] = {}
         if limit is not None:
@@ -110,4 +143,6 @@ class SessionsAPI:
             params["with_asset_public_url"] = "true" if with_asset_public_url else "false"
         if format is not None:
             params["format"] = format
+        if time_desc is not None:
+            params["time_desc"] = "true" if time_desc else "false"
         return self._requester.request("GET", f"/session/{session_id}/messages", params=params or None)

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/resources/spaces.py

@@ -2,7 +2,8 @@
 Spaces endpoints.
 """
 
-from typing import Any, Mapping, MutableMapping
+from collections.abc import Mapping, MutableMapping
+from typing import Any
 
 from ..client_types import RequesterProtocol
 

{acontext-0.0.1.dev1 → acontext-0.0.1.dev2}/src/acontext/uploads.py

@@ -4,7 +4,7 @@ Utilities for working with file uploads.
 
 import io
 from dataclasses import dataclass
-from typing import BinaryIO, Tuple
+from typing import BinaryIO
 
 
 @dataclass(slots=True)
@@ -19,7 +19,7 @@ class FileUpload:
     content: BinaryIO | bytes
     content_type: str | None = None
 
-    def as_httpx(self) -> Tuple[str, BinaryIO, str | None]:
+    def as_httpx(self) -> tuple[str, BinaryIO, str | None]:
         """
         Convert to the tuple format expected by ``httpx``.
         """

acontext-0.0.1.dev1/src/acontext/messages.py

@@ -1,94 +0,0 @@
-"""
-Support for constructing session messages.
-"""
-
-from dataclasses import dataclass
-from typing import Any, BinaryIO, Mapping, MutableMapping, Sequence, Tuple
-
-from .uploads import FileUpload, normalize_file_upload
-
-
-@dataclass(slots=True)
-class MessagePart:
-    """
-    Represents a single message part for ``/session/{id}/messages``.
-
-    Args:
-        type: One of ``text``, ``image``, ``audio``, ``video``, ``file``, ``tool-call``,
-            ``tool-result`` or ``data``.
-        text: Optional textual payload for ``text`` parts.
-        meta: Optional metadata dictionary accepted by the API.
-        file: Optional file attachment; required for binary part types.
-        file_field: Optional field name to use in the multipart body. When omitted the
-            client will auto-generate deterministic field names.
-    """
-
-    type: str
-    text: str | None = None
-    meta: Mapping[str, Any] | None = None
-    file: FileUpload | tuple[str, BinaryIO | bytes] | tuple[str, BinaryIO | bytes, str | None] | None = None
-    file_field: str | None = None
-
-    @classmethod
-    def text_part(cls, text: str, *, meta: Mapping[str, Any] | None = None) -> "MessagePart":
-        return cls(type="text", text=text, meta=meta)
-
-    @classmethod
-    def file_part(
-        cls,
-        upload: FileUpload | tuple[str, BinaryIO | bytes] | tuple[str, BinaryIO | bytes, str | None],
-        *,
-        meta: Mapping[str, Any] | None = None,
-        type: str = "file",
-    ) -> "MessagePart":
-        return cls(type=type, file=upload, meta=meta)
-
-
-def normalize_message_part(part: MessagePart | str | Mapping[str, Any]) -> MessagePart:
-    if isinstance(part, MessagePart):
-        return part
-    if isinstance(part, str):
-        return MessagePart(type="text", text=part)
-    if isinstance(part, Mapping):
-        if "type" not in part:
-            raise ValueError("mapping message parts must include a 'type'")
-        file = part.get("file")
-        normalized_file: FileUpload | tuple[str, BinaryIO | bytes] | tuple[str, BinaryIO | bytes, str | None] | None
-        if file is None:
-            normalized_file = None
-        else:
-            normalized_file = file  # type: ignore[assignment]
-        return MessagePart(
-            type=str(part["type"]),
-            text=part.get("text"),
-            meta=part.get("meta"),
-            file=normalized_file,
-            file_field=part.get("file_field"),
-        )
-    raise TypeError("unsupported message part type")
-
-
-def build_message_payload(
-    parts: Sequence[MessagePart | str | Mapping[str, Any]],
-) -> tuple[list[MutableMapping[str, Any]], dict[str, Tuple[str, BinaryIO, str | None]]]:
-    payload_parts: list[MutableMapping[str, Any]] = []
-    files: dict[str, Tuple[str, BinaryIO, str | None]] = {}
-
-    for idx, raw_part in enumerate(parts):
-        part = normalize_message_part(raw_part)
-        payload: MutableMapping[str, Any] = {"type": part.type}
-
-        if part.meta is not None:
-            payload["meta"] = dict(part.meta)
-        if part.text is not None:
-            payload["text"] = part.text
-
-        if part.file is not None:
-            upload = normalize_file_upload(part.file)
-            field_name = part.file_field or f"file_{idx}"
-            payload["file_field"] = field_name
-            files[field_name] = upload.as_httpx()
-
-        payload_parts.append(payload)
-
-    return payload_parts, files