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

undo change to always setting role to assistant

Browse source
This commit is contained in:
ishaan-jaff 2023-09-15 15:16:58 -07:00
parent 437c2b2316
commit b966baa1f0
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -2451,7 +2451,7 @@ class CustomStreamWrapper:
def __next__(self): def __next__(self):
try: try:
completion_obj = {"content": "", "role": "assistant"} # default to role being assistant completion_obj = {"content": "", "role": ""} # default to role being assistant
if self.model in litellm.anthropic_models: if self.model in litellm.anthropic_models:
chunk = next(self.completion_stream) chunk = next(self.completion_stream)
completion_obj["content"] = self.handle_anthropic_chunk(chunk) completion_obj["content"] = self.handle_anthropic_chunk(chunk)

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.643" version = "0.1.644"
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"