Skip to content

Commit 11f6bae

Browse files
committed
fix: add functionality to retrieve authorized shared tools
1 parent 95838a2 commit 11f6bae

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

‎apps/application/serializers/application.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,12 @@ def get_authorized_tool_ids(user_id: str, workspace_id: str, tool_ids: List[str]
219219
)
220220
if "VIEW" in permission.permission_list or "ROLE" in permission.permission_list
221221
}
222+
get_authorized_tool = DatabaseModelManage.get_model("get_authorized_tool")
223+
if get_authorized_tool is not None:
224+
shared_tool_query_set = QuerySet(Tool).filter(id__in=tool_ids, scope=ToolScope.SHARED)
225+
granted_tool_ids.update(
226+
str(tool.id) for tool in get_authorized_tool(shared_tool_query_set, workspace_id)
227+
)
222228
return [tool_id for tool_id in tool_ids if tool_id in granted_tool_ids]
223229

224230

0 commit comments

Comments
 (0)