Добавлена обратока автризации через composio
This commit is contained in:
parent
a02d7d64b9
commit
ad89138afc
1 changed files with 10 additions and 4 deletions
|
|
@ -45,10 +45,16 @@ class AgentService:
|
||||||
|
|
||||||
# 2. Инструмент завершил работу и вернул результат
|
# 2. Инструмент завершил работу и вернул результат
|
||||||
elif kind == "on_tool_end":
|
elif kind == "on_tool_end":
|
||||||
yield MsgEventToolResult(
|
result = event["data"].get("output")
|
||||||
tool_name=event["name"],
|
|
||||||
result=event["data"].get("output")
|
# Перехватываем ссылку на авторизацию Composio v3
|
||||||
)
|
if result and "connect.composio.dev" in str(result):
|
||||||
|
yield MsgEventTextChunk(text=f"\n⚠️ Для выполнения действия требуется авторизация. Перейдите по ссылке: {result}\n")
|
||||||
|
else:
|
||||||
|
yield MsgEventToolResult(
|
||||||
|
tool_name=event["name"],
|
||||||
|
result=str(result) # Страховка от ошибки сериализации JSON
|
||||||
|
)
|
||||||
|
|
||||||
# 3. В конце генерации отправляем событие завершения
|
# 3. В конце генерации отправляем событие завершения
|
||||||
yield MsgEventEnd(tokens_used=0) # потом заменить на метадату
|
yield MsgEventEnd(tokens_used=0) # потом заменить на метадату
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue