From e4fbc8d9089d1491880360148ca50112fed78211 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 16 Sep 2023 19:36:41 -0700
Subject: [PATCH] fix cohere streaming

---
 litellm/__pycache__/main.cpython-311.pyc | Bin 34057 -> 34060 bytes
 litellm/main.py                          |   2 +-
 pyproject.toml                           |   2 +-
 3 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
index 7bed614a6cea6ed3d3a0cd67ca58596db8503b83..593e38f0d537c848251d0faa19a1397c3840e422 100644
GIT binary patch
delta 425
zcmeC|V(RH);$6<m%f$c$jA?ACUpDdv@iS&@PT-GXWqh=Gx5#BC#){4T;$N8=Yd5z_
zyD~D~+I(6@pOMjN^KV&cM#e3h#pRtD8HF}SD=0HFHcYNpmS*gnJX`tr<PEY~j8T&V
zMI<K+$g^x-q2eOJR3yLogUw<d#v7Zbcz)yMi%PmsP<}zW;-YlL73qr2Q^Ox{*9$Q0
z^yGB(6JR*S>?{tV&q{0fu`rxtW(0B0v2gjbGoRx#@n>Q_&%_F3pJ!JGNu1YrRRPgg
zMO>vp^eqD?u<pAYPP`!geK99R5dF}?Ljy#=Qgjvo(Ql2MltA<cQLyxfdUKWn8-@?o
zVB0>}2o~}(f8=$R2PyfY>%<14znQR<Su%XH04w=sDOk?M{GH1g?B-u0&UPRr|AZaE
zrv5i`lmzh)Itn<7f*DR~PW)hoi<v9fAqTzWoejVuega&Meu9R~2mPdk9Q|~Fv<|DI
LpW)_~lvV`*2Euq<

delta 448
zcmeC_V(RQ-;$6<m%f$c$X$fqp*EaG7@iV4vPT-GXWqi1Kx5#BC#`4Ym;$N8=Yc{t^
zyD~D~+<aO_pOMjV^KV&cM#jyX#pRtD83i{-D=0HF)=#cimS*gjJX`rVqr&7y8F6Wm
z5`B=-5KyC0!%(BXCTbZo1H)?O$&B*Cn^&p0NU&%!7PW8wV6&Kq@#^L&p5J)6gCZ|T
zM_rVTx*{F5c}DmH?s|TPot~VIe*6rln4QHz^jT>QKNf~_%#0w;ITkK|cII<@CjLy!
z=b2c6?DOpEAc^z(t|}n<s)(yJh`wdu1lE0*!-*Hfzc1#b2%;Zac!1TvQg9Xk@!uLc
zDS_w@B4Ftc^=2#u)(ju4z_xv`7A)jp{>bAj4^r|)$B7L@e=}w&vtamU4p#EbLa>~Z
z`8%gG*v-F$o$WwM{s}pPP5p1`C<)>pbmVsw1v8vfo%q2F7gJXaFvCmE*#OM&<L7ep
b6Et8x=qD-U=%)>&wOJkg3^uo?v?>4qkdt~d

diff --git a/litellm/main.py b/litellm/main.py
index d099c1c177..60c4d802dd 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -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 (
diff --git a/pyproject.toml b/pyproject.toml
index 812377984b..843bdea018 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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"