From 5e3aa7938fdd68b5eeabae86539dad221caeb36b Mon Sep 17 00:00:00 2001 From: kerry Date: Fri, 2 Oct 2026 21:06:48 +0000 Subject: [PATCH] test(spend): fix chaos helper import name Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../spend/test_spend_log_request_tags_chaos.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/tests/integration/spend/test_spend_log_request_tags_chaos.py b/tests/integration/spend/test_spend_log_request_tags_chaos.py index 26b33a9c68c..5069c6d0d59 100644 --- a/tests/integration/spend/test_spend_log_request_tags_chaos.py +++ b/tests/integration/spend/test_spend_log_request_tags_chaos.py @@ -11,8 +11,8 @@ from integration._support.database import read_rows from integration._support.process import owned_proxy_process from integration._support.wire import Reply, Request, wire_server from integration.spend._request_tag_helpers import ( - MODEL, - OPENAI_MODEL, + ANTHROPIC_ANTHROPIC_MODEL, + OPENAI_ANTHROPIC_MODEL, T3, provider_env, provider_reply, @@ -47,14 +47,19 @@ def _tagged_requests( candidate.request( "POST", "/anthropic/v1/messages", - {"model": MODEL, "max_tokens": 16, "messages": [{"role": "user", "content": marker}], "stream": stream}, + { + "model": ANTHROPIC_MODEL, + "max_tokens": 16, + "messages": [{"role": "user", "content": marker}], + "stream": stream, + }, key=key, headers=ANTHROPIC_HEADERS, ), candidate.request( "POST", "/openai/v1/chat/completions", - {"model": OPENAI_MODEL, "messages": [{"role": "user", "content": marker}], "stream": stream}, + {"model": OPENAI_ANTHROPIC_MODEL, "messages": [{"role": "user", "content": marker}], "stream": stream}, key=key, headers=HEADERS, ),