feat(vertex_ai_partner.py): initial working commit for calling vertex ai mistral

Closes https://github.com/BerriAI/litellm/issues/4874
This commit is contained in:
Krrish Dholakia 2024-07-27 12:54:14 -07:00
parent 1a8f45e8da
commit 5b71421a7b
10 changed files with 343 additions and 140 deletions

View file

@ -160,7 +160,7 @@ class MistralConfig:
optional_params["max_tokens"] = value
if param == "tools":
optional_params["tools"] = value
if param == "stream" and value == True:
if param == "stream" and value is True:
optional_params["stream"] = value
if param == "temperature":
optional_params["temperature"] = value