Skip to content

Commit 36fd0d5

Browse files
fix merge
1 parent 5fd4efe commit 36fd0d5

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

tests/integrations/litellm/test_litellm.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def test_nonstreaming_chat_completion(
187187

188188
messages = [{"role": "user", "content": "Hello!"}]
189189

190-
client = OpenAI(api_key="z")
190+
client = OpenAI(api_key="test-key")
191191

192192
model_response = get_model_response(
193193
nonstreaming_chat_completions_model_response,
@@ -270,7 +270,7 @@ def test_streaming_chat_completion(
270270

271271
messages = [{"role": "user", "content": "Hello!"}]
272272

273-
client = OpenAI(api_key="z")
273+
client = OpenAI(api_key="test-key")
274274

275275
model_response = get_model_response(
276276
server_side_event_chunks(
@@ -499,7 +499,7 @@ def test_span_origin(
499499

500500
messages = [{"role": "user", "content": "Hello!"}]
501501

502-
client = OpenAI(api_key="z")
502+
client = OpenAI(api_key="test-key")
503503

504504
model_response = get_model_response(
505505
nonstreaming_chat_completions_model_response,
@@ -545,7 +545,7 @@ def test_multiple_providers(
545545

546546
messages = [{"role": "user", "content": "Hello!"}]
547547

548-
openai_client = OpenAI(api_key="z")
548+
openai_client = OpenAI(api_key="test-key")
549549
openai_model_response = get_model_response(
550550
nonstreaming_chat_completions_model_response,
551551
serialize_pydantic=True,
@@ -585,7 +585,7 @@ def test_multiple_providers(
585585
model="claude-3-opus-20240229",
586586
messages=messages,
587587
client=anthropic_client,
588-
api_key="z",
588+
api_key="test-key",
589589
)
590590

591591
litellm_utils.executor.shutdown(wait=True)
@@ -608,7 +608,7 @@ def test_multiple_providers(
608608
model="gemini/gemini-pro",
609609
messages=messages,
610610
client=gemini_client,
611-
api_key="z",
611+
api_key="test-key",
612612
)
613613

614614
litellm_utils.executor.shutdown(wait=True)
@@ -636,7 +636,7 @@ def test_additional_parameters(
636636
events = capture_events()
637637

638638
messages = [{"role": "user", "content": "Hello!"}]
639-
client = OpenAI(api_key="z")
639+
client = OpenAI(api_key="test-key")
640640

641641
model_response = get_model_response(
642642
nonstreaming_chat_completions_model_response,
@@ -679,15 +679,15 @@ def test_additional_parameters(
679679
assert span["data"][SPANDATA.GEN_AI_REQUEST_PRESENCE_PENALTY] == 0.5
680680

681681

682-
def test_no_integration(sentry_init, capture_events):
682+
def test_no_integration(sentry_init, capture_events, get_model_response, nonstreaming_chat_completions_model_response):
683683
"""Test that when integration is not enabled, callbacks don't break."""
684684
sentry_init(
685685
traces_sample_rate=1.0,
686686
)
687687
events = capture_events()
688688

689689
messages = [{"role": "user", "content": "Hello!"}]
690-
client = OpenAI(api_key="z")
690+
client = OpenAI(api_key="test-key")
691691

692692
model_response = get_model_response(
693693
nonstreaming_chat_completions_model_response,
@@ -863,7 +863,7 @@ def test_binary_content_encoding_image_url(
863863
],
864864
}
865865
]
866-
client = OpenAI(api_key="z")
866+
client = OpenAI(api_key="test-key")
867867

868868
model_response = get_model_response(
869869
nonstreaming_chat_completions_model_response,
@@ -942,7 +942,7 @@ def test_binary_content_encoding_mixed_content(
942942
],
943943
}
944944
]
945-
client = OpenAI(api_key="z")
945+
client = OpenAI(api_key="test-key")
946946

947947
model_response = get_model_response(
948948
nonstreaming_chat_completions_model_response,
@@ -1007,7 +1007,7 @@ def test_binary_content_encoding_uri_type(
10071007
],
10081008
}
10091009
]
1010-
client = OpenAI(api_key="z")
1010+
client = OpenAI(api_key="test-key")
10111011

10121012
model_response = get_model_response(
10131013
nonstreaming_chat_completions_model_response,

0 commit comments

Comments
 (0)