fix(dynamo_db.py): add cost tracking support for key + user

litellm/proxy/db/base_client.py

@@ -1,5 +1,7 @@
 from typing import Any, Literal, List
-class CustomDB: 
+
+
+class CustomDB:
     """
     Implements a base class that we expect any custom db implementation (e.g. DynamoDB) to follow
     """
@@ -7,37 +9,45 @@ class CustomDB:
     def __init__(self) -> None:
         pass
 
-    def get_data(self, key: str, value: str, table_name: Literal["user", "key", "config"]):
+    def get_data(self, key: str, table_name: Literal["user", "key", "config"]):
         """
         Check if key valid
         """
         pass
-    
-    def insert_data(self, value: Any, table_name: Literal["user", "key", "config"]): 
+
+    def insert_data(self, value: Any, table_name: Literal["user", "key", "config"]):
         """
         For new key / user logic
         """
         pass
 
-    def update_data(self, key: str, value: Any, table_name: Literal["user", "key", "config"]): 
+    def update_data(
+        self, key: str, value: Any, table_name: Literal["user", "key", "config"]
+    ):
         """
         For cost tracking logic
         """
         pass
 
-    def delete_data(self, keys: List[str], table_name: Literal["user", "key", "config"]): 
+    def delete_data(
+        self, keys: List[str], table_name: Literal["user", "key", "config"]
+    ):
         """
         For /key/delete endpoint s
         """
-    
-    def connect(self, ):
+
+    def connect(
+        self,
+    ):
+        """
+        For connecting to db and creating / updating any tables
         """
-        For connecting to db and creating / updating any tables 
-        """ 
         pass
 
-    def disconnect(self, ): 
+    def disconnect(
+        self,
+    ):
         """
         For closing connection on server shutdown
         """
-        pass
+        pass