Add memory_store support to agent lifecycle classes

Introduces an optional memory_store (DatabaseBase) parameter to MCPEnabledBase, AzureAgentBase, FoundryAgentTemplate, and ReasoningAgentTemplate classes. This enables agents to utilize a shared database for memory storage, improving extensibility and future integration with persistent memory features.

Author: Fr4nc3

src/backend/common/utils/utils_af.py

@@ -51,6 +51,7 @@ async def create_RAI_agent() -> FoundryAgentTemplate:
         "- Is completely meaningless, incoherent, or appears to be spam\n"
         "Respond with 'TRUE' if the input violates any rules and should be blocked, otherwise respond with 'FALSE'."
     )
+    
     model_deployment_name = config.AZURE_OPENAI_DEPLOYMENT_NAME
     agent = FoundryAgentTemplate(
         agent_name=agent_name,

src/backend/v4/api/router.py

@@ -44,9 +44,6 @@
 )
 
 
-
-
-
 @app_v4.websocket("/socket/{process_id}")
 async def start_comms(
     websocket: WebSocket, process_id: str, user_id: str = Query(None)

src/backend/v4/magentic_agents/common/lifecycle.py

@@ -25,6 +25,7 @@
 from azure.ai.agents.aio import AgentsClient
 from azure.identity.aio import DefaultAzureCredential
 from common.models.messages_af import TeamConfiguration
+from common.database.database_base import DatabaseBase
 from v4.common.services.team_service import TeamService
 from v4.config.agent_registry import agent_registry
 from v4.magentic_agents.models.agent_models import MCPConfig
@@ -43,6 +44,7 @@ def __init__(
         team_service: TeamService | None = None,
         team_config: TeamConfiguration | None = None,
         project_endpoint: str | None = None,
+        memory_store: DatabaseBase| None = None,
     ) -> None:
         self._stack: AsyncExitStack | None = None
         self.mcp_cfg: MCPConfig | None = mcp
@@ -53,6 +55,7 @@ def __init__(
         self.client: Optional[AzureAIAgentClient] = None
         self.project_endpoint = project_endpoint
         self.creds: Optional[DefaultAzureCredential] = None
+        self.memory_store: Optional[DatabaseBase] = memory_store
 
     async def open(self) -> "MCPEnabledBase":
         if self._stack is not None:
@@ -154,8 +157,9 @@ def __init__(
         project_endpoint: str | None = None,
         team_service: TeamService | None = None,
         team_config: TeamConfiguration | None = None,
+        memory_store: DatabaseBase | None = None,
     ) -> None:
-        super().__init__(mcp=mcp, team_service=team_service, team_config=team_config, project_endpoint=project_endpoint)
+        super().__init__(mcp=mcp, team_service=team_service, team_config=team_config, project_endpoint=project_endpoint, memory_store=memory_store)
 
         self._created_ephemeral: bool = (
             False  # reserved if you add ephemeral agent cleanup

src/backend/v4/magentic_agents/foundry_agent.py

@@ -8,6 +8,7 @@
 from azure.ai.projects.models import ConnectionType
 from common.config.app_config import config
 from common.models.messages_af import TeamConfiguration
+from src.backend.common.database.database_base import DatabaseBase
 from v4.common.services.team_service import TeamService
 from v4.config.agent_registry import agent_registry
 from v4.magentic_agents.common.lifecycle import AzureAgentBase
@@ -35,13 +36,15 @@ def __init__(
         search_config: SearchConfig | None = None,
         team_service: TeamService | None = None,
         team_config: TeamConfiguration | None = None,
+        memory_store: DatabaseBase | None = None,
     ) -> None:
         super().__init__(
             mcp=mcp_config,
             model_deployment_name=model_deployment_name,
             project_endpoint=project_endpoint,
             team_service=team_service,
             team_config=team_config,
+            memory_store=memory_store,
         )
         self.agent_name = agent_name
         self.agent_description = agent_description

src/backend/v4/magentic_agents/reasoning_agent.py

@@ -16,6 +16,7 @@
 #from agent_framework.azure import AzureOpenAIChatClient
 
 from common.models.messages_af import TeamConfiguration
+from src.backend.common.database.database_base import DatabaseBase
 from v4.common.services.team_service import TeamService
 from v4.magentic_agents.common.lifecycle import MCPEnabledBase
 from v4.magentic_agents.models.agent_models import MCPConfig, SearchConfig
@@ -48,6 +49,7 @@ def __init__(
         max_search_docs: int = 3,
         team_service: TeamService | None = None,
         team_config: TeamConfiguration | None = None,
+        memory_store: DatabaseBase | None = None,
     ) -> None:
         """Initialize reasoning agent.
         
@@ -61,7 +63,7 @@ def __init__(
             mcp_config: Optional MCP server configuration
             max_search_docs: Maximum number of search documents to retrieve
         """
-        super().__init__(mcp=mcp_config, team_service=team_service, team_config=team_config, project_endpoint=project_endpoint)
+        super().__init__(mcp=mcp_config, team_service=team_service, team_config=team_config, project_endpoint=project_endpoint, memory_store=memory_store)
         self.agent_name = agent_name
         self.agent_description = agent_description
         self.base_instructions = agent_instructions