Files
goose/test_lead_worker.sh
2025-06-05 13:55:32 +10:00

31 lines
1.1 KiB
Bash
Executable File

#!/bin/bash
# Test script for lead/worker provider functionality
# Set up test environment variables
export GOOSE_PROVIDER="openai"
export GOOSE_MODEL="gpt-4o-mini"
export OPENAI_API_KEY="test-key"
# Test 1: Default behavior (no lead/worker)
echo "Test 1: Default behavior (no lead/worker)"
unset GOOSE_LEAD_MODEL
unset GOOSE_WORKER_MODEL
unset GOOSE_LEAD_TURNS
# Test 2: Lead/worker with same provider
echo -e "\nTest 2: Lead/worker with same provider"
export GOOSE_LEAD_MODEL="gpt-4o"
export GOOSE_WORKER_MODEL="gpt-4o-mini"
export GOOSE_LEAD_TURNS="3"
# Test 3: Lead/worker with default worker (uses main model)
echo -e "\nTest 3: Lead/worker with default worker"
export GOOSE_LEAD_MODEL="gpt-4o"
unset GOOSE_WORKER_MODEL
export GOOSE_LEAD_TURNS="5"
echo -e "\nConfiguration examples:"
echo "- Default: Uses GOOSE_MODEL for all turns"
echo "- Lead/Worker: Set GOOSE_LEAD_MODEL to use a different model for initial turns"
echo "- GOOSE_LEAD_TURNS: Number of turns to use lead model (default: 5)"
echo "- GOOSE_WORKER_MODEL: Model to use after lead turns (default: GOOSE_MODEL)"