Scope document tools to caller owner

Co-authored-by: Lohinth <lohinth25@proton.me>
This commit is contained in:
Lohinth
2026-06-02 02:30:02 +05:30
committed by GitHub
parent d42e6a7acc
commit a8d9a180d9
3 changed files with 191 additions and 16 deletions

View File

@@ -651,15 +651,15 @@ async def execute_tool_block(
elif tool == "create_document":
title = content.split("\n")[0].strip()[:60]
desc = f"create_document: {title}"
result = await do_create_document(content, session_id=session_id)
result = await do_create_document(content, session_id=session_id, owner=owner)
elif tool == "update_document":
desc = f"update_document: {content.split(chr(10))[0][:60]}"
result = await do_update_document(content)
result = await do_update_document(content, owner=owner)
elif tool == "edit_document":
result = await do_edit_document(content)
result = await do_edit_document(content, owner=owner)
desc = f"edit_document: {result.get('title', '')}"
elif tool == "suggest_document":
result = await do_suggest_document(content)
result = await do_suggest_document(content, owner=owner)
desc = f"suggest_document: {result.get('count', 0)} suggestions"
elif tool == "search_chats":
query = content.split("\n")[0].strip()