036ad54ac6
This PR ensures that `arcade.core` does not show up anywhere in "user space". This is crucial for helping developers understand what objects are safe to use, and helps maintain a good developer experience. Specific changes: - `ToolAuthorizationContext` and `ToolContext` are now visible via `arcade.sdk` - `ToolCatalog` is now visible via `arcade.sdk` - `Toolkit` is now visible via `arcade.sdk` - `config` is now visible via `arcade.sdk.config`
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import functools
|
|
import inspect
|
|
from typing import Any, Callable, TypeVar, Union
|
|
|
|
from arcade.core.utils import snake_to_pascal_case
|
|
from arcade.sdk.auth import ToolAuthorization
|
|
from arcade.sdk.errors import ToolExecutionError
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def tool(
|
|
func: Callable | None = None,
|
|
desc: str | None = None,
|
|
name: str | None = None,
|
|
requires_auth: Union[ToolAuthorization, None] = None,
|
|
) -> Callable:
|
|
def decorator(func: Callable) -> Callable:
|
|
func_name = str(getattr(func, "__name__", None))
|
|
tool_name = name or snake_to_pascal_case(func_name)
|
|
|
|
func.__tool_name__ = tool_name # type: ignore[attr-defined]
|
|
func.__tool_description__ = desc or inspect.cleandoc(func.__doc__ or "") # type: ignore[attr-defined]
|
|
func.__tool_requires_auth__ = requires_auth # type: ignore[attr-defined]
|
|
|
|
if inspect.iscoroutinefunction(func):
|
|
|
|
@functools.wraps(func)
|
|
async def func_with_error_handling(*args: Any, **kwargs: Any) -> Any:
|
|
try:
|
|
return await func(*args, **kwargs)
|
|
|
|
# make sure developer raised ToolExecutionError is not
|
|
# reraised incorrectly.
|
|
except ToolExecutionError:
|
|
raise
|
|
except Exception as e:
|
|
raise ToolExecutionError(
|
|
message=f"Error in execution of {tool_name}",
|
|
developer_message=f"Error in {func_name}: {e!s}",
|
|
) from e
|
|
|
|
else:
|
|
|
|
@functools.wraps(func)
|
|
def func_with_error_handling(*args: Any, **kwargs: Any) -> Any:
|
|
try:
|
|
return func(*args, **kwargs)
|
|
except ToolExecutionError:
|
|
raise
|
|
except Exception as e:
|
|
raise ToolExecutionError(
|
|
message=f"Error in execution of {tool_name}",
|
|
developer_message=f"Error in {func_name}: {e!s}",
|
|
) from e
|
|
|
|
return func_with_error_handling
|
|
|
|
if func:
|
|
return decorator(func)
|
|
return decorator
|