From d2ff7a17df43ba52200da40b3c4b66e7eb7b91cb Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 28 Oct 2023 15:07:45 -0700
Subject: [PATCH] (feat) track cost for responses easily

---
 litellm/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/litellm/utils.py b/litellm/utils.py
index 76a43c677..99e901437 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -201,6 +201,10 @@ class ModelResponse(OpenAIObject):
         d["choices"] = [choice.to_dict_recursive() for choice in self.choices]
         return d
 
+    def cost(self):
+        # for non streaming responses
+        return completion_cost(completion_response=self)
+
 class EmbeddingResponse(OpenAIObject):
     def __init__(self, id=None, choices=None, created=None, model=None, usage=None, stream=False, response_ms=None, **params):
         self.object = "list"