Fix throttle symbol availability

miquido · Aug 29, 2024 · abf0476 · abf0476
1 parent be9ee9d
commit abf0476
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git 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 = "[email protected]" },

diff --git 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
@@ -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 → src/draive/utils/throttling.py b/src/draive/utils/throttle.py → src/draive/utils/throttling.py