From c489f41964a3300f3ff66f7f27ded49721f9197b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 13 Nov 2023 17:13:41 -0800 Subject: [PATCH] test(utils.py): additional logging --- litellm/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index ff746a920e3..e2593b4a222 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -4377,11 +4377,12 @@ class CustomStreamWrapper: return {"text": text, "is_finished": is_finished, "finish_reason": finish_reason} elif str_line.startswith("data:"): data_json = json.loads(str_line[5:]) - print_verbose(f"delta content: {data_json['choices'][0]['text']}") + print_verbose(f"delta content: {data_json}") text = data_json["choices"][0].get("text", "") if data_json["choices"][0].get("finish_reason", None): is_finished = True finish_reason = data_json["choices"][0]["finish_reason"] + print_verbose(f"text: {text}; is_finished: {is_finished}; finish_reason: {finish_reason}") return {"text": text, "is_finished": is_finished, "finish_reason": finish_reason} elif "error" in str_line: raise ValueError(f"Unable to parse response. Original response: {str_line}")