phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

clean up unused func

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-06 20:58:51 -07:00
parent 22ca6d6d30
commit 7bcb088ac0
2 changed files with 5 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

58
litellm/proxy/common_utils/admin_ui_utils.py
View file

@ -166,61 +166,3 @@ def missing_keys_form(missing_key_names: str):
</html>
"""
return missing_keys_html_form.format(missing_keys=missing_key_names)
def setup_admin_ui_on_server_root_path(server_root_path: str):
"""
Helper util to setup Admin UI on Server root path
"""
from litellm._logging import verbose_proxy_logger
if server_root_path != "":
print("setting proxy base url to server root path") # noqa
if os.getenv("PROXY_BASE_URL") is None:
os.environ["PROXY_BASE_URL"] = server_root_path
# re-build admin UI on server root path
# Save the original directory
original_dir = os.getcwd()
current_dir = (
os.path.dirname(os.path.abspath(__file__))
+ "/../../../ui/litellm-dashboard/"
)
build_ui_path = os.path.join(current_dir, "build_ui_custom_path.sh")
package_path = os.path.join(current_dir, "package.json")
print(f"Setting up Admin UI on {server_root_path}/ui .......") # noqa
try:
# Change the current working directory
os.chdir(current_dir)
# Make the script executable
subprocess.run(["chmod", "+x", "build_ui_custom_path.sh"], check=True)
# Run npm install
subprocess.run(["npm", "install"], check=True)
# Run npm run build
subprocess.run(["npm", "run", "build"], check=True)
# Run the custom build script with the argument
subprocess.run(
["./build_ui_custom_path.sh", f"{server_root_path}/ui"], check=True
)
print("Admin UI setup completed successfully.") # noqa
except subprocess.CalledProcessError as e:
print(f"An error occurred during the Admin UI setup: {e}") # noqa
except Exception as e:
print(f"An unexpected error occurred: {e}") # noqa
finally:
# Always return to the original directory, even if an error occurred
os.chdir(original_dir)
print(f"Returned to original directory: {original_dir}") # noqa
pass

5
litellm/proxy/proxy_server.py
View file

@ -138,7 +138,6 @@ from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
from litellm.proxy.caching_routes import router as caching_router
from litellm.proxy.common_utils.admin_ui_utils import (
html_form,
setup_admin_ui_on_server_root_path,
show_missing_vars_in_env,
)
from litellm.proxy.common_utils.debug_utils import init_verbose_loggers
@ -388,6 +387,10 @@ try:
os.rename(src, dst)
if server_root_path != "":
print( # noqa
f"server_root_path is set, forwarding any /ui requests to {server_root_path}/ui"
) # noqa
if os.getenv("PROXY_BASE_URL") is None:
os.environ["PROXY_BASE_URL"] = server_root_path
@app.middleware("http")