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

bump version

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-15 12:24:00 -07:00
parent 9a815528b7
commit be69b42428
4 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -89,10 +89,11 @@ class Message(OpenAIObject):
self.logprobs = logprobs
class Delta(OpenAIObject):
def __init__(self, content=" ", logprobs=None, **params):
def __init__(self, content=" ", logprobs=None, role="assistant", **params):
super(Delta, self).__init__(**params)
self.content = content
self.logprobs = logprobs
self.role = role
class Choices(OpenAIObject):
@ -1033,7 +1034,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None):
elif model in litellm.ai21_models:
custom_llm_provider = "ai21"
## together_ai
elif model in litellm.together_ai_models:
elif model in litellm.together_ai_models or "togethercomputer":
custom_llm_provider = "together_ai"
## aleph_alpha
elif model in litellm.aleph_alpha_models: