Merge pull request #690 from Undertone0809/v1.16.6/add-tooltype-dec

perf: add tooltype declare
Undertone0809 · May 25, 2024 · 5a64460 · 5a64460
2 parents 1b93e41 + 7d1e488
commit 5a64460
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/promptulate/__init__.py b/promptulate/__init__.py
@@ -36,7 +36,7 @@
     SystemMessage,
     UserMessage,
 )
-from promptulate.tools.base import BaseTool, Tool, define_tool
+from promptulate.tools.base import BaseTool, Tool, ToolTypes, define_tool
 from promptulate.utils.logger import enable_log
 from promptulate.utils.string_template import StringTemplate
 
@@ -60,6 +60,7 @@
     "Tool",
     "define_tool",
     "BaseTool",
+    "ToolTypes",
 ]
 
 _agent_fields = [

diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ name = "promptulate"
 readme = "README.md"
 homepage = "https://github.com/Undertone0809/promptulate"
 repository = "https://github.com/Undertone0809/promptulate"
-version = "1.16.5"
+version = "1.16.6"
 keywords = [
     "promptulate",
     "pne",

diff --git a/tests/test_import.py b/tests/test_import.py
@@ -1,3 +1,9 @@
+def test_import_init():
+    import promptulate as pne
+
+    pne.ToolTypes  # noqa
+
+
 def test_import_utils():
     from promptulate.utils import (
         StringTemplate,