scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

(test) custom logger

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 12:35:12 -08:00
parent 89d9a5d38d
commit e23700dd19
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litellm/tests/test_aamazing_proxy_custom_logger.py
View file

@ -16,12 +16,6 @@ from litellm import embedding, completion, completion_cost, Timeout
from litellm import RateLimitError
import importlib, inspect
# init callbacks to be empty
litellm.success_callback = []
litellm.callbacks = []
litellm.failure_callback = []
litellm._async_success_callback = []
# test /chat/completion request to the proxy
from fastapi.testclient import TestClient
from fastapi import FastAPI
@ -54,6 +48,12 @@ headers = {
"Authorization": f"Bearer {token}"
}
# init callbacks to be empty
litellm.success_callback = []
litellm.callbacks = []
litellm.failure_callback = []
litellm._async_success_callback = []
def test_embedding(client):
try:
# Your test data
@ -61,7 +61,8 @@ def test_embedding(client):
# import the initialized custom logger
print(litellm.callbacks)
assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
# assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
my_custom_logger = litellm.callbacks[0]
assert my_custom_logger.async_success_embedding == False
@ -98,7 +99,7 @@ def test_chat_completion(client):
# import the initialized custom logger
print(litellm.callbacks)
assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
# assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
my_custom_logger = litellm.callbacks[0]
assert my_custom_logger.async_success == False
@ -145,7 +146,7 @@ def test_chat_completion_stream(client):
# import the initialized custom logger
print(litellm.callbacks)
assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
# assert len(litellm.callbacks) == 1 # assert litellm is initialized with 1 callback
my_custom_logger = litellm.callbacks[0]
assert my_custom_logger.streaming_response_obj == None # no streaming response obj is set pre call