Merge pull request #2475 from BerriAI/litellm_azure_dall_e_3_cost_tracking

fix(azure.py): support cost tracking for azure/dall-e-3
2024-03-12 12:57:31 -07:00 · 2024-03-12 12:57:31 -07:00 · 1ba102c618
commit 1ba102c618
parent a18c941621 7dd94c802e
3 changed files with 44 additions and 1 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -3841,7 +3841,9 @@ def completion_cost(
                    * n
                )
            else:
-                raise Exception(f"Model={model} not found in completion cost model map")
+                raise Exception(
+                    f"Model={image_gen_model_name} not found in completion cost model map"
+                )
        # Calculate cost based on prompt_tokens, completion_tokens
        if (
            "togethercomputer" in model