changes v1

Author: Rafi-Microsoft

src/ContentProcessor/requirements.txt

@@ -2,9 +2,9 @@ azure-appconfiguration>=1.7.1
 azure-identity>=1.19.0
 azure-storage-blob>=12.24.1
 azure-storage-queue>=12.12.0
+azure-ai-projects>=1.0.0b5
 certifi>=2024.12.14
 charset-normalizer>=3.4.1
-openai==2.0.0
 pandas>=2.2.3
 pdf2image>=1.17.0
 poppler-utils>=0.1.0

src/ContentProcessor/src/libs/application/application_configuration.py

@@ -25,6 +25,7 @@ class AppConfiguration(ModelBaseSettings):
         app_cps_processes (str): Folder name CPS processes name in Blob Container.
         app_cps_configuration (str): Folder CPS configuration name Blob Container.
         app_content_understanding_endpoint (str): The endpoint for content understanding Service.
+        app_ai_project_endpoint (str): The AI Foundry project endpoint.
         app_azure_openai_endpoint (str): The endpoint for Azure OpenAI.
         app_azure_openai_model (str): The model for Azure OpenAI.
         app_cosmos_connstr (str): The connection string for Cosmos DB.
@@ -44,6 +45,7 @@ class AppConfiguration(ModelBaseSettings):
     app_cps_processes: str
     app_cps_configuration: str
     app_content_understanding_endpoint: str
+    app_ai_project_endpoint: str
     app_azure_openai_endpoint: str
     app_azure_openai_model: str
     app_cosmos_connstr: str

src/ContentProcessor/src/libs/azure_helper/azure_openai.py

@@ -1,18 +1,32 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from azure.identity import get_bearer_token_provider
+from azure.ai.projects import AIProjectClient
 from helpers.azure_credential_utils import get_azure_credential
-from openai import AzureOpenAI
 
 
-def get_openai_client(azure_openai_endpoint: str) -> AzureOpenAI:
+def get_foundry_client(ai_project_endpoint: str):
+    """
+    Create an OpenAI-compatible client via Azure AI Foundry (Projects SDK).
+
+    This function uses the Azure AI Foundry approach:
+    1. Create an AIProjectClient with the AI project endpoint
+    2. Call .get_openai_client() to get an OpenAI-compatible client
+
+    Args:
+        ai_project_endpoint: The AI Foundry project endpoint URL (e.g., https://aif-xyz.services.ai.azure.com)
+
+    Returns:
+        An OpenAI-compatible client from the AI Foundry project
+    """
     credential = get_azure_credential()
-    token_provider = get_bearer_token_provider(
-        credential, "https://cognitiveservices.azure.com/.default"
-    )
-    return AzureOpenAI(
-        azure_endpoint=azure_openai_endpoint,
-        azure_ad_token_provider=token_provider,
-        api_version="2024-10-01-preview",
+    
+    # Create the AI Foundry Project client
+    project_client = AIProjectClient(
+        endpoint=ai_project_endpoint,
+        credential=credential
     )
+    
+    # Get the OpenAI-compatible client from the project
+    # This client supports .beta.chat.completions.parse() and other OpenAI methods
+    return project_client.get_openai_client()

src/ContentProcessor/src/libs/pipeline/handlers/map_handler.py

@@ -8,7 +8,7 @@
 from pdf2image import convert_from_bytes
 
 from libs.application.application_context import AppContext
-from libs.azure_helper.azure_openai import get_openai_client
+from libs.azure_helper.azure_openai import get_foundry_client
 from libs.azure_helper.model.content_understanding import AnalyzedResult
 from libs.pipeline.entities.mime_types import MimeTypes
 from libs.pipeline.entities.pipeline_file import ArtifactType, PipelineLogEntry
@@ -82,8 +82,8 @@ async def execute(self, context: MessageContext) -> StepResult:
         )
 
         # Invoke GPT with the prompt
-        gpt_response = get_openai_client(
-            self.application_context.configuration.app_azure_openai_endpoint
+        gpt_response = get_foundry_client(
+            self.application_context.configuration.app_ai_project_endpoint
         ).beta.chat.completions.parse(
             model=self.application_context.configuration.app_azure_openai_model,
             messages=[

src/ContentProcessor/src/tests/test_main.py

@@ -55,6 +55,7 @@ async def test_application_run(mocker):
             ConfigItem(
                 "app_content_understanding_endpoint", "https://example.com/content"
             ),
+            ConfigItem("app_ai_project_endpoint", "https://example.ai.azure.com"),
             ConfigItem("app_azure_openai_endpoint", "https://example.com/openai"),
             ConfigItem("app_azure_openai_model", "model-name"),
             ConfigItem(