Merge branch 'main' into dev

dreadnode · May 30, 2024 · 1b6bb7d · 1b6bb7d
2 parents c8c80b4 + 1148334
commit 1b6bb7d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rigging/generator/__init__.py b/rigging/generator/__init__.py
@@ -7,6 +7,8 @@
     GeneratedText,
     GenerateParams,
     Generator,
+    StopReason,
+    Usage,
     chat,
     complete,
     get_generator,
@@ -31,6 +33,8 @@
     "GenerateParams",
     "GeneratedMessage",
     "GeneratedText",
+    "StopReason",
+    "Usage",
     "chat",
     "complete",
     "get_generator",