From ff0e5c29679ce97eed20d9ad19cc5efbb54958f6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 9 Sep 2023 19:05:19 -0700 Subject: [PATCH] make succes_callback in Callable --- litellm/__init__.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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"