Add Poetry JFrog source configuration to all workflows

The previous commit only set PIP_INDEX_URL, but Poetry uses its own
resolver and needs explicit source configuration. Add a
"Configure Poetry for JFrog" step after poetry install in every job
that sets up the JFrog repository and credentials, then adds it as
the primary source for the project.

Co-authored-by: Isaac
Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>

Author: vikrantpuppala

.github/actions/setup-jfrog/action.yml

@@ -1,5 +1,5 @@
 name: Setup JFrog OIDC
-description: Obtain a JFrog access token via GitHub OIDC and configure pip to use JFrog PyPI proxy
+description: Obtain a JFrog access token via GitHub OIDC and configure pip and poetry to use JFrog PyPI proxy
 
 runs:
   using: composite
@@ -30,3 +30,4 @@ runs:
         set -euo pipefail
         echo "PIP_INDEX_URL=https://gha-service-account:${JFROG_ACCESS_TOKEN}@databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple" >> "$GITHUB_ENV"
         echo "pip configured to use JFrog registry"
+

.github/workflows/code-coverage.yml

@@ -50,6 +50,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists

.github/workflows/code-quality-checks.yml

@@ -47,6 +47,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists
@@ -135,6 +140,11 @@ jobs:
             virtualenvs-create: true
             virtualenvs-in-project: true
             installer-parallel: true
+      -   name: Configure Poetry for JFrog
+          run: |
+            poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+            poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+            poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists
@@ -217,6 +227,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists
@@ -274,6 +289,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists

.github/workflows/daily-telemetry-e2e.yml

@@ -56,6 +56,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists

.github/workflows/integration.yml

@@ -45,6 +45,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
 
       #----------------------------------------------
       #       load cached venv if cache exists
@@ -107,6 +112,11 @@ jobs:
           virtualenvs-create: true
           virtualenvs-in-project: true
           installer-parallel: true
+      - name: Configure Poetry for JFrog
+        run: |
+          poetry config repositories.jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
+          poetry config http-basic.jfrog gha-service-account "${JFROG_ACCESS_TOKEN}"
+          poetry source add --priority=primary jfrog https://databricks.jfrog.io/artifactory/api/pypi/db-pypi/simple
       - name: Load cached venv
         id: cached-poetry-dependencies
         uses: actions/cache@0057852bfaa89a56745cba8c7296529d2fc39830 # v4