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

fix(cohere/chat.py): fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-30 16:01:04 -08:00
parent 2fbc71a62c
commit 927f9fa4eb
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/cohere/chat.py
View file

@ -378,7 +378,7 @@ def completion(
raise CohereError(message=response.text, status_code=response.status_code) raise CohereError(message=response.text, status_code=response.status_code)
if "stream" in optional_params and optional_params["stream"] is True: 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, client=client,
api_base=api_base, api_base=api_base,
headers=headers, # type: ignore headers=headers, # type: ignore
@ -393,7 +393,7 @@ def completion(
model=model, model=model,
custom_llm_provider="cohere_chat", custom_llm_provider="cohere_chat",
logging_obj=logging_obj, logging_obj=logging_obj,
_response_headers=headers, _response_headers=dict(cohere_headers),
) )
else: else:
## LOGGING ## LOGGING