94598fbd25
- Delete test_minimax_provider.py which imports non-existent `create_llm` function (should be `create_llm_provider`), causing pytest collection errors - Add scripts/smoke-test-slim.sh: shared retain + recall validation script used by both Docker slim and pip slim CI jobs - Update docker/test-image.sh to run retain/recall after health check for all API targets - Update test-pip-slim CI job to run the shared smoke test script
62 lines
1.9 KiB
Bash
Executable file
62 lines
1.9 KiB
Bash
Executable file
#!/bin/bash
|
|
#
|
|
# Slim variant smoke test: retain + recall
|
|
#
|
|
# Verifies that a Hindsight API endpoint can store and retrieve memories.
|
|
# Used by both Docker slim and pip slim CI jobs.
|
|
#
|
|
# Usage:
|
|
# ./scripts/smoke-test-slim.sh [base_url]
|
|
#
|
|
# Arguments:
|
|
# base_url - API base URL (default: http://localhost:8888)
|
|
#
|
|
# Exit codes:
|
|
# 0 - Success
|
|
# 1 - Failure
|
|
#
|
|
|
|
set -euo pipefail
|
|
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
NC='\033[0m'
|
|
|
|
BASE_URL="${1:-http://localhost:8888}"
|
|
BANK_ID="smoke-test-$$"
|
|
|
|
echo "Running retain/recall smoke test against: $BASE_URL"
|
|
echo "Bank: $BANK_ID"
|
|
|
|
# Retain
|
|
echo ""
|
|
echo "--- Retain ---"
|
|
RETAIN_RESPONSE=$(curl -sf -X POST "$BASE_URL/v1/default/banks/$BANK_ID/memories" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"items": [{"content": "Alice is a software engineer who loves Python and distributed systems."}]}')
|
|
echo "$RETAIN_RESPONSE" | python3 -m json.tool 2>/dev/null || echo "$RETAIN_RESPONSE"
|
|
|
|
SUCCESS=$(echo "$RETAIN_RESPONSE" | python3 -c "import sys, json; d = json.load(sys.stdin); print(d.get('success', False))" 2>/dev/null || echo "False")
|
|
if [ "$SUCCESS" != "True" ]; then
|
|
echo -e "${RED}FAIL: retain did not return success=true${NC}"
|
|
exit 1
|
|
fi
|
|
echo "Retain: OK"
|
|
|
|
# Recall
|
|
echo ""
|
|
echo "--- Recall ---"
|
|
RECALL_RESPONSE=$(curl -sf -X POST "$BASE_URL/v1/default/banks/$BANK_ID/memories/recall" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"query": "What does Alice do?"}')
|
|
echo "$RECALL_RESPONSE" | python3 -m json.tool 2>/dev/null || echo "$RECALL_RESPONSE"
|
|
|
|
RESULTS_COUNT=$(echo "$RECALL_RESPONSE" | python3 -c "import sys, json; d = json.load(sys.stdin); print(len(d.get('results', [])))" 2>/dev/null || echo "0")
|
|
if [ -z "$RESULTS_COUNT" ] || [ "$RESULTS_COUNT" -eq 0 ]; then
|
|
echo -e "${RED}FAIL: recall returned no results${NC}"
|
|
exit 1
|
|
fi
|
|
echo "Recall: OK ($RESULTS_COUNT results)"
|
|
|
|
echo ""
|
|
echo -e "${GREEN}PASS: retain/recall smoke test${NC}"
|