From f3d25d2c27a8d818c35026addb1a661b0900fc92 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 13 Jan 2024 11:10:23 -0800 Subject: [PATCH] (test) hidden params in stream_chunk builder --- litellm/tests/test_stream_chunk_builder.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py index b7bd0d2163..001ae07e09 100644 --- a/litellm/tests/test_stream_chunk_builder.py +++ b/litellm/tests/test_stream_chunk_builder.py @@ -120,7 +120,7 @@ def test_stream_chunk_builder_litellm_tool_call(): def test_stream_chunk_builder_litellm_tool_call_regular_message(): try: messages = [{"role": "user", "content": "Hey, how's it going?"}] - litellm.set_verbose = False + # litellm.set_verbose = True response = litellm.completion( model="gpt-3.5-turbo", messages=messages, @@ -138,6 +138,10 @@ def test_stream_chunk_builder_litellm_tool_call_regular_message(): == response.usage.completion_tokens + response.usage.prompt_tokens ) + # check provider is in hidden params + print("hidden params", response._hidden_params) + assert response._hidden_params["custom_llm_provider"] == "openai" + except Exception as e: pytest.fail(f"An exception occurred - {str(e)}")