phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Revert "(feat) track used api_base in response"

Browse source 
This reverts commit 3aba9019ac.
This commit is contained in:
ishaan-jaff 2024-03-02 11:12:09 -08:00
parent 127bc743b2
commit 868a415aa0
1 changed files with 0 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

34
litellm/router.py
View file

@ -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(