From 868a415aa0160985ee27850f0faef04e06b1dfb1 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 2 Mar 2024 11:12:09 -0800
Subject: [PATCH] Revert "(feat) track used api_base in response"

This reverts commit 3aba9019ac27481361c3fa87956907f5c41dd4ca.
---
 litellm/router.py | 34 ----------------------------------
 1 file changed, 34 deletions(-)

diff --git a/litellm/router.py b/litellm/router.py
index 59cbf43d5..6f33d0b0d 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -357,11 +357,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.completion(model={model_name})\033[32m 200 OK\033[0m"
             )
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(
@@ -451,12 +446,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.acompletion(model={model_name})\033[32m 200 OK\033[0m"
             )
-
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(
@@ -613,12 +602,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.aimage_generation(model={model_name})\033[32m 200 OK\033[0m"
             )
-
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(
@@ -711,11 +694,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.amoderation(model={model_name})\033[32m 200 OK\033[0m"
             )
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(
@@ -847,12 +825,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.atext_completion(model={model_name})\033[32m 200 OK\033[0m"
             )
-
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(
@@ -982,12 +954,6 @@ class Router:
             verbose_router_logger.info(
                 f"litellm.aembedding(model={model_name})\033[32m 200 OK\033[0m"
             )
-
-            # set used api_base in response
-            if hasattr(response, "_hidden_params"):
-                response._hidden_params["api_base"] = deployment.get(
-                    "litellm_params", {}
-                ).get("api_base", None)
             return response
         except Exception as e:
             verbose_router_logger.info(