47 lines
1.3 KiB
Bash
Executable file
47 lines
1.3 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 OPENAI_API_KEY=sk-xxx
|
|
# export COHERE_API_KEY=xxx
|
|
# ./docker/test-slim-local.sh
|
|
#
|
|
# Or inline:
|
|
# OPENAI_API_KEY=sk_xxx COHERE_API_KEY=xxx ./docker/test-slim-local.sh
|
|
#
|
|
|
|
set -euo pipefail
|
|
|
|
# Check for required API keys
|
|
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 LLM and external providers
|
|
export HINDSIGHT_API_LLM_PROVIDER=openai
|
|
export HINDSIGHT_API_LLM_API_KEY=$OPENAI_API_KEY
|
|
export HINDSIGHT_API_LLM_MODEL=gpt-4o-mini
|
|
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
|