diff --git a/litellm/main.py b/litellm/main.py index ee9c9a423..16d6bd29d 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -105,7 +105,7 @@ async def acompletion(*args, **kwargs): else: return response -def mock_completion(model: str, messages: List, stream: bool = False, mock_response: str = "This is a mock request", **kwargs): +def mock_completion(model: str, messages: List, stream: Optional[bool] = False, mock_response: str = "This is a mock request", **kwargs): """ Generate a mock completion response for testing or debugging purposes. diff --git a/pyproject.toml b/pyproject.toml index cc845f058..44398321a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.810" +version = "0.1.811" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"