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

test proxy server.py

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-15 15:09:49 -07:00
parent 6a5a9d02a9
commit ae86722a8d
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_proxy_server.py
View file

@ -26,7 +26,7 @@ logging.basicConfig(
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from fastapi import FastAPI from fastapi import FastAPI
from litellm.proxy.proxy_server import ( from litellm.proxy.proxy_server import (
router, app,
save_worker_config, save_worker_config,
initialize, initialize,
) # Replace with the actual module where your FastAPI router is defined ) # Replace with the actual module where your FastAPI router is defined
@ -119,9 +119,6 @@ def client_no_auth(fake_env_vars):
config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml" config_fp = f"{filepath}/test_configs/test_config_no_auth.yaml"
# initialize can get run in parallel, it sets specific variables for the fast api app, sinc eit gets run in parallel different tests use the wrong variables # initialize can get run in parallel, it sets specific variables for the fast api app, sinc eit gets run in parallel different tests use the wrong variables
asyncio.run(initialize(config=config_fp, debug=True)) asyncio.run(initialize(config=config_fp, debug=True))
app = FastAPI()
app.include_router(router) # Include your router in the test app
return TestClient(app) return TestClient(app)
@ -426,6 +423,10 @@ def test_add_new_model(client_no_auth):
def test_health(client_no_auth): def test_health(client_no_auth):
global headers global headers
import time import time
from litellm._logging import verbose_logger, verbose_proxy_logger
import logging
verbose_proxy_logger.setLevel(logging.DEBUG)
try: try:
response = client_no_auth.get("/health") response = client_no_auth.get("/health")