(feat) track cost for responses easily

2026-07-15 22:22:12 +00:00 · 2023-10-28 15:08:35 -07:00
parent 59e47fac78
commit d2ff7a17df
1 changed files with 4 additions and 0 deletions
@@ -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"