docs(streaming_logging.md): fix example

nidhishs · Jul 15, 2024 · ff1bcdd · ff1bcdd
1 parent e8e31c4
commit ff1bcdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: