add pre-down hooks to unblock azd down

Author: hund030

.devcontainer/devcontainer.json

@@ -23,7 +23,7 @@
     "forwardPorts": [
         50505
     ],
-    "postCreateCommand": "",
+    "postCreateCommand": "python3 -m pip install -r requirements-dev.txt",
     "remoteUser": "vscode",
     "hostRequirements": {
         "memory": "8gb"

azure.yaml

@@ -132,3 +132,12 @@ hooks:
         run: ./scripts/auth_update.sh;./scripts/prepdocs.sh
         interactive: true
         continueOnError: false
+    predown:
+      windows:
+        shell: pwsh
+        run: ./scripts/pre-down.ps1
+        continueOnError: true
+      posix:
+        shell: sh
+        run: ./scripts/pre-down.sh
+        continueOnError: true

requirements-dev.txt

@@ -1,4 +1,5 @@
 -r app/backend/requirements.txt
+azure-mgmt-cognitiveservices
 ruff
 black
 pytest

scripts/pre-down.ps1

@@ -0,0 +1,10 @@
+# Get the directory of the current script
+$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Definition
+
+# Load environment variables from azd env
+$subscriptionId = azd env get-value AZURE_SUBSCRIPTION_ID
+$resourceName = azd env get-value AZURE_OPENAI_SERVICE
+$resourceGroup = azd env get-value AZURE_OPENAI_RESOURCE_GROUP
+
+# Run the Python script with the retrieved values
+python "$scriptDir/pre-down.py" --subscription-id $subscriptionId --resource-name $resourceName --resource-group $resourceGroup

scripts/pre-down.py

@@ -0,0 +1,32 @@
+import argparse
+
+from azure.identity import DefaultAzureCredential
+from azure.mgmt.cognitiveservices import CognitiveServicesManagementClient
+
+# Set up argument parsing
+parser = argparse.ArgumentParser(description="Delete an Azure OpenAI deployment.")
+parser.add_argument("--resource-name", required=True, help="The name of the Azure OpenAI resource.")
+parser.add_argument("--resource-group", required=True, help="The name of the Azure resource group.")
+parser.add_argument("--subscription-id", required=True, help="The Azure subscription ID.")
+
+args = parser.parse_args()
+
+# Authenticate using DefaultAzureCredential
+credential = DefaultAzureCredential()
+
+# Initialize the Cognitive Services client
+client = CognitiveServicesManagementClient(credential, subscription_id=args.subscription_id)
+
+# List all deployments
+deployments = client.deployments.list(
+    resource_group_name=args.resource_group, account_name=args.resource_name
+)
+
+# Delete each deployment and wait for the operation to complete
+for deployment in deployments:
+    deployment_name = deployment.name
+    poller = client.deployments.begin_delete(
+        resource_group_name=args.resource_group, account_name=args.resource_name, deployment_name=deployment_name
+    )
+    poller.result()
+    print(f"Deployment {deployment_name} deleted successfully.")

scripts/pre-down.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Get the directory of the current script
+script_dir=$(dirname "$0")
+
+# Load environment variables from azd env
+subscription_id=$(azd env get-value AZURE_SUBSCRIPTION_ID)
+resource_name=$(azd env get-value AZURE_OPENAI_SERVICE)
+resource_group=$(azd env get-value AZURE_OPENAI_RESOURCE_GROUP)
+
+# Run the Python script with the retrieved values
+python "$script_dir/pre-down.py" --subscription-id $subscription_id --resource-name $resource_name --resource-group $resource_group