a3s-code 0.1.0__tar.gz → 0.2.0__tar.gz

{a3s_code-0.1.0 → a3s_code-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: a3s-code
-Version: 0.1.0
+Version: 0.2.0
 Summary: A3S Code Agent Python SDK
 Project-URL: Homepage, https://github.com/A3S-Lab/Code
 Project-URL: Documentation, https://github.com/A3S-Lab/Code#readme

{a3s_code-0.1.0 → a3s_code-0.2.0}/a3s_code/__init__.py

@@ -28,6 +28,7 @@ from .types import (
     MemoryType,
     CronJobStatus,
     CronExecutionStatus,
+    StorageType,
     # Types
     AgentInfo,
     ToolCapability,
@@ -106,6 +107,7 @@ __all__ = [
     "MemoryType",
     "CronJobStatus",
     "CronExecutionStatus",
+    "StorageType",
     # Types
     "AgentInfo",
     "ToolCapability",

{a3s_code-0.1.0 → a3s_code-0.2.0}/a3s_code/client.py

@@ -8,7 +8,7 @@ import grpc
 import json
 import os
 from pathlib import Path
-from typing import Optional, List, Dict, Iterator, Any
+from typing import Optional, List, Dict, Iterator, Any, Union
 
 from .types import (
     ProviderInfo,
@@ -17,6 +17,8 @@ from .types import (
     ModelLimitInfo,
     ModelModalitiesInfo,
     SessionConfig,
+    LLMConfig,
+    StorageType,
     Message,
     Todo,
     ConfirmationPolicy,
@@ -290,8 +292,91 @@ class A3sClient:
         config: Optional[SessionConfig] = None,
         session_id: Optional[str] = None,
         initial_context: Optional[List[Message]] = None,
+        *,
+        name: str = "",
+        workspace: str = "",
+        llm: Optional[Union[LLMConfig, Dict[str, Any]]] = None,
+        system_prompt: Optional[str] = None,
+        max_context_length: Optional[int] = None,
+        auto_compact: Optional[bool] = None,
+        storage_type: Optional[StorageType] = None,
     ) -> Dict[str, Any]:
-        """Create a new session."""
+        """Create a new session.
+
+        Accepts either a SessionConfig object or keyword arguments to build one.
+        If both are provided, keyword arguments override the config fields.
+
+        Args:
+            config: Optional SessionConfig object.
+            session_id: Optional session ID (server generates one if omitted).
+            initial_context: Optional list of messages to seed the session.
+            name: Session name.
+            workspace: Working directory for tool sandboxing. Each session gets
+                its own workspace-scoped ToolContext. If empty, falls back to
+                the server-level default workspace.
+            llm: LLM configuration (LLMConfig object or plain dict with keys:
+                provider, model, api_key, base_url, temperature, max_tokens).
+            system_prompt: Custom system prompt for the session.
+            max_context_length: Maximum context window length.
+            auto_compact: Enable automatic context compaction.
+            storage_type: Storage backend (StorageType.MEMORY or StorageType.FILE).
+
+        Returns:
+            Dict with session_id and session.
+        """
+        # Build config from kwargs when no config object is provided
+        has_kwargs = any(v for v in [name, workspace, llm, system_prompt,
+                                     max_context_length, auto_compact, storage_type])
+        if config is None and has_kwargs:
+            llm_config = None
+            if llm is not None:
+                if isinstance(llm, dict):
+                    llm_config = LLMConfig(
+                        provider=llm.get("provider", ""),
+                        model=llm.get("model", ""),
+                        api_key=llm.get("api_key"),
+                        base_url=llm.get("base_url"),
+                        temperature=llm.get("temperature"),
+                        max_tokens=llm.get("max_tokens"),
+                    )
+                else:
+                    llm_config = llm
+            config = SessionConfig(
+                name=name,
+                workspace=workspace,
+                llm=llm_config,
+                system_prompt=system_prompt,
+                max_context_length=max_context_length,
+                auto_compact=auto_compact,
+                storage_type=storage_type,
+            )
+        elif config is not None and has_kwargs:
+            # kwargs override config fields
+            if name:
+                config.name = name
+            if workspace:
+                config.workspace = workspace
+            if llm is not None:
+                if isinstance(llm, dict):
+                    config.llm = LLMConfig(
+                        provider=llm.get("provider", ""),
+                        model=llm.get("model", ""),
+                        api_key=llm.get("api_key"),
+                        base_url=llm.get("base_url"),
+                        temperature=llm.get("temperature"),
+                        max_tokens=llm.get("max_tokens"),
+                    )
+                else:
+                    config.llm = llm
+            if system_prompt is not None:
+                config.system_prompt = system_prompt
+            if max_context_length is not None:
+                config.max_context_length = max_context_length
+            if auto_compact is not None:
+                config.auto_compact = auto_compact
+            if storage_type is not None:
+                config.storage_type = storage_type
+
         request = {
             "session_id": session_id,
             "config": self._session_config_to_proto(config) if config else None,
@@ -1741,18 +1826,43 @@ class A3sClient:
             "workspace": config.workspace,
         }
         if config.llm:
-            result["llm"] = {
-                "provider": config.llm.provider,
-                "model": config.llm.model,
-                "api_key": config.llm.api_key or "",
-                "base_url": config.llm.base_url or "",
-            }
+            if isinstance(config.llm, dict):
+                llm_proto = {
+                    "provider": config.llm.get("provider", ""),
+                    "model": config.llm.get("model", ""),
+                    "api_key": config.llm.get("api_key", ""),
+                    "base_url": config.llm.get("base_url", ""),
+                }
+                if config.llm.get("temperature") is not None:
+                    llm_proto["temperature"] = config.llm["temperature"]
+                if config.llm.get("max_tokens") is not None:
+                    llm_proto["max_tokens"] = config.llm["max_tokens"]
+            else:
+                llm_proto = {
+                    "provider": config.llm.provider,
+                    "model": config.llm.model,
+                    "api_key": config.llm.api_key or "",
+                    "base_url": config.llm.base_url or "",
+                }
+                if config.llm.temperature is not None:
+                    llm_proto["temperature"] = config.llm.temperature
+                if config.llm.max_tokens is not None:
+                    llm_proto["max_tokens"] = config.llm.max_tokens
+            result["llm"] = llm_proto
         if config.system_prompt:
             result["system_prompt"] = config.system_prompt
         if config.max_context_length:
             result["max_context_length"] = config.max_context_length
         if config.auto_compact is not None:
             result["auto_compact"] = config.auto_compact
+        if config.storage_type is not None:
+            # Map enum value to proto int; use .value to normalize aliases
+            storage_value_map = {
+                "STORAGE_TYPE_UNSPECIFIED": 0,
+                "STORAGE_TYPE_MEMORY": 1,
+                "STORAGE_TYPE_FILE": 2,
+            }
+            result["storage_type"] = storage_value_map.get(config.storage_type.value, 0)
         return result
 
     def _message_to_proto(self, msg: Message) -> Dict[str, Any]:

{a3s_code-0.1.0 → a3s_code-0.2.0}/a3s_code/types.py

@@ -155,6 +155,18 @@ class ResourceLimits:
 # ============================================================================
 
 
+class StorageType(Enum):
+    """Storage backend type for session persistence."""
+    UNSPECIFIED = "STORAGE_TYPE_UNSPECIFIED"
+    MEMORY = "STORAGE_TYPE_MEMORY"
+    FILE = "STORAGE_TYPE_FILE"
+
+    # Aliases for proto-style naming (used by examples and TS SDK parity)
+    STORAGE_TYPE_UNSPECIFIED = "STORAGE_TYPE_UNSPECIFIED"
+    STORAGE_TYPE_MEMORY = "STORAGE_TYPE_MEMORY"
+    STORAGE_TYPE_FILE = "STORAGE_TYPE_FILE"
+
+
 @dataclass
 class LLMConfig:
     provider: str = ""
@@ -173,6 +185,7 @@ class SessionConfig:
     system_prompt: Optional[str] = None
     max_context_length: Optional[int] = None
     auto_compact: Optional[bool] = None
+    storage_type: Optional[StorageType] = None
 
 
 @dataclass

{a3s_code-0.1.0 → a3s_code-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "a3s-code"
-version = "0.1.0"
+version = "0.2.0"
 description = "A3S Code Agent Python SDK"
 readme = "README.md"
 license = "MIT"