diff --git a/pyproject.toml b/pyproject.toml index ea9f6e5..06b14aa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "superagent-py" -version = "v0.2.18" +version = "v0.2.19" description = "" readme = "README.md" authors = [] diff --git a/src/superagent/core/client_wrapper.py b/src/superagent/core/client_wrapper.py index d72c5ea..e9b79fc 100644 --- a/src/superagent/core/client_wrapper.py +++ b/src/superagent/core/client_wrapper.py @@ -16,7 +16,7 @@ def get_headers(self) -> typing.Dict[str, str]: headers: typing.Dict[str, str] = { "X-Fern-Language": "Python", "X-Fern-SDK-Name": "superagent-py", - "X-Fern-SDK-Version": "v0.2.18", + "X-Fern-SDK-Version": "v0.2.19", } token = self._get_token() if token is not None: diff --git a/src/superagent/types/tool_type.py b/src/superagent/types/tool_type.py index 9795bc0..f202eba 100644 --- a/src/superagent/types/tool_type.py +++ b/src/superagent/types/tool_type.py @@ -30,6 +30,7 @@ class ToolType(str, enum.Enum): FUNCTION = "FUNCTION" HTTP = "HTTP" SUPERRAG = "SUPERRAG" + RESEARCH = "RESEARCH" def visit( self, @@ -52,6 +53,7 @@ def visit( function: typing.Callable[[], T_Result], http: typing.Callable[[], T_Result], superrag: typing.Callable[[], T_Result], + research: typing.Callable[[], T_Result], ) -> T_Result: if self is ToolType.ALGOLIA: return algolia() @@ -91,3 +93,5 @@ def visit( return http() if self is ToolType.SUPERRAG: return superrag() + if self is ToolType.RESEARCH: + return research()