spring-ai-integration-tests.yml

name: Spring AI Integration Tests
run-name: Spring AI Integration Tests
on:
  workflow_dispatch:
  schedule:
    - cron: '0 4 * * 1-5'
    - cron: '0 10 * * 1-5'

jobs:

  test-ollama:
    runs-on: ubuntu-latest
    services:
      ollama:
        image: ollama/ollama:latest
        ports:
          - 11434:11434
    env:
      OLLAMA_WITH_REUSE: true
    steps:
      - name: Checkout the action
        uses: actions/checkout@v4

#      - name: Install Ollama
#        run: |
#          curl -fsSL https://ollama.com/install.sh | sh
#          ollama serve &
#          sleep 10  # Give Ollama time to start up

      - name: Configure Testcontainers
        run: |
          mkdir -p $HOME
          echo "testcontainers.reuse.enable = true" >> $HOME/.testcontainers.properties       

      - name: Integration Test
        uses: ./.github/actions/do-integration-test
        with:
          model-name: ollama
          maven-properties: "-Dskip.model.ollama=false"

# Anthrohpic is overloaded :(
#
#  test-anthropic:
#    runs-on: ubuntu-latest
#    env:
#      ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
#    steps:
#      - name: Check secret
#        id: secret_check
#        if: ${{ env.ANTHROPIC_API_KEY != '' }}
#        run: echo "Secret exists"
#
#      - uses: actions/checkout@v4
#        if: steps.secret_check.conclusion == 'success'
#
#      - name: Integration Test
#        if: steps.secret_check.conclusion == 'success'
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: anthropic

  test-azure-openai:
    runs-on: ubuntu-latest
    env:
      AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
      AZURE_OPENAI_ENDPOINT: ${{ secrets.AZURE_OPENAI_ENDPOINT }}
      AZURE_OPENAI_TRANSCRIPTION_API_KEY: ${{ secrets.AZURE_OPENAI_TRANSCRIPTION_API_KEY }}
      AZURE_OPENAI_TRANSCRIPTION_ENDPOINT: ${{ secrets.AZURE_OPENAI_TRANSCRIPTION_ENDPOINT }}
      AZURE_OPENAI_IMAGE_API_KEY: ${{ secrets.AZURE_OPENAI_IMAGE_API_KEY }}
      AZURE_OPENAI_IMAGE_ENDPOINT: ${{ secrets.AZURE_OPENAI_IMAGE_ENDPOINT }}
    steps:
      - name: Check secrets
        id: secret_check
        if: ${{ env.AZURE_OPENAI_API_KEY != '' && env.AZURE_OPENAI_ENDPOINT != '' && env.AZURE_OPENAI_TRANSCRIPTION_API_KEY != '' && env.AZURE_OPENAI_TRANSCRIPTION_ENDPOINT != '' && env.AZURE_OPENAI_IMAGE_API_KEY != '' && env.AZURE_OPENAI_IMAGE_ENDPOINT != '' }}
        run: echo "Secrets exist"

      - uses: actions/checkout@v4
        if: steps.secret_check.conclusion == 'success'

      - name: Integration Test
        if: steps.secret_check.conclusion == 'success'
        uses: ./.github/actions/do-integration-test
        with:
          model-name: azure-openai

#  test-bedrock:
#    runs-on: ubuntu-latest
#    if: ${{ secrets.AWS_ACCESS_KEY_ID != '' && secrets.AWS_SECRET_ACCESS_KEY != '' }}
#    env:
#      AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
#      AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: bedrock

#  test-huggingface:
#    runs-on: ubuntu-latest
#    env:
#      HUGGINGFACE_API_KEY: ${{ secrets.HUGGINGFACE_API_KEY }}
#      HUGGINGFACE_CHAT_URL: ${{ secrets.HUGGINGFACE_CHAT_URL }}
#    if: ${{ env.HUGGINGFACE_API_KEY != '' && env.HUGGINGFACE_CHAT_URL != '' }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: huggingface
#
#  test-mistral-ai:
#    runs-on: ubuntu-latest
#    env:
#      MISTRAL_AI_API_KEY: ${{ secrets.MISTRAL_AI_API_KEY }}
#    if: ${{ env.MISTRAL_AI_API_KEY != '' }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: mistral-ai

