mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-23 23:40:07 +00:00
Krish Dholakia
e0fa33f099
* fix(ui_sso.py): update user as proxy admin in db table, when checking for proxy_admin_id Fixes issue where existing internal user, unable to make calls when set as proxy admin id * fix(utils.py): fix custom base path
22 lines
518 B
Python
22 lines
518 B
Python
import json
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../../..")
|
|
) # Adds the parent directory to the system path
|
|
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from litellm.proxy.utils import get_custom_url
|
|
|
|
|
|
def test_get_custom_url(monkeypatch):
|
|
monkeypatch.setenv("SERVER_ROOT_PATH", "/litellm")
|
|
custom_url = get_custom_url(request_base_url="http://0.0.0.0:4000", route="ui/")
|
|
assert custom_url == "http://0.0.0.0:4000/litellm/ui/"
|