autobyteus 1.2.3__py3-none-any.whl → 1.3.0__py3-none-any.whl

autobyteus/agent/bootstrap_steps/__init__.py

@@ -9,6 +9,7 @@ from .workspace_context_initialization_step import WorkspaceContextInitializatio
 # ToolInitializationStep is no longer a bootstrap step.
 from .system_prompt_processing_step import SystemPromptProcessingStep
 from .mcp_server_prewarming_step import McpServerPrewarmingStep
+from .working_context_snapshot_restore_step import WorkingContextSnapshotRestoreStep
 # LLMConfigFinalizationStep and LLMInstanceCreationStep removed.
 
 __all__ = [
@@ -16,4 +17,5 @@ __all__ = [
     "WorkspaceContextInitializationStep",
     "SystemPromptProcessingStep",
     "McpServerPrewarmingStep",
+    "WorkingContextSnapshotRestoreStep",
 ]

autobyteus/agent/bootstrap_steps/agent_bootstrapper.py

@@ -5,6 +5,7 @@ from typing import List, Optional
 from .base_bootstrap_step import BaseBootstrapStep
 from .workspace_context_initialization_step import WorkspaceContextInitializationStep
 from .system_prompt_processing_step import SystemPromptProcessingStep
+from .working_context_snapshot_restore_step import WorkingContextSnapshotRestoreStep
 from .mcp_server_prewarming_step import McpServerPrewarmingStep
 
 logger = logging.getLogger(__name__)
@@ -27,6 +28,7 @@ class AgentBootstrapper:
                 WorkspaceContextInitializationStep(),
                 McpServerPrewarmingStep(),
                 SystemPromptProcessingStep(),
+                WorkingContextSnapshotRestoreStep(),
             ]
             logger.debug("AgentBootstrapper initialized with default steps.")
         else:

autobyteus/agent/bootstrap_steps/working_context_snapshot_restore_step.py

@@ -0,0 +1,38 @@
+import logging
+from typing import TYPE_CHECKING
+
+from .base_bootstrap_step import BaseBootstrapStep
+from autobyteus.memory.restore.working_context_snapshot_bootstrapper import WorkingContextSnapshotBootstrapper
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+
+logger = logging.getLogger(__name__)
+
+
+class WorkingContextSnapshotRestoreStep(BaseBootstrapStep):
+    def __init__(self, bootstrapper: WorkingContextSnapshotBootstrapper | None = None) -> None:
+        self._bootstrapper = bootstrapper or WorkingContextSnapshotBootstrapper()
+        logger.debug("WorkingContextSnapshotRestoreStep initialized.")
+
+    async def execute(self, context: "AgentContext") -> bool:
+        restore_options = getattr(context.state, "restore_options", None)
+        if not restore_options:
+            return True
+
+        memory_manager = getattr(context.state, "memory_manager", None)
+        if not memory_manager:
+            logger.error("WorkingContextSnapshotRestoreStep requires a memory manager to restore working context snapshot.")
+            return False
+
+        system_prompt = context.state.processed_system_prompt
+        if not system_prompt:
+            llm_instance = context.llm_instance
+            system_prompt = llm_instance.config.system_message if llm_instance else ""
+
+        try:
+            self._bootstrapper.bootstrap(memory_manager, system_prompt, restore_options)
+            return True
+        except Exception as exc:  # pragma: no cover - defensive
+            logger.error("WorkingContextSnapshotRestoreStep failed: %s", exc, exc_info=True)
+            return False

autobyteus/agent/context/agent_config.py

@@ -48,7 +48,8 @@ class AgentConfig:
                  workspace: Optional['BaseAgentWorkspace'] = None,
                  lifecycle_processors: Optional[List['BaseLifecycleEventProcessor']] = None,
                  initial_custom_data: Optional[Dict[str, Any]] = None,
-                 skills: Optional[List[str]] = None):
+                 skills: Optional[List[str]] = None,
+                 memory_dir: Optional[str] = None):
         """
         Initializes the AgentConfig.
 
@@ -71,6 +72,7 @@ class AgentConfig:
             initial_custom_data: An optional dictionary of data to pre-populate
                                  the agent's runtime state `custom_data`.
             skills: An optional list of skill names or paths to be preloaded for this agent.
+            memory_dir: Optional override for the agent memory base directory.
         """
         self.name = name
         self.role = role
@@ -91,6 +93,7 @@ class AgentConfig:
         self.lifecycle_processors = lifecycle_processors or []
         self.initial_custom_data = initial_custom_data
         self.skills = skills or []
+        self.memory_dir = memory_dir
 
         # Filter out ToolManifestInjectorProcessor if in API_TOOL_CALL mode
         tool_call_format = resolve_tool_call_format()
@@ -133,6 +136,7 @@ class AgentConfig:
             lifecycle_processors=self.lifecycle_processors.copy(), # Shallow copy the list
             initial_custom_data=copy.deepcopy(self.initial_custom_data), # Deep copy for simple data
             skills=self.skills.copy(), # Shallow copy the list
+            memory_dir=self.memory_dir,
         )
 
     def __repr__(self) -> str:

autobyteus/agent/context/agent_runtime_state.py

@@ -21,6 +21,7 @@ if TYPE_CHECKING:
     from autobyteus.tools.base_tool import BaseTool 
     from autobyteus.agent.tool_invocation import ToolInvocationTurn
     from autobyteus.memory.memory_manager import MemoryManager
+    from autobyteus.memory.restore.working_context_snapshot_bootstrapper import WorkingContextSnapshotBootstrapOptions
 
 logger = logging.getLogger(__name__)
 
@@ -62,6 +63,7 @@ class AgentRuntimeState:
         # NEW: Memory manager and active turn tracking
         self.memory_manager: Optional["MemoryManager"] = None
         self.active_turn_id: Optional[str] = None
+        self.restore_options: Optional["WorkingContextSnapshotBootstrapOptions"] = None
         
         self.processed_system_prompt: Optional[str] = None
         # self.final_llm_config_for_creation removed

autobyteus/agent/factory/agent_factory.py

@@ -1,8 +1,6 @@
 # file: autobyteus/autobyteus/agent/factory/agent_factory.py
 import logging
 import random
-import os
-from pathlib import Path
 from typing import Optional, TYPE_CHECKING, Dict, List
 
 # LLMFactory is no longer needed here.
@@ -15,7 +13,9 @@ from autobyteus.agent.workspace.base_workspace import BaseAgentWorkspace
 from autobyteus.agent.handlers import *
 from autobyteus.utils.singleton import SingletonMeta
 from autobyteus.tools.base_tool import BaseTool
-from autobyteus.memory import FileMemoryStore, MemoryManager
+from autobyteus.memory import FileMemoryStore, MemoryManager, resolve_memory_base_dir
+from autobyteus.memory.store.working_context_snapshot_store import WorkingContextSnapshotStore
+from autobyteus.memory.restore.working_context_snapshot_bootstrapper import WorkingContextSnapshotBootstrapOptions
 from autobyteus.agent.input_processor.memory_ingest_input_processor import MemoryIngestInputProcessor
 from autobyteus.agent.tool_execution_result_processor.memory_ingest_tool_result_processor import (
     MemoryIngestToolResultProcessor,
@@ -105,10 +105,12 @@ class AgentFactory(metaclass=SingletonMeta):
         
         config.skills = updated_skills
 
-    def _create_runtime(self, 
-                        agent_id: str, 
-                        config: AgentConfig
-                        ) -> 'AgentRuntime': 
+    def _create_runtime_with_id(self,
+                                agent_id: str,
+                                config: AgentConfig,
+                                memory_dir_override: Optional[str] = None,
+                                restore_options: Optional[WorkingContextSnapshotBootstrapOptions] = None
+                                ) -> 'AgentRuntime':
         from autobyteus.agent.runtime.agent_runtime import AgentRuntime 
 
         # Prepare skills (resolve paths to names and register them)
@@ -122,11 +124,11 @@ class AgentFactory(metaclass=SingletonMeta):
         )
 
         # Memory manager (file-backed) initialization
-        memory_dir = os.getenv("AUTOBYTEUS_MEMORY_DIR")
-        if memory_dir is None:
-            memory_dir = str(Path.cwd() / "memory")
+        memory_dir = resolve_memory_base_dir(override_dir=memory_dir_override or config.memory_dir)
         memory_store = FileMemoryStore(base_dir=memory_dir, agent_id=agent_id)
-        runtime_state.memory_manager = MemoryManager(store=memory_store)
+        working_context_snapshot_store = WorkingContextSnapshotStore(base_dir=memory_dir, agent_id=agent_id)
+        runtime_state.memory_manager = MemoryManager(store=memory_store, working_context_snapshot_store=working_context_snapshot_store)
+        runtime_state.restore_options = restore_options
 
         # Ensure memory ingest processors are present
         if not any(isinstance(p, MemoryIngestInputProcessor) for p in config.input_processors):
@@ -146,7 +148,7 @@ class AgentFactory(metaclass=SingletonMeta):
         logger.info(f"Instantiating AgentRuntime for agent_id: '{agent_id}' with config: '{config.name}'.")
         
         return AgentRuntime(
-            context=context, 
+            context=context,
             event_handler_registry=event_handler_registry
         )
 
@@ -166,7 +168,7 @@ class AgentFactory(metaclass=SingletonMeta):
         while agent_id in self._active_agents:
             agent_id = f"{config.name}_{config.role}_{random.randint(1000, 9999)}"
 
-        runtime = self._create_runtime(
+        runtime = self._create_runtime_with_id(
             agent_id=agent_id,
             config=config,
         )
@@ -176,6 +178,29 @@ class AgentFactory(metaclass=SingletonMeta):
         logger.info(f"Agent '{agent_id}' created and stored successfully.")
         return agent
 
+    def restore_agent(
+        self,
+        agent_id: str,
+        config: AgentConfig,
+        memory_dir: Optional[str] = None,
+    ) -> Agent:
+        if not agent_id or not isinstance(agent_id, str):
+            raise ValueError("restore_agent requires a non-empty string agent_id.")
+        if agent_id in self._active_agents:
+            raise ValueError(f"Agent '{agent_id}' is already active.")
+
+        restore_options = WorkingContextSnapshotBootstrapOptions()
+        runtime = self._create_runtime_with_id(
+            agent_id=agent_id,
+            config=config,
+            memory_dir_override=memory_dir,
+            restore_options=restore_options,
+        )
+        agent = Agent(runtime=runtime)
+        self._active_agents[agent_id] = agent
+        logger.info(f"Agent '{agent_id}' restored and stored successfully.")
+        return agent
+
     def get_agent(self, agent_id: str) -> Optional[Agent]:
         """Retrieves an active agent instance by its ID."""
         return self._active_agents.get(agent_id)

autobyteus/agent/input_processor/memory_ingest_input_processor.py

@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING
 
 from autobyteus.agent.input_processor.base_user_input_processor import BaseAgentUserInputMessageProcessor
 from autobyteus.agent.message.multimodal_message_builder import build_llm_user_message
+from autobyteus.agent.sender_type import SenderType
 
 if TYPE_CHECKING:
     from autobyteus.agent.message.agent_input_user_message import AgentInputUserMessage
@@ -26,6 +27,9 @@ class MemoryIngestInputProcessor(BaseAgentUserInputMessageProcessor):
         memory_manager = getattr(context.state, "memory_manager", None)
         if not memory_manager:
             return message
+        if message.sender_type == SenderType.TOOL:
+            logger.debug("MemoryIngestInputProcessor skipping TOOL-originated message to avoid duplicate tool results.")
+            return message
 
         turn_id = memory_manager.start_turn()
         context.state.active_turn_id = turn_id

autobyteus/agent/llm_request_assembler.py

@@ -60,12 +60,12 @@ class LLMRequestAssembler:
                     bundle=bundle,
                     raw_tail=raw_tail,
                 )
-                self.memory_manager.reset_transcript(snapshot_messages)
+                self.memory_manager.reset_working_context_snapshot(snapshot_messages)
                 self.memory_manager.clear_compaction_request()
                 did_compact = True
 