# Removed since not able to get keys from ORACLE
# When keys are available uncomment this code and verify no other structural changes in the other jobs
#  test-oci-genai:
#    runs-on: ubuntu-latest
#    env:
#      OCI_COMPARTMENT_ID_KEY: ${{ secrets.OCI_COMPARTMENT_ID_KEY }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: oci-genai

#  test-openai:
#    runs-on: ubuntu-latest
#    env:
#      OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
#    if: ${{ env.OPENAI_API_KEY != '' }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: openai
#
#  test-stability-ai:
#    runs-on: ubuntu-latest
#    env:
#      STABILITYAI_API_KEY: ${{ secrets.STABILITYAI_API_KEY }}
#    if: ${{ env.STABILITYAI_API_KEY != '' }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: stability-ai

#  test-vertex-ai-embedding:
#    runs-on: ubuntu-latest
#    env:
#      VERTEX_AI_GEMINI_PROJECT_ID: ${{ secrets.VERTEX_AI_GEMINI_PROJECT_ID }}
#      VERTEX_AI_GEMINI_LOCATION: ${{ secrets.VERTEX_AI_GEMINI_LOCATION }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: vertex-ai-embedding
#
#  test-vertex-ai-gemini:
#    runs-on: ubuntu-latest
#    env:
#      VERTEX_AI_GEMINI_PROJECT_ID: ${{ secrets.VERTEX_AI_GEMINI_PROJECT_ID }}
#      VERTEX_AI_GEMINI_LOCATION: ${{ secrets.VERTEX_AI_GEMINI_LOCATION }}
#    steps:
#      - name: Checkout the action
#        uses: actions/checkout@v4
#
#      - name: Integration Test
#        uses: ./.github/actions/do-integration-test
#        with:
#          model-name: vertex-ai-gemini

  test-vectorstores:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - name: Configure Testcontainers
        run: |
          mkdir -p $HOME
          echo "testcontainers.reuse.enable = true" >> $HOME/.testcontainers.properties
      - name: Integration Test
        uses: ./.github/actions/do-multi-module-test
        with:
          modules: spring-ai-vector-store/spring-ai-vector-store-chroma, spring-ai-vector-store/spring-ai-vector-store-neo4j, spring-ai-vector-store/spring-ai-vector-store-pgvector, spring-ai-vector-store/spring-ai-vector-store-redis

  test-autoconfigure:
    env:
      # TODO: uncomment keys that have values
#      AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
#      AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
      AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
      AZURE_OPENAI_ENDPOINT: ${{ secrets.AZURE_OPENAI_ENDPOINT }}
#      AZURE_OPENAI_TRANSCRIPTION_DEPLOYMENT_NAME: ${{ secrets.AZURE_OPENAI_TRANSCRIPTION_DEPLOYMENT_NAME }}
#      ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
#      HUGGINGFACE_API_KEY: ${{ secrets.HUGGINGFACE_API_KEY }}
#      HUGGINGFACE_CHAT_URL: ${{ secrets.HUGGINGFACE_CHAT_URL }}
#      MISTRAL_AI_API_KEY: ${{ secrets.MISTRAL_AI_API_KEY }}
#      OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
#      PALM_API_KEY: ${{ secrets.PALM_API_KEY }}
#      STABILITYAI_API_KEY: ${{ secrets.STABILITYAI_API_KEY }}
#      VERTEX_AI_GEMINI_PROJECT_ID: ${{ secrets.VERTEX_AI_GEMINI_PROJECT_ID }}
#      VERTEX_AI_GEMINI_LOCATION: ${{ secrets.VERTEX_AI_GEMINI_LOCATION }}
    runs-on: ubuntu-latest
    steps:
      - name: Checkout the action
        uses: actions/checkout@v4

      - name: Autoconfigure Test
        uses: ./.github/actions/do-autoconfigure-test