Added permission to enable/disable image capture and attch knownledge

Author: cableman

backend/open_webui/config.py

@@ -1357,6 +1357,14 @@ def feishu_oauth_register(oauth: 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"
 )
@@ -1481,6 +1489,8 @@ def feishu_oauth_register(oauth: 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

@@ -199,6 +199,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/EditGroupModal.svelte

@@ -65,6 +65,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

@@ -36,6 +36,8 @@
 			valves: true,
 			system_prompt: true,
 			params: true,
+			image_capture: true,
+            attach_knowledge: true,
 			file_upload: true,
 			delete: true,
 			delete_message: true,
@@ -493,6 +495,38 @@
 			{/if}
 		</div>
 
+        <div class="flex flex-col w-full">
+			<div class="flex w-full justify-between my-1">
+				<div class=" self-center text-xs font-medium">
+					{$i18n.t('Allow Image capture')}
+				</div>
+				<Switch bind:state={permissions.chat.image_capture} />
+			</div>
+			{#if defaultPermissions?.chat?.image_capture && !permissions.chat.image_capture}
+				<div>
+					<div class="text-xs text-gray-500">
+						{$i18n.t('This is a default user permission and will remain enabled.')}
+					</div>
+				</div>
+			{/if}
+		</div>
+
+        <div class="flex flex-col w-full">
+			<div class="flex w-full justify-between my-1">
+				<div class=" self-center text-xs font-medium">
+					{$i18n.t('Allow Attach Knowledge')}
+				</div>
+				<Switch bind:state={permissions.chat.attach_knowledge} />
+			</div>
+			{#if defaultPermissions?.chat?.attach_knowledge && !permissions.chat.attach_knowledge}
+				<div>
+					<div class="text-xs text-gray-500">
+						{$i18n.t('This is a default user permission and will remain enabled.')}
+					</div>
+				</div>
+			{/if}
+		</div>
+
 		<div class="flex flex-col w-full">
 			<div class="flex w-full justify-between my-1">
 				<div class=" self-center text-xs font-medium">

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

@@ -163,6 +163,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')
@@ -193,6 +194,7 @@
 							<div class=" line-clamp-1">{$i18n.t('Capture')}</div>
 						</DropdownMenu.Item>
 					</Tooltip>
+                    {/if}
 
 					<Tooltip
 						content={fileUploadCapableModels.length !== selectedModels.length
@@ -250,6 +252,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')
@@ -279,6 +282,7 @@
 								</div>
 							</button>
 						</Tooltip>
+                        {/if}
 					{/if}
 
 					{#if ($chats ?? []).length > 0}