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

fix(vertex_ai_partner.py): add /chat/completion codestral support

Browse source 
Closes https://github.com/BerriAI/litellm/issues/4984
This commit is contained in:
Krrish Dholakia 2024-08-01 16:12:05 -07:00
parent f04160df46
commit 6e163d3c8a
5 changed files with 30 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -2074,7 +2074,11 @@ def completion(
timeout=timeout,
client=client,
)
elif model.startswith("meta/") or model.startswith("mistral"):
elif (
model.startswith("meta/")
or model.startswith("mistral")
or model.startswith("codestral")
):
model_response = vertex_partner_models_chat_completion.completion(
model=model,
messages=messages,