feat: Add batch testing mode for conversational agent and implement conversation test cases
This commit is contained in:
130
scripts/test_agent_conversation_live.sh
Executable file
130
scripts/test_agent_conversation_live.sh
Executable file
@@ -0,0 +1,130 @@
|
||||
#!/bin/bash
|
||||
# Live testing script for conversational agent
|
||||
# Tests agent function calling with real Ollama/Gemini backends
|
||||
|
||||
set -e
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"
|
||||
Z3ED="${PROJECT_ROOT}/build/bin/z3ed"
|
||||
ROM_FILE="${PROJECT_ROOT}/assets/zelda3.sfc"
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m'
|
||||
|
||||
echo "========================================="
|
||||
echo "Live Conversational Agent Test"
|
||||
echo "========================================="
|
||||
echo ""
|
||||
|
||||
# Prerequisites check
|
||||
if [ ! -f "$Z3ED" ]; then
|
||||
echo -e "${RED}✗ z3ed not found at $Z3ED${NC}"
|
||||
echo "Build with: cmake --build build --target z3ed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ ! -f "$ROM_FILE" ]; then
|
||||
echo -e "${RED}✗ ROM file not found at $ROM_FILE${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}✓ Prerequisites met${NC}"
|
||||
echo ""
|
||||
|
||||
# Check for AI backends
|
||||
BACKEND_AVAILABLE=false
|
||||
|
||||
echo "Checking AI Backends..."
|
||||
echo "-----------------------"
|
||||
|
||||
# Check Ollama
|
||||
if command -v ollama &> /dev/null; then
|
||||
if curl -s http://localhost:11434/api/tags > /dev/null 2>&1; then
|
||||
echo -e "${GREEN}✓ Ollama server running${NC}"
|
||||
if ollama list | grep -q "qwen2.5-coder"; then
|
||||
echo -e "${GREEN}✓ qwen2.5-coder model available${NC}"
|
||||
BACKEND_AVAILABLE=true
|
||||
AI_BACKEND="Ollama"
|
||||
else
|
||||
echo -e "${YELLOW}⚠ Recommended model qwen2.5-coder:7b not installed${NC}"
|
||||
echo " Install with: ollama pull qwen2.5-coder:7b"
|
||||
fi
|
||||
else
|
||||
echo -e "${YELLOW}⚠ Ollama not running${NC}"
|
||||
echo " Start with: ollama serve"
|
||||
fi
|
||||
else
|
||||
echo -e "${YELLOW}⚠ Ollama not installed${NC}"
|
||||
fi
|
||||
|
||||
# Check Gemini
|
||||
if [ -n "$GEMINI_API_KEY" ]; then
|
||||
echo -e "${GREEN}✓ Gemini API key set${NC}"
|
||||
BACKEND_AVAILABLE=true
|
||||
if [ "$AI_BACKEND" != "Ollama" ]; then
|
||||
AI_BACKEND="Gemini"
|
||||
fi
|
||||
else
|
||||
echo -e "${YELLOW}⚠ GEMINI_API_KEY not set${NC}"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
|
||||
if [ "$BACKEND_AVAILABLE" = false ]; then
|
||||
echo -e "${RED}✗ No AI backend available${NC}"
|
||||
echo ""
|
||||
echo "Please set up at least one backend:"
|
||||
echo " - Ollama: brew install ollama && ollama serve && ollama pull qwen2.5-coder:7b"
|
||||
echo " - Gemini: export GEMINI_API_KEY='your-key-here'"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}✓ Using AI Backend: $AI_BACKEND${NC}"
|
||||
echo ""
|
||||
|
||||
# Run the test-conversation command with default test cases
|
||||
echo "========================================="
|
||||
echo "Running Automated Conversation Tests"
|
||||
echo "========================================="
|
||||
echo ""
|
||||
echo "This will run 5 default test cases:"
|
||||
echo " 1. Simple ROM introspection (dungeon query)"
|
||||
echo " 2. Overworld tile search"
|
||||
echo " 3. Multi-step conversation"
|
||||
echo " 4. Command generation (tile placement)"
|
||||
echo " 5. Map description"
|
||||
echo ""
|
||||
|
||||
read -p "Press Enter to start tests (or Ctrl+C to cancel)..."
|
||||
echo ""
|
||||
|
||||
# Run the tests
|
||||
"$Z3ED" agent test-conversation --rom "$ROM_FILE" --verbose
|
||||
|
||||
TEST_EXIT_CODE=$?
|
||||
|
||||
echo ""
|
||||
echo "========================================="
|
||||
echo "Test Results"
|
||||
echo "========================================="
|
||||
|
||||
if [ $TEST_EXIT_CODE -eq 0 ]; then
|
||||
echo -e "${GREEN}✅ All tests completed successfully${NC}"
|
||||
else
|
||||
echo -e "${RED}❌ Tests failed with exit code $TEST_EXIT_CODE${NC}"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo "Next Steps:"
|
||||
echo " - Review the output above for any warnings"
|
||||
echo " - Check if tool calls are being invoked correctly"
|
||||
echo " - Verify JSON/table formatting is working"
|
||||
echo " - Test with custom conversation file: z3ed agent test-conversation --file my_tests.json"
|
||||
echo ""
|
||||
|
||||
exit $TEST_EXIT_CODE
|
||||
Reference in New Issue
Block a user