fix(tests): use maximize disk space action (i-am-bee#1607)

Signed-off-by: Radek Ježek <[email protected]>
Signed-off-by: Eden Gilbert <[email protected]>

Author: jezekra1

.github/workflows/e2e-test.yml

@@ -20,7 +20,16 @@ on:
 jobs:
   e2e-test:
     runs-on: ubuntu-latest
+    env:
+      AGENTSTACK__HOME: ${{ github.workspace }}/.agentstack
     steps:
+      - name: Maximize build space
+        uses: easimon/maximize-build-space@master
+        with:
+          root-reserve-mb: 15360
+          temp-reserve-mb: 2048
+          swap-size-mb: 1024
+          remove-dotnet: 'true'
       - uses: actions/checkout@v4
       - name: "Set up Lima"
         uses: lima-vm/lima-actions/setup@v1
@@ -35,9 +44,9 @@ jobs:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
       - run: mise run agentstack-server:test:e2e
         env:
-          LLM_API_BASE: "${{ secrets.OPENAI_API_BASE }}"
-          LLM_MODEL: "${{ secrets.OPENAI_MODEL }}"
-          LLM_API_KEY: "${{ secrets.OPENAI_API_KEY }}"
-          # LLM_API_BASE: "https://api.groq.com/openai/v1"
-          # LLM_MODEL: "groq:meta-llama/llama-4-maverick-17b-128e-instruct"
-          # LLM_API_KEY: "${{ secrets.GROQ_API_KEY }}"
+          # LLM_API_BASE: "${{ secrets.OPENAI_API_BASE }}"
+          # LLM_MODEL: "${{ secrets.OPENAI_MODEL }}"
+          # LLM_API_KEY: "${{ secrets.OPENAI_API_KEY }}"
+          LLM_API_BASE: "https://api.groq.com/openai/v1"
+          LLM_MODEL: "groq:meta-llama/llama-4-maverick-17b-128e-instruct"
+          LLM_API_KEY: "${{ secrets.GROQ_API_KEY }}"

apps/agentstack-server/tasks.toml

@@ -260,9 +260,6 @@ VM_NAME=e2e-test-run
 {{ mise_bin }} run agentstack:delete --vm-name=${VM_NAME}
 curl http://localhost:8333 >/dev/null 2>&1 && echo "Another instance at localhost:8333 is already running" && exit 2
 
-#   TODO: docling is too heavy, enable when pipeline has more resources
-#   --set docling.enabled=true
-
 {{ mise_bin }} run agentstack:start \
     --vm-name=${VM_NAME} \
     --set externalRegistries=null \
@@ -271,6 +268,7 @@ curl http://localhost:8333 >/dev/null 2>&1 && echo "Another instance at localhos
     --set auth.basic.enabled="true" \
     --set auth.basic.adminPassword="test-password" \
     --set auth.jwtSecretKey="test-secret-key" \
+    --set docling.enabled=true \
     --set providerBuilds.enabled=true \
     --set localDockerRegistry.enabled=true
 

apps/agentstack-server/tests/conftest.py

@@ -27,7 +27,7 @@ class TestConfiguration(BaseSettings):
     llm_api_base: Secret[str] = Secret("http://localhost:11434/v1")
     llm_model: str = "other:llama3.1:8b"
     llm_api_key: Secret[str] = Secret("dummy")
-    test_agent_image: str = "ghcr.io/i-am-bee/agentstack/agents/chat:0.4.0-rc1"
+    test_agent_image: str = "ghcr.io/i-am-bee/agentstack/agents/chat:0.4.2-rc1"
     test_agent_build_repo: str = "https://github.com/i-am-bee/agentstack-starter"
     server_url: str = "http://agentstack-server-svc:8333"
     db_url: str = "postgresql+asyncpg://agentstack-user:password@postgresql:5432/agentstack"

apps/agentstack-server/tests/e2e/routes/test_files.py

@@ -62,7 +62,6 @@ def create_fn(text: str) -> BytesIO:
 
 
 @pytest.mark.usefixtures("clean_up", "setup_platform_client")
-@pytest.mark.skip(reason="TODO: Docling is temporarily disabled because the image is too heavy")
 async def test_text_extraction_pdf_workflow(subtests, test_configuration, test_pdf: Callable[[str], BytesIO]):
     """Test complete PDF text extraction workflow: upload -> extract -> wait -> verify"""
 

tasks.toml

@@ -171,8 +171,8 @@ export __OLD_PS1="$PS1"
 export __OLD_LIMA_HOME="$LIMA_HOME"
 export __OLD_KUBECONFIG="$KUBECONFIG"
 
-export KUBECONFIG="${HOME}/.agentstack/lima/${VM_NAME}/copied-from-guest/kubeconfig.yaml"
-export LIMA_HOME=~/.agentstack/lima
+export KUBECONFIG="${AGENTSTACK__HOME:-${HOME}/.agentstack}/lima/${VM_NAME}/copied-from-guest/kubeconfig.yaml"
+export LIMA_HOME=${AGENTSTACK__HOME:-${HOME}/.agentstack}/lima
 export PS1="(${VM_NAME}) ${__OLD_PS1}"
 
 """