Added permission to enable/disable image capture and attch knownledge

Author: cableman

backend/open_webui/config.py

@@ -1251,6 +1251,14 @@ def feishu_oauth_register(client: OAuth):
     os.environ.get("USER_PERMISSIONS_CHAT_SYSTEM_PROMPT", "True").lower() == "true"
 )
 
+USER_PERMISSIONS_CHAT_IMAGE_CAPTURE = (
+    os.environ.get("USER_PERMISSIONS_CHAT_IMAGE_CAPTURE", "True").lower() == "true"
+)
+
+USER_PERMISSIONS_CHAT_ATTACH_KONWLEDGE = (
+    os.environ.get("USER_PERMISSIONS_CHAT_ATTACH_KONWLEDGE", "True").lower() == "true"
+)
+
 USER_PERMISSIONS_CHAT_PARAMS = (
     os.environ.get("USER_PERMISSIONS_CHAT_PARAMS", "True").lower() == "true"
 )
@@ -1359,6 +1367,8 @@ def feishu_oauth_register(client: OAuth):
         "controls": USER_PERMISSIONS_CHAT_CONTROLS,
         "valves": USER_PERMISSIONS_CHAT_VALVES,
         "system_prompt": USER_PERMISSIONS_CHAT_SYSTEM_PROMPT,
+        "image_capture": USER_PERMISSIONS_CHAT_IMAGE_CAPTURE,
+        "attach_knowledge": USER_PERMISSIONS_CHAT_ATTACH_KONWLEDGE,
         "params": USER_PERMISSIONS_CHAT_PARAMS,
         "file_upload": USER_PERMISSIONS_CHAT_FILE_UPLOAD,
         "delete": USER_PERMISSIONS_CHAT_DELETE,

backend/open_webui/routers/users.py

@@ -163,6 +163,8 @@ class ChatPermissions(BaseModel):
     controls: bool = True
     valves: bool = True
     system_prompt: bool = True
+    image_capture: bool = True
+    attach_knowledge: bool = True
     params: bool = True
     file_upload: bool = True
     delete: bool = True

src/lib/components/admin/Users/Groups.svelte

@@ -68,6 +68,8 @@
 			controls: true,
 			valves: true,
 			system_prompt: true,
+			image_capture: true,
+            attach_knowledge: true,
 			params: true,
 			file_upload: true,
 			delete: true,

src/lib/components/admin/Users/Groups/EditGroupModal.svelte

@@ -50,6 +50,8 @@
 			controls: true,
 			valves: true,
 			system_prompt: true,
+			image_capture: true,
+            attach_knowledge: true,
 			params: true,
 			file_upload: true,
 			delete: true,

src/lib/components/admin/Users/Groups/Permissions.svelte

@@ -24,6 +24,8 @@
 			valves: true,
 			system_prompt: true,
 			params: true,
+			image_capture: true,
+            attach_knowledge: true,
 			file_upload: true,
 			delete: true,
 			delete_message: true,
@@ -304,6 +306,22 @@
 			<Switch bind:state={permissions.chat.edit} />
 		</div>
 
+		<div class="  flex w-full justify-between my-2 pr-2">
+			<div class=" self-center text-xs font-medium">
+				{$i18n.t('Allow Image capture')}
+			</div>
+
+			<Switch bind:state={permissions.chat.image_capture} />
+		</div>
+
+        <div class="  flex w-full justify-between my-2 pr-2">
+			<div class=" self-center text-xs font-medium">
+				{$i18n.t('Allow Attach Knowledge')}
+			</div>
+
+			<Switch bind:state={permissions.chat.attach_knowledge} />
+		</div>
+
 		<div class="  flex w-full justify-between my-2 pr-2">
 			<div class=" self-center text-xs font-medium">
 				{$i18n.t('Allow Chat Delete')}

src/lib/components/chat/MessageInput/InputMenu.svelte

@@ -159,6 +159,7 @@
 						</DropdownMenu.Item>
 					</Tooltip>
 
+                    {#if $user?.role === 'admin' || $user?.permissions.chat?.image_capture}
 					<Tooltip
 						content={fileUploadCapableModels.length !== selectedModels.length
 							? $i18n.t('Model(s) do not support file upload')
@@ -189,6 +190,7 @@
 							<div class=" line-clamp-1">{$i18n.t('Capture')}</div>
 						</DropdownMenu.Item>
 					</Tooltip>
+                    {/if}
 
 					<Tooltip
 						content={fileUploadCapableModels.length !== selectedModels.length
@@ -244,6 +246,7 @@
 					{/if}
 
 					{#if ($knowledge ?? []).length > 0}
+                        {#if $user?.role === 'admin' || $user?.permissions.chat?.attach_knowledge}
 						<Tooltip
 							content={fileUploadCapableModels.length !== selectedModels.length
 								? $i18n.t('Model(s) do not support file upload')
@@ -273,6 +276,7 @@
 								</div>
 							</button>
 						</Tooltip>
+                        {/if}
 					{/if}
 
 					{#if ($chats ?? []).length > 0}