mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-21 06:27:00 +00:00
test(test_async_fn.py): adding additional logging for tests
This commit is contained in:
@@ -30,6 +30,8 @@ def test_get_response_streaming():
|
||||
user_message = "Hello, how are you?"
|
||||
messages = [{"content": user_message, "role": "user"}]
|
||||
try:
|
||||
import litellm
|
||||
litellm.set_verbose = True
|
||||
response = await acompletion(model="gpt-3.5-turbo", messages=messages, stream=True)
|
||||
print(type(response))
|
||||
|
||||
|
||||
@@ -311,7 +311,6 @@ def test_redis_cache_completion():
|
||||
def custom_get_cache_key(*args, **kwargs):
|
||||
# return key to use for your cache:
|
||||
key = kwargs.get("model", "") + str(kwargs.get("messages", "")) + str(kwargs.get("temperature", "")) + str(kwargs.get("logit_bias", ""))
|
||||
print("key for cache", key)
|
||||
return key
|
||||
|
||||
def test_custom_redis_cache_with_key():
|
||||
|
||||
Reference in New Issue
Block a user