diff --git a/litellm/__init__.py b/litellm/__init__.py
index 704b36e3bb..920833c113 100644
--- a/litellm/__init__.py
+++ b/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[
diff --git a/pyproject.toml b/pyproject.toml
index f255a1b231..8f8b312b2c 100644
--- a/pyproject.toml
+++ b/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"