diff --git a/azure.yaml b/azure.yaml index aab8bf16..c099a770 100644 --- a/azure.yaml +++ b/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 diff --git a/scripts/setup_postgres_azurerole.ps1 b/scripts/setup_postgres_azurerole.ps1 index 9a6d2562..ac87f45b 100644 --- a/scripts/setup_postgres_azurerole.ps1 +++ b/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 \ No newline at end of file +python ./src/backend/fastapi_app/setup_postgres_azurerole.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --app-identity-name $APP_IDENTITY_NAME diff --git a/scripts/setup_postgres_azurerole.sh b/scripts/setup_postgres_azurerole.sh index 64b065f4..d14cd5b6 100755 --- a/scripts/setup_postgres_azurerole.sh +++ b/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." diff --git a/scripts/setup_postgres_database.ps1 b/scripts/setup_postgres_database.ps1 index a81b15cd..93a4393a 100644 --- a/scripts/setup_postgres_database.ps1 +++ b/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 \ No newline at end of file +python ./src/backend/fastapi_app/setup_postgres_database.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --database $POSTGRES_DATABASE diff --git a/scripts/setup_postgres_database.sh b/scripts/setup_postgres_database.sh index 8f0e9df6..40da59f7 100755 --- a/scripts/setup_postgres_database.sh +++ b/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 diff --git a/scripts/setup_postgres_seeddata.ps1 b/scripts/setup_postgres_seeddata.ps1 index bb56f2b8..3d9b906b 100644 --- a/scripts/setup_postgres_seeddata.ps1 +++ b/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 \ No newline at end of file +python ./src/backend/fastapi_app/setup_postgres_seeddata.py --host $POSTGRES_HOST --username $POSTGRES_USERNAME --database $POSTGRES_DATABASE diff --git a/scripts/setup_postgres_seeddata.sh b/scripts/setup_postgres_seeddata.sh index b88ff396..38db5262 100755 --- a/scripts/setup_postgres_seeddata.sh +++ b/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 diff --git a/src/backend/fastapi_app/postgres_engine.py b/src/backend/fastapi_app/postgres_engine.py index b3970e1a..0dcaf814 100644 --- a/src/backend/fastapi_app/postgres_engine.py +++ b/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")