phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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

BIN
dist/litellm-0.1.636-py3-none-any.whl vendored Normal file
View file

Binary file not shown.

BIN
dist/litellm-0.1.636.tar.gz vendored Normal file
View file

Binary file not shown.

5
litellm/utils.py
View file

@ -89,10 +89,11 @@ class Message(OpenAIObject):
self.logprobs = logprobs self.logprobs = logprobs
class Delta(OpenAIObject): class Delta(OpenAIObject):
def __init__(self, content=" ", logprobs=None, **params): def __init__(self, content=" ", logprobs=None, role="assistant", **params):
super(Delta, self).__init__(**params) super(Delta, self).__init__(**params)
self.content = content self.content = content
self.logprobs = logprobs self.logprobs = logprobs
self.role = role
class Choices(OpenAIObject): 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: elif model in litellm.ai21_models:
custom_llm_provider = "ai21" custom_llm_provider = "ai21"
## together_ai ## together_ai
elif model in litellm.together_ai_models: elif model in litellm.together_ai_models or "togethercomputer":
custom_llm_provider = "together_ai" custom_llm_provider = "together_ai"
## aleph_alpha ## aleph_alpha
elif model in litellm.aleph_alpha_models: elif model in litellm.aleph_alpha_models:

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.638" version = "0.1.639"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"