Support for Redis-based session store to run across multiple nodes

agents-core/pyproject.toml

@@ -71,6 +71,7 @@ decart = ["vision-agents-plugins-decart"]
 twilio = ["vision-agents-plugins-twilio"]
 turbopuffer = ["vision-agents-plugins-turbopuffer"]
 mistral = ["vision-agents-plugins-mistral"]
+redis = ["redis[hiredis]>=5.0.0"]
 
 all-plugins = [
     "vision-agents-plugins-anthropic",

agents-core/vision_agents/core/__init__.py

@@ -1,13 +1,33 @@
 from vision_agents.core.agents import Agent
 from vision_agents.core.agents.agent_launcher import AgentLauncher, AgentSession
+from vision_agents.core.agents.session_registry import (
+    InMemorySessionKVStore,
+    SessionInfo,
+    SessionKVStore,
+    SessionRegistry,
+)
 from vision_agents.core.edge.types import User
 from vision_agents.core.runner import Runner, ServeOptions
 
 __all__ = [
     "Agent",
-    "User",
     "AgentLauncher",
     "AgentSession",
+    "InMemorySessionKVStore",
     "Runner",
     "ServeOptions",
+    "SessionInfo",
+    "SessionKVStore",
+    "SessionRegistry",
+    "User",
 ]
+
+try:
+    from vision_agents.core.agents.session_registry import RedisSessionKVStore
+
+    __all__ += ["RedisSessionKVStore"]
+except ModuleNotFoundError as exc:
+    # Only swallow a missing `redis` package; re-raise anything else
+    # so real import errors in redis_store.py surface immediately.
+    if not exc.name or not exc.name.startswith("redis"):
+        raise

agents-core/vision_agents/core/agents/__init__.py

@@ -8,10 +8,18 @@
 from .conversation import Conversation as Conversation
 from .agent_launcher import AgentLauncher as AgentLauncher
 from .agent_types import AgentOptions as AgentOptions
+from .session_registry import InMemorySessionKVStore as InMemorySessionKVStore
+from .session_registry import SessionInfo as SessionInfo
+from .session_registry import SessionKVStore as SessionKVStore
+from .session_registry import SessionRegistry as SessionRegistry
 
 __all__ = [
     "Agent",
-    "Conversation",
     "AgentLauncher",
     "AgentOptions",
+    "Conversation",
+    "InMemorySessionKVStore",
+    "SessionInfo",
+    "SessionKVStore",
+    "SessionRegistry",
 ]