Skip to content

Commit f12b7d3

Browse files
remove test
1 parent 6626a1b commit f12b7d3

File tree

1 file changed

+0
-36
lines changed

1 file changed

+0
-36
lines changed

tests/integrations/litellm/test_litellm.py

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -1237,42 +1237,6 @@ async def test_async_additional_parameters(
12371237
assert span["data"][SPANDATA.GEN_AI_REQUEST_PRESENCE_PENALTY] == 0.5
12381238

12391239

1240-
def test_litellm_specific_parameters(sentry_init, capture_events):
1241-
"""Test that LiteLLM-specific parameters are captured."""
1242-
sentry_init(
1243-
integrations=[LiteLLMIntegration()],
1244-
traces_sample_rate=1.0,
1245-
)
1246-
events = capture_events()
1247-
1248-
messages = [{"role": "user", "content": "Hello!"}]
1249-
mock_response = MockCompletionResponse()
1250-
1251-
with start_transaction(name="litellm test"):
1252-
kwargs = {
1253-
"model": "gpt-3.5-turbo",
1254-
"messages": messages,
1255-
"api_base": "https://custom-api.example.com",
1256-
"api_version": "2023-01-01",
1257-
"custom_llm_provider": "custom_provider",
1258-
}
1259-
1260-
_input_callback(kwargs)
1261-
_success_callback(
1262-
kwargs,
1263-
mock_response,
1264-
datetime.now(),
1265-
datetime.now(),
1266-
)
1267-
1268-
(event,) = events
1269-
(span,) = event["spans"]
1270-
1271-
assert span["data"]["gen_ai.litellm.api_base"] == "https://custom-api.example.com"
1272-
assert span["data"]["gen_ai.litellm.api_version"] == "2023-01-01"
1273-
assert span["data"]["gen_ai.litellm.custom_llm_provider"] == "custom_provider"
1274-
1275-
12761240
def test_no_integration(
12771241
reset_litellm_executor,
12781242
sentry_init,

0 commit comments

Comments
 (0)