make succes_callback in Callable

litellm/__init__.py

@@ -1,9 +1,9 @@
 import threading, requests
-from typing import Callable, List, Optional, Dict
+from typing import Callable, List, Optional, Dict, Union
 from litellm.caching import Cache
 
 input_callback: List[str] = []
-success_callback: List[str] = []
+success_callback: List[Union[str, Callable]] = []
 failure_callback: List[str] = []
 set_verbose = False
 email: Optional[

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.578"
+version = "0.1.580"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"