Merge pull request #2044 from PolicyEngine/fix/provide-hf-token

Add Hugging Face token to CI/CD, Docker
PolicyEngine · Dec 4, 2024 · a88fca9 · a88fca9
2 parents 544b0d4 + f1b2f81
commit a88fca9
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 0 deletions.
diff --git a/.github/workflows/pr.yml b/.github/workflows/pr.yml
@@ -77,3 +77,4 @@ jobs:
           POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN: ${{ secrets.POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN }}
           ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
@@ -76,6 +76,7 @@ jobs:
           POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN: ${{ secrets.POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN }}
           ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
   docker:
     name: Docker
     runs-on: ubuntu-latest

diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    added:
+    - Hugging Face connection token to CI/CD scripts and Docker image build script
diff --git a/gcp/export.py b/gcp/export.py
@@ -13,6 +13,7 @@
 GITHUB_MICRODATA_TOKEN = os.environ["POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN"]
 ANTHROPIC_API_KEY = os.environ["ANTHROPIC_API_KEY"]
 OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
+HUGGING_FACE_TOKEN = os.environ["HUGGING_FACE_TOKEN"]
 
 # Export GAE to to .gac.json and DB_PD to .dbpw in the current directory
 
@@ -35,6 +36,9 @@
             ".anthropic_api_key", ANTHROPIC_API_KEY
         )
         dockerfile = dockerfile.replace(".openai_api_key", OPENAI_API_KEY)
+        dockerfile = dockerfile.replace(
+            ".hugging_face_token", HUGGING_FACE_TOKEN
+        )
 
     with open(dockerfile_location, "w") as f:
         f.write(dockerfile)
diff --git a/gcp/policyengine_api/Dockerfile b/gcp/policyengine_api/Dockerfile
@@ -5,6 +5,7 @@ ENV POLICYENGINE_DB_PASSWORD .dbpw
 ENV POLICYENGINE_GITHUB_MICRODATA_AUTH_TOKEN .github_microdata_token
 ENV ANTHROPIC_API_KEY .anthropic_api_key
 ENV OPENAI_API_KEY .openai_api_key
+ENV HUGGING_FACE_TOKEN .hugging_face_token
 
 WORKDIR /app