scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

refactor(tests): streamline Claude Code request execution in Docker test script

Browse source 
Introduce a new function, run_claude_request, to encapsulate the logic for executing Claude Code requests within a Docker container. This refactor simplifies the test script by reducing redundancy and improving readability, while maintaining the same functionality for back-to-back requests.

Made-with: Cursor
This commit is contained in:
Darcy Liu 2026-04-25 23:50:59 +00:00
parent bd310d8c8a
commit 81abd35c85
1 changed files with 28 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
tests/proxy_e2e_anthropic_messages_tests/claude_code/run_claude_code_docker_test.sh
View file

@ -193,18 +193,38 @@ curl -fsS --connect-timeout 5 --max-time 20 \
-H "Authorization: Bearer ${LITELLM_MASTER_KEY}" "${PROXY_URL}/v1/models" | grep -q "\"${MODEL_NAME}\""
echo "[4/5] Running back-to-back Claude Code requests in an isolated non-root container..."
run_claude_request() {
local prompt="$1"
local output_file="$2"
local run_output
if ! run_output="$(
docker compose -p "${COMPOSE_PROJECT_NAME}" -f "${COMPOSE_FILE}" run --rm -T \
-e CLAUDE_CODE_PROMPT="${prompt}" \
-e CLAUDE_CODE_OUTPUT_FILE="${output_file}" \
claude-code 2>&1
)"; then
echo "Claude Code container run failed."
echo "---- claude-code container output ----"
printf '%s\n' "${run_output}" | sed -n '1,200p'
echo "---- litellm logs tail ----"
docker compose -p "${COMPOSE_PROJECT_NAME}" -f "${COMPOSE_FILE}" logs --tail=200 litellm || true
return 1
fi
printf '%s\n' "${run_output}"
}
FIRST_RESPONSE="$(
docker compose -p "${COMPOSE_PROJECT_NAME}" -f "${COMPOSE_FILE}" run --rm \
-e CLAUDE_CODE_PROMPT="Respond with exactly this text and nothing else: Hello from LiteLLM Claude Code request one." \
-e CLAUDE_CODE_OUTPUT_FILE="/tmp/claude-output-1.txt" \
claude-code
run_claude_request \
"Respond with exactly this text and nothing else: Hello from LiteLLM Claude Code request one." \
"/tmp/claude-output-1.txt"
)"
SECOND_RESPONSE="$(
docker compose -p "${COMPOSE_PROJECT_NAME}" -f "${COMPOSE_FILE}" run --rm \
-e CLAUDE_CODE_PROMPT="Respond with exactly this text and nothing else: Hello from LiteLLM Claude Code request two." \
-e CLAUDE_CODE_OUTPUT_FILE="/tmp/claude-output-2.txt" \
claude-code
run_claude_request \
"Respond with exactly this text and nothing else: Hello from LiteLLM Claude Code request two." \
"/tmp/claude-output-2.txt"
)"
[[ -n "${FIRST_RESPONSE}" ]] || {