diff --git a/litellm/llms/novita/chat/transformation.py b/litellm/llms/novita/chat/transformation.py index bf30ba455b..c05d2d7b2c 100644 --- a/litellm/llms/novita/chat/transformation.py +++ b/litellm/llms/novita/chat/transformation.py @@ -19,6 +19,7 @@ class NovitaConfig(OpenAIGPTConfig): model: str, messages: List[AllMessageValues], optional_params: dict, + litellm_params: dict, api_key: Optional[str] = None, api_base: Optional[str] = None, ) -> dict: @@ -26,10 +27,7 @@ class NovitaConfig(OpenAIGPTConfig): raise ValueError( "Missing Novita AI API Key - A call is being made to novita but no key is set either in the environment variables or via params" ) - - headers = { - "Authorization": f"Bearer {api_key}", - "Content-Type": "application/json", - "X-Novita-Source": "litellm", - } + headers["Authorization"] = f"Bearer {api_key}" + headers["Content-Type"] = "application/json" + headers["X-Novita-Source"] = "litellm" return headers diff --git a/tests/litellm/llms/novita/chat/test_transformation.py b/tests/litellm/llms/novita/chat/test_transformation.py index 25b87e24dc..5216ea9488 100644 --- a/tests/litellm/llms/novita/chat/test_transformation.py +++ b/tests/litellm/llms/novita/chat/test_transformation.py @@ -33,6 +33,7 @@ class TestNovitaConfig: model="novita/meta-llama/llama-3.3-70b-instruct", messages=[{"role": "user", "content": "Hello"}], optional_params={}, + litellm_params={}, api_key=api_key, api_base="https://api.novita.ai/v3/openai" ) @@ -52,6 +53,7 @@ class TestNovitaConfig: model="novita/meta-llama/llama-3.3-70b-instruct", messages=[{"role": "user", "content": "Hello"}], optional_params={}, + litellm_params={}, api_key=None, api_base="https://api.novita.ai/v3/openai" )