Pass project_endpoint to agent constructors

Updated agent creation logic to explicitly pass the Azure AI project endpoint from config to agent constructors. Refactored AzureAgentBase to accept project_endpoint as an argument instead of reading from the environment variable. Also enabled Azure Search connection lookup in FoundryAgentTemplate.

Author: Fr4nc3

src/backend/common/utils/utils_af.py

@@ -33,6 +33,7 @@ async def create_RAI_agent() -> FoundryAgentTemplate:
         agent_instructions=agent_instructions,
         model_deployment_name=model_deployment_name,
         enable_code_interpreter=False,
+        project_endpoint=config.AZURE_AI_PROJECT_ENDPOINT,
         mcp_config=None,
         search_config=None,
     )

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

@@ -111,27 +111,22 @@ class AzureAgentBase(MCPEnabledBase):
       - optionally register themselves via agent_registry
     """
 
-    def __init__(self, mcp: MCPConfig | None = None, model_deployment_name: str | None = None) -> None:
+    def __init__(self, mcp: MCPConfig | None = None, model_deployment_name: str | None = None, project_endpoint: str | None = None) -> None:
         super().__init__(mcp=mcp)
         self.creds: Optional[DefaultAzureCredential] = None
         self.client: Optional[AzureAIAgentClient] = None
-        self.project_endpoint: Optional[str] = None
+       
         self._created_ephemeral: bool = (
             False  # reserved if you add ephemeral agent cleanup
         )
+        self.project_endpoint = project_endpoint
         self.model_deployment_name = model_deployment_name
 
     async def open(self) -> "AzureAgentBase":
         if self._stack is not None:
             return self
         self._stack = AsyncExitStack()
 
-        # Resolve Azure AI Project endpoint (mirrors old SK env var usage)
-        self.project_endpoint = os.getenv("AZURE_AI_PROJECT_ENDPOINT")
-        if not self.project_endpoint:
-            raise RuntimeError(
-                "AZURE_AI_PROJECT_ENDPOINT environment variable is required for AzureAgentBase."
-            )
 
         # Acquire credential
         self.creds = DefaultAzureCredential()

src/backend/v4/magentic_agents/foundry_agent.py

@@ -33,11 +33,12 @@ def __init__(
         agent_description: str,
         agent_instructions: str,
         model_deployment_name: str,
+        project_endpoint: str,
         enable_code_interpreter: bool = False,
         mcp_config: MCPConfig | None = None,
         search_config: SearchConfig | None = None,
     ) -> None:
-        super().__init__(mcp=mcp_config, model_deployment_name=model_deployment_name)
+        super().__init__(mcp=mcp_config, model_deployment_name=model_deployment_name, project_endpoint=project_endpoint)
         self.agent_name = agent_name
         self.agent_description = agent_description
         self.agent_instructions = agent_instructions
@@ -110,10 +111,10 @@ async def _create_azure_search_enabled_client(self):
         query_type = getattr(self.search, "search_query_type", "vector")
 
         # ai_search_conn_id = ""
-        # async for connection in self.client.project_client.connections.list():
-        #     if connection.type == ConnectionType.AZURE_AI_SEARCH:
-        #         ai_search_conn_id = connection.id
-        #         break
+        async for connection in self.client.project_client.connections.list():
+            if connection.type == ConnectionType.AZURE_AI_SEARCH:
+                connection_id = connection.id
+                break
         if not connection_id or not index_name:
             self.logger.error(
                 "Missing azure_search_connection_id or azure_search_index_name in search_config; aborting Azure Search path."

src/backend/v4/magentic_agents/magentic_agent_factory.py

@@ -118,6 +118,7 @@ async def create_agent_from_config(self, user_id: str, agent_obj: SimpleNamespac
                 agent_instructions=getattr(agent_obj, "system_message", ""),
                 model_deployment_name=deployment_name,
                 enable_code_interpreter=getattr(agent_obj, "coding_tools", False),
+                project_endpoint=config.AZURE_AI_PROJECT_ENDPOINT,
                 mcp_config=mcp_config,
                 search_config=search_config,
             )