Pydantic Agent - Agent Tool/Deformator Decorator не работает над методами, которые имеют Runcontext ⇐ Python

[Anonymous]

Я пытаюсь создать агент Pydantic AI
при определении моих инструментов, когда я передаю свои инструменты вместе с моим агентом как параметр как < /p>
chat_agent = Agent(
model=create_chat_model(),
output_type=ChatResponse,
deps_type=ChatAgentDeps,
tools=[xyz, abc, tst],
):
< /code>
Это работает совершенно хорошо, как и ожидалось < /p>
Однако, если я добавлю декоратор, как показано ниже. Он никогда не работает и не регистрирует инструменты. Каковы могут быть причины и как я могу это исправить? Я хочу, чтобы мой код выглядел чище и легче расширить, таким образом, добавление декоратора казалось мне лучшим способом зарегистрировать инструмент < /p>
@chat_agent.tool
def xyz(ctx: RunContext[Deps],
qry,
):
pass
< /code>
@chat_agent.tool
def abc(
_: RunContext[Deps]
):
pass

< /code>
I tried tools, toolset and tool_plain decorators and none of them seem to work

Подробнее здесь: https://stackoverflow.com/questions/797 ... -of-method