diff --git a/litellm/utils.py b/litellm/utils.py index 1475dcb586..445da0db55 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -4443,6 +4443,9 @@ class CustomStreamWrapper: is_finished = True finish_reason = stop_reason ######## bedrock.cohere mappings ############### + # meta mapping + elif "generation" in chunk_data: + text = chunk_data['generation'] # bedrock.meta # cohere mapping elif "text" in chunk_data: text = chunk_data["text"] # bedrock.cohere