From ff1bcdd3b5ce32812bc349f27c22defce58cb17e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 15 Jul 2024 14:58:15 -0700 Subject: [PATCH] docs(streaming_logging.md): fix example --- docs/my-website/docs/proxy/streaming_logging.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/my-website/docs/proxy/streaming_logging.md b/docs/my-website/docs/proxy/streaming_logging.md index eb248424d9..a9d577677c 100644 --- a/docs/my-website/docs/proxy/streaming_logging.md +++ b/docs/my-website/docs/proxy/streaming_logging.md @@ -26,7 +26,7 @@ class MyCustomHandler(CustomLogger): datefmt='%Y-%m-%d %H:%M:%S' ) - response_cost = litellm.completion_cost(completion_response=response_obj) + response_cost: Optional[float] = kwargs.get("response_cost", None) print("regular response_cost", response_cost) logging.info(f"Model {response_obj.model} Cost: ${response_cost:.8f}") except: