diff --git a/pyproject.toml b/pyproject.toml index f20201d..1f0e288 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta" [project] name = "draive" description = "Framework designed to simplify and accelerate the development of LLM-based applications." -version = "0.28.1" +version = "0.28.2" readme = "README.md" maintainers = [ { name = "Kacper KaliƄski", email = "kacper.kalinski@miquido.com" }, diff --git a/src/draive/__init__.py b/src/draive/__init__.py index 7d7ac43..d594cab 100644 --- a/src/draive/__init__.py +++ b/src/draive/__init__.py @@ -182,6 +182,7 @@ not_missing, setup_logging, split_sequence, + throttle, with_timeout, ) @@ -320,6 +321,7 @@ "TextGeneration", "TextGenerator", "TextTokenizer", + "throttle", "Tokenization", "tokenize_text", "TokenUsage", diff --git a/src/draive/utils/__init__.py b/src/draive/utils/__init__.py index 9b63e2b..b7caa7d 100644 --- a/src/draive/utils/__init__.py +++ b/src/draive/utils/__init__.py @@ -11,6 +11,7 @@ from draive.utils.queue import AsyncQueue from draive.utils.split_sequence import split_sequence from draive.utils.stream import AsyncStream +from draive.utils.throttling import throttle from draive.utils.timeout import with_timeout __all__ = [ @@ -37,5 +38,6 @@ "not_missing", "setup_logging", "split_sequence", + "throttle", "with_timeout", ] diff --git a/src/draive/utils/throttle.py b/src/draive/utils/throttling.py similarity index 100% rename from src/draive/utils/throttle.py rename to src/draive/utils/throttling.py