Fix upload button (#939)

Co-authored-by: Petr Bulánek <bulanek.petr@gmail.com>

Author: jezekra1

agents/community/aider/src/aider_agent/agent.py

@@ -21,6 +21,7 @@
 
 
 @server.agent(
+    input_content_types=["none"],
     metadata=Metadata(
         annotations=Annotations(
             beeai_ui=PlatformUIAnnotation(

agents/community/gpt-researcher/gpt_researcher_agent/agent.py

@@ -25,6 +25,7 @@
 
 
 @server.agent(
+    input_content_types=["none"],
     metadata=Metadata(
         annotations=Annotations(
             beeai_ui=PlatformUIAnnotation(

agents/official/beeai-framework/chat/src/chat/agent.py

@@ -40,6 +40,7 @@ def to_framework_message(role: str, content: str) -> beeai_framework.backend.Mes
 
 
 @server.agent(
+    input_content_types=["none"],
     metadata=Metadata(
         annotations=Annotations(
             beeai_ui=PlatformUIAnnotation(

apps/beeai-server/src/beeai_server/infrastructure/persistence/migrations/alembic/versions/975813acde2a_.py

@@ -0,0 +1,37 @@
+# Copyright 2025 © BeeAI a Series of LF Projects, LLC
+# SPDX-License-Identifier: Apache-2.0
+
+"""empty message
+
+Revision ID: 975813acde2a
+Revises: 327991a3c7a7
+Create Date: 2025-07-17 15:39:56.507771
+
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "975813acde2a"
+down_revision: str | None = "327991a3c7a7"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("agents", sa.Column("input_content_types", sa.JSON(), nullable=True))
+    op.add_column("agents", sa.Column("output_content_types", sa.JSON(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("agents", "output_content_types")
+    op.drop_column("agents", "input_content_types")
+    # ### end Alembic commands ###

apps/beeai-server/src/beeai_server/infrastructure/persistence/repositories/agent.py

@@ -25,6 +25,8 @@
     Column("description", Text, nullable=True),
     Column("provider_id", ForeignKey("providers.id", ondelete="CASCADE"), nullable=False),
     Column("metadata", JSON, nullable=False),
+    Column("input_content_types", JSON, nullable=True),
+    Column("output_content_types", JSON, nullable=True),
 )
 
 agent_requests_table = Table(
@@ -55,6 +57,8 @@ async def bulk_create(self, agents: list[Agent]) -> None:
                     "description": agent.description,
                     "provider_id": agent.metadata.provider_id,
                     "metadata": agent.metadata.model_dump(mode="json"),
+                    "input_content_types": agent.input_content_types,
+                    "output_content_types": agent.output_content_types,
                 }
                 for agent in agents
             ]
@@ -84,6 +88,8 @@ def _to_agent(self, row: Row) -> Agent:
                 "name": row.name,
                 "description": row.description,
                 "metadata": row.metadata,
+                "input_content_types": row.input_content_types or ["*/*"],
+                "output_content_types": row.output_content_types or ["*/*"],
             }
         )
 

apps/beeai-ui/src/modules/runs/chat/ChatView.tsx

@@ -20,7 +20,7 @@ interface Props {
 
 export function ChatView({ agent }: Props) {
   return (
-    <FileUploadProvider allowedContentTypes={agent.input_content_types ?? []}>
+    <FileUploadProvider allowedContentTypes={agent.input_content_types}>
       <AgentRunProvider agent={agent}>
         <Chat />
       </AgentRunProvider>

apps/beeai-ui/src/modules/runs/files/constants.ts

@@ -12,3 +12,5 @@ export const FILE_CONTENT_URL = '/api/v1/files/{file_id}/content';
 export const FILE_CONTENT_URL_BASE = 'http://{platform_url}';
 
 export const ALL_FILES_CONTENT_TYPE = '*/*';
+
+export const NO_FILES_CONTENT_TYPE = 'none';

apps/beeai-ui/src/modules/runs/files/contexts/FileUploadProvider.tsx

@@ -11,7 +11,7 @@ import { useToast } from '#contexts/Toast/index.ts';
 
 import { useDeleteFile } from '../api/mutations/useDeleteFile';
 import { useUploadFile } from '../api/mutations/useUploadFile';
-import { ALL_FILES_CONTENT_TYPE, MAX_FILE_SIZE, MAX_FILES } from '../constants';
+import { ALL_FILES_CONTENT_TYPE, MAX_FILE_SIZE, MAX_FILES, NO_FILES_CONTENT_TYPE } from '../constants';
 import { type FileEntity, FileStatus } from '../types';
 import { FileUploadContext } from './file-upload-context';
 
@@ -88,11 +88,19 @@ export function FileUploadProvider({ allowedContentTypes, children }: PropsWithC
     [files],
   );
 
-  const isDisabled = !allowedContentTypes.length;
-  const accept = allowedContentTypes.reduce(
-    (value, mimeType) => (mimeType === ALL_FILES_CONTENT_TYPE ? value : { ...value, [mimeType]: [] }),
-    {} as Record<string, string[]>,
-  );
+  const isDisabled = allowedContentTypes.includes(NO_FILES_CONTENT_TYPE) || !allowedContentTypes.length;
+  const accept = isDisabled
+    ? {}
+    : allowedContentTypes.reduce(
+        (value, mimeType) =>
+          mimeType === ALL_FILES_CONTENT_TYPE
+            ? value
+            : {
+                ...value,
+                [mimeType]: [],
+              },
+        {} as Record<string, string[]>,
+      );
 
   const dropzone = useDropzone({
     accept,

apps/beeai-ui/src/modules/runs/hands-off/HandsOffView.tsx

@@ -20,7 +20,7 @@ interface Props {
 
 export function HandsOffView({ agent }: Props) {
   return (
-    <FileUploadProvider allowedContentTypes={agent.input_content_types ?? []}>
+    <FileUploadProvider allowedContentTypes={agent.input_content_types}>
       <AgentRunProvider agent={agent}>
         <HandsOff />
       </AgentRunProvider>