mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 05:55:16 +08:00
21 lines
616 B
Python
21 lines
616 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import inspect
|
||
|
|
from typing import TYPE_CHECKING, Any, cast
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from plyngent.agent import ToolDefinition
|
||
|
|
|
||
|
|
|
||
|
|
def call_sync(definition: ToolDefinition, *args: object, **kwargs: object) -> str:
|
||
|
|
result: Any = definition.handler(*args, **kwargs)
|
||
|
|
assert not inspect.isawaitable(result)
|
||
|
|
return cast("str", result)
|
||
|
|
|
||
|
|
|
||
|
|
async def call_async(definition: ToolDefinition, *args: object, **kwargs: object) -> str:
|
||
|
|
result: Any = definition.handler(*args, **kwargs)
|
||
|
|
if inspect.isawaitable(result):
|
||
|
|
result = await result
|
||
|
|
return cast("str", result)
|