Merge pull request #526 from microsoft/psl-asyncsfi

refactor: Changed Async and Sync Azure credential retrieval

Author: Prajwal-Microsoft

src/app.py

@@ -170,7 +170,7 @@ async def init_ai_foundry_client():
 
         ai_project_client = AIProjectClient(
             endpoint=app_settings.azure_ai.agent_endpoint,
-            credential=get_azure_credential()
+            credential=await get_azure_credential_async()
         )
         track_event_if_configured("AIFoundryAgentEndpointUsed", {
             "endpoint": app_settings.azure_ai.agent_endpoint
@@ -1164,7 +1164,7 @@ async def fetch_azure_search_content():
             return jsonify({"error": "URL and title are required"}), 400
 
         # Get Azure AD token
-        credential = await get_azure_credential_async()
+        credential = get_azure_credential()
         token = credential.get_token("https://search.azure.com/.default")
         access_token = token.token
 

src/backend/api/agent/browse_agent_factory.py

@@ -1,6 +1,6 @@
 from azure.ai.projects.aio import AIProjectClient
 from azure.ai.agents.models import AzureAISearchTool, AzureAISearchQueryType
-from backend.helpers.azure_credential_utils import get_azure_credential
+from backend.helpers.azure_credential_utils import get_azure_credential_async
 from backend.settings import app_settings
 from event_utils import track_event_if_configured
 
@@ -20,7 +20,7 @@ async def create_or_get_agent(cls):
         """
         project_client = AIProjectClient(
             endpoint=app_settings.azure_ai.agent_endpoint,
-            credential=get_azure_credential(),
+            credential=await get_azure_credential_async(),
             api_version=app_settings.azure_ai.agent_api_version
         )
 

src/backend/api/agent/section_agent_factory.py

@@ -1,6 +1,6 @@
 from azure.ai.projects.aio import AIProjectClient
 from azure.ai.agents.models import AzureAISearchTool, AzureAISearchQueryType
-from backend.helpers.azure_credential_utils import get_azure_credential
+from backend.helpers.azure_credential_utils import get_azure_credential_async
 from backend.settings import app_settings
 from event_utils import track_event_if_configured
 
@@ -19,7 +19,7 @@ async def create_or_get_agent(cls):
         """
         project_client = AIProjectClient(
             endpoint=app_settings.azure_ai.agent_endpoint,
-            credential=get_azure_credential(),
+            credential=await get_azure_credential_async(),
             api_version=app_settings.azure_ai.agent_api_version
         )
 

src/backend/api/agent/template_agent_factory.py

@@ -1,6 +1,6 @@
 from azure.ai.projects.aio import AIProjectClient
 from azure.ai.agents.models import AzureAISearchTool, AzureAISearchQueryType
-from backend.helpers.azure_credential_utils import get_azure_credential
+from backend.helpers.azure_credential_utils import get_azure_credential_async
 from backend.settings import app_settings
 from event_utils import track_event_if_configured
 
@@ -20,7 +20,7 @@ async def create_or_get_agent(cls):
         """
         project_client = AIProjectClient(
             endpoint=app_settings.azure_ai.agent_endpoint,
-            credential=get_azure_credential(),
+            credential=await get_azure_credential_async(),
             api_version=app_settings.azure_ai.agent_api_version
         )