ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix cohere streaming

Browse Source
This commit is contained in:
Krrish Dholakia 2023-09-16 19:36:41 -07:00
parent e44c218c1b
commit e4fbc8d908
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
litellm/__pycache__/main.cpython-311.pyc
View File

Binary file not shown.

2
litellm/main.py
View File

@ -571,7 +571,7 @@ def completion(
if "stream" in optional_params and optional_params["stream"] == True:
# don't try to access stream object,
response = CustomStreamWrapper(model_response, model, logging_obj=logging)
response = CustomStreamWrapper(model_response, model, custom_llm_provider="cohere", logging_obj=logging)
return response
response = model_response
elif (

2
pyproject.toml
View File

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