fdp code level changes- hardcoded endpoint

Author: Shreyas-Microsoft

src/backend/common/config/config.py

@@ -38,6 +38,7 @@ def __init__(self):
         self.azure_queue_name = os.getenv("AZURE_QUEUE_NAME")
 
         self.azure_openai_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
+        self.ai_project_endpoint = os.getenv("AI_PROJECT_ENDPOINT")
         self.migrator_agent_model_deploy = os.getenv("MIGRATOR_AGENT_MODEL_DEPLOY")
         self.picker_agent_model_deploy = os.getenv("PICKER_AGENT_MODEL_DEPLOY")
         self.fixer_agent_model_deploy = os.getenv("FIXER_AGENT_MODEL_DEPLOY")
@@ -62,3 +63,4 @@ def get_azure_credentials(self):
 
 
 app_config = Config()
+print(f"[DEBUG] AI_PROJECT_ENDPOINT: '{os.getenv('AI_PROJECT_ENDPOINT')}'")

src/backend/sql_agents/agents/agent_base.py

@@ -3,11 +3,10 @@
 import logging
 from abc import ABC, abstractmethod
 from typing import Any, Generic, List, Optional, TypeVar, Union
+import json
 
-from azure.ai.projects.models import (
-    ResponseFormatJsonSchema,
-    ResponseFormatJsonSchemaType,
-)
+# Removed broken import:
+# from azure.ai.projects.models import ResponseFormatJsonSchema, ResponseFormatJsonSchemaType
 
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
 from semantic_kernel.functions import KernelArguments
@@ -107,23 +106,30 @@ async def setup(self) -> AzureAIAgent:
 
         kernel_args = self.get_kernel_arguments()
 
+        # Define an agent on the Azure AI agent service
+        # Construct response_format manually
+        response_format = None
+        if self.response_object:
+            response_format = {
+                "type": "json_schema",
+                "json_schema": {
+                    "name": self.response_object.__name__,
+                    "description": f"respond with {self.response_object.__name__.lower()}",
+                    "schema": self.response_object.model_json_schema()
+                }
+            }
+
         try:
-            # Define an agent on the Azure AI agent service
             agent_definition = await self.config.ai_project_client.agents.create_agent(
                 model=_deployment_name,
                 name=_name,
                 instructions=template_content,
                 temperature=self.temperature,
-                response_format=ResponseFormatJsonSchemaType(
-                    json_schema=ResponseFormatJsonSchema(
-                        name=self.response_object.__name__,
-                        description=f"respond with {self.response_object.__name__.lower()}",
-                        schema=self.response_object.model_json_schema(),
-                    )
-                ),
+                response_format=response_format,
             )
         except Exception as exc:
             logger.error("Error creating agent definition: %s", exc)
+            raise
         # Set the agent definition with the response format
 
         # Create a Semantic Kernel agent based on the agent definition

src/backend/sql_agents/process_batch.py

@@ -24,7 +24,7 @@
 from fastapi import HTTPException
 
 
-from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent  # pylint: disable=E0611
+from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent, AzureAIAgentSettings  # pylint: disable=E0611
 from semantic_kernel.contents import AuthorRole
 from semantic_kernel.exceptions.service_exceptions import ServiceResponseException
 
@@ -57,10 +57,11 @@ async def process_batch_async(
     except Exception as exc:
         logger.error("Error updating batch status. %s", exc)
 
+    ai_agent_settings = AzureAIAgentSettings()
     # Add client and auto cleanup
     async with (
         DefaultAzureCredential() as creds,
-        AzureAIAgent.create_client(credential=creds) as client,
+        AzureAIAgent.create_client(credential=creds, endpoint="https://fdp-rg-psl-codmod-main-resource.services.ai.azure.com/api/projects/fdp-rg-psl-codmod-main") as client,
     ):
 
         # setup all agent settings and agents per batch