tPj(ddlmZmZGddZy))AnyLiteralceZdZdZddZdededfdZdededfd Z dedededfd Z d e ededfd Z d Z dZy)CustomDBzg Implements a base class that we expect any custom db implementation (e.g. DynamoDB) to follow Ncy)Nselfs a/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/db/base_client.py__init__zCustomDB.__init__ s key table_name)userrconfigcy)z$ Check if key valid Nr)r rrs r get_datazCustomDB.get_data r valuecy)z* For new key / user logic Nr)r rrs r insert_datazCustomDB.insert_datarr cy)z) For cost tracking logic Nr)r rrrs r update_datazCustomDB.update_datarr keyscy)z, For /key/delete endpoint s Nr)r rrs r delete_datazCustomDB.delete_datarr cy)zI For connecting to db and creating / updating any tables Nrr s r connectzCustomDB.connect rr cy)z; For closing connection on server shutdown Nrr s r disconnectzCustomDB.disconnect'rr )returnN)__name__ __module__ __qualname____doc__r strrrrrrlistrrr rr r rrs  C W5L-M   ':Q2R  s 3 GD[<\  S  w?V7W   r rN)typingrrrrr r r)s( ( r