updated import path

Harmanpreet Kaur · Harmanpreet Kaur · commit 9dd852630c23 · 2025-04-23T12:56:45.000+05:30
diff --git a/src/backend/sql_agents/helpers/agents_manager.py b/src/backend/sql_agents/helpers/agents_manager.py
@@ -2,8 +2,7 @@
 
 import logging
 
-from semantic_kernel.agents.azure_ai import AzureAIAgent  # pylint: disable=E0611
-
+from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent  # pylint: disable=E0611
 
 from sql_agents.agents.agent_config import AgentBaseConfig
 from sql_agents.agents.fixer.setup import setup_fixer_agent
diff --git a/src/backend/sql_agents/process_batch.py b/src/backend/sql_agents/process_batch.py
@@ -23,7 +23,7 @@
 from fastapi import HTTPException
 
 
-from semantic_kernel.agents.azure_ai import AzureAIAgent  # pylint: disable=E0611
+from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent # pylint: disable=E0611
 from semantic_kernel.contents import AuthorRole
 from semantic_kernel.exceptions.service_exceptions import ServiceResponseException
 
