fix import

This commit is contained in:
ishaan-jaff 2023-08-28 14:21:25 -07:00
parent ecec028f38
commit 43fea52d7c

View file

@ -1,5 +1,3 @@
import litellm, openai
def get_prompt(*args, **kwargs): def get_prompt(*args, **kwargs):
# make this safe checks, it should not throw any exceptions # make this safe checks, it should not throw any exceptions
if len(args) > 1: if len(args) > 1:
@ -13,8 +11,8 @@ def get_prompt(*args, **kwargs):
return None return None
class RedisCache(): class RedisCache():
import redis
def __init__(self, host, port, password): def __init__(self, host, port, password):
import redis
# if users don't provider one, use the default litellm cache # if users don't provider one, use the default litellm cache
self.redis_client = redis.Redis(host=host, port=port, password=password) self.redis_client = redis.Redis(host=host, port=port, password=password)