phase 2 of updates for symbiont 0.3.1 changes

Author: jaschadub

pyproject.toml

@@ -36,6 +36,7 @@ dependencies = [
     "cryptography>=41.0.0",
     "pyyaml>=6.0.0",
     "pydantic-settings>=2.0.0",
+    "redis>=5.0.0",
 ]
 
 [project.optional-dependencies]

requirements.txt

@@ -3,5 +3,6 @@ pydantic>=2.0.0
 python-dotenv>=1.0.0
 pyjwt>=2.8.0
 cryptography>=41.0.0
+redis>=5.0.0
 pyyaml>=6.0.0
 pydantic-settings>=2.0.0

setup.py

@@ -1,14 +1,16 @@
-from setuptools import setup, find_packages
 import os
 
+from setuptools import find_packages, setup
+
+
 # Read README.md for long description
 def read_readme():
-    with open(os.path.join(os.path.dirname(__file__), 'README.md'), 'r', encoding='utf-8') as f:
+    with open(os.path.join(os.path.dirname(__file__), 'README.md'), encoding='utf-8') as f:
         return f.read()
 
 # Read requirements.txt for install_requires
 def read_requirements():
-    with open(os.path.join(os.path.dirname(__file__), 'requirements.txt'), 'r', encoding='utf-8') as f:
+    with open(os.path.join(os.path.dirname(__file__), 'requirements.txt'), encoding='utf-8') as f:
         return [line.strip() for line in f if line.strip() and not line.startswith('#')]
 
 setup(
@@ -40,4 +42,4 @@ def read_requirements():
         'Programming Language :: Python :: 3.11',
         'Programming Language :: Python :: 3.12',
     ],
-)
+)

symbiont/client.py

@@ -26,9 +26,11 @@
     AgentMetrics,
     AgentStatusResponse,
     AnalysisResults,
+    ConsolidationResponse,
     # Configuration models (Phase 1)
     ContextQuery,
     ContextResponse,
+    ConversationContext,
     # Agent DSL models
     DslCompileRequest,
     DslCompileResponse,
@@ -45,6 +47,12 @@
     # MCP Management models
     McpServerConfig,
     McpToolInfo,
+    MemoryQuery,
+    MemoryResponse,
+    MemorySearchRequest,
+    MemorySearchResponse,
+    # Phase 2 Memory System models
+    MemoryStoreRequest,
     ReviewSession,
     ReviewSessionCreate,
     ReviewSessionList,
@@ -1032,3 +1040,110 @@ def get_http_input_metrics(self, server_id: str) -> Dict[str, Any]:
         """
         response = self._request("GET", f"http-input/servers/{server_id}/metrics")
         return response.json()
+
+    # =============================================================================
+    # Phase 2 Memory System Methods
+    # =============================================================================
+
+    def add_memory(self, memory_request: Union[MemoryStoreRequest, Dict[str, Any]]) -> MemoryResponse:
+        """Store a new memory in the system.
+
+        Args:
+            memory_request: Memory storage request
+
+        Returns:
+            MemoryResponse: Memory storage response
+        """
+        if isinstance(memory_request, dict):
+            memory_request = MemoryStoreRequest(**memory_request)
+
+        response = self._request("POST", "memory", json=memory_request.model_dump())
+        return MemoryResponse(**response.json())
+
+    def get_memory(self, memory_query: Union[MemoryQuery, Dict[str, Any]]) -> MemoryResponse:
+        """Retrieve a specific memory.
+
+        Args:
+            memory_query: Memory query parameters
+
+        Returns:
+            MemoryResponse: Memory retrieval response
+        """
+        if isinstance(memory_query, dict):
+            memory_query = MemoryQuery(**memory_query)
+
+        response = self._request("GET", "memory", params=memory_query.model_dump(exclude_none=True))
+        return MemoryResponse(**response.json())
+
+    def search_memory(self, search_request: Union[MemorySearchRequest, Dict[str, Any]]) -> MemorySearchResponse:
+        """Search for memories matching criteria.
+
+        Args:
+            search_request: Memory search request
+
+        Returns:
+            MemorySearchResponse: Search results
+        """
+        if isinstance(search_request, dict):
+            search_request = MemorySearchRequest(**search_request)
+
+        response = self._request("POST", "memory/search", json=search_request.model_dump())
+        return MemorySearchResponse(**response.json())
+
+    def consolidate_memory(self, agent_id: str) -> ConsolidationResponse:
+        """Consolidate memories for an agent.
+
+        Args:
+            agent_id: The agent identifier
+
+        Returns:
+            ConsolidationResponse: Consolidation process results
+        """
+        response = self._request("POST", f"memory/consolidate/{agent_id}")
+        return ConsolidationResponse(**response.json())
+
+    def get_conversation_context(self, conversation_id: str, agent_id: str) -> ConversationContext:
+        """Get conversation context with associated memories.
+
+        Args:
+            conversation_id: The conversation identifier
+            agent_id: The agent identifier
+
+        Returns:
+            ConversationContext: Conversation context with memories
+        """
+        params = {
+            "conversation_id": conversation_id,
+            "agent_id": agent_id
+        }
+        response = self._request("GET", "memory/conversation", params=params)
+        return ConversationContext(**response.json())
+
+    def delete_memory(self, memory_id: str) -> Dict[str, Any]:
+        """Delete a memory by ID.
+
+        Args:
+            memory_id: The memory identifier
+
+        Returns:
+            Dict[str, Any]: Deletion confirmation
+        """
+        response = self._request("DELETE", f"memory/{memory_id}")
+        return response.json()
+
+    def list_agent_memories(self, agent_id: str, limit: int = 100) -> MemorySearchResponse:
+        """List all memories for an agent.
+
+        Args:
+            agent_id: The agent identifier
+            limit: Maximum number of memories to return
+
+        Returns:
+            MemorySearchResponse: List of agent memories
+        """
+        params = {
+            "agent_id": agent_id,
+            "limit": limit
+        }
+        response = self._request("GET", "memory/agent", params=params)
+        return MemorySearchResponse(**response.json())

symbiont/exceptions.py

@@ -128,3 +128,40 @@ def __init__(self, message: str = "Insufficient permissions for this operation",
         """
         super().__init__(message, 403)
         self.required_permission = required_permission
+
+
+# =============================================================================
+# Phase 2 Memory System Exception Classes
+# =============================================================================
+
+class MemoryError(SymbiontError):
+    """Base exception for memory system errors."""
+    pass
+
+
+class MemoryStorageError(MemoryError):
+    """Raised when memory storage operations fail."""
+
+    def __init__(self, message: str = "Memory storage error", storage_type: str = None):
+        """Initialize the MemoryStorageError.
+
+        Args:
+            message: Error message describing the storage failure.
+            storage_type: Optional storage backend type that failed.
+        """
+        super().__init__(message)
+        self.storage_type = storage_type
+
+
+class MemoryRetrievalError(MemoryError):
+    """Raised when memory retrieval operations fail."""
+
+    def __init__(self, message: str = "Memory retrieval error", memory_id: str = None):
+        """Initialize the MemoryRetrievalError.
+
+        Args:
+            message: Error message describing the retrieval failure.
+            memory_id: Optional memory ID that failed to retrieve.
+        """
+        super().__init__(message)
+        self.memory_id = memory_id