-        self.memory_manager.active_transcript.append_message(user_message)
-        final_messages = self.memory_manager.get_transcript_messages()
+        self.memory_manager.working_context_snapshot.append_message(user_message)
+        final_messages = self.memory_manager.get_working_context_messages()
         rendered_payload = await self.render_payload(final_messages)
 
         return RequestPackage(
@@ -91,8 +91,8 @@ class LLMRequestAssembler:
     def _ensure_system_prompt(self, system_prompt: Optional[str]) -> None:
         if not system_prompt:
             return
-        existing = self.memory_manager.get_transcript_messages()
+        existing = self.memory_manager.get_working_context_messages()
         if not existing:
-            self.memory_manager.active_transcript.append_message(
+            self.memory_manager.working_context_snapshot.append_message(
                 Message(role=MessageRole.SYSTEM, content=system_prompt)
             )

autobyteus/agent/streaming/segments/segment_events.py

@@ -17,6 +17,7 @@ class SegmentType(str, Enum):
     PATCH_FILE = "patch_file"
     RUN_BASH = "run_bash"
     REASONING = "reasoning"
+    MEDIA = "media"
 
 
 class SegmentEventType(str, Enum):

autobyteus/memory/__init__.py

@@ -12,6 +12,7 @@ from autobyteus.memory.compaction.compaction_result import CompactionResult
 from autobyteus.memory.compaction.summarizer import Summarizer
 from autobyteus.memory.retrieval.memory_bundle import MemoryBundle
 from autobyteus.memory.retrieval.retriever import Retriever
+from autobyteus.memory.path_resolver import resolve_memory_base_dir, resolve_agent_memory_dir
 
 __all__ = [
     "MemoryType",
@@ -29,4 +30,6 @@ __all__ = [
     "Summarizer",
     "MemoryBundle",
     "Retriever",
+    "resolve_memory_base_dir",
+    "resolve_agent_memory_dir",
 ]

autobyteus/memory/memory_manager.py

@@ -13,8 +13,10 @@ from autobyteus.memory.compaction.compactor import Compactor
 from autobyteus.memory.retrieval.retriever import Retriever
 from autobyteus.memory.store.base_store import MemoryStore
 from autobyteus.memory.turn_tracker import TurnTracker
-from autobyteus.memory.active_transcript import ActiveTranscript
+from autobyteus.memory.working_context_snapshot import WorkingContextSnapshot
 from autobyteus.memory.tool_interaction_builder import build_tool_interactions
+from autobyteus.memory.working_context_snapshot_serializer import WorkingContextSnapshotSerializer
+from autobyteus.memory.store.working_context_snapshot_store import WorkingContextSnapshotStore
 
 
 class MemoryManager:
@@ -25,7 +27,8 @@ class MemoryManager:
         compaction_policy: Optional[CompactionPolicy] = None,
         compactor: Optional[Compactor] = None,
         retriever: Optional[Retriever] = None,
-        active_transcript: Optional[ActiveTranscript] = None,
+        working_context_snapshot: Optional[WorkingContextSnapshot] = None,
+        working_context_snapshot_store: Optional[WorkingContextSnapshotStore] = None,
     ):
         self.store = store
         self.turn_tracker = turn_tracker or TurnTracker()
@@ -34,8 +37,9 @@ class MemoryManager:
         self.retriever = retriever or Retriever(store=store)
         self.memory_types = MemoryType
         self._seq_by_turn: dict[str, int] = {}
-        self.active_transcript = active_transcript or ActiveTranscript()
+        self.working_context_snapshot = working_context_snapshot or WorkingContextSnapshot()
         self.compaction_required: bool = False
+        self.working_context_snapshot_store = working_context_snapshot_store
 
     def start_turn(self) -> str:
         return self.turn_tracker.next_turn_id()
@@ -86,7 +90,7 @@ class MemoryManager:
             tool_args=tool_invocation.arguments,
         )
         self.store.add([trace])
-        self.active_transcript.append_tool_calls(
+        self.working_context_snapshot.append_tool_calls(
             [ToolCallSpec(id=tool_invocation.id, name=tool_invocation.name, arguments=tool_invocation.arguments)]
         )
 
@@ -109,7 +113,7 @@ class MemoryManager:
             tool_error=event.error,
         )
         self.store.add([trace])
-        self.active_transcript.append_tool_result(
+        self.working_context_snapshot.append_tool_result(
             tool_call_id=event.tool_invocation_id or "",
             tool_name=event.tool_name,
             tool_result=event.result,
@@ -129,10 +133,11 @@ class MemoryManager:
         )
         self.store.add([trace])
         if response.content or response.reasoning:
-            self.active_transcript.append_assistant(
+            self.working_context_snapshot.append_assistant(
                 content=response.content,
                 reasoning=response.reasoning,
             )
+        self.persist_working_context_snapshot()
 
     def _get_raw_tail(self, tail_turns: int, exclude_turn_id: Optional[str] = None) -> List[RawTraceItem]:
         raw_items = self.store.list(MemoryType.RAW_TRACE)
@@ -167,11 +172,28 @@ class MemoryManager:
     def get_raw_tail(self, tail_turns: int, exclude_turn_id: Optional[str] = None) -> List[RawTraceItem]:
         return self._get_raw_tail(tail_turns, exclude_turn_id=exclude_turn_id)
 
-    def get_transcript_messages(self):
-        return self.active_transcript.build_messages()
+    def get_working_context_messages(self):
+        return self.working_context_snapshot.build_messages()
+
+    def reset_working_context_snapshot(self, snapshot_messages):
+        self.working_context_snapshot.reset(snapshot_messages)
+        self.persist_working_context_snapshot()
+
+    def persist_working_context_snapshot(self) -> None:
+        if not self.working_context_snapshot_store:
+            return
+        agent_id = getattr(self.working_context_snapshot_store, "agent_id", None) or getattr(self.store, "agent_id", None)
+        if not agent_id:
+            return
+        metadata = {
+            "schema_version": 1,
+            "agent_id": agent_id,
+            "epoch_id": self.working_context_snapshot.epoch_id,
+            "last_compaction_ts": self.working_context_snapshot.last_compaction_ts,
+        }
+        payload = WorkingContextSnapshotSerializer.serialize(self.working_context_snapshot, metadata)
+        self.working_context_snapshot_store.write(agent_id, payload)
 
-    def reset_transcript(self, snapshot_messages):
-        self.active_transcript.reset(snapshot_messages)
 
     def get_tool_interactions(self, turn_id: Optional[str] = None):
         raw_items = self.store.list(MemoryType.RAW_TRACE)

autobyteus/memory/path_resolver.py

@@ -0,0 +1,27 @@
+import os
+from pathlib import Path
+from typing import Mapping, Optional, Union
+
+
+def resolve_memory_base_dir(
+    override_dir: Optional[str] = None,
+    env: Optional[Mapping[str, str]] = None,
+    fallback_dir: Optional[Union[str, Path]] = None,
+) -> str:
+    override_value = override_dir.strip() if override_dir else ""
+    if override_value:
+        return override_value
+
+    env_values = env if env is not None else os.environ
+    env_value = env_values.get("AUTOBYTEUS_MEMORY_DIR", "").strip()
+    if env_value:
+        return env_value
+
+    if fallback_dir is not None:
+        return str(fallback_dir)
+
+    return str(Path.cwd() / "memory")
+
+
+def resolve_agent_memory_dir(base_dir: Union[str, Path], agent_id: str) -> str:
+    return str(Path(base_dir) / "agents" / agent_id)

autobyteus/memory/restore/__init__.py

@@ -0,0 +1 @@
+# restore package

autobyteus/memory/restore/working_context_snapshot_bootstrapper.py

@@ -0,0 +1,61 @@
+from dataclasses import dataclass
+from typing import Optional
+
+from autobyteus.memory.working_context_snapshot_serializer import WorkingContextSnapshotSerializer
+from autobyteus.memory.compaction_snapshot_builder import CompactionSnapshotBuilder
+from autobyteus.memory.store.working_context_snapshot_store import WorkingContextSnapshotStore
+
+
+@dataclass
+class WorkingContextSnapshotBootstrapOptions:
+    max_episodic: int = 3
+    max_semantic: int = 20
+    raw_tail_turns: Optional[int] = None
+
+
+class WorkingContextSnapshotBootstrapper:
+    def __init__(
+        self,
+        working_context_snapshot_store: Optional[WorkingContextSnapshotStore] = None,
+        snapshot_builder: Optional[CompactionSnapshotBuilder] = None,
+    ) -> None:
+        self.working_context_snapshot_store = working_context_snapshot_store
+        self.snapshot_builder = snapshot_builder or CompactionSnapshotBuilder()
+
+    def bootstrap(self, memory_manager, system_prompt: str, options: WorkingContextSnapshotBootstrapOptions) -> None:
+        store = self._resolve_store(memory_manager)
+        agent_id = self._resolve_agent_id(memory_manager, store)
+
+        if store and agent_id and store.exists(agent_id):
+            payload = store.read(agent_id)
+            if payload and WorkingContextSnapshotSerializer.validate(payload):
+                snapshot, _meta = WorkingContextSnapshotSerializer.deserialize(payload)
+                memory_manager.reset_working_context_snapshot(snapshot.build_messages())
+                return
+
+        bundle = memory_manager.retriever.retrieve(
+            max_episodic=options.max_episodic,
+            max_semantic=options.max_semantic,
+        )
+        tail_turns = options.raw_tail_turns
+        if tail_turns is None:
+            policy = getattr(memory_manager, "compaction_policy", None)
+            tail_turns = getattr(policy, "raw_tail_turns", 0) if policy else 0
+        raw_tail = memory_manager.get_raw_tail(tail_turns or 0, exclude_turn_id=None)
+        snapshot_messages = self.snapshot_builder.build(
+            system_prompt=system_prompt,
+            bundle=bundle,
+            raw_tail=raw_tail,
+        )
+        memory_manager.reset_working_context_snapshot(snapshot_messages)
+
+    def _resolve_store(self, memory_manager) -> Optional[WorkingContextSnapshotStore]:
+        if self.working_context_snapshot_store is not None:
+            return self.working_context_snapshot_store
+        return getattr(memory_manager, "working_context_snapshot_store", None)
+
+    def _resolve_agent_id(self, memory_manager, store: Optional[WorkingContextSnapshotStore]) -> Optional[str]:
+        if store and getattr(store, "agent_id", None):
+            return store.agent_id
+        store_obj = getattr(memory_manager, "store", None)
+        return getattr(store_obj, "agent_id", None)

autobyteus/memory/store/__init__.py

@@ -1,7 +1,9 @@
 from autobyteus.memory.store.base_store import MemoryStore
 from autobyteus.memory.store.file_store import FileMemoryStore
+from autobyteus.memory.store.working_context_snapshot_store import WorkingContextSnapshotStore
 
 __all__ = [
     "MemoryStore",
     "FileMemoryStore",
+    "WorkingContextSnapshotStore",
 ]

autobyteus/memory/store/working_context_snapshot_store.py

@@ -0,0 +1,28 @@
+import json
+from pathlib import Path
+from typing import Optional, Union, Dict, Any
+
+
+class WorkingContextSnapshotStore:
+    def __init__(self, base_dir: Union[str, Path], agent_id: str) -> None:
+        self.base_dir = Path(base_dir)
+        self.agent_id = agent_id
+
+    def exists(self, agent_id: str) -> bool:
+        return self._get_path(agent_id).exists()
+
+    def read(self, agent_id: str) -> Optional[Dict[str, Any]]:
+        path = self._get_path(agent_id)
+        if not path.exists():
+            return None
+        with path.open("r", encoding="utf-8") as handle:
+            return json.load(handle)
+
+    def write(self, agent_id: str, payload: Dict[str, Any]) -> None:
+        path = self._get_path(agent_id)
+        path.parent.mkdir(parents=True, exist_ok=True)
+        with path.open("w", encoding="utf-8") as handle:
+            json.dump(payload, handle)
+
+    def _get_path(self, agent_id: str) -> Path:
+        return self.base_dir / "agents" / agent_id / "working_context_snapshot.json"

autobyteus/memory/{active_transcript.py → working_context_snapshot.py}

@@ -10,7 +10,7 @@ from autobyteus.llm.utils.messages import (
 )
 
 
-class ActiveTranscript:
+class WorkingContextSnapshot:
     def __init__(self, initial_messages: Optional[Iterable[Message]] = None):
         self._messages: List[Message] = list(initial_messages) if initial_messages else []
         self.epoch_id: int = 1

autobyteus/memory/working_context_snapshot_serializer.py

@@ -0,0 +1,135 @@
+import json
+from typing import Any, Dict, Iterable, List, Optional, Tuple
+
+from autobyteus.llm.utils.messages import (
+    Message,
+    MessageRole,
+    ToolCallPayload,
+    ToolCallSpec,
+    ToolResultPayload,
+)
+from autobyteus.memory.working_context_snapshot import WorkingContextSnapshot
+
+
+class WorkingContextSnapshotSerializer:
+    @staticmethod
+    def serialize(working_context_snapshot: WorkingContextSnapshot, metadata: Dict[str, Any]) -> Dict[str, Any]:
+        payload = {
+            "schema_version": metadata.get("schema_version", 1),
+            "agent_id": metadata.get("agent_id"),
+            "epoch_id": metadata.get("epoch_id", working_context_snapshot.epoch_id),
+            "last_compaction_ts": metadata.get("last_compaction_ts", working_context_snapshot.last_compaction_ts),
+            "messages": [WorkingContextSnapshotSerializer._serialize_message(msg) for msg in working_context_snapshot.build_messages()],
+        }
+        return payload
+
+    @staticmethod
+    def deserialize(payload: Dict[str, Any]) -> Tuple[WorkingContextSnapshot, Dict[str, Any]]:
+        messages = [
+            WorkingContextSnapshotSerializer._deserialize_message(msg)
+            for msg in payload.get("messages", [])
+            if isinstance(msg, dict)
+        ]
+        snapshot = WorkingContextSnapshot(initial_messages=messages)
+        metadata = {
+            "schema_version": payload.get("schema_version"),
+            "agent_id": payload.get("agent_id"),
+            "epoch_id": payload.get("epoch_id"),
+            "last_compaction_ts": payload.get("last_compaction_ts"),
+        }
+        if isinstance(metadata["epoch_id"], int):
+            snapshot.epoch_id = metadata["epoch_id"]
+        if metadata["last_compaction_ts"] is not None:
+            snapshot.last_compaction_ts = metadata["last_compaction_ts"]
+        return snapshot, metadata
+
+    @staticmethod
+    def validate(payload: Dict[str, Any]) -> bool:
+        if not isinstance(payload, dict):
+            return False
+        if not isinstance(payload.get("schema_version"), int):
+            return False
+        if not isinstance(payload.get("agent_id"), str):
+            return False
+        messages = payload.get("messages")
+        if not isinstance(messages, list):
+            return False
+        for msg in messages:
+            if not isinstance(msg, dict):
+                return False
+            if not isinstance(msg.get("role"), str):
+                return False
+        return True
+
+    @staticmethod
+    def _serialize_message(message: Message) -> Dict[str, Any]:
+        base = message.to_dict()
+        if base.get("tool_payload"):
+            base["tool_payload"] = WorkingContextSnapshotSerializer._normalize_tool_payload(base["tool_payload"])
+        return base
+
+    @staticmethod
+    def _deserialize_message(data: Dict[str, Any]) -> Message:
+        role = MessageRole(data.get("role"))
+        tool_payload = WorkingContextSnapshotSerializer._deserialize_tool_payload(data.get("tool_payload"))
+        return Message(
+            role=role,
+            content=data.get("content"),
+            reasoning_content=data.get("reasoning_content"),
+            image_urls=data.get("image_urls") or [],
+            audio_urls=data.get("audio_urls") or [],
+            video_urls=data.get("video_urls") or [],
+            tool_payload=tool_payload,
+        )
+
+    @staticmethod
+    def _normalize_tool_payload(payload: Dict[str, Any]) -> Dict[str, Any]:
+        if "tool_calls" in payload:
+            return {
+                "tool_calls": [
+                    {
+                        "id": call.get("id"),
+                        "name": call.get("name"),
+                        "arguments": WorkingContextSnapshotSerializer._safe_json_value(call.get("arguments")),
+                    }
+                    for call in payload.get("tool_calls", [])
+                ]
+            }
+        return {
+            "tool_call_id": payload.get("tool_call_id"),
+            "tool_name": payload.get("tool_name"),
+            "tool_result": WorkingContextSnapshotSerializer._safe_json_value(payload.get("tool_result")),
+            "tool_error": payload.get("tool_error"),
+        }
+
+    @staticmethod
+    def _deserialize_tool_payload(payload: Optional[Dict[str, Any]]) -> Optional[Any]:
+        if not payload:
+            return None
+        if "tool_calls" in payload:
+            calls = []
+            for call in payload.get("tool_calls", []) or []:
+                calls.append(
+                    ToolCallSpec(
+                        id=str(call.get("id")),
+                        name=str(call.get("name")),
+                        arguments=call.get("arguments") or {},
+                    )
+                )
+            return ToolCallPayload(tool_calls=calls)
+        if "tool_call_id" in payload:
+            return ToolResultPayload(
+                tool_call_id=str(payload.get("tool_call_id")),
+                tool_name=str(payload.get("tool_name")),
+                tool_result=payload.get("tool_result"),
+                tool_error=payload.get("tool_error"),
+            )
+        return None
+
+    @staticmethod
+    def _safe_json_value(value: Any) -> Any:
+        try:
+            json.dumps(value)
+            return value
+        except TypeError:
+            return str(value)

autobyteus/multimedia/audio/api/gemini_audio_client.py

@@ -96,8 +96,9 @@ class GeminiAudioClient(BaseAudioClient):
     An audio client that uses Google's Gemini models for audio tasks.
 
     **Setup Requirements:**
-    1.  **AI Studio Mode:** Set `GEMINI_API_KEY`.
+    1.  **Vertex AI Express Mode:** Set `VERTEX_AI_API_KEY`.
     2.  **Vertex AI Mode:** Set `VERTEX_AI_PROJECT` and `VERTEX_AI_LOCATION`.
+    3.  **AI Studio Mode:** Set `GEMINI_API_KEY`.
     """
 
     def __init__(self, model: "AudioModel", config: "MultimediaConfig"):

autobyteus/multimedia/image/api/gemini_image_client.py

@@ -20,8 +20,9 @@ class GeminiImageClient(BaseImageClient):
     An image client that uses Google's Gemini models for image generation tasks.
 
     **Setup Requirements:**
-    1.  **AI Studio Mode:** Set `GEMINI_API_KEY`.
+    1.  **Vertex AI Express Mode:** Set `VERTEX_AI_API_KEY`.
     2.  **Vertex AI Mode:** Set `VERTEX_AI_PROJECT` and `VERTEX_AI_LOCATION`.
+    3.  **AI Studio Mode:** Set `GEMINI_API_KEY`.
     """
 
     def __init__(self, model: "ImageModel", config: "MultimediaConfig"):

autobyteus/utils/gemini_helper.py

@@ -15,11 +15,12 @@ class GeminiRuntimeInfo:
 def initialize_gemini_client_with_runtime() -> tuple[genai.Client, GeminiRuntimeInfo]:
     """
     Initializes the Google GenAI Client based on available environment variables.
-    Supports both Vertex AI (GCP) and AI Studio (API Key) modes.
+    Supports Vertex AI (GCP), Vertex AI Express (API Key), and AI Studio (API Key) modes.
 
     Priority:
-    1. Vertex AI (requires VERTEX_AI_PROJECT and VERTEX_AI_LOCATION)
-    2. AI Studio (requires GEMINI_API_KEY)
+    1. Vertex AI Express (requires VERTEX_AI_API_KEY)
+    2. Vertex AI (requires VERTEX_AI_PROJECT and VERTEX_AI_LOCATION)
+    3. AI Studio (requires GEMINI_API_KEY)
 
     Returns:
         (client, runtime_info)
@@ -27,7 +28,14 @@ def initialize_gemini_client_with_runtime() -> tuple[genai.Client, GeminiRuntime
     Raises:
         ValueError: If neither configuration set is found.
     """
-    # 1. Try Vertex AI Configuration
+    # 1. Try Vertex AI Express Configuration (API Key)
+    vertex_api_key = os.environ.get("VERTEX_AI_API_KEY")
+    if vertex_api_key:
+        logger.info("Initializing Gemini Client in Vertex AI Express mode (API key).")
+        client = genai.Client(vertexai=True, api_key=vertex_api_key)
+        return client, GeminiRuntimeInfo(runtime="vertex", project=None, location=None)
+
+    # 2. Try Vertex AI Configuration (Project/Location + ADC)
     project = os.environ.get("VERTEX_AI_PROJECT")
     location = os.environ.get("VERTEX_AI_LOCATION")
 
@@ -38,7 +46,7 @@ def initialize_gemini_client_with_runtime() -> tuple[genai.Client, GeminiRuntime
         client = genai.Client(vertexai=True, project=project, location=location)
         return client, GeminiRuntimeInfo(runtime="vertex", project=project, location=location)
 
-    # 2. Try AI Studio Configuration (API Key)
+    # 3. Try AI Studio Configuration (API Key)
     api_key = os.environ.get("GEMINI_API_KEY")
     if api_key:
         logger.info("Initializing Gemini Client in AI Studio mode.")
@@ -48,9 +56,9 @@ def initialize_gemini_client_with_runtime() -> tuple[genai.Client, GeminiRuntime
     # 3. Fallback / Error
     error_msg = (
         "Failed to initialize Gemini Client: Missing configuration. "
-        "Please set 'GEMINI_API_KEY' for AI Studio mode, OR set both "
-        "'VERTEX_AI_PROJECT' and 'VERTEX_AI_LOCATION' for Vertex AI mode."
+        "Please set 'VERTEX_AI_API_KEY' for Vertex AI Express mode, OR set both "
+        "'VERTEX_AI_PROJECT' and 'VERTEX_AI_LOCATION' for Vertex AI mode, OR set "
+        "'GEMINI_API_KEY' for AI Studio mode."
     )
     logger.error(error_msg)
     raise ValueError(error_msg)
-

{autobyteus-1.2.3.dist-info → autobyteus-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autobyteus
-Version: 1.2.3
+Version: 1.3.0
 Summary: Multi-Agent framework
 Author-email: Ryan Zheng <ryan.zheng.work@gmail.com>
 License: MIT License with Additional Terms for Commercial Use
@@ -34,7 +34,7 @@ Requires-Dist: botocore
 Requires-Dist: certifi==2025.4.26
 Requires-Dist: cryptography
 Requires-Dist: google-api-python-client
-Requires-Dist: google-genai==1.52.0
+Requires-Dist: google-genai
 Requires-Dist: httpx
 Requires-Dist: Jinja2
 Requires-Dist: mcp[cli]

{autobyteus-1.2.3.dist-info → autobyteus-1.3.0.dist-info}/RECORD

@@ -3,22 +3,23 @@ autobyteus/check_requirements.py,sha256=nLWmqMlGiAToQuub68IpL2EhRxFZ1CyCwRCMi2C5
 autobyteus/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autobyteus/agent/agent.py,sha256=-xTkRSXjW1CChqSkgdlYrGEBok-4wjJhnM4f-7_faJ0,5211
 autobyteus/agent/exceptions.py,sha256=tfXvey5SkT70X6kcu29o8YO91KB0hI9_uLrba91_G2s,237
-autobyteus/agent/llm_request_assembler.py,sha256=X4nJqsm1nCOpFPKE10-Xs6xzu50YiUxNykOQ08UstV4,3946
+autobyteus/agent/llm_request_assembler.py,sha256=RB-mBAIE02uNa3kAH-ejlP56M7WZUjoLj7q2SP9omaY,3984
 autobyteus/agent/processor_option.py,sha256=kQhrp2AfwTWMVYPwq0Dg_vvlPDGBdcmO12IecCw4rEI,472
 autobyteus/agent/sender_type.py,sha256=kCtm9BnJoXv3Ab6wBOToFliqGHQy9HEkIhP0GsXhyZI,641
 autobyteus/agent/token_budget.py,sha256=_enFEGTDjVif0l99ztqqLHnH6NzAOmBq8jj8J5xR9x8,1786
 autobyteus/agent/tool_invocation.py,sha256=D9HT0spLLmstzT-gtGCMK9BANdUywL9oDd6aR1Ilg70,2093
-autobyteus/agent/bootstrap_steps/__init__.py,sha256=mkOXVVENq5XJTAP6hgNo6UKXLmiBmo4kjU7MVX4hgD0,749
-autobyteus/agent/bootstrap_steps/agent_bootstrapper.py,sha256=yCqILTIHLojIgDBmt13cUmkwftvATh8buxrHibl1qQQ,1346
+autobyteus/agent/bootstrap_steps/__init__.py,sha256=Mg3Nu11TmnpBYTUhCPJBlhqyo0mmwyrt-rBrWFGCHL8,875
+autobyteus/agent/bootstrap_steps/agent_bootstrapper.py,sha256=zsGiFhnULL6i-vjORdU7bWSoF7R2I1ygzosrzlrT5UU,1484
 autobyteus/agent/bootstrap_steps/base_bootstrap_step.py,sha256=7tZVk1S4IqWAxs25eNxRA4wIElEHSFz6-rlIT39ejxs,1147
 autobyteus/agent/bootstrap_steps/mcp_server_prewarming_step.py,sha256=MK-WXZ2966TrVAtmAbMdLLAsAlMryoEW2SunRPpP3BI,3316
 autobyteus/agent/bootstrap_steps/system_prompt_processing_step.py,sha256=r7NRlX7SnVjHO5YgJ8GQ-AkVkqva2CBOhDxafpUMsrM,5933
+autobyteus/agent/bootstrap_steps/working_context_snapshot_restore_step.py,sha256=YXYBwbdRGLF27E9PvbTgiOUqZHfcIimpTYPhOepNx7k,1618
 autobyteus/agent/bootstrap_steps/workspace_context_initialization_step.py,sha256=csrBDoUu5qYfYRI5xCJkto0ifluVMRd9gkv1s5nr7BI,1997
 autobyteus/agent/context/__init__.py,sha256=1an2L4sKJ1tYbJKAhCLSw-oYzt5_lmUwh1SYClA5c3M,447
-autobyteus/agent/context/agent_config.py,sha256=EtntGBBgECwlkKRsCANOwyWckzlpKJHOSCqJg3AFBBY,7755
+autobyteus/agent/context/agent_config.py,sha256=lRoSCtXbrRGVVXqXpFOt71kv9QP_tNNwDYxqZv4-bJc,7962
 autobyteus/agent/context/agent_context.py,sha256=j9h7ytxlNvYCZSsdmRBGMSyfw-D7zYnuKlctfyn9nAU,5833
 autobyteus/agent/context/agent_context_registry.py,sha256=GqwKn0EKKTRv6Vwwrb5kMRrwD9uH5NCh_Nvtmw82QTo,2748
-autobyteus/agent/context/agent_runtime_state.py,sha256=R8xrEfElqn-AEkl5hn_vYtaXnjdjfp6vrk2QImtijDQ,5538
+autobyteus/agent/context/agent_runtime_state.py,sha256=7Cps1ZIf0n9B5uews5ZQlSTPxlGRPJidc9yCKcudmgI,5745
 autobyteus/agent/events/__init__.py,sha256=zqWRWijkNPc_MyTEu8V74DxQS0M5tGve_DsquQxJUIM,1960
 autobyteus/agent/events/agent_events.py,sha256=5li1ZB9jGcofkCzVaebvhYyoG22kpgzM1k_4XZYLe-k,5022
 autobyteus/agent/events/agent_input_event_queue_manager.py,sha256=cgUNvZs71otaGWMFkeeNxYehkZue2aP0QDXIMjcvBhs,12485
@@ -26,7 +27,7 @@ autobyteus/agent/events/event_store.py,sha256=cDOAQK8R5Mk8cS9_sE_RcL780iOQN1Gjkr
 autobyteus/agent/events/notifiers.py,sha256=x5YmRKSKOs1F5WUzzzYK20wRyiu5QJOZgDQHqoUj0QQ,7368
 autobyteus/agent/events/worker_event_dispatcher.py,sha256=f1lHfem02xY9yCPGFdJkZZEuuZVAxl-mfGkPeh-tQaA,3574
 autobyteus/agent/factory/__init__.py,sha256=4_PxMM-S_BRuYoQBHIffY6bXpBdEv-zFyuB6YaK93Yw,204
-autobyteus/agent/factory/agent_factory.py,sha256=90zI6OKhcVQFk5rzMuBqPFL9KnFZ7oNx2gse5s1k0BM,9316
+autobyteus/agent/factory/agent_factory.py,sha256=PFKzJSjr9S3ZErc4jkYEy2T0BMv3GnC8XU8eTeJDy7M,10783
 autobyteus/agent/handlers/__init__.py,sha256=YIupwVsl_fIHLWdjn_dgrNziZYrvaOr3F8D3hhRbdS4,1598
 autobyteus/agent/handlers/approved_tool_invocation_event_handler.py,sha256=s_D4v1FrvgCZJQR9n0bA_1tH61qIkWEU8DsRg0PU8D4,8505
 autobyteus/agent/handlers/base_event_handler.py,sha256=G2vtFJT_vyObWTgrgwIgRwOssBQ-6A3gxHtc7S4SEuQ,1264
@@ -43,7 +44,7 @@ autobyteus/agent/handlers/tool_result_event_handler.py,sha256=U5rYPpGN2evffVYDK5
 autobyteus/agent/handlers/user_input_message_event_handler.py,sha256=BlnsOAcMSk_7bYuRymE1vEnwbmlROwOTVjS9b0faB8Q,5926
 autobyteus/agent/input_processor/__init__.py,sha256=uyxNlVWQXkHshNFp-9MkjRjMK0f7Ve0Mjx_d-q8C8Ic,330
 autobyteus/agent/input_processor/base_user_input_processor.py,sha256=7y11eGo4gT25nt1F-c3cNqkyBQwq1sCOkDkQ5SPhg2A,2650
-autobyteus/agent/input_processor/memory_ingest_input_processor.py,sha256=PVlECkAZbBQFidUG5h6ZVkw_jfo1dhgzcIdeZ3Mn-Yw,1423
+autobyteus/agent/input_processor/memory_ingest_input_processor.py,sha256=Shcnp8dm6qjk5b1DJe9sA2v7HAh8VCP1PNp2Gc3OSg8,1674
 autobyteus/agent/input_processor/processor_definition.py,sha256=ISRHvjbBhlG2DYdgSK4hN3i8w1DJkgPOvCDY06TuslQ,2101
 autobyteus/agent/input_processor/processor_meta.py,sha256=Ns_VcPbK4-xLlpbdFIC_xWfgoXDHVduVTNN7EUCUPgU,2241
 autobyteus/agent/input_processor/processor_registry.py,sha256=GMTw-XAeHvNaHcWEOpSQ9mCDeb4iT94slQa-pNH44M0,4651
@@ -148,7 +149,7 @@ autobyteus/agent/streaming/parser/strategies/registry.py,sha256=PtfuG63fFIH8QZUN
 autobyteus/agent/streaming/parser/strategies/sentinel_strategy.py,sha256=wLT8_6t19kLPe1dI_k4sPQvV7aopBYCio1YPcxxJHbY,672
 autobyteus/agent/streaming/parser/strategies/xml_tag_strategy.py,sha256=Szn__0LajOlkbfcxokK1QEbdnF9d_cENM8tjiQ-_4_E,640
 autobyteus/agent/streaming/segments/__init__.py,sha256=BmTIdAoPz5nslRfmgB46BDbrOjSHP9vCTK45rFVm6gc,183
-autobyteus/agent/streaming/segments/segment_events.py,sha256=y7PxbqaFFJb7ZjEa9jBTAd7L-o4_l6U4suN-FXSTO7c,2634
+autobyteus/agent/streaming/segments/segment_events.py,sha256=U8l7tG5jTsrZEzJuuusa0fWHtgDaVRgHJy4cCBFLImI,2654
 autobyteus/agent/streaming/streams/__init__.py,sha256=BYIZoE3wHPk4aot4Sa8Zmf8_ndAnvxGYn-gP17EwP_M,129
 autobyteus/agent/streaming/streams/agent_event_stream.py,sha256=JcXPRDib77dB7VqdBwchqlOoNziaMi6yAPdFya-Y9EY,11110
 autobyteus/agent/streaming/utils/__init__.py,sha256=nYG9HyjmMltHYtUhWXafa_wR5bHuQBEVZ4Im0gJZw-I,109
@@ -334,12 +335,14 @@ autobyteus/llm/utils/response_types.py,sha256=WXVAJCb1wKBp3IvNaNGth_gEyEyejV92mj
 autobyteus/llm/utils/token_usage.py,sha256=C8YJH_-sYKJHVcE47hwwk-p0VTAFF0ezFGjSbc0mbzI,601
 autobyteus/llm/utils/token_usage_tracker.py,sha256=RC4zL5k343-wLm0jXc-rwAOMhpxs_1klnrh-xi2J7W8,4220
 autobyteus/llm/utils/tool_call_delta.py,sha256=9kzf_fl9Jozs2UupZgNP003CHNtaYS_J2KZJt_MIDaE,1123
-autobyteus/memory/__init__.py,sha256=uspTA6Iky46kyfShbmRsaaAMDPiBbSyaHUy9RUVj3Gc,1253
-autobyteus/memory/active_transcript.py,sha256=CCNbr7Mej6WkD9gipAReN6NqmOvEo_bMXnNlysfHuHY,2237
+autobyteus/memory/__init__.py,sha256=_IFapdlcEVllmoYfpEXDPxSTPE5ndeM8TCkp2mm9ckY,1410
 autobyteus/memory/compaction_snapshot_builder.py,sha256=FqRdNjtZV3QobUbutBQZb57524fMSA-4phPNc-0cRNY,3461
-autobyteus/memory/memory_manager.py,sha256=bekNsGdR6lubOfQf7HAoIeCNF_phNttJT4pwU_1pMI0,7388
+autobyteus/memory/memory_manager.py,sha256=oVAp6pwzgkFG2VS2F6mHMqD8gAPMAR0qdnxNS2oAsO8,8654
+autobyteus/memory/path_resolver.py,sha256=Cl9e-0pzJ2mbPBM1qX4xnuCfIhUuo-9cubi42S7oMEs,786
 autobyteus/memory/tool_interaction_builder.py,sha256=nGmzx6h_9BxD22CLxF_gFxlwt_du8fa0WXKRK8EuX0U,1737
 autobyteus/memory/turn_tracker.py,sha256=Gf_7vXfz6ZQ48AIOdHNgb6zifhgAwQD88N9_i3MUy14,272
+autobyteus/memory/working_context_snapshot.py,sha256=tX40sjmsL6Rw--TLRvSPWNlMEFZpXhJD4swMvNZW5fg,2243
+autobyteus/memory/working_context_snapshot_serializer.py,sha256=I8G8RdqsYP2GjhBNZPjuirhZ_MdqTIzamgNlsaSHJlw,5310
 autobyteus/memory/compaction/__init__.py,sha256=2Huatvj1mbKZYj_U7Xle5tnIE4IZ0WA5oxUib0KeLGM,274
 autobyteus/memory/compaction/compaction_result.py,sha256=25prRrghtExjmtHgN0cMfOafyDGq1_wwM2KxUtD4ERE,210
 autobyteus/memory/compaction/compactor.py,sha256=cY3Riwyd5Ya2VYUPg-7mE8ZeUKSoNmxjD7a2v-3he40,3315
@@ -352,12 +355,15 @@ autobyteus/memory/models/semantic_item.py,sha256=Vy2tpUYPiJFPBGCf2a4sRcj-gnhnPyR
 autobyteus/memory/models/tool_interaction.py,sha256=8SKN0vOzuf1_wPF6jRIf4YVI_cJArPzdmz3Oy156K5g,439
 autobyteus/memory/policies/__init__.py,sha256=mCBbjgFFWzsw2bqkL4uLk7goHIfZQL7SPcmTFKAjpu4,113
 autobyteus/memory/policies/compaction_policy.py,sha256=WYk9VIu_oUNkLjUvgoBJ6dhcE9Cm3BFQMaxpFvpRr0E,466
+autobyteus/memory/restore/__init__.py,sha256=Kob-S-xqEm8PhHhhw35fJVaxEbhCd4TlJGAao0enpMI,18
+autobyteus/memory/restore/working_context_snapshot_bootstrapper.py,sha256=Ps3LX6Xjb3i0K9RQCW-WNvtfwls2-a91klpszv2YhQ0,2818
 autobyteus/memory/retrieval/__init__.py,sha256=phIGpifCU6wmURstz2H_ZMFziNRyEvI1nXf5D-Kn3oQ,179
 autobyteus/memory/retrieval/memory_bundle.py,sha256=P6g9FPWevsHmqLldojqi5qiv2ZUXwWdedvfruMmL_gw,353
 autobyteus/memory/retrieval/retriever.py,sha256=n0-FnSTlEWP8W7dfBdQv9Y3y0jzETM7eJgbRZDUg4cI,574
-autobyteus/memory/store/__init__.py,sha256=fs4J2WsSeMsNr8A2TvHEcHf4-i8a9lHAOsoN7PWXfXY,179
+autobyteus/memory/store/__init__.py,sha256=mhQBJHpOkuNJ3uLsfnXX6aEsI2aGP8dbz1VAb36WzPw,309
 autobyteus/memory/store/base_store.py,sha256=eUGpTdOgPISa3hFIvX0OK5YYagauknXcd-20ghoS4vM,419
 autobyteus/memory/store/file_store.py,sha256=Ay60DkLMkZdtbNRYj5tncXn6zXpGA8aWDRyzypxZ1g4,4128
+autobyteus/memory/store/working_context_snapshot_store.py,sha256=62PSwJ3NGWPDLWcftq6GY7Z_nwuQhILhfrVD92iyGGc,1009
 autobyteus/multimedia/__init__.py,sha256=8sssScdnrek2-1-s7wStlOc37ZilLTf0cJzMqVzW4Zw,589
 autobyteus/multimedia/providers.py,sha256=AUlct_V9SGtmrMpKXmzaB8b2luvxRO2_QSP0NNO3ce8,147
 autobyteus/multimedia/runtimes.py,sha256=mgmA06Ng7Gh6UW6YNi0O5CmwV6oDw3kX2qxLssUqG0o,180
@@ -368,7 +374,7 @@ autobyteus/multimedia/audio/autobyteus_audio_provider.py,sha256=p4vfp8orsKiYGtrt
 autobyteus/multimedia/audio/base_audio_client.py,sha256=zl12JZrvDUmKEnNhEtdY7jXK0TYmil9rUgIkU_VWbL0,1498
 autobyteus/multimedia/audio/api/__init__.py,sha256=M6A3fmXx5nn3R_0wozcph4ANTA25RGG4L4-a-M4OAoc,240
 autobyteus/multimedia/audio/api/autobyteus_audio_client.py,sha256=QoVgVTRklmLGtcOKkrYe_nBJXB2OVwlz_LUuPjkDggo,3313
-autobyteus/multimedia/audio/api/gemini_audio_client.py,sha256=wVxf4a2gbrAzSyqzzTOEVAKMbupN84I29lXpOwSAr-Y,10100
+autobyteus/multimedia/audio/api/gemini_audio_client.py,sha256=40Xa8ixZA_NII1B3J7c5B1DL_y9PdsuJxhVaWcy_iGo,10161
 autobyteus/multimedia/audio/api/openai_audio_client.py,sha256=wNuNBgD1CdTjs7IV-bOD6wBeqf_NYR5YO4vz_QyeVIA,4017
 autobyteus/multimedia/image/__init__.py,sha256=YWwPtRgbTtPoZBgAmjt87P-pTREOZEpJzDbKhD9jSRg,287
 autobyteus/multimedia/image/autobyteus_image_provider.py,sha256=aCukfKyk4E3Xf1s0nFK69kwzTOGMf6uej648HhvO5PY,5115
@@ -377,7 +383,7 @@ autobyteus/multimedia/image/image_client_factory.py,sha256=8yu1KnXIAyioSgXIGUcue
 autobyteus/multimedia/image/image_model.py,sha256=850haQxRbKnt0ST7pVS-3q4IEnB7f8MnTSoQBh0ZplY,4225
 autobyteus/multimedia/image/api/__init__.py,sha256=Vh_FC_6rxcPhJqFpAvgv3yBqHYVmxrzjyVSSrCM7rww,255
 autobyteus/multimedia/image/api/autobyteus_image_client.py,sha256=LW6hi5DxtPYoKx8freb9YK5wnpcjhxFbeo-jBwlC1ks,4894
-autobyteus/multimedia/image/api/gemini_image_client.py,sha256=xR9caTH055jx4h7tkjwcQDmHXvunqkASWhatha4RUJg,7112
+autobyteus/multimedia/image/api/gemini_image_client.py,sha256=23kHGs9HrsnLID9KzAnLRuBZ9LTjWGIJopk23IZN_Ws,7173
 autobyteus/multimedia/image/api/openai_image_client.py,sha256=EKntmL66XauILnFwLnppN0oCPQcNrbYHzvF0O6d5YoU,9589
 autobyteus/multimedia/utils/__init__.py,sha256=pH2c5CB2V-QXVl6SgL6N70KKH0VShByhZHS8m7L9TCg,298
 autobyteus/multimedia/utils/api_utils.py,sha256=dbrIQzRnoz66CwaZOIG4353h5ccbTEIvUPUSRfgIWOQ,613
@@ -533,7 +539,7 @@ autobyteus/utils/diff_utils.py,sha256=YMP1GI64L_YH6ezEcY8st09lO3ecb2IGIPvGxDUpdV
 autobyteus/utils/download_utils.py,sha256=9KftlNZ9p8ktHOksc9ghXnnknCpV67jaAEOTYDY9SnU,4225
 autobyteus/utils/dynamic_enum.py,sha256=c_mgKtKrjb958LlCWeAApl1LMvVB7U_0SWl-8XFhRag,1339
 autobyteus/utils/file_utils.py,sha256=5tJQT4LdFS6BmRyQwCaL7g2hiK4G2vMCbdX9l3EETCg,2253
-autobyteus/utils/gemini_helper.py,sha256=ucfqWZNvBulu0qKor9LbKJn56GYyTDZOsoKgnQOutVQ,1882
+autobyteus/utils/gemini_helper.py,sha256=KnpQDDNg6sNeHJLEZ_mXNuhBsgsPypJlkK91xA1O9CQ,2431
 autobyteus/utils/gemini_model_mapping.py,sha256=2BsJwb-Vd48DXNvBISJvV19vAXqigPuBRyzsi_eTXYU,2496
 autobyteus/utils/html_cleaner.py,sha256=mI2V83yFRfQe8NnN6fr6Ujpa0bPlq25NW5tTKaz2WGk,8672
 autobyteus/utils/llm_output_formatter.py,sha256=QwWP8VltDx0n0ssL2DsOMaik8NzYpB6Odj1ofS6bO9w,3035
@@ -593,8 +599,8 @@ autobyteus/workflow/streaming/workflow_stream_event_payloads.py,sha256=NVaC4Ak84
 autobyteus/workflow/streaming/workflow_stream_events.py,sha256=j3uKwdV6Xl7JW0Qgawusk8L10atDfP8h5TsC-4IF8B0,2212
 autobyteus/workflow/utils/__init__.py,sha256=SzaMZHnJBIJKcT_r-HOeyIcuxzRu2bGeFkOcMLJaalk,222
 autobyteus/workflow/utils/wait_for_idle.py,sha256=EYko8MLjzFYLHijIG1qOMRmh6kVqZfNCI5Q3zA0qPCg,1953
-autobyteus-1.2.3.dist-info/licenses/LICENSE,sha256=Ompok_c8HRsXRwmax-pGR9OZRRxZC9RPp4JB6eTJd0M,1360
-autobyteus-1.2.3.dist-info/METADATA,sha256=W0B9Dnfrho_FhPqXPlb6aKxT6b-VH-Wrkvy_IE0-Jww,14883
-autobyteus-1.2.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-autobyteus-1.2.3.dist-info/top_level.txt,sha256=OeVeFlKcnysp6uMGe8TDaoFeuh4NUK4wZIfNjXCWKTE,11
-autobyteus-1.2.3.dist-info/RECORD,,
+autobyteus-1.3.0.dist-info/licenses/LICENSE,sha256=Ompok_c8HRsXRwmax-pGR9OZRRxZC9RPp4JB6eTJd0M,1360
+autobyteus-1.3.0.dist-info/METADATA,sha256=Y6LbGRSNAEEqfwGH9kbSI07G1_FCaQNUImWHmi7mlbc,14875
+autobyteus-1.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+autobyteus-1.3.0.dist-info/top_level.txt,sha256=OeVeFlKcnysp6uMGe8TDaoFeuh4NUK4wZIfNjXCWKTE,11
+autobyteus-1.3.0.dist-info/RECORD,,