fix(utils.py): enable cost tracking for image gen models on proxy

2024-01-26 20:51:13 -08:00 · 2024-01-26 20:51:13 -08:00 · a299ac2328
commit a299ac2328
parent 511510a1cc
3 changed files with 43 additions and 42 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -2029,14 +2029,15 @@ def client(original_function):
                start_time=start_time,
            )
            ## check if metadata is passed in
+            litellm_params = {}
            if "metadata" in kwargs:
-                litellm_params = {"metadata": kwargs["metadata"]}
-                logging_obj.update_environment_variables(
-                    model=model,
-                    user="",
-                    optional_params={},
-                    litellm_params=litellm_params,
-                )
+                litellm_params["metadata"] = kwargs["metadata"]
+            logging_obj.update_environment_variables(
+                model=model,
+                user="",
+                optional_params={},
+                litellm_params=litellm_params,
+            )
            return logging_obj
        except Exception as e:
            import logging