o Æ°iiã@s&ddlmZmZmZGdd„dƒZdS)é)ÚAnyÚLiteralÚListc@sŠeZdZdZddd„Zdededfd d „Zd ededfd d „Z ded ededfdd„Z de ededfdd„Z dd„Z dd„ZdS)ÚCustomDBzg Implements a base class that we expect any custom db implementation (e.g. DynamoDB) to follow ÚreturnNcCsdS)N©©ÚselfrrúS/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/db/base_client.pyÚ__init__ szCustomDB.__init__ÚkeyÚ table_name)Úuserr ÚconfigcCódS)z$ Check if key valid Nr)r r r rrr Úget_data ózCustomDB.get_dataÚvaluecCr)z* For new key / user logic Nr)r rr rrr Ú insert_datarzCustomDB.insert_datacCr)z) For cost tracking logic Nr)r r rr rrr Ú update_dataózCustomDB.update_dataÚkeyscCr)z, For /key/delete endpoint s Nr)r rr rrr Ú delete_data szCustomDB.delete_datacCr)zI For connecting to db and creating / updating any tables Nrrrrr Úconnect'rzCustomDB.connectcCr)z; For closing connection on server shutdown Nrrrrr Ú disconnect/rzCustomDB.disconnect)rN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r Ústrrrrrrrrrrrrrr rs& ÿÿ ÿÿ ÿ rN)Útypingrrrrrrrr Ús