Агент Pydantic — декоратор инструмента/набора инструментов агента не работает поверх методов, имеющих контекст запуска. ⇐ Python

[Anonymous]

Я пытаюсь создать пидантический ИИ-агент
При определении моих инструментов, когда я передаю свои инструменты вместе с моим агентом в качестве параметра как

Код: Выделить всё

chat_agent = Agent(
model=create_chat_model(),
output_type=ChatResponse,
deps_type=ChatAgentDeps,
tools=[xyz, abc, tst],
):

все работает нормально, как и ожидалось.
Однако, если я добавлю декоратор, как показано ниже. он никогда не работает и не регистрирует инструменты. В чем могут быть причины и как это исправить? Я хочу, чтобы мой код выглядел чище и его было легче расширять, поэтому простое добавление декоратора показалось мне лучшим способом зарегистрировать инструмент.

Код: Выделить всё

@chat_agent.tool
def xyz(ctx: RunContext[Deps],
qry,
):
pass

Код: Выделить всё

@chat_agent.tool
def abc(
_: RunContext[Deps]
):
pass

Я пробовал инструменты, набор инструментов и декораторыtool_plain, но ни один из них не работает