docs(streaming_logging.md): fix example

2026-07-14 18:20:55 +00:00 · 2024-07-15 14:58:15 -07:00
parent e8e31c4029
commit ff1bcdd3b5
1 changed files with 1 additions and 1 deletions
@@ -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: