fix for agent_id issue (#27)

update biceps to add AZURE_AI_AGENT_ID as an env variable

Author: sophia-ramsey

azure.yaml

@@ -43,6 +43,7 @@ pipeline:
     - USE_APPLICATION_INSIGHTS
     - USE_SEARCH_SERVICE
     - AZURE_AI_AGENT_NAME
+    - AZURE_AI_AGENT_ID
     - AZURE_AI_AGENT_DEPLOYMENT_NAME
     - AZURE_AI_AGENT_DEPLOYMENT_SKU
     - AZURE_AI_AGENT_DEPLOYMENT_CAPACITY

infra/api.bicep

@@ -10,6 +10,7 @@ param exists bool
 param projectConnectionString string
 param agentDeploymentName string
 param agentName string
+param agentID string
 
 resource apiIdentity 'Microsoft.ManagedIdentity/userAssignedIdentities@2023-01-31' = {
   name: identityName
@@ -29,6 +30,10 @@ var env = [
     name: 'AZURE_AI_AGENT_NAME'
     value: agentName
   }
+  {
+    name: 'AZURE_AI_AGENT_ID'
+    value: agentID
+  }
   {
     name: 'AZURE_AI_AGENT_DEPLOYMENT_NAME'
     value: agentDeploymentName

infra/main.bicep

@@ -75,6 +75,8 @@ param principalId string = ''
 param agentModelFormat string
 @description('Name of agent to deploy')
 param agentName string
+@description('ID of agent to deploy')
+param aiAgentID string = ''
 @description('Name of the chat model to deploy')
 param agentModelName string
 @description('Name of the model deployment')
@@ -126,6 +128,8 @@ var resourceToken = toLower(uniqueString(subscription().id, environmentName, loc
 var projectName = !empty(aiProjectName) ? aiProjectName : 'ai-project-${resourceToken}'
 var tags = { 'azd-env-name': environmentName }
 
+var agentID = !empty(aiAgentID) ? aiAgentID : ''
+
 var aiDeployments = [
   {
     name: agentDeploymentName
@@ -331,6 +335,7 @@ module api 'api.bicep' = {
     projectConnectionString: projectConnectionString
     agentDeploymentName: agentDeploymentName
     agentName: agentName
+    agentID: agentID
     exists: apiAppExists
   }
 }
@@ -342,6 +347,7 @@ output AZURE_TENANT_ID string = tenant().tenantId
 output AZURE_AIPROJECT_CONNECTION_STRING string = projectConnectionString
 output AZURE_AI_AGENT_DEPLOYMENT_NAME string = agentDeploymentName
 output AZURE_AI_AGENT_NAME string = agentName
+output AZURE_AI_AGENT_ID string = agentID
 
 // Outputs required by azd for ACA
 output AZURE_CONTAINER_ENVIRONMENT_NAME string = containerApps.outputs.environmentName

infra/main.parameters.json

@@ -53,6 +53,9 @@
     "agentName": {
       "value": "${AZURE_AI_AGENT_NAME=agent-template-assistant}"
     },
+    "aiAgentID": {
+      "value": "${AZURE_AI_AGENT_ID}"
+    },
     "agentDeploymentName": {
       "value": "${AZURE_AI_AGENT_MODEL_NAME=gpt-4o-mini}"
     },