From eb25e9f15f0f0c69007189b1c7776308e36d3817 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Sep 2023 15:19:42 -0700 Subject: [PATCH] add bedrock anthropic streaming --- litellm/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index b5c9ad2fe..897ca0be1 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3026,6 +3026,8 @@ class CustomStreamWrapper: finish_reason = "" if "outputText" in chunk_data: text = chunk_data['outputText'] + if "completion" in chunk_data: + text = chunk_data['completion'] # bedrock.anthropic if chunk_data.get("completionReason", None): is_finished = True finish_reason = chunk_data["completionReason"]