From 7ee50062991db3fd4fe00db28fae9e174050b7b5 Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Wed, 6 Aug 2025 04:21:00 -0700
Subject: [PATCH] Use different base image

---
 .github/scripts/run_langgraph_cli_test.py |  2 --
 .github/workflows/_integration_test.yml   | 25 ++++++++++++++++++-----
 2 files changed, 20 insertions(+), 7 deletions(-)

diff --git a/.github/scripts/run_langgraph_cli_test.py b/.github/scripts/run_langgraph_cli_test.py
index 478a215ab..96025f632 100644
--- a/.github/scripts/run_langgraph_cli_test.py
+++ b/.github/scripts/run_langgraph_cli_test.py
@@ -1,6 +1,4 @@
 import asyncio
-import json
-import os
 import pathlib
 import sys
 import langgraph_cli
diff --git a/.github/workflows/_integration_test.yml b/.github/workflows/_integration_test.yml
index 5857609ec..e363172be 100644
--- a/.github/workflows/_integration_test.yml
+++ b/.github/workflows/_integration_test.yml
@@ -43,28 +43,43 @@ jobs:
       - name: Build and test service A
         if: steps.changed-files.outputs.all
         working-directory: libs/cli/examples
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
         run: |
           # The build-arg isn't used; just testing that we accept other args
-          langgraph build -t langgraph-test-a  --base-image "langchain/langgraph-trial"
-          cp .env.example .envg
+          langgraph build -t langgraph-test-a
+          cp .env.example .env
+          echo $LANGSMITH_API_KEY >> .env
           timeout 60 python ../../../.github/scripts/run_langgraph_cli_test.py -c langgraph.json -t langgraph-test-a
       - name: Build and test service B
         if: steps.changed-files.outputs.all
         working-directory: libs/cli/examples/graphs
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
         run: |
-          langgraph build -t langgraph-test-b  --base-image "langchain/langgraph-trial"
+          langgraph build -t langgraph-test-b
+          cp .env.example .env
+          echo $LANGSMITH_API_KEY >> .env
           timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-b
       - name: Build and test service C
         if: steps.changed-files.outputs.all
         working-directory: libs/cli/examples/graphs_reqs_a
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
         run: |
-          langgraph build -t langgraph-test-c  --base-image "langchain/langgraph-trial"
+          langgraph build -t langgraph-test-c
+          cp .env.example .env
+          echo $LANGSMITH_API_KEY >> .env
           timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-c
       - name: Build and test service D
         if: steps.changed-files.outputs.all
         working-directory: libs/cli/examples/graphs_reqs_b
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
         run: |
-          langgraph build -t langgraph-test-d  --base-image "langchain/langgraph-trial"
+          langgraph build -t langgraph-test-d
+          cp .env.example .env
+          echo $LANGSMITH_API_KEY >> .env
           timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-d
 
       - name: Build JS service