From 927f9fa4eb94138fca068da191affaeb050a4cc0 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 30 Nov 2024 16:01:04 -0800
Subject: [PATCH] fix(cohere/chat.py): fix linting errors

---
 litellm/llms/cohere/chat.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/llms/cohere/chat.py b/litellm/llms/cohere/chat.py
index d65c4f895..d2288b3b1 100644
--- a/litellm/llms/cohere/chat.py
+++ b/litellm/llms/cohere/chat.py
@@ -378,7 +378,7 @@ def completion(
         raise CohereError(message=response.text, status_code=response.status_code)
 
     if "stream" in optional_params and optional_params["stream"] is True:
-        completion_stream, headers = make_sync_call(
+        completion_stream, cohere_headers = make_sync_call(
             client=client,
             api_base=api_base,
             headers=headers,  # type: ignore
@@ -393,7 +393,7 @@ def completion(
             model=model,
             custom_llm_provider="cohere_chat",
             logging_obj=logging_obj,
-            _response_headers=headers,
+            _response_headers=dict(cohere_headers),
         )
     else:
         ## LOGGING