Port to azd env get value

azure.yaml

@@ -2,6 +2,8 @@
 name: rag-postgres-openai-python
 metadata:
     template: rag-postgres-openai-python@0.0.1
+requiredVersions:
+    azd: ">= 1.9.5"
 services:
     web:
         project: ./src/backend

scripts/setup_postgres_azurerole.ps1

@@ -1,10 +1,14 @@
-$POSTGRES_HOST = ((azd env get-values | Select-String -Pattern "POSTGRES_HOST") -replace '^POSTGRES_HOST=', '')
-$POSTGRES_USERNAME = ((azd env get-values | Select-String -Pattern "POSTGRES_USERNAME") -replace '^POSTGRES_USERNAME=', '')
-$APP_IDENTITY_NAME = ((azd env get-values | Select-String -Pattern "SERVICE_WEB_IDENTITY_NAME") -replace '^SERVICE_WEB_IDENTITY_NAME=', '')
+$POSTGRES_HOST = (azd env get-value POSTGRES_HOST)
+if (-not $?) {
+    Write-Host "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+}
+$POSTGRES_USERNAME = (azd env get-value POSTGRES_USERNAME)
+$APP_IDENTITY_NAME = (azd env get-value SERVICE_WEB_IDENTITY_NAME)
 
 if ([string]::IsNullOrEmpty($POSTGRES_HOST) -or [string]::IsNullOrEmpty($POSTGRES_USERNAME) -or [string]::IsNullOrEmpty($APP_IDENTITY_NAME)) {
     Write-Host "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and SERVICE_WEB_IDENTITY_NAME environment variables. Make sure you run azd up first."
     exit 1
 }
 
-python ./src/backend/fastapi_app/setup_postgres_azurerole.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --app-identity-name $APP_IDENTITY_NAME
+python ./src/backend/fastapi_app/setup_postgres_azurerole.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --app-identity-name $APP_IDENTITY_NAME

scripts/setup_postgres_azurerole.sh

@@ -1,6 +1,10 @@
-POSTGRES_HOST=$(azd env get-values | grep POSTGRES_HOST | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_HOST=//')
-POSTGRES_USERNAME=$(azd env get-values | grep POSTGRES_USERNAME | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_USERNAME=//')
-APP_IDENTITY_NAME=$(azd env get-values | grep SERVICE_WEB_IDENTITY_NAME | sed 's/="/=/' | sed 's/"$//' | sed 's/^SERVICE_WEB_IDENTITY_NAME=//')
+POSTGRES_HOST=$(azd env get-value POSTGRES_HOST)
+if [ $? -ne 0 ]; then
+    echo "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+fi
+POSTGRES_USERNAME=$(azd env get-value POSTGRES_USERNAME)
+APP_IDENTITY_NAME=$(azd env get-value SERVICE_WEB_IDENTITY_NAME)
 
 if [ -z "$POSTGRES_HOST" ] || [ -z "$POSTGRES_USERNAME" ] || [ -z "$APP_IDENTITY_NAME" ]; then
     echo "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and SERVICE_WEB_IDENTITY_NAME environment variables. Make sure you run azd up first."

scripts/setup_postgres_database.ps1

@@ -1,10 +1,14 @@
-$POSTGRES_HOST = ((azd env get-values | Select-String -Pattern "POSTGRES_HOST") -replace '^POSTGRES_HOST=', '')
-$POSTGRES_USERNAME = ((azd env get-values | Select-String -Pattern "POSTGRES_USERNAME") -replace '^POSTGRES_USERNAME=', '')
-$POSTGRES_PASSWORD = ((azd env get-values | Select-String -Pattern "POSTGRES_PASSWORD") -replace '^POSTGRES_PASSWORD=', '')
+$POSTGRES_HOST = (azd env get-value POSTGRES_HOST)
+if (-not $?) {
+    Write-Host "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+}
+$POSTGRES_USERNAME = (azd env get-value POSTGRES_USERNAME)
+$POSTGRES_DATABASE = (azd env get-value POSTGRES_DATABASE)
 
-if ([string]::IsNullOrEmpty($POSTGRES_HOST) -or [string]::IsNullOrEmpty($POSTGRES_USERNAME) -or [string]::IsNullOrEmpty($POSTGRES_PASSWORD)) {
-    Write-Host "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and POSTGRES_PASSWORD environment variables. Make sure you run azd up first."
+if ([string]::IsNullOrEmpty($POSTGRES_HOST) -or [string]::IsNullOrEmpty($POSTGRES_USERNAME) -or [string]::IsNullOrEmpty($POSTGRES_DATABASE)) {
+    Write-Host "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and POSTGRES_DATABASE environment variables. Make sure you run azd up first."
     exit 1
 }
 
-python ./backend/src/fastapi_app/setup_postgres_database.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --password $POSTGRES_PASSWORD
+python ./src/backend/fastapi_app/setup_postgres_database.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --database $POSTGRES_DATABASE

scripts/setup_postgres_database.sh

@@ -1,6 +1,10 @@
-POSTGRES_HOST=$(azd env get-values | grep POSTGRES_HOST | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_HOST=//')
-POSTGRES_USERNAME=$(azd env get-values | grep POSTGRES_USERNAME | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_USERNAME=//')
-POSTGRES_DATABASE=$(azd env get-values | grep POSTGRES_DATABASE | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_DATABASE=//')
+POSTGRES_HOST=$(azd env get-value POSTGRES_HOST)
+if [ $? -ne 0 ]; then
+    echo "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+fi
+POSTGRES_USERNAME=$(azd env get-value POSTGRES_USERNAME)
+POSTGRES_DATABASE=$(azd env get-value POSTGRES_DATABASE)
 
 . ./scripts/load_python_env.sh
 

scripts/setup_postgres_seeddata.ps1

@@ -1,10 +1,14 @@
-$POSTGRES_HOST = ((azd env get-values | Select-String -Pattern "POSTGRES_HOST") -replace '^POSTGRES_HOST=', '')
-$POSTGRES_USERNAME = ((azd env get-values | Select-String -Pattern "POSTGRES_USERNAME") -replace '^POSTGRES_USERNAME=', '')
-$POSTGRES_PASSWORD = ((azd env get-values | Select-String -Pattern "POSTGRES_PASSWORD") -replace '^POSTGRES_PASSWORD=', '')
+$POSTGRES_HOST = (azd env get-value POSTGRES_HOST)
+if (-not $?) {
+    Write-Host "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+}
+$POSTGRES_USERNAME = (azd env get-value POSTGRES_USERNAME)
+$POSTGRES_DATABASE = (azd env get-value POSTGRES_DATABASE)
 
-if ([string]::IsNullOrEmpty($POSTGRES_HOST) -or [string]::IsNullOrEmpty($POSTGRES_USERNAME) -or [string]::IsNullOrEmpty($POSTGRES_PASSWORD)) {
-    Write-Host "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and POSTGRES_PASSWORD environment variables. Make sure you run azd up first."
+if ([string]::IsNullOrEmpty($POSTGRES_HOST) -or [string]::IsNullOrEmpty($POSTGRES_USERNAME) -or [string]::IsNullOrEmpty($POSTGRES_DATABASE)) {
+    Write-Host "Can't find POSTGRES_HOST, POSTGRES_USERNAME, and POSTGRES_DATABASE environment variables. Make sure you run azd up first."
     exit 1
 }
 
-python ./src/backend/fastapi_app/setup_postgres_seeddata.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --password $POSTGRES_PASSWORD
+python ./src/backend/fastapi_app/setup_postgres_seeddata.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --database $POSTGRES_DATABASE

scripts/setup_postgres_seeddata.sh

@@ -1,6 +1,10 @@
-POSTGRES_HOST=$(azd env get-values | grep POSTGRES_HOST | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_HOST=//')
-POSTGRES_USERNAME=$(azd env get-values | grep POSTGRES_USERNAME | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_USERNAME=//')
-POSTGRES_DATABASE=$(azd env get-values | grep POSTGRES_DATABASE | sed 's/="/=/' | sed 's/"$//' | sed 's/^POSTGRES_DATABASE=//')
+POSTGRES_HOST=$(azd env get-value POSTGRES_HOST)
+if [ $? -ne 0 ]; then
+    echo "Failed to find a value or POSTGRES_HOST in your azd environment. Make sure you run azd up first."
+    exit 1
+fi
+POSTGRES_USERNAME=$(azd env get-value POSTGRES_USERNAME)
+POSTGRES_DATABASE=$(azd env get-value POSTGRES_DATABASE)
 
 . ./scripts/load_python_env.sh
 

src/backend/fastapi_app/postgres_engine.py

@@ -30,7 +30,7 @@ def get_password_from_azure_credential():
 
     engine = create_async_engine(
         DATABASE_URI,
-        echo=True,
+        echo=False,
     )
 
     @event.listens_for(engine.sync_engine, "do_connect")