diff --git a/litellm/tests/test_aamazing_proxy_custom_logger.py b/litellm/tests/test_aamazing_proxy_custom_logger.py index 84883776e10..9325d5f88c4 100644 --- a/litellm/tests/test_aamazing_proxy_custom_logger.py +++ b/litellm/tests/test_aamazing_proxy_custom_logger.py @@ -33,7 +33,7 @@ async def wrapper_startup_event(): initialize(config=config_fp) # Use the app fixture in your client fixture -@pytest.fixture() +@pytest.fixture(scope="function") def client(): with TestClient(app) as client: yield client @@ -51,6 +51,8 @@ litellm.callbacks = [] litellm.failure_callback = [] litellm._async_success_callback = [] +print("Testing proxy custom logger") + def test_embedding(client): try: # Your test data @@ -93,6 +95,7 @@ def test_embedding(client): assert model_info == {'input_cost_per_token': 0.002, 'mode': 'embedding', 'id': 'hello'} result = response.json() print(f"Received response: {result}") + print("Passed Embedding custom logger on proxy!") except Exception as e: pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}") @@ -129,7 +132,8 @@ def test_chat_completion(client): response = client.post("/chat/completions", json=test_data, headers=headers) print("made request", response.status_code, response.text) print("LiteLLM Callbacks", litellm.callbacks) - print("my_custom_logger", my_custom_logger) + print("my_custom_logger in /chat/completions", my_custom_logger) + print("vars my custom logger, ", vars(my_custom_logger)) assert my_custom_logger.async_success == True # checks if the status of async_success is True, only the async_log_success_event can set this to true assert my_custom_logger.async_completion_kwargs["model"] == "chatgpt-v-2" # checks if kwargs passed to async_log_success_event are correct print("\n\n Custom Logger Async Completion args", my_custom_logger.async_completion_kwargs) @@ -221,3 +225,4 @@ def test_chat_completion_stream(client): except Exception as e: pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}") +