Skip to content

Commit

Permalink
perf: add tooltype declare
Browse files Browse the repository at this point in the history
Undertone0809 committed May 25, 2024
1 parent 1b93e41 commit 7d1e488
Showing 3 changed files with 9 additions and 2 deletions.
3 changes: 2 additions & 1 deletion promptulate/__init__.py
Original file line number Diff line number Diff line change
@@ -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 = [
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -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",
6 changes: 6 additions & 0 deletions tests/test_import.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
def test_import_init():
import promptulate as pne

pne.ToolTypes # noqa


def test_import_utils():
from promptulate.utils import (
StringTemplate,

0 comments on commit 7d1e488

Please sign in to comment.