diff --git a/litellm/utils.py b/litellm/utils.py index 76a43c6779..99e9014372 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"