File tree Expand file tree Collapse file tree 1 file changed +0
-36
lines changed
tests/integrations/litellm Expand file tree Collapse file tree 1 file changed +0
-36
lines changed Original file line number Diff line number Diff 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-
12761240def test_no_integration (
12771241 reset_litellm_executor ,
12781242 sentry_init ,
You can’t perform that action at this time.
0 commit comments