51 lines
1.4 KiB
Bash
Executable file
51 lines
1.4 KiB
Bash
Executable file
#!/bin/bash
|
|
#
|
|
# Local Test Script for Slim Docker Images
|
|
#
|
|
# This script makes it easy to test slim images locally with external providers.
|
|
# It expects API keys to be set in environment variables.
|
|
#
|
|
# Usage:
|
|
# export GROQ_API_KEY=gsk_xxx
|
|
# export OPENAI_API_KEY=sk-xxx
|
|
# export COHERE_API_KEY=xxx
|
|
# ./docker/test-slim-local.sh
|
|
#
|
|
# Or inline:
|
|
# GROQ_API_KEY=gsk_xxx OPENAI_API_KEY=sk_xxx COHERE_API_KEY=xxx ./docker/test-slim-local.sh
|
|
#
|
|
|
|
set -euo pipefail
|
|
|
|
# Check for required API keys
|
|
if [ -z "${GROQ_API_KEY:-}" ]; then
|
|
echo "❌ Error: GROQ_API_KEY environment variable is required"
|
|
echo "Set it with: export GROQ_API_KEY=gsk_xxx"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z "${OPENAI_API_KEY:-}" ]; then
|
|
echo "❌ Error: OPENAI_API_KEY environment variable is required"
|
|
echo "Set it with: export OPENAI_API_KEY=sk-xxx"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z "${COHERE_API_KEY:-}" ]; then
|
|
echo "❌ Error: COHERE_API_KEY environment variable is required"
|
|
echo "Set it with: export COHERE_API_KEY=xxx"
|
|
exit 1
|
|
fi
|
|
|
|
# Configuration
|
|
IMAGE="${1:-hindsight-slim:test}"
|
|
echo "Testing image: $IMAGE"
|
|
echo ""
|
|
|
|
# Set up external providers
|
|
export HINDSIGHT_API_EMBEDDINGS_PROVIDER=openai
|
|
export HINDSIGHT_API_EMBEDDINGS_OPENAI_API_KEY=$OPENAI_API_KEY
|
|
export HINDSIGHT_API_RERANKER_PROVIDER=cohere
|
|
export HINDSIGHT_API_COHERE_API_KEY=$COHERE_API_KEY
|
|
|
|
# Run the test
|
|
exec "$(dirname "$0")/test-image.sh" "$IMAGE" standalone
|