phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix novita transformation and test (#10056)

Browse source
This commit is contained in:
Jason 2025-04-23 12:50:02 +08:00 • committed by GitHub
parent ba03736077
commit d23e1d5a9d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/novita/chat/transformation.py
View file

@ -19,6 +19,7 @@ class NovitaConfig(OpenAIGPTConfig):
model: str, model: str,
messages: List[AllMessageValues], messages: List[AllMessageValues],
optional_params: dict, optional_params: dict,
litellm_params: dict,
api_key: Optional[str] = None, api_key: Optional[str] = None,
api_base: Optional[str] = None, api_base: Optional[str] = None,
) -> dict: ) -> dict:
@ -26,10 +27,7 @@ class NovitaConfig(OpenAIGPTConfig):
raise ValueError( 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" "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}"
headers = { headers["Content-Type"] = "application/json"
"Authorization": f"Bearer {api_key}", headers["X-Novita-Source"] = "litellm"
"Content-Type": "application/json",
"X-Novita-Source": "litellm",
}
return headers return headers

2
tests/litellm/llms/novita/chat/test_transformation.py
View file

@ -33,6 +33,7 @@ class TestNovitaConfig:
model="novita/meta-llama/llama-3.3-70b-instruct", model="novita/meta-llama/llama-3.3-70b-instruct",
messages=[{"role": "user", "content": "Hello"}], messages=[{"role": "user", "content": "Hello"}],
optional_params={}, optional_params={},
litellm_params={},
api_key=api_key, api_key=api_key,
api_base="https://api.novita.ai/v3/openai" api_base="https://api.novita.ai/v3/openai"
) )
@ -52,6 +53,7 @@ class TestNovitaConfig:
model="novita/meta-llama/llama-3.3-70b-instruct", model="novita/meta-llama/llama-3.3-70b-instruct",
messages=[{"role": "user", "content": "Hello"}], messages=[{"role": "user", "content": "Hello"}],
optional_params={}, optional_params={},
litellm_params={},
api_key=None, api_key=None,
api_base="https://api.novita.ai/v3/openai" api_base="https://api.novita.ai/v3/openai"
) )