add updated_at as an option

autogpt_platform/backend/backend/blocks/agent.py

@@ -7,7 +7,6 @@ from backend.data.block import (
     BlockInput,
     BlockOutput,
     BlockSchema,
-    BlockSchemaInput,
     BlockType,
     get_block,
 )
@@ -20,7 +19,7 @@ _logger = logging.getLogger(__name__)
 
 
 class AgentExecutorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         user_id: str = SchemaField(description="User ID")
         graph_id: str = SchemaField(description="Graph ID")
         graph_version: int = SchemaField(description="Graph Version")
@@ -54,7 +53,6 @@ class AgentExecutorBlock(Block):
             return validate_with_jsonschema(cls.get_input_schema(data), data)
 
     class Output(BlockSchema):
-        # Use BlockSchema to avoid automatic error field that could clash with graph outputs
         pass
 
     def __init__(self):
@@ -67,13 +65,7 @@ class AgentExecutorBlock(Block):
             categories={BlockCategory.AGENT},
         )
 
-    async def run(
-        self,
-        input_data: Input,
-        *,
-        graph_exec_id: str,
-        **kwargs,
-    ) -> BlockOutput:
+    async def run(self, input_data: Input, **kwargs) -> BlockOutput:
 
         from backend.executor import utils as execution_utils
 
@@ -83,7 +75,6 @@ class AgentExecutorBlock(Block):
             user_id=input_data.user_id,
             inputs=input_data.inputs,
             nodes_input_masks=input_data.nodes_input_masks,
-            parent_graph_exec_id=graph_exec_id,
         )
 
         logger = execution_utils.LogMetadata(

autogpt_platform/backend/backend/blocks/ai_condition.py

@@ -10,12 +10,7 @@ from backend.blocks.llm import (
     LLMResponse,
     llm_call,
 )
-from backend.data.block import (
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import APIKeyCredentials, NodeExecutionStats, SchemaField
 
 
@@ -28,7 +23,7 @@ class AIConditionBlock(AIBlockBase):
     It provides the same yes/no data pass-through functionality as the standard ConditionBlock.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         input_value: Any = SchemaField(
             description="The input value to evaluate with the AI condition",
             placeholder="Enter the value to be evaluated (text, number, or any data)",
@@ -55,7 +50,7 @@ class AIConditionBlock(AIBlockBase):
         )
         credentials: AICredentials = AICredentialsField()
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: bool = SchemaField(
             description="The result of the AI condition evaluation (True or False)"
         )

autogpt_platform/backend/backend/blocks/ai_image_customizer.py

@@ -5,13 +5,7 @@ from pydantic import SecretStr
 from replicate.client import Client as ReplicateClient
 from replicate.helpers import FileOutput
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -48,7 +42,7 @@ TEST_CREDENTIALS_INPUT = {
 
 
 class AIImageCustomizerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REPLICATE], Literal["api_key"]
         ] = CredentialsField(
@@ -74,8 +68,9 @@ class AIImageCustomizerBlock(Block):
             title="Output Format",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         image_url: MediaFileType = SchemaField(description="URL of the generated image")
+        error: str = SchemaField(description="Error message if generation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/ai_image_generator_block.py

@@ -5,7 +5,7 @@ from pydantic import SecretStr
 from replicate.client import Client as ReplicateClient
 from replicate.helpers import FileOutput
 
-from backend.data.block import Block, BlockCategory, BlockSchemaInput, BlockSchemaOutput
+from backend.data.block import Block, BlockCategory, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -101,7 +101,7 @@ class ImageGenModel(str, Enum):
 
 
 class AIImageGeneratorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REPLICATE], Literal["api_key"]
         ] = CredentialsField(
@@ -135,8 +135,9 @@ class AIImageGeneratorBlock(Block):
             title="Image Style",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         image_url: str = SchemaField(description="URL of the generated image")
+        error: str = SchemaField(description="Error message if generation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/ai_music_generator.py

@@ -6,13 +6,7 @@ from typing import Literal
 from pydantic import SecretStr
 from replicate.client import Client as ReplicateClient
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -60,7 +54,7 @@ class NormalizationStrategy(str, Enum):
 
 
 class AIMusicGeneratorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REPLICATE], Literal["api_key"]
         ] = CredentialsField(
@@ -113,8 +107,9 @@ class AIMusicGeneratorBlock(Block):
             title="Normalization Strategy",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: str = SchemaField(description="URL of the generated audio file")
+        error: str = SchemaField(description="Error message if the model run failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/ai_shortform_video_block.py

@@ -6,13 +6,7 @@ from typing import Literal
 
 from pydantic import SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -154,7 +148,7 @@ logger = logging.getLogger(__name__)
 class AIShortformVideoCreatorBlock(Block):
     """Creates a short‑form text‑to‑video clip using stock or AI imagery."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REVID], Literal["api_key"]
         ] = CredentialsField(
@@ -193,8 +187,9 @@ class AIShortformVideoCreatorBlock(Block):
             placeholder=VisualMediaType.STOCK_VIDEOS,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_url: str = SchemaField(description="The URL of the created video")
+        error: str = SchemaField(description="Error message if the request failed")
 
     async def create_webhook(self) -> tuple[str, str]:
         """Create a new webhook URL for receiving notifications."""
@@ -341,7 +336,7 @@ class AIShortformVideoCreatorBlock(Block):
 class AIAdMakerVideoCreatorBlock(Block):
     """Generates a 30‑second vertical AI advert using optional user‑supplied imagery."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REVID], Literal["api_key"]
         ] = CredentialsField(
@@ -369,8 +364,9 @@ class AIAdMakerVideoCreatorBlock(Block):
             description="Restrict visuals to supplied images only.", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_url: str = SchemaField(description="URL of the finished advert")
+        error: str = SchemaField(description="Error message on failure")
 
     async def create_webhook(self) -> tuple[str, str]:
         """Create a new webhook URL for receiving notifications."""
@@ -528,7 +524,7 @@ class AIAdMakerVideoCreatorBlock(Block):
 class AIScreenshotToVideoAdBlock(Block):
     """Creates an advert where the supplied screenshot is narrated by an AI avatar."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REVID], Literal["api_key"]
         ] = CredentialsField(description="Revid.ai API key")
@@ -546,8 +542,9 @@ class AIScreenshotToVideoAdBlock(Block):
             default=AudioTrack.DONT_STOP_ME_ABSTRACT_FUTURE_BASS
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_url: str = SchemaField(description="Rendered video URL")
+        error: str = SchemaField(description="Error, if encountered")
 
     async def create_webhook(self) -> tuple[str, str]:
         """Create a new webhook URL for receiving notifications."""

autogpt_platform/backend/backend/blocks/airtable/bases.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -24,7 +23,7 @@ class AirtableCreateBaseBlock(Block):
     Creates a new base in an Airtable workspace, or returns existing base if one with the same name exists.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -54,7 +53,7 @@ class AirtableCreateBaseBlock(Block):
             ],
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         base_id: str = SchemaField(description="The ID of the created or found base")
         tables: list[dict] = SchemaField(description="Array of table objects")
         table: dict = SchemaField(description="A single table object")
@@ -119,7 +118,7 @@ class AirtableListBasesBlock(Block):
     Lists all bases in an Airtable workspace that the user has access to.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -130,7 +129,7 @@ class AirtableListBasesBlock(Block):
             description="Pagination offset from previous request", default=""
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         bases: list[dict] = SchemaField(description="Array of base objects")
         offset: Optional[str] = SchemaField(
             description="Offset for next page (null if no more bases)", default=None

autogpt_platform/backend/backend/blocks/airtable/records.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -32,7 +31,7 @@ class AirtableListRecordsBlock(Block):
     Lists records from an Airtable table with optional filtering, sorting, and pagination.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -66,7 +65,7 @@ class AirtableListRecordsBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         records: list[dict] = SchemaField(description="Array of record objects")
         offset: Optional[str] = SchemaField(
             description="Offset for next page (null if no more records)", default=None
@@ -138,7 +137,7 @@ class AirtableGetRecordBlock(Block):
     Retrieves a single record from an Airtable table by its ID.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -154,7 +153,7 @@ class AirtableGetRecordBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         id: str = SchemaField(description="The record ID")
         fields: dict = SchemaField(description="The record fields")
         created_time: str = SchemaField(description="The record created time")
@@ -218,7 +217,7 @@ class AirtableCreateRecordsBlock(Block):
     Creates one or more records in an Airtable table.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -240,7 +239,7 @@ class AirtableCreateRecordsBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         records: list[dict] = SchemaField(description="Array of created record objects")
         details: dict = SchemaField(description="Details of the created records")
 
@@ -291,7 +290,7 @@ class AirtableUpdateRecordsBlock(Block):
     Updates one or more existing records in an Airtable table.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -307,7 +306,7 @@ class AirtableUpdateRecordsBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         records: list[dict] = SchemaField(description="Array of updated record objects")
 
     def __init__(self):
@@ -340,7 +339,7 @@ class AirtableDeleteRecordsBlock(Block):
     Deletes one or more records from an Airtable table.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -352,7 +351,7 @@ class AirtableDeleteRecordsBlock(Block):
             description="Array of upto 10 record IDs to delete"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         records: list[dict] = SchemaField(description="Array of deletion results")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/airtable/schema.py

@@ -7,8 +7,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -24,13 +23,13 @@ class AirtableListSchemaBlock(Block):
     fields, and views.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
         base_id: str = SchemaField(description="The Airtable base ID")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         base_schema: dict = SchemaField(
             description="Complete base schema with tables, fields, and views"
         )
@@ -67,7 +66,7 @@ class AirtableCreateTableBlock(Block):
     Creates a new table in an Airtable base with specified fields and views.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -78,7 +77,7 @@ class AirtableCreateTableBlock(Block):
             default=[{"name": "Name", "type": "singleLineText"}],
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         table: dict = SchemaField(description="Created table object")
         table_id: str = SchemaField(description="ID of the created table")
 
@@ -110,7 +109,7 @@ class AirtableUpdateTableBlock(Block):
     Updates an existing table's properties such as name or description.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -126,7 +125,7 @@ class AirtableUpdateTableBlock(Block):
             description="The date dependency of the table to update", default=None
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         table: dict = SchemaField(description="Updated table object")
 
     def __init__(self):
@@ -158,7 +157,7 @@ class AirtableCreateFieldBlock(Block):
     Adds a new field (column) to an existing Airtable table.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -177,7 +176,7 @@ class AirtableCreateFieldBlock(Block):
             description="The options of the field to create", default=None
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         field: dict = SchemaField(description="Created field object")
         field_id: str = SchemaField(description="ID of the created field")
 
@@ -210,7 +209,7 @@ class AirtableUpdateFieldBlock(Block):
     Updates an existing field's properties in an Airtable table.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -226,7 +225,7 @@ class AirtableUpdateFieldBlock(Block):
             advanced=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         field: dict = SchemaField(description="Updated field object")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/airtable/triggers.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     BlockWebhookConfig,
     CredentialsMetaInput,
@@ -33,7 +32,7 @@ class AirtableWebhookTriggerBlock(Block):
     Thin wrapper just forwards the payloads one at a time to the next block.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = airtable.credentials_field(
             description="Airtable API credentials"
         )
@@ -44,7 +43,7 @@ class AirtableWebhookTriggerBlock(Block):
             description="Airtable webhook event filter"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         payload: WebhookPayload = SchemaField(description="Airtable webhook payload")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/apollo/organization.py

@@ -10,20 +10,14 @@ from backend.blocks.apollo.models import (
     PrimaryPhone,
     SearchOrganizationsRequest,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import CredentialsField, SchemaField
 
 
 class SearchOrganizationsBlock(Block):
     """Search for organizations in Apollo"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         organization_num_employees_range: list[int] = SchemaField(
             description="""The number range of employees working for the company. This enables you to find companies based on headcount. You can add multiple ranges to expand your search results.
 
@@ -75,7 +69,7 @@ To find IDs, identify the values for organization_id when you call this endpoint
             description="Apollo credentials",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         organizations: list[Organization] = SchemaField(
             description="List of organizations found",
             default_factory=list,

autogpt_platform/backend/backend/blocks/apollo/people.py

@@ -14,20 +14,14 @@ from backend.blocks.apollo.models import (
     SearchPeopleRequest,
     SenorityLevels,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import CredentialsField, SchemaField
 
 
 class SearchPeopleBlock(Block):
     """Search for people in Apollo"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         person_titles: list[str] = SchemaField(
             description="""Job titles held by the people you want to find. For a person to be included in search results, they only need to match 1 of the job titles you add. Adding more job titles expands your search results.
 
@@ -115,7 +109,7 @@ class SearchPeopleBlock(Block):
             description="Apollo credentials",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         people: list[Contact] = SchemaField(
             description="List of people found",
             default_factory=list,

autogpt_platform/backend/backend/blocks/apollo/person.py

@@ -6,20 +6,14 @@ from backend.blocks.apollo._auth import (
     ApolloCredentialsInput,
 )
 from backend.blocks.apollo.models import Contact, EnrichPersonRequest
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import CredentialsField, SchemaField
 
 
 class GetPersonDetailBlock(Block):
     """Get detailed person data with Apollo API, including email reveal"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         person_id: str = SchemaField(
             description="Apollo person ID to enrich (most accurate method)",
             default="",
@@ -74,7 +68,7 @@ class GetPersonDetailBlock(Block):
             description="Apollo credentials",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         contact: Contact = SchemaField(
             description="Enriched contact information",
         )

autogpt_platform/backend/backend/blocks/ayrshare/_util.py

@@ -3,7 +3,7 @@ from typing import Optional
 
 from pydantic import BaseModel, Field
 
-from backend.data.block import BlockSchemaInput
+from backend.data.block import BlockSchema
 from backend.data.model import SchemaField, UserIntegrations
 from backend.integrations.ayrshare import AyrshareClient
 from backend.util.clients import get_database_manager_async_client
@@ -17,7 +17,7 @@ async def get_profile_key(user_id: str):
     return user_integrations.managed_credentials.ayrshare_profile_key
 
 
-class BaseAyrshareInput(BlockSchemaInput):
+class BaseAyrshareInput(BlockSchema):
     """Base input model for Ayrshare social media posts with common fields."""
 
     post: str = SchemaField(

autogpt_platform/backend/backend/blocks/ayrshare/post_to_bluesky.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -38,7 +38,7 @@ class PostToBlueskyBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_facebook.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -101,7 +101,7 @@ class PostToFacebookBlock(Block):
             description="URL for custom link preview", default="", advanced=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_gmb.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -94,7 +94,7 @@ class PostToGMBBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_instagram.py

@@ -5,7 +5,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -94,7 +94,7 @@ class PostToInstagramBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_linkedin.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -94,7 +94,7 @@ class PostToLinkedInBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_pinterest.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -73,7 +73,7 @@ class PostToPinterestBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_reddit.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -19,7 +19,7 @@ class PostToRedditBlock(Block):
 
         pass  # Uses all base fields
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_snapchat.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -43,7 +43,7 @@ class PostToSnapchatBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_telegram.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -38,7 +38,7 @@ class PostToTelegramBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_threads.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -31,7 +31,7 @@ class PostToThreadsBlock(Block):
             advanced=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_tiktok.py

@@ -5,7 +5,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -98,7 +98,7 @@ class PostToTikTokBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_x.py

@@ -3,7 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -97,7 +97,7 @@ class PostToXBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/ayrshare/post_to_youtube.py

@@ -6,7 +6,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     SchemaField,
 )
@@ -119,7 +119,7 @@ class PostToYouTubeBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_result: PostResponse = SchemaField(description="The result of the post")
         post: PostIds = SchemaField(description="The result of the post")
 

autogpt_platform/backend/backend/blocks/baas/bots.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -24,7 +23,7 @@ class BaasBotJoinMeetingBlock(Block):
     Deploy a bot immediately or at a scheduled start_time to join and record a meeting.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = baas.credentials_field(
             description="Meeting BaaS API credentials"
         )
@@ -58,7 +57,7 @@ class BaasBotJoinMeetingBlock(Block):
             description="Custom metadata to attach to the bot", default={}
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         bot_id: str = SchemaField(description="UUID of the deployed bot")
         join_response: dict = SchemaField(
             description="Full response from join operation"
@@ -104,13 +103,13 @@ class BaasBotLeaveMeetingBlock(Block):
     Force the bot to exit the call.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = baas.credentials_field(
             description="Meeting BaaS API credentials"
         )
         bot_id: str = SchemaField(description="UUID of the bot to remove from meeting")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         left: bool = SchemaField(description="Whether the bot successfully left")
 
     def __init__(self):
@@ -139,7 +138,7 @@ class BaasBotFetchMeetingDataBlock(Block):
     Pull MP4 URL, transcript & metadata for a completed meeting.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = baas.credentials_field(
             description="Meeting BaaS API credentials"
         )
@@ -148,7 +147,7 @@ class BaasBotFetchMeetingDataBlock(Block):
             description="Include transcript data in response", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         mp4_url: str = SchemaField(
             description="URL to download the meeting recording (time-limited)"
         )
@@ -186,13 +185,13 @@ class BaasBotDeleteRecordingBlock(Block):
     Purge MP4 + transcript data for privacy or storage management.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = baas.credentials_field(
             description="Meeting BaaS API credentials"
         )
         bot_id: str = SchemaField(description="UUID of the bot whose data to delete")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         deleted: bool = SchemaField(
             description="Whether the data was successfully deleted"
         )

autogpt_platform/backend/backend/blocks/bannerbear/text_overlay.py

@@ -11,8 +11,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -28,7 +27,7 @@ TEST_CREDENTIALS = APIKeyCredentials(
 )
 
 
-class TextModification(BlockSchemaInput):
+class TextModification(BlockSchema):
     name: str = SchemaField(
         description="The name of the layer to modify in the template"
     )
@@ -61,7 +60,7 @@ class TextModification(BlockSchemaInput):
 
 
 class BannerbearTextOverlayBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = bannerbear.credentials_field(
             description="API credentials for Bannerbear"
         )
@@ -97,7 +96,7 @@ class BannerbearTextOverlayBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         success: bool = SchemaField(
             description="Whether the image generation was successfully initiated"
         )
@@ -106,6 +105,7 @@ class BannerbearTextOverlayBlock(Block):
         )
         uid: str = SchemaField(description="Unique identifier for the generated image")
         status: str = SchemaField(description="Status of the image generation")
+        error: str = SchemaField(description="Error message if the operation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/basic.py

@@ -1,21 +1,14 @@
 import enum
 from typing import Any
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-    BlockType,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema, BlockType
 from backend.data.model import SchemaField
 from backend.util.file import store_media_file
 from backend.util.type import MediaFileType, convert
 
 
 class FileStoreBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         file_in: MediaFileType = SchemaField(
             description="The file to store in the temporary directory, it can be a URL, data URI, or local path."
         )
@@ -26,7 +19,7 @@ class FileStoreBlock(Block):
             title="Produce Base64 Output",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         file_out: MediaFileType = SchemaField(
             description="The relative path to the stored file in the temporary directory."
         )
@@ -64,7 +57,7 @@ class StoreValueBlock(Block):
     The block output will be static, the output can be consumed multiple times.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         input: Any = SchemaField(
             description="Trigger the block to produce the output. "
             "The value is only used when `data` is None."
@@ -75,7 +68,7 @@ class StoreValueBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         output: Any = SchemaField(description="The stored data retained in the block.")
 
     def __init__(self):
@@ -101,10 +94,10 @@ class StoreValueBlock(Block):
 
 
 class PrintToConsoleBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: Any = SchemaField(description="The data to print to the console.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         output: Any = SchemaField(description="The data printed to the console.")
         status: str = SchemaField(description="The status of the print operation.")
 
@@ -128,10 +121,10 @@ class PrintToConsoleBlock(Block):
 
 
 class NoteBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: str = SchemaField(description="The text to display in the sticky note.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         output: str = SchemaField(description="The text to display in the sticky note.")
 
     def __init__(self):
@@ -161,14 +154,15 @@ class TypeOptions(enum.Enum):
 
 
 class UniversalTypeConverterBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         value: Any = SchemaField(
             description="The value to convert to a universal type."
         )
         type: TypeOptions = SchemaField(description="The type to convert the value to.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         value: Any = SchemaField(description="The converted value.")
+        error: str = SchemaField(description="Error message if conversion failed.")
 
     def __init__(self):
         super().__init__(
@@ -201,10 +195,10 @@ class ReverseListOrderBlock(Block):
     A block which takes in a list and returns it in the opposite order.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         input_list: list[Any] = SchemaField(description="The list to reverse")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         reversed_list: list[Any] = SchemaField(description="The list in reversed order")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/block.py

@@ -2,13 +2,7 @@ import os
 import re
 from typing import Type
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
@@ -21,12 +15,12 @@ class BlockInstallationBlock(Block):
         for development purposes only.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         code: str = SchemaField(
             description="Python code of the block to be installed",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         success: str = SchemaField(
             description="Success message if the block is installed successfully",
         )

autogpt_platform/backend/backend/blocks/branching.py

@@ -1,13 +1,7 @@
 from enum import Enum
 from typing import Any
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.type import convert
 
@@ -22,7 +16,7 @@ class ComparisonOperator(Enum):
 
 
 class ConditionBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         value1: Any = SchemaField(
             description="Enter the first value for comparison",
             placeholder="For example: 10 or 'hello' or True",
@@ -46,7 +40,7 @@ class ConditionBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: bool = SchemaField(
             description="The result of the condition evaluation (True or False)"
         )
@@ -117,7 +111,7 @@ class ConditionBlock(Block):
 
 
 class IfInputMatchesBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         input: Any = SchemaField(
             description="The input to match against",
             placeholder="For example: 10 or 'hello' or True",
@@ -137,7 +131,7 @@ class IfInputMatchesBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: bool = SchemaField(
             description="The result of the condition evaluation (True or False)"
         )

autogpt_platform/backend/backend/blocks/code_executor.py

@@ -4,15 +4,9 @@ from typing import Any, Literal, Optional
 from e2b_code_interpreter import AsyncSandbox
 from e2b_code_interpreter import Result as E2BExecutionResult
 from e2b_code_interpreter.charts import Chart as E2BExecutionResultChart
-from pydantic import BaseModel, Field, JsonValue, SecretStr
+from pydantic import BaseModel, JsonValue, SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -67,7 +61,7 @@ class MainCodeExecutionResult(BaseModel):
     jpeg: Optional[str] = None
     pdf: Optional[str] = None
     latex: Optional[str] = None
-    json_data: Optional[JsonValue] = Field(None, alias="json")
+    json: Optional[JsonValue] = None  # type: ignore (reportIncompatibleMethodOverride)
     javascript: Optional[str] = None
     data: Optional[dict] = None
     chart: Optional[Chart] = None
@@ -165,7 +159,7 @@ class ExecuteCodeBlock(Block, BaseE2BExecutorMixin):
     # TODO : Add support to upload and download files
     # NOTE: Currently, you can only customize the CPU and Memory
     #       by creating a pre customized sandbox template
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.E2B], Literal["api_key"]
         ] = CredentialsField(
@@ -223,7 +217,7 @@ class ExecuteCodeBlock(Block, BaseE2BExecutorMixin):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         main_result: MainCodeExecutionResult = SchemaField(
             title="Main Result", description="The main result from the code execution"
         )
@@ -238,6 +232,7 @@ class ExecuteCodeBlock(Block, BaseE2BExecutorMixin):
             description="Standard output logs from execution"
         )
         stderr_logs: str = SchemaField(description="Standard error logs from execution")
+        error: str = SchemaField(description="Error message if execution failed")
 
     def __init__(self):
         super().__init__(
@@ -301,7 +296,7 @@ class ExecuteCodeBlock(Block, BaseE2BExecutorMixin):
 
 
 class InstantiateCodeSandboxBlock(Block, BaseE2BExecutorMixin):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.E2B], Literal["api_key"]
         ] = CredentialsField(
@@ -351,7 +346,7 @@ class InstantiateCodeSandboxBlock(Block, BaseE2BExecutorMixin):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         sandbox_id: str = SchemaField(description="ID of the sandbox instance")
         response: str = SchemaField(
             title="Text Result",
@@ -361,6 +356,7 @@ class InstantiateCodeSandboxBlock(Block, BaseE2BExecutorMixin):
             description="Standard output logs from execution"
         )
         stderr_logs: str = SchemaField(description="Standard error logs from execution")
+        error: str = SchemaField(description="Error message if execution failed")
 
     def __init__(self):
         super().__init__(
@@ -425,7 +421,7 @@ class InstantiateCodeSandboxBlock(Block, BaseE2BExecutorMixin):
 
 
 class ExecuteCodeStepBlock(Block, BaseE2BExecutorMixin):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.E2B], Literal["api_key"]
         ] = CredentialsField(
@@ -458,7 +454,7 @@ class ExecuteCodeStepBlock(Block, BaseE2BExecutorMixin):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         main_result: MainCodeExecutionResult = SchemaField(
             title="Main Result", description="The main result from the code execution"
         )
@@ -473,6 +469,7 @@ class ExecuteCodeStepBlock(Block, BaseE2BExecutorMixin):
             description="Standard output logs from execution"
         )
         stderr_logs: str = SchemaField(description="Standard error logs from execution")
+        error: str = SchemaField(description="Error message if execution failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/code_extraction_block.py

@@ -1,23 +1,17 @@
 import re
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
 class CodeExtractionBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: str = SchemaField(
             description="Text containing code blocks to extract (e.g., AI response)",
             placeholder="Enter text containing code blocks",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         html: str = SchemaField(description="Extracted HTML code")
         css: str = SchemaField(description="Extracted CSS code")
         javascript: str = SchemaField(description="Extracted JavaScript code")

autogpt_platform/backend/backend/blocks/compass/triggers.py

@@ -5,8 +5,7 @@ from backend.data.block import (
     BlockCategory,
     BlockManualWebhookConfig,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
 )
 from backend.data.model import SchemaField
 from backend.integrations.providers import ProviderName
@@ -28,10 +27,10 @@ class TranscriptionDataModel(BaseModel):
 
 
 class CompassAITriggerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         payload: TranscriptionDataModel = SchemaField(hidden=True)
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         transcription: str = SchemaField(
             description="The contents of the compass transcription."
         )

autogpt_platform/backend/backend/blocks/count_words_and_char_block.py

@@ -1,22 +1,16 @@
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
 class WordCharacterCountBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: str = SchemaField(
             description="Input text to count words and characters",
             placeholder="Enter your text here",
             advanced=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         word_count: int = SchemaField(description="Number of words in the input text")
         character_count: int = SchemaField(
             description="Number of characters in the input text"

autogpt_platform/backend/backend/blocks/data_manipulation.py

@@ -1,12 +1,6 @@
 from typing import Any, List
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.json import loads
 from backend.util.mock import MockObject
@@ -18,13 +12,13 @@ from backend.util.prompt import estimate_token_count_str
 
 
 class CreateDictionaryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         values: dict[str, Any] = SchemaField(
             description="Key-value pairs to create the dictionary with",
             placeholder="e.g., {'name': 'Alice', 'age': 25}",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         dictionary: dict[str, Any] = SchemaField(
             description="The created dictionary containing the specified key-value pairs"
         )
@@ -68,7 +62,7 @@ class CreateDictionaryBlock(Block):
 
 
 class AddToDictionaryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         dictionary: dict[Any, Any] = SchemaField(
             default_factory=dict,
             description="The dictionary to add the entry to. If not provided, a new dictionary will be created.",
@@ -92,10 +86,11 @@ class AddToDictionaryBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_dictionary: dict = SchemaField(
             description="The dictionary with the new entry added."
         )
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -146,11 +141,11 @@ class AddToDictionaryBlock(Block):
 
 
 class FindInDictionaryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         input: Any = SchemaField(description="Dictionary to lookup from")
         key: str | int = SchemaField(description="Key to lookup in the dictionary")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         output: Any = SchemaField(description="Value found for the given key")
         missing: Any = SchemaField(
             description="Value of the input that missing the key"
@@ -206,7 +201,7 @@ class FindInDictionaryBlock(Block):
 
 
 class RemoveFromDictionaryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         dictionary: dict[Any, Any] = SchemaField(
             description="The dictionary to modify."
         )
@@ -215,11 +210,12 @@ class RemoveFromDictionaryBlock(Block):
             default=False, description="Whether to return the removed value."
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_dictionary: dict[Any, Any] = SchemaField(
             description="The dictionary after removal."
         )
         removed_value: Any = SchemaField(description="The removed value if requested.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -255,18 +251,19 @@ class RemoveFromDictionaryBlock(Block):
 
 
 class ReplaceDictionaryValueBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         dictionary: dict[Any, Any] = SchemaField(
             description="The dictionary to modify."
         )
         key: str | int = SchemaField(description="Key to replace the value for.")
         value: Any = SchemaField(description="The new value for the given key.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_dictionary: dict[Any, Any] = SchemaField(
             description="The dictionary after replacement."
         )
         old_value: Any = SchemaField(description="The value that was replaced.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -303,10 +300,10 @@ class ReplaceDictionaryValueBlock(Block):
 
 
 class DictionaryIsEmptyBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         dictionary: dict[Any, Any] = SchemaField(description="The dictionary to check.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         is_empty: bool = SchemaField(description="True if the dictionary is empty.")
 
     def __init__(self):
@@ -330,7 +327,7 @@ class DictionaryIsEmptyBlock(Block):
 
 
 class CreateListBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         values: List[Any] = SchemaField(
             description="A list of values to be combined into a new list.",
             placeholder="e.g., ['Alice', 25, True]",
@@ -346,10 +343,11 @@ class CreateListBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         list: List[Any] = SchemaField(
             description="The created list containing the specified values."
         )
+        error: str = SchemaField(description="Error message if list creation failed.")
 
     def __init__(self):
         super().__init__(
@@ -406,7 +404,7 @@ class CreateListBlock(Block):
 
 
 class AddToListBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(
             default_factory=list,
             advanced=False,
@@ -427,10 +425,11 @@ class AddToListBlock(Block):
             description="The position to insert the new entry. If not provided, the entry will be appended to the end of the list.",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_list: List[Any] = SchemaField(
             description="The list with the new entry added."
         )
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -485,11 +484,11 @@ class AddToListBlock(Block):
 
 
 class FindInListBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(description="The list to search in.")
         value: Any = SchemaField(description="The value to search for.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         index: int = SchemaField(description="The index of the value in the list.")
         found: bool = SchemaField(
             description="Whether the value was found in the list."
@@ -527,14 +526,15 @@ class FindInListBlock(Block):
 
 
 class GetListItemBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(description="The list to get the item from.")
         index: int = SchemaField(
             description="The 0-based index of the item (supports negative indices)."
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         item: Any = SchemaField(description="The item at the specified index.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -561,7 +561,7 @@ class GetListItemBlock(Block):
 
 
 class RemoveFromListBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(description="The list to modify.")
         value: Any = SchemaField(
             default=None, description="Value to remove from the list."
@@ -574,9 +574,10 @@ class RemoveFromListBlock(Block):
             default=False, description="Whether to return the removed item."
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_list: List[Any] = SchemaField(description="The list after removal.")
         removed_item: Any = SchemaField(description="The removed item if requested.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -617,16 +618,17 @@ class RemoveFromListBlock(Block):
 
 
 class ReplaceListItemBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(description="The list to modify.")
         index: int = SchemaField(
             description="Index of the item to replace (supports negative indices)."
         )
         value: Any = SchemaField(description="The new value for the given index.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         updated_list: List[Any] = SchemaField(description="The list after replacement.")
         old_item: Any = SchemaField(description="The item that was replaced.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(
@@ -661,10 +663,10 @@ class ReplaceListItemBlock(Block):
 
 
 class ListIsEmptyBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         list: List[Any] = SchemaField(description="The list to check.")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         is_empty: bool = SchemaField(description="True if the list is empty.")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/dataforseo/keyword_suggestions.py

@@ -8,8 +8,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
     UserPasswordCredentials,
@@ -19,7 +18,7 @@ from ._api import DataForSeoClient
 from ._config import dataforseo
 
 
-class KeywordSuggestion(BlockSchemaInput):
+class KeywordSuggestion(BlockSchema):
     """Schema for a keyword suggestion result."""
 
     keyword: str = SchemaField(description="The keyword suggestion")
@@ -46,7 +45,7 @@ class KeywordSuggestion(BlockSchemaInput):
 class DataForSeoKeywordSuggestionsBlock(Block):
     """Block for getting keyword suggestions from DataForSEO Labs."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = dataforseo.credentials_field(
             description="DataForSEO credentials (username and password)"
         )
@@ -78,7 +77,7 @@ class DataForSeoKeywordSuggestionsBlock(Block):
             le=3000,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         suggestions: List[KeywordSuggestion] = SchemaField(
             description="List of keyword suggestions with metrics"
         )
@@ -91,6 +90,7 @@ class DataForSeoKeywordSuggestionsBlock(Block):
         seed_keyword: str = SchemaField(
             description="The seed keyword used for the query"
         )
+        error: str = SchemaField(description="Error message if the API call failed")
 
     def __init__(self):
         super().__init__(
@@ -213,12 +213,12 @@ class DataForSeoKeywordSuggestionsBlock(Block):
 class KeywordSuggestionExtractorBlock(Block):
     """Extracts individual fields from a KeywordSuggestion object."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         suggestion: KeywordSuggestion = SchemaField(
             description="The keyword suggestion object to extract fields from"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         keyword: str = SchemaField(description="The keyword suggestion")
         search_volume: Optional[int] = SchemaField(
             description="Monthly search volume", default=None

autogpt_platform/backend/backend/blocks/dataforseo/related_keywords.py

@@ -8,8 +8,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
     UserPasswordCredentials,
@@ -19,7 +18,7 @@ from ._api import DataForSeoClient
 from ._config import dataforseo
 
 
-class RelatedKeyword(BlockSchemaInput):
+class RelatedKeyword(BlockSchema):
     """Schema for a related keyword result."""
 
     keyword: str = SchemaField(description="The related keyword")
@@ -46,7 +45,7 @@ class RelatedKeyword(BlockSchemaInput):
 class DataForSeoRelatedKeywordsBlock(Block):
     """Block for getting related keywords from DataForSEO Labs."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = dataforseo.credentials_field(
             description="DataForSEO credentials (username and password)"
         )
@@ -86,7 +85,7 @@ class DataForSeoRelatedKeywordsBlock(Block):
             le=4,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         related_keywords: List[RelatedKeyword] = SchemaField(
             description="List of related keywords with metrics"
         )
@@ -99,6 +98,7 @@ class DataForSeoRelatedKeywordsBlock(Block):
         seed_keyword: str = SchemaField(
             description="The seed keyword used for the query"
         )
+        error: str = SchemaField(description="Error message if the API call failed")
 
     def __init__(self):
         super().__init__(
@@ -231,12 +231,12 @@ class DataForSeoRelatedKeywordsBlock(Block):
 class RelatedKeywordExtractorBlock(Block):
     """Extracts individual fields from a RelatedKeyword object."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         related_keyword: RelatedKeyword = SchemaField(
             description="The related keyword object to extract fields from"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         keyword: str = SchemaField(description="The related keyword")
         search_volume: Optional[int] = SchemaField(
             description="Monthly search volume", default=None

autogpt_platform/backend/backend/blocks/decoder_block.py

@@ -1,23 +1,17 @@
 import codecs
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
 class TextDecoderBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: str = SchemaField(
             description="A string containing escaped characters to be decoded",
             placeholder='Your entire text block with \\n and \\" escaped characters',
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         decoded_text: str = SchemaField(
             description="The decoded text with escape sequences processed"
         )

autogpt_platform/backend/backend/blocks/discord/bot_blocks.py

@@ -7,13 +7,7 @@ from typing import Any
 import discord
 from pydantic import SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import APIKeyCredentials, SchemaField
 from backend.util.file import store_media_file
 from backend.util.request import Requests
@@ -34,10 +28,10 @@ TEST_CREDENTIALS_INPUT = TEST_BOT_CREDENTIALS_INPUT
 
 
 class ReadDiscordMessagesBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         message_content: str = SchemaField(
             description="The content of the message received"
         )
@@ -170,7 +164,7 @@ class ReadDiscordMessagesBlock(Block):
 
 
 class SendDiscordMessageBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         message_content: str = SchemaField(
             description="The content of the message to send"
@@ -184,7 +178,7 @@ class SendDiscordMessageBlock(Block):
             default="",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="The status of the operation (e.g., 'Message sent', 'Error')"
         )
@@ -316,7 +310,7 @@ class SendDiscordMessageBlock(Block):
 
 
 class SendDiscordDMBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         user_id: str = SchemaField(
             description="The Discord user ID to send the DM to (e.g., '123456789012345678')"
@@ -325,7 +319,7 @@ class SendDiscordDMBlock(Block):
             description="The content of the direct message to send"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="The status of the operation")
         message_id: str = SchemaField(description="The ID of the sent message")
 
@@ -405,7 +399,7 @@ class SendDiscordDMBlock(Block):
 
 
 class SendDiscordEmbedBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         channel_identifier: str = SchemaField(
             description="Channel ID or channel name to send the embed to"
@@ -442,7 +436,7 @@ class SendDiscordEmbedBlock(Block):
             default=[],
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Operation status")
         message_id: str = SchemaField(description="ID of the sent embed message")
 
@@ -592,7 +586,7 @@ class SendDiscordEmbedBlock(Block):
 
 
 class SendDiscordFileBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         channel_identifier: str = SchemaField(
             description="Channel ID or channel name to send the file to"
@@ -613,7 +607,7 @@ class SendDiscordFileBlock(Block):
             description="Optional message to send with the file", default=""
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Operation status")
         message_id: str = SchemaField(description="ID of the sent message")
 
@@ -794,7 +788,7 @@ class SendDiscordFileBlock(Block):
 
 
 class ReplyToDiscordMessageBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         channel_id: str = SchemaField(
             description="The channel ID where the message to reply to is located"
@@ -805,7 +799,7 @@ class ReplyToDiscordMessageBlock(Block):
             description="Whether to mention the original message author", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Operation status")
         reply_id: str = SchemaField(description="ID of the reply message")
 
@@ -919,13 +913,13 @@ class ReplyToDiscordMessageBlock(Block):
 
 
 class DiscordUserInfoBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         user_id: str = SchemaField(
             description="The Discord user ID to get information about"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         user_id: str = SchemaField(
             description="The user's ID (passed through for chaining)"
         )
@@ -1036,7 +1030,7 @@ class DiscordUserInfoBlock(Block):
 
 
 class DiscordChannelInfoBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordCredentials = DiscordCredentialsField()
         channel_identifier: str = SchemaField(
             description="Channel name or channel ID to look up"
@@ -1047,7 +1041,7 @@ class DiscordChannelInfoBlock(Block):
             default="",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         channel_id: str = SchemaField(description="The channel's ID")
         channel_name: str = SchemaField(description="The channel's name")
         server_id: str = SchemaField(description="The server's ID")

autogpt_platform/backend/backend/blocks/discord/oauth_blocks.py

@@ -2,13 +2,7 @@
 Discord OAuth-based blocks.
 """
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import DiscordOAuthUser, get_current_user
@@ -27,12 +21,12 @@ class DiscordGetCurrentUserBlock(Block):
     This block requires Discord OAuth2 credentials (not bot tokens).
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: DiscordOAuthCredentialsInput = DiscordOAuthCredentialsField(
             ["identify"]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         user_id: str = SchemaField(description="The authenticated user's Discord ID")
         username: str = SchemaField(description="The user's username")
         avatar_url: str = SchemaField(description="URL to the user's avatar image")

autogpt_platform/backend/backend/blocks/email_block.py

@@ -5,13 +5,7 @@ from typing import Literal
 
 from pydantic import BaseModel, ConfigDict, SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     CredentialsField,
     CredentialsMetaInput,
@@ -57,7 +51,7 @@ class SMTPConfig(BaseModel):
 
 
 class SendEmailBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         to_email: str = SchemaField(
             description="Recipient email address", placeholder="recipient@example.com"
         )
@@ -73,7 +67,7 @@ class SendEmailBlock(Block):
         )
         credentials: SMTPCredentialsInput = SMTPCredentialsField()
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Status of the email sending operation")
         error: str = SchemaField(
             description="Error message if the email sending failed"

autogpt_platform/backend/backend/blocks/enrichlayer/linkedin.py

@@ -8,13 +8,7 @@ which provides access to LinkedIn profile data and related information.
 import logging
 from typing import Optional
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import APIKeyCredentials, CredentialsField, SchemaField
 from backend.util.type import MediaFileType
 
@@ -35,7 +29,7 @@ logger = logging.getLogger(__name__)
 class GetLinkedinProfileBlock(Block):
     """Block to fetch LinkedIn profile data using Enrichlayer API."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         """Input schema for GetLinkedinProfileBlock."""
 
         linkedin_url: str = SchemaField(
@@ -86,12 +80,13 @@ class GetLinkedinProfileBlock(Block):
             description="Enrichlayer API credentials"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         """Output schema for GetLinkedinProfileBlock."""
 
         profile: PersonProfileResponse = SchemaField(
             description="LinkedIn profile data"
         )
+        error: str = SchemaField(description="Error message if the request failed")
 
     def __init__(self):
         """Initialize GetLinkedinProfileBlock."""
@@ -204,7 +199,7 @@ class GetLinkedinProfileBlock(Block):
 class LinkedinPersonLookupBlock(Block):
     """Block to look up LinkedIn profiles by person's information using Enrichlayer API."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         """Input schema for LinkedinPersonLookupBlock."""
 
         first_name: str = SchemaField(
@@ -247,12 +242,13 @@ class LinkedinPersonLookupBlock(Block):
             description="Enrichlayer API credentials"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         """Output schema for LinkedinPersonLookupBlock."""
 
         lookup_result: PersonLookupResponse = SchemaField(
             description="LinkedIn profile lookup result"
         )
+        error: str = SchemaField(description="Error message if the request failed")
 
     def __init__(self):
         """Initialize LinkedinPersonLookupBlock."""
@@ -350,7 +346,7 @@ class LinkedinPersonLookupBlock(Block):
 class LinkedinRoleLookupBlock(Block):
     """Block to look up LinkedIn profiles by role in a company using Enrichlayer API."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         """Input schema for LinkedinRoleLookupBlock."""
 
         role: str = SchemaField(
@@ -370,12 +366,13 @@ class LinkedinRoleLookupBlock(Block):
             description="Enrichlayer API credentials"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         """Output schema for LinkedinRoleLookupBlock."""
 
         role_lookup_result: RoleLookupResponse = SchemaField(
             description="LinkedIn role lookup result"
         )
+        error: str = SchemaField(description="Error message if the request failed")
 
     def __init__(self):
         """Initialize LinkedinRoleLookupBlock."""
@@ -452,7 +449,7 @@ class LinkedinRoleLookupBlock(Block):
 class GetLinkedinProfilePictureBlock(Block):
     """Block to get LinkedIn profile pictures using Enrichlayer API."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         """Input schema for GetLinkedinProfilePictureBlock."""
 
         linkedin_profile_url: str = SchemaField(
@@ -463,12 +460,13 @@ class GetLinkedinProfilePictureBlock(Block):
             description="Enrichlayer API credentials"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         """Output schema for GetLinkedinProfilePictureBlock."""
 
         profile_picture_url: MediaFileType = SchemaField(
             description="LinkedIn profile picture URL"
         )
+        error: str = SchemaField(description="Error message if the request failed")
 
     def __init__(self):
         """Initialize GetLinkedinProfilePictureBlock."""

autogpt_platform/backend/backend/blocks/exa/answers.py

@@ -4,8 +4,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -50,7 +49,7 @@ class CostDollars(BaseModel):
 
 
 class ExaAnswerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -70,7 +69,7 @@ class ExaAnswerBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         answer: str = SchemaField(
             description="The generated answer based on search results"
         )

autogpt_platform/backend/backend/blocks/exa/contents.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -15,7 +14,7 @@ from .helpers import ContentSettings
 
 
 class ExaContentsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -28,7 +27,7 @@ class ExaContentsBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: list = SchemaField(
             description="List of document contents", default_factory=list
         )

autogpt_platform/backend/backend/blocks/exa/search.py

@@ -5,8 +5,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -17,7 +16,7 @@ from .helpers import ContentSettings
 
 
 class ExaSearchBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -64,7 +63,7 @@ class ExaSearchBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: list = SchemaField(
             description="List of search results", default_factory=list
         )

autogpt_platform/backend/backend/blocks/exa/similar.py

@@ -6,8 +6,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -18,7 +17,7 @@ from .helpers import ContentSettings
 
 
 class ExaFindSimilarBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -66,7 +65,7 @@ class ExaFindSimilarBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: list[Any] = SchemaField(
             description="List of similar documents with title, URL, published date, author, and score",
             default_factory=list,

autogpt_platform/backend/backend/blocks/exa/webhook_blocks.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockType,
     BlockWebhookConfig,
     CredentialsMetaInput,
@@ -85,7 +84,7 @@ class ExaWebsetWebhookBlock(Block):
     including creation, updates, searches, and exports.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="Exa API credentials for webhook management"
         )
@@ -105,7 +104,7 @@ class ExaWebsetWebhookBlock(Block):
             description="Webhook payload data", default={}, hidden=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         event_type: str = SchemaField(description="Type of event that occurred")
         event_id: str = SchemaField(description="Unique identifier for this event")
         webset_id: str = SchemaField(description="ID of the affected webset")

autogpt_platform/backend/backend/blocks/exa/websets.py

@@ -31,8 +31,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     Requests,
     SchemaField,
@@ -105,7 +104,7 @@ class Webset(BaseModel):
 
 
 class ExaCreateWebsetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -220,7 +219,7 @@ class ExaCreateWebsetBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         webset: Webset = SchemaField(
             description="The unique identifier for the created webset"
         )
@@ -405,7 +404,7 @@ class ExaCreateWebsetBlock(Block):
 
 
 class ExaUpdateWebsetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -418,7 +417,7 @@ class ExaUpdateWebsetBlock(Block):
             description="Key-value pairs to associate with this webset (set to null to clear)",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         webset_id: str = SchemaField(description="The unique identifier for the webset")
         status: str = SchemaField(description="The status of the webset")
         external_id: Optional[str] = SchemaField(
@@ -476,7 +475,7 @@ class ExaUpdateWebsetBlock(Block):
 
 
 class ExaListWebsetsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -498,7 +497,7 @@ class ExaListWebsetsBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         websets: list[Webset] = SchemaField(
             description="List of websets", default_factory=list
         )
@@ -551,7 +550,7 @@ class ExaListWebsetsBlock(Block):
 
 
 class ExaGetWebsetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -560,7 +559,7 @@ class ExaGetWebsetBlock(Block):
             placeholder="webset-id-or-external-id",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         webset_id: str = SchemaField(description="The unique identifier for the webset")
         status: str = SchemaField(description="The status of the webset")
         external_id: Optional[str] = SchemaField(
@@ -638,7 +637,7 @@ class ExaGetWebsetBlock(Block):
 
 
 class ExaDeleteWebsetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -647,7 +646,7 @@ class ExaDeleteWebsetBlock(Block):
             placeholder="webset-id-or-external-id",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         webset_id: str = SchemaField(
             description="The unique identifier for the deleted webset"
         )
@@ -696,7 +695,7 @@ class ExaDeleteWebsetBlock(Block):
 
 
 class ExaCancelWebsetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = exa.credentials_field(
             description="The Exa integration requires an API Key."
         )
@@ -705,7 +704,7 @@ class ExaCancelWebsetBlock(Block):
             placeholder="webset-id-or-external-id",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         webset_id: str = SchemaField(description="The unique identifier for the webset")
         status: str = SchemaField(
             description="The status of the webset after cancellation"

autogpt_platform/backend/backend/blocks/fal/ai_video_generator.py

@@ -10,13 +10,7 @@ from backend.blocks.fal._auth import (
     FalCredentialsField,
     FalCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import ClientResponseError, Requests
 
@@ -30,7 +24,7 @@ class FalModel(str, Enum):
 
 
 class AIVideoGeneratorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         prompt: str = SchemaField(
             description="Description of the video to generate.",
             placeholder="A dog running in a field.",
@@ -42,7 +36,7 @@ class AIVideoGeneratorBlock(Block):
         )
         credentials: FalCredentialsInput = FalCredentialsField()
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_url: str = SchemaField(description="The URL of the generated video.")
         error: str = SchemaField(
             description="Error message if video generation failed."

autogpt_platform/backend/backend/blocks/firecrawl/crawl.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -20,7 +19,7 @@ from ._format_utils import convert_to_format_options
 
 
 class FirecrawlCrawlBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = firecrawl.credentials_field()
         url: str = SchemaField(description="The URL to crawl")
         limit: int = SchemaField(description="The number of pages to crawl", default=10)
@@ -40,7 +39,7 @@ class FirecrawlCrawlBlock(Block):
             description="The format of the crawl", default=[ScrapeFormat.MARKDOWN]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         data: list[dict[str, Any]] = SchemaField(description="The result of the crawl")
         markdown: str = SchemaField(description="The markdown of the crawl")
         html: str = SchemaField(description="The html of the crawl")
@@ -56,10 +55,6 @@ class FirecrawlCrawlBlock(Block):
         change_tracking: dict[str, Any] = SchemaField(
             description="The change tracking of the crawl"
         )
-        error: str = SchemaField(
-            description="Error message if the crawl failed",
-            default="",
-        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/firecrawl/extract.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     BlockCost,
     BlockCostType,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
     cost,
@@ -21,7 +20,7 @@ from ._config import firecrawl
 
 @cost(BlockCost(2, BlockCostType.RUN))
 class FirecrawlExtractBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = firecrawl.credentials_field()
         urls: list[str] = SchemaField(
             description="The URLs to crawl - at least one is required. Wildcards are supported. (/*)"
@@ -38,12 +37,8 @@ class FirecrawlExtractBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         data: dict[str, Any] = SchemaField(description="The result of the crawl")
-        error: str = SchemaField(
-            description="Error message if the extraction failed",
-            default="",
-        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/firecrawl/map.py

@@ -7,8 +7,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -17,20 +16,16 @@ from ._config import firecrawl
 
 
 class FirecrawlMapWebsiteBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = firecrawl.credentials_field()
 
         url: str = SchemaField(description="The website url to map")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         links: list[str] = SchemaField(description="List of URLs found on the website")
         results: list[dict[str, Any]] = SchemaField(
             description="List of search results with url, title, and description"
         )
-        error: str = SchemaField(
-            description="Error message if the map failed",
-            default="",
-        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/firecrawl/scrape.py

@@ -8,8 +8,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -19,7 +18,7 @@ from ._format_utils import convert_to_format_options
 
 
 class FirecrawlScrapeBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = firecrawl.credentials_field()
         url: str = SchemaField(description="The URL to crawl")
         limit: int = SchemaField(description="The number of pages to crawl", default=10)
@@ -39,7 +38,7 @@ class FirecrawlScrapeBlock(Block):
             description="The format of the crawl", default=[ScrapeFormat.MARKDOWN]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         data: dict[str, Any] = SchemaField(description="The result of the crawl")
         markdown: str = SchemaField(description="The markdown of the crawl")
         html: str = SchemaField(description="The html of the crawl")
@@ -55,10 +54,6 @@ class FirecrawlScrapeBlock(Block):
         change_tracking: dict[str, Any] = SchemaField(
             description="The change tracking of the crawl"
         )
-        error: str = SchemaField(
-            description="Error message if the scrape failed",
-            default="",
-        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/firecrawl/search.py

@@ -9,8 +9,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     SchemaField,
 )
@@ -20,7 +19,7 @@ from ._format_utils import convert_to_format_options
 
 
 class FirecrawlSearchBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = firecrawl.credentials_field()
         query: str = SchemaField(description="The query to search for")
         limit: int = SchemaField(description="The number of pages to crawl", default=10)
@@ -36,13 +35,9 @@ class FirecrawlSearchBlock(Block):
             description="Returns the content of the search if specified", default=[]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         data: dict[str, Any] = SchemaField(description="The result of the search")
         site: dict[str, Any] = SchemaField(description="The site of the search")
-        error: str = SchemaField(
-            description="Error message if the search failed",
-            default="",
-        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/flux_kontext.py

@@ -5,13 +5,7 @@ from pydantic import SecretStr
 from replicate.client import Client as ReplicateClient
 from replicate.helpers import FileOutput
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -63,7 +57,7 @@ class AspectRatio(str, Enum):
 
 
 class AIImageEditorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.REPLICATE], Literal["api_key"]
         ] = CredentialsField(
@@ -96,10 +90,11 @@ class AIImageEditorBlock(Block):
             title="Model",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         output_image: MediaFileType = SchemaField(
             description="URL of the transformed image"
         )
+        error: str = SchemaField(description="Error message if generation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/generic_webhook/triggers.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     BlockCategory,
     BlockManualWebhookConfig,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     ProviderBuilder,
     ProviderName,
     SchemaField,
@@ -20,14 +19,14 @@ generic_webhook = (
 
 
 class GenericWebhookTriggerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         payload: dict = SchemaField(hidden=True, default_factory=dict)
         constants: dict = SchemaField(
             description="The constants to be set when the block is put on the graph",
             default_factory=dict,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         payload: dict = SchemaField(
             description="The complete webhook payload that was received from the generic webhook."
         )

autogpt_platform/backend/backend/blocks/github/checks.py

@@ -3,13 +3,7 @@ from typing import Optional
 
 from pydantic import BaseModel
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -45,7 +39,7 @@ class ChecksConclusion(Enum):
 class GithubCreateCheckRunBlock(Block):
     """Block for creating a new check run on a GitHub repository."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo:status")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -82,7 +76,7 @@ class GithubCreateCheckRunBlock(Block):
             default="",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class CheckRunResult(BaseModel):
             id: int
             html_url: str
@@ -217,7 +211,7 @@ class GithubCreateCheckRunBlock(Block):
 class GithubUpdateCheckRunBlock(Block):
     """Block for updating an existing check run on a GitHub repository."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo:status")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -245,7 +239,7 @@ class GithubUpdateCheckRunBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class CheckRunResult(BaseModel):
             id: int
             html_url: str
@@ -255,6 +249,7 @@ class GithubUpdateCheckRunBlock(Block):
         check_run: CheckRunResult = SchemaField(
             description="Details of the updated check run"
         )
+        error: str = SchemaField(description="Error message if check run update failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/github/ci.py

@@ -5,13 +5,7 @@ from typing import Optional
 
 from typing_extensions import TypedDict
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -43,7 +37,7 @@ class CheckRunConclusion(Enum):
 
 
 class GithubGetCIResultsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description="GitHub repository",
@@ -66,7 +60,7 @@ class GithubGetCIResultsBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class CheckRunItem(TypedDict, total=False):
             id: int
             name: str
@@ -110,6 +104,7 @@ class GithubGetCIResultsBlock(Block):
         total_checks: int = SchemaField(description="Total number of CI checks")
         passed_checks: int = SchemaField(description="Number of passed checks")
         failed_checks: int = SchemaField(description="Number of failed checks")
+        error: str = SchemaField(description="Error message if the operation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/github/issues.py

@@ -3,13 +3,7 @@ from urllib.parse import urlparse
 
 from typing_extensions import TypedDict
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import convert_comment_url_to_api_endpoint, get_api
@@ -30,7 +24,7 @@ def is_github_url(url: str) -> bool:
 
 # --8<-- [start:GithubCommentBlockExample]
 class GithubCommentBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue or pull request",
@@ -41,7 +35,7 @@ class GithubCommentBlock(Block):
             placeholder="Enter your comment",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         id: int = SchemaField(description="ID of the created comment")
         url: str = SchemaField(description="URL to the comment on GitHub")
         error: str = SchemaField(
@@ -118,7 +112,7 @@ class GithubCommentBlock(Block):
 
 
 class GithubUpdateCommentBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         comment_url: str = SchemaField(
             description="URL of the GitHub comment",
@@ -141,7 +135,7 @@ class GithubUpdateCommentBlock(Block):
             placeholder="Enter your comment",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         id: int = SchemaField(description="ID of the updated comment")
         url: str = SchemaField(description="URL to the comment on GitHub")
         error: str = SchemaField(
@@ -225,14 +219,14 @@ class GithubUpdateCommentBlock(Block):
 
 
 class GithubListCommentsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue or pull request",
             placeholder="https://github.com/owner/repo/issues/1",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class CommentItem(TypedDict):
             id: int
             body: str
@@ -245,6 +239,7 @@ class GithubListCommentsBlock(Block):
         comments: list[CommentItem] = SchemaField(
             description="List of comments with their ID, body, user, and URL"
         )
+        error: str = SchemaField(description="Error message if listing comments failed")
 
     def __init__(self):
         super().__init__(
@@ -340,7 +335,7 @@ class GithubListCommentsBlock(Block):
 
 
 class GithubMakeIssueBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -353,7 +348,7 @@ class GithubMakeIssueBlock(Block):
             description="Body of the issue", placeholder="Enter the issue body"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         number: int = SchemaField(description="Number of the created issue")
         url: str = SchemaField(description="URL of the created issue")
         error: str = SchemaField(
@@ -415,14 +410,14 @@ class GithubMakeIssueBlock(Block):
 
 
 class GithubReadIssueBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue",
             placeholder="https://github.com/owner/repo/issues/1",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         title: str = SchemaField(description="Title of the issue")
         body: str = SchemaField(description="Body of the issue")
         user: str = SchemaField(description="User who created the issue")
@@ -488,14 +483,14 @@ class GithubReadIssueBlock(Block):
 
 
 class GithubListIssuesBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class IssueItem(TypedDict):
             title: str
             url: str
@@ -506,6 +501,7 @@ class GithubListIssuesBlock(Block):
         issues: list[IssueItem] = SchemaField(
             description="List of issues with their title and URL"
         )
+        error: str = SchemaField(description="Error message if listing issues failed")
 
     def __init__(self):
         super().__init__(
@@ -577,7 +573,7 @@ class GithubListIssuesBlock(Block):
 
 
 class GithubAddLabelBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue or pull request",
@@ -588,7 +584,7 @@ class GithubAddLabelBlock(Block):
             placeholder="Enter the label",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Status of the label addition operation")
         error: str = SchemaField(
             description="Error message if the label addition failed"
@@ -637,7 +633,7 @@ class GithubAddLabelBlock(Block):
 
 
 class GithubRemoveLabelBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue or pull request",
@@ -648,7 +644,7 @@ class GithubRemoveLabelBlock(Block):
             placeholder="Enter the label",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Status of the label removal operation")
         error: str = SchemaField(
             description="Error message if the label removal failed"
@@ -698,7 +694,7 @@ class GithubRemoveLabelBlock(Block):
 
 
 class GithubAssignIssueBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue",
@@ -709,7 +705,7 @@ class GithubAssignIssueBlock(Block):
             placeholder="Enter the username",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="Status of the issue assignment operation"
         )
@@ -764,7 +760,7 @@ class GithubAssignIssueBlock(Block):
 
 
 class GithubUnassignIssueBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         issue_url: str = SchemaField(
             description="URL of the GitHub issue",
@@ -775,7 +771,7 @@ class GithubUnassignIssueBlock(Block):
             placeholder="Enter the username",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="Status of the issue unassignment operation"
         )

autogpt_platform/backend/backend/blocks/github/pull_requests.py

@@ -2,13 +2,7 @@ import re
 
 from typing_extensions import TypedDict
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -22,14 +16,14 @@ from ._auth import (
 
 
 class GithubListPullRequestsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class PRItem(TypedDict):
             title: str
             url: str
@@ -114,7 +108,7 @@ class GithubListPullRequestsBlock(Block):
 
 
 class GithubMakePullRequestBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -141,7 +135,7 @@ class GithubMakePullRequestBlock(Block):
             placeholder="Enter the base branch",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         number: int = SchemaField(description="Number of the created pull request")
         url: str = SchemaField(description="URL of the created pull request")
         error: str = SchemaField(
@@ -215,7 +209,7 @@ class GithubMakePullRequestBlock(Block):
 
 
 class GithubReadPullRequestBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         pr_url: str = SchemaField(
             description="URL of the GitHub pull request",
@@ -227,7 +221,7 @@ class GithubReadPullRequestBlock(Block):
             advanced=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         title: str = SchemaField(description="Title of the pull request")
         body: str = SchemaField(description="Body of the pull request")
         author: str = SchemaField(description="User who created the pull request")
@@ -331,7 +325,7 @@ class GithubReadPullRequestBlock(Block):
 
 
 class GithubAssignPRReviewerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         pr_url: str = SchemaField(
             description="URL of the GitHub pull request",
@@ -342,7 +336,7 @@ class GithubAssignPRReviewerBlock(Block):
             placeholder="Enter the reviewer's username",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="Status of the reviewer assignment operation"
         )
@@ -398,7 +392,7 @@ class GithubAssignPRReviewerBlock(Block):
 
 
 class GithubUnassignPRReviewerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         pr_url: str = SchemaField(
             description="URL of the GitHub pull request",
@@ -409,7 +403,7 @@ class GithubUnassignPRReviewerBlock(Block):
             placeholder="Enter the reviewer's username",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="Status of the reviewer unassignment operation"
         )
@@ -465,14 +459,14 @@ class GithubUnassignPRReviewerBlock(Block):
 
 
 class GithubListPRReviewersBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         pr_url: str = SchemaField(
             description="URL of the GitHub pull request",
             placeholder="https://github.com/owner/repo/pull/1",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class ReviewerItem(TypedDict):
             username: str
             url: str

autogpt_platform/backend/backend/blocks/github/repo.py

@@ -2,13 +2,7 @@ import base64
 
 from typing_extensions import TypedDict
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -22,14 +16,14 @@ from ._auth import (
 
 
 class GithubListTagsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class TagItem(TypedDict):
             name: str
             url: str
@@ -40,6 +34,7 @@ class GithubListTagsBlock(Block):
         tags: list[TagItem] = SchemaField(
             description="List of tags with their name and file tree browser URL"
         )
+        error: str = SchemaField(description="Error message if listing tags failed")
 
     def __init__(self):
         super().__init__(
@@ -116,14 +111,14 @@ class GithubListTagsBlock(Block):
 
 
 class GithubListBranchesBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class BranchItem(TypedDict):
             name: str
             url: str
@@ -135,6 +130,7 @@ class GithubListBranchesBlock(Block):
         branches: list[BranchItem] = SchemaField(
             description="List of branches with their name and file tree browser URL"
         )
+        error: str = SchemaField(description="Error message if listing branches failed")
 
     def __init__(self):
         super().__init__(
@@ -211,7 +207,7 @@ class GithubListBranchesBlock(Block):
 
 
 class GithubListDiscussionsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -221,7 +217,7 @@ class GithubListDiscussionsBlock(Block):
             description="Number of discussions to fetch", default=5
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class DiscussionItem(TypedDict):
             title: str
             url: str
@@ -327,14 +323,14 @@ class GithubListDiscussionsBlock(Block):
 
 
 class GithubListReleasesBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class ReleaseItem(TypedDict):
             name: str
             url: str
@@ -346,6 +342,7 @@ class GithubListReleasesBlock(Block):
         releases: list[ReleaseItem] = SchemaField(
             description="List of releases with their name and file tree browser URL"
         )
+        error: str = SchemaField(description="Error message if listing releases failed")
 
     def __init__(self):
         super().__init__(
@@ -417,7 +414,7 @@ class GithubListReleasesBlock(Block):
 
 
 class GithubReadFileBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -433,7 +430,7 @@ class GithubReadFileBlock(Block):
             default="master",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         text_content: str = SchemaField(
             description="Content of the file (decoded as UTF-8 text)"
         )
@@ -441,6 +438,7 @@ class GithubReadFileBlock(Block):
             description="Raw base64-encoded content of the file"
         )
         size: int = SchemaField(description="The size of the file (in bytes)")
+        error: str = SchemaField(description="Error message if the file reading failed")
 
     def __init__(self):
         super().__init__(
@@ -503,7 +501,7 @@ class GithubReadFileBlock(Block):
 
 
 class GithubReadFolderBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -519,7 +517,7 @@ class GithubReadFolderBlock(Block):
             default="master",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class DirEntry(TypedDict):
             name: str
             path: str
@@ -627,7 +625,7 @@ class GithubReadFolderBlock(Block):
 
 
 class GithubMakeBranchBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -642,7 +640,7 @@ class GithubMakeBranchBlock(Block):
             placeholder="source_branch_name",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Status of the branch creation operation")
         error: str = SchemaField(
             description="Error message if the branch creation failed"
@@ -707,7 +705,7 @@ class GithubMakeBranchBlock(Block):
 
 
 class GithubDeleteBranchBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -718,7 +716,7 @@ class GithubDeleteBranchBlock(Block):
             placeholder="branch_name",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(description="Status of the branch deletion operation")
         error: str = SchemaField(
             description="Error message if the branch deletion failed"
@@ -768,7 +766,7 @@ class GithubDeleteBranchBlock(Block):
 
 
 class GithubCreateFileBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -791,7 +789,7 @@ class GithubCreateFileBlock(Block):
             default="Create new file",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         url: str = SchemaField(description="URL of the created file")
         sha: str = SchemaField(description="SHA of the commit")
         error: str = SchemaField(
@@ -870,7 +868,7 @@ class GithubCreateFileBlock(Block):
 
 
 class GithubUpdateFileBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
@@ -893,9 +891,10 @@ class GithubUpdateFileBlock(Block):
             default="Update file",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         url: str = SchemaField(description="URL of the updated file")
         sha: str = SchemaField(description="SHA of the commit")
+        error: str = SchemaField(description="Error message if the file update failed")
 
     def __init__(self):
         super().__init__(
@@ -975,7 +974,7 @@ class GithubUpdateFileBlock(Block):
 
 
 class GithubCreateRepositoryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         name: str = SchemaField(
             description="Name of the repository to create",
@@ -999,7 +998,7 @@ class GithubCreateRepositoryBlock(Block):
             default="",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         url: str = SchemaField(description="URL of the created repository")
         clone_url: str = SchemaField(description="Git clone URL of the repository")
         error: str = SchemaField(
@@ -1078,14 +1077,14 @@ class GithubCreateRepositoryBlock(Block):
 
 
 class GithubListStargazersBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo_url: str = SchemaField(
             description="URL of the GitHub repository",
             placeholder="https://github.com/owner/repo",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class StargazerItem(TypedDict):
             username: str
             url: str

autogpt_platform/backend/backend/blocks/github/reviews.py

@@ -4,13 +4,7 @@ from typing import Any, List, Optional
 
 from typing_extensions import TypedDict
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -32,7 +26,7 @@ class ReviewEvent(Enum):
 
 
 class GithubCreatePRReviewBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         class ReviewComment(TypedDict, total=False):
             path: str
             position: Optional[int]
@@ -67,7 +61,7 @@ class GithubCreatePRReviewBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         review_id: int = SchemaField(description="ID of the created review")
         state: str = SchemaField(
             description="State of the review (e.g., PENDING, COMMENTED, APPROVED, CHANGES_REQUESTED)"
@@ -203,7 +197,7 @@ class GithubCreatePRReviewBlock(Block):
 
 
 class GithubListPRReviewsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description="GitHub repository",
@@ -214,7 +208,7 @@ class GithubListPRReviewsBlock(Block):
             placeholder="123",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class ReviewItem(TypedDict):
             id: int
             user: str
@@ -229,6 +223,7 @@ class GithubListPRReviewsBlock(Block):
         reviews: list[ReviewItem] = SchemaField(
             description="List of all reviews on the pull request"
         )
+        error: str = SchemaField(description="Error message if listing reviews failed")
 
     def __init__(self):
         super().__init__(
@@ -322,7 +317,7 @@ class GithubListPRReviewsBlock(Block):
 
 
 class GithubSubmitPendingReviewBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description="GitHub repository",
@@ -341,7 +336,7 @@ class GithubSubmitPendingReviewBlock(Block):
             default=ReviewEvent.COMMENT,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         state: str = SchemaField(description="State of the submitted review")
         html_url: str = SchemaField(description="URL of the submitted review")
         error: str = SchemaField(
@@ -420,7 +415,7 @@ class GithubSubmitPendingReviewBlock(Block):
 
 
 class GithubResolveReviewDiscussionBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description="GitHub repository",
@@ -439,8 +434,9 @@ class GithubResolveReviewDiscussionBlock(Block):
             default=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         success: bool = SchemaField(description="Whether the operation was successful")
+        error: str = SchemaField(description="Error message if the operation failed")
 
     def __init__(self):
         super().__init__(
@@ -583,7 +579,7 @@ class GithubResolveReviewDiscussionBlock(Block):
 
 
 class GithubGetPRReviewCommentsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description="GitHub repository",
@@ -600,7 +596,7 @@ class GithubGetPRReviewCommentsBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class CommentItem(TypedDict):
             id: int
             user: str
@@ -620,6 +616,7 @@ class GithubGetPRReviewCommentsBlock(Block):
         comments: list[CommentItem] = SchemaField(
             description="List of all review comments on the pull request"
         )
+        error: str = SchemaField(description="Error message if getting comments failed")
 
     def __init__(self):
         super().__init__(
@@ -747,7 +744,7 @@ class GithubGetPRReviewCommentsBlock(Block):
 
 
 class GithubCreateCommentObjectBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         path: str = SchemaField(
             description="The file path to comment on",
             placeholder="src/main.py",
@@ -784,7 +781,7 @@ class GithubCreateCommentObjectBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         comment_object: dict = SchemaField(
             description="The comment object formatted for GitHub API"
         )

autogpt_platform/backend/backend/blocks/github/statuses.py

@@ -3,13 +3,7 @@ from typing import Optional
 
 from pydantic import BaseModel
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 from ._api import get_api
@@ -32,7 +26,7 @@ class StatusState(Enum):
 class GithubCreateStatusBlock(Block):
     """Block for creating a commit status on a GitHub repository."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubFineGrainedAPICredentialsInput = (
             GithubFineGrainedAPICredentialsField("repo:status")
         )
@@ -60,7 +54,7 @@ class GithubCreateStatusBlock(Block):
             advanced=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         class StatusResult(BaseModel):
             id: int
             url: str
@@ -72,6 +66,7 @@ class GithubCreateStatusBlock(Block):
             updated_at: str
 
         status: StatusResult = SchemaField(description="Details of the created status")
+        error: str = SchemaField(description="Error message if status creation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/github/triggers.py

@@ -8,8 +8,7 @@ from backend.data.block import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     BlockWebhookConfig,
 )
 from backend.data.model import SchemaField
@@ -27,7 +26,7 @@ logger = logging.getLogger(__name__)
 
 # --8<-- [start:GithubTriggerExample]
 class GitHubTriggerBase:
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GithubCredentialsInput = GithubCredentialsField("repo")
         repo: str = SchemaField(
             description=(
@@ -41,7 +40,7 @@ class GitHubTriggerBase:
         payload: dict = SchemaField(hidden=True, default_factory=dict)
         # --8<-- [end:example-payload-field]
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         payload: dict = SchemaField(
             description="The complete webhook payload that was received from GitHub. "
             "Includes information about the affected resource (e.g. pull request), "

autogpt_platform/backend/backend/blocks/google/calendar.py

@@ -8,13 +8,7 @@ from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 from pydantic import BaseModel
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.settings import Settings
 
@@ -49,7 +43,7 @@ class CalendarEvent(BaseModel):
 
 
 class GoogleCalendarReadEventsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/calendar.readonly"]
         )
@@ -79,7 +73,7 @@ class GoogleCalendarReadEventsBlock(Block):
             description="Include events you've declined", default=False
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         events: list[CalendarEvent] = SchemaField(
             description="List of calendar events in the requested time range",
             default_factory=list,
@@ -385,7 +379,7 @@ class RecurringEvent(BaseModel):
 
 
 class GoogleCalendarCreateEventBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/calendar"]
         )
@@ -439,11 +433,12 @@ class GoogleCalendarCreateEventBlock(Block):
             default_factory=lambda: [ReminderPreset.TEN_MINUTES],
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         event_id: str = SchemaField(description="ID of the created event")
         event_link: str = SchemaField(
             description="Link to view the event in Google Calendar"
         )
+        error: str = SchemaField(description="Error message if event creation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/google/gmail.py

@@ -14,13 +14,7 @@ from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 from pydantic import BaseModel, Field
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.file import MediaFileType, get_exec_file_path, store_media_file
 from backend.util.settings import Settings
@@ -326,7 +320,7 @@ class GmailBase(Block, ABC):
 
 
 class GmailReadBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.readonly"]
         )
@@ -339,7 +333,7 @@ class GmailReadBlock(GmailBase):
             default=10,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         email: Email = SchemaField(
             description="Email data",
         )
@@ -522,7 +516,7 @@ class GmailSendBlock(GmailBase):
     - Attachment support for multiple files
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.send"]
         )
@@ -546,7 +540,7 @@ class GmailSendBlock(GmailBase):
             description="Files to attach", default_factory=list, advanced=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: GmailSendResult = SchemaField(
             description="Send confirmation",
         )
@@ -624,7 +618,7 @@ class GmailCreateDraftBlock(GmailBase):
     - Attachment support for multiple files
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.modify"]
         )
@@ -648,7 +642,7 @@ class GmailCreateDraftBlock(GmailBase):
             description="Files to attach", default_factory=list, advanced=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: GmailDraftResult = SchemaField(
             description="Draft creation result",
         )
@@ -727,12 +721,12 @@ class GmailCreateDraftBlock(GmailBase):
 
 
 class GmailListLabelsBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.labels"]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: list[dict] = SchemaField(
             description="List of labels",
         )
@@ -785,7 +779,7 @@ class GmailListLabelsBlock(GmailBase):
 
 
 class GmailAddLabelBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.modify"]
         )
@@ -796,7 +790,7 @@ class GmailAddLabelBlock(GmailBase):
             description="Label name to add",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: GmailLabelResult = SchemaField(
             description="Label addition result",
         )
@@ -871,7 +865,7 @@ class GmailAddLabelBlock(GmailBase):
 
 
 class GmailRemoveLabelBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.modify"]
         )
@@ -882,7 +876,7 @@ class GmailRemoveLabelBlock(GmailBase):
             description="Label name to remove",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: GmailLabelResult = SchemaField(
             description="Label removal result",
         )
@@ -947,16 +941,17 @@ class GmailRemoveLabelBlock(GmailBase):
 
 
 class GmailGetThreadBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.readonly"]
         )
         threadId: str = SchemaField(description="Gmail thread ID")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         thread: Thread = SchemaField(
             description="Gmail thread with decoded message bodies"
         )
+        error: str = SchemaField(description="Error message if any")
 
     def __init__(self):
         super().__init__(
@@ -1223,7 +1218,7 @@ class GmailReplyBlock(GmailBase):
     - Full Unicode/emoji support with UTF-8 encoding
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             [
                 "https://www.googleapis.com/auth/gmail.send",
@@ -1251,13 +1246,14 @@ class GmailReplyBlock(GmailBase):
             description="Files to attach", default_factory=list, advanced=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         messageId: str = SchemaField(description="Sent message ID")
         threadId: str = SchemaField(description="Thread ID")
         message: dict = SchemaField(description="Raw Gmail message object")
         email: Email = SchemaField(
             description="Parsed email object with decoded body and attachments"
         )
+        error: str = SchemaField(description="Error message if any")
 
     def __init__(self):
         super().__init__(
@@ -1372,7 +1368,7 @@ class GmailDraftReplyBlock(GmailBase):
     - Full Unicode/emoji support with UTF-8 encoding
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             [
                 "https://www.googleapis.com/auth/gmail.modify",
@@ -1400,11 +1396,12 @@ class GmailDraftReplyBlock(GmailBase):
             description="Files to attach", default_factory=list, advanced=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         draftId: str = SchemaField(description="Created draft ID")
         messageId: str = SchemaField(description="Draft message ID")
         threadId: str = SchemaField(description="Thread ID")
         status: str = SchemaField(description="Draft creation status")
+        error: str = SchemaField(description="Error message if any")
 
     def __init__(self):
         super().__init__(
@@ -1485,13 +1482,14 @@ class GmailDraftReplyBlock(GmailBase):
 
 
 class GmailGetProfileBlock(GmailBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/gmail.readonly"]
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         profile: Profile = SchemaField(description="Gmail user profile information")
+        error: str = SchemaField(description="Error message if any")
 
     def __init__(self):
         super().__init__(
@@ -1557,7 +1555,7 @@ class GmailForwardBlock(GmailBase):
     - Manual content type override option
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             [
                 "https://www.googleapis.com/auth/gmail.send",
@@ -1591,10 +1589,11 @@ class GmailForwardBlock(GmailBase):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         messageId: str = SchemaField(description="Forwarded message ID")
         threadId: str = SchemaField(description="Thread ID")
         status: str = SchemaField(description="Forward status")
+        error: str = SchemaField(description="Error message if any")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/google/sheets.py

@@ -5,13 +5,7 @@ from typing import Any
 from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.settings import Settings
 
@@ -201,7 +195,7 @@ class BatchOperationType(str, Enum):
     CLEAR = "clear"
 
 
-class BatchOperation(BlockSchemaInput):
+class BatchOperation(BlockSchema):
     type: BatchOperationType = SchemaField(
         description="The type of operation to perform"
     )
@@ -212,7 +206,7 @@ class BatchOperation(BlockSchemaInput):
 
 
 class GoogleSheetsReadBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets.readonly"]
         )
@@ -224,7 +218,7 @@ class GoogleSheetsReadBlock(Block):
             description="The A1 notation of the range to read",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: list[list[str]] = SchemaField(
             description="The data read from the spreadsheet",
         )
@@ -280,7 +274,7 @@ class GoogleSheetsReadBlock(Block):
 
 
 class GoogleSheetsWriteBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -295,7 +289,7 @@ class GoogleSheetsWriteBlock(Block):
             description="The data to write to the spreadsheet",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result of the write operation",
         )
@@ -369,7 +363,7 @@ class GoogleSheetsWriteBlock(Block):
 
 
 class GoogleSheetsAppendBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -409,8 +403,9 @@ class GoogleSheetsAppendBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(description="Append API response")
+        error: str = SchemaField(description="Error message, if any")
 
     def __init__(self):
         super().__init__(
@@ -508,7 +503,7 @@ class GoogleSheetsAppendBlock(Block):
 
 
 class GoogleSheetsClearBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -520,7 +515,7 @@ class GoogleSheetsClearBlock(Block):
             description="The A1 notation of the range to clear",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result of the clear operation",
         )
@@ -576,7 +571,7 @@ class GoogleSheetsClearBlock(Block):
 
 
 class GoogleSheetsMetadataBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets.readonly"]
         )
@@ -585,7 +580,7 @@ class GoogleSheetsMetadataBlock(Block):
             title="Spreadsheet ID or URL",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The metadata of the spreadsheet including sheets info",
         )
@@ -657,7 +652,7 @@ class GoogleSheetsMetadataBlock(Block):
 
 
 class GoogleSheetsManageSheetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -677,8 +672,9 @@ class GoogleSheetsManageSheetBlock(Block):
             description="New sheet name for copy", default=""
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(description="Operation result")
+        error: str = SchemaField(description="Error message, if any")
 
     def __init__(self):
         super().__init__(
@@ -764,7 +760,7 @@ class GoogleSheetsManageSheetBlock(Block):
 
 
 class GoogleSheetsBatchOperationsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -776,7 +772,7 @@ class GoogleSheetsBatchOperationsBlock(Block):
             description="List of operations to perform",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result of the batch operations",
         )
@@ -881,7 +877,7 @@ class GoogleSheetsBatchOperationsBlock(Block):
 
 
 class GoogleSheetsFindReplaceBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -908,7 +904,7 @@ class GoogleSheetsFindReplaceBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result of the find/replace operation including number of replacements",
         )
@@ -991,7 +987,7 @@ class GoogleSheetsFindReplaceBlock(Block):
 
 
 class GoogleSheetsFindBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets.readonly"]
         )
@@ -1024,7 +1020,7 @@ class GoogleSheetsFindBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result of the find operation including locations and count",
         )
@@ -1259,7 +1255,7 @@ class GoogleSheetsFindBlock(Block):
 
 
 class GoogleSheetsFormatBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -1274,8 +1270,9 @@ class GoogleSheetsFormatBlock(Block):
         italic: bool = SchemaField(default=False)
         font_size: int = SchemaField(default=10)
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(description="API response or success flag")
+        error: str = SchemaField(description="Error message, if any")
 
     def __init__(self):
         super().__init__(
@@ -1386,7 +1383,7 @@ class GoogleSheetsFormatBlock(Block):
 
 
 class GoogleSheetsCreateSpreadsheetBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: GoogleCredentialsInput = GoogleCredentialsField(
             ["https://www.googleapis.com/auth/spreadsheets"]
         )
@@ -1398,7 +1395,7 @@ class GoogleSheetsCreateSpreadsheetBlock(Block):
             default=["Sheet1"],
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(
             description="The result containing spreadsheet ID and URL",
         )

autogpt_platform/backend/backend/blocks/google_maps.py

@@ -3,13 +3,7 @@ from typing import Literal
 import googlemaps
 from pydantic import BaseModel, SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -43,7 +37,7 @@ class Place(BaseModel):
 
 
 class GoogleMapsSearchBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.GOOGLE_MAPS], Literal["api_key"]
         ] = CredentialsField(description="Google Maps API Key")
@@ -64,8 +58,9 @@ class GoogleMapsSearchBlock(Block):
             le=60,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         place: Place = SchemaField(description="Place found")
+        error: str = SchemaField(description="Error message if the search failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/http.py

@@ -8,13 +8,7 @@ from typing import Literal
 import aiofiles
 from pydantic import SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     CredentialsField,
     CredentialsMetaInput,
@@ -68,7 +62,7 @@ class HttpMethod(Enum):
 
 
 class SendWebRequestBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         url: str = SchemaField(
             description="The URL to send the request to",
             placeholder="https://api.example.com",
@@ -99,7 +93,7 @@ class SendWebRequestBlock(Block):
             default_factory=list,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         response: object = SchemaField(description="The response from the server")
         client_error: object = SchemaField(description="Errors on 4xx status codes")
         server_error: object = SchemaField(description="Errors on 5xx status codes")

autogpt_platform/backend/backend/blocks/hubspot/company.py

@@ -3,19 +3,13 @@ from backend.blocks.hubspot._auth import (
     HubSpotCredentialsField,
     HubSpotCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
 
 class HubSpotCompanyBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: HubSpotCredentialsInput = HubSpotCredentialsField()
         operation: str = SchemaField(
             description="Operation to perform (create, update, get)", default="get"
@@ -28,7 +22,7 @@ class HubSpotCompanyBlock(Block):
             description="Company domain for get/update operations", default=""
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         company: dict = SchemaField(description="Company information")
         status: str = SchemaField(description="Operation status")
 

autogpt_platform/backend/backend/blocks/hubspot/contact.py

@@ -3,19 +3,13 @@ from backend.blocks.hubspot._auth import (
     HubSpotCredentialsField,
     HubSpotCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
 
 class HubSpotContactBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: HubSpotCredentialsInput = HubSpotCredentialsField()
         operation: str = SchemaField(
             description="Operation to perform (create, update, get)", default="get"
@@ -28,7 +22,7 @@ class HubSpotContactBlock(Block):
             description="Email address for get/update operations", default=""
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         contact: dict = SchemaField(description="Contact information")
         status: str = SchemaField(description="Operation status")
 

autogpt_platform/backend/backend/blocks/hubspot/engagement.py

@@ -5,19 +5,13 @@ from backend.blocks.hubspot._auth import (
     HubSpotCredentialsField,
     HubSpotCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
 
 class HubSpotEngagementBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: HubSpotCredentialsInput = HubSpotCredentialsField()
         operation: str = SchemaField(
             description="Operation to perform (send_email, track_engagement)",
@@ -35,7 +29,7 @@ class HubSpotEngagementBlock(Block):
             default=30,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: dict = SchemaField(description="Operation result")
         status: str = SchemaField(description="Operation status")
 

autogpt_platform/backend/backend/blocks/ideogram.py

@@ -4,13 +4,7 @@ from typing import Any, Dict, Literal, Optional
 from pydantic import SecretStr
 from requests.exceptions import RequestException
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -90,7 +84,7 @@ class UpscaleOption(str, Enum):
 
 
 class IdeogramModelBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.IDEOGRAM], Literal["api_key"]
         ] = CredentialsField(
@@ -160,8 +154,9 @@ class IdeogramModelBlock(Block):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: str = SchemaField(description="Generated image URL")
+        error: str = SchemaField(description="Error message if the model run failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/io.py

@@ -2,14 +2,7 @@ import copy
 from datetime import date, time
 from typing import Any, Optional
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchema,
-    BlockSchemaInput,
-    BlockType,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema, BlockType
 from backend.data.model import SchemaField
 from backend.util.file import store_media_file
 from backend.util.mock import MockObject
@@ -29,7 +22,7 @@ class AgentInputBlock(Block):
     It Outputs the value passed as input.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         name: str = SchemaField(description="The name of the input.")
         value: Any = SchemaField(
             description="The value to be passed as input.",
@@ -67,7 +60,6 @@ class AgentInputBlock(Block):
             return schema
 
     class Output(BlockSchema):
-        # Use BlockSchema to avoid automatic error field for interface definition
         result: Any = SchemaField(description="The value passed as input.")
 
     def __init__(self, **kwargs):
@@ -117,7 +109,7 @@ class AgentOutputBlock(Block):
         If formatting fails or no `format` is provided, the raw `value` is output.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         value: Any = SchemaField(
             description="The value to be recorded as output.",
             default=None,
@@ -159,7 +151,6 @@ class AgentOutputBlock(Block):
             return self.get_field_schema("value")
 
     class Output(BlockSchema):
-        # Use BlockSchema to avoid automatic error field for interface definition
         output: Any = SchemaField(description="The value recorded as output.")
         name: Any = SchemaField(description="The name of the value recorded as output.")
 

autogpt_platform/backend/backend/blocks/iteration.py

@@ -1,18 +1,12 @@
 from typing import Any
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.json import loads
 
 
 class StepThroughItemsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         items: list = SchemaField(
             advanced=False,
             description="The list or dictionary of items to iterate over",
@@ -32,7 +26,7 @@ class StepThroughItemsBlock(Block):
             default="",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         item: Any = SchemaField(description="The current item in the iteration")
         key: Any = SchemaField(
             description="The key or index of the current item in the iteration",

autogpt_platform/backend/backend/blocks/jina/chunking.py

@@ -3,19 +3,13 @@ from backend.blocks.jina._auth import (
     JinaCredentialsField,
     JinaCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
 
 class JinaChunkingBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         texts: list = SchemaField(description="List of texts to chunk")
 
         credentials: JinaCredentialsInput = JinaCredentialsField()
@@ -26,7 +20,7 @@ class JinaChunkingBlock(Block):
             description="Whether to return token information", default=False
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         chunks: list = SchemaField(description="List of chunked texts")
         tokens: list = SchemaField(
             description="List of token information for each chunk",

autogpt_platform/backend/backend/blocks/jina/embeddings.py

@@ -3,19 +3,13 @@ from backend.blocks.jina._auth import (
     JinaCredentialsField,
     JinaCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
 
 class JinaEmbeddingBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         texts: list = SchemaField(description="List of texts to embed")
         credentials: JinaCredentialsInput = JinaCredentialsField()
         model: str = SchemaField(
@@ -23,7 +17,7 @@ class JinaEmbeddingBlock(Block):
             default="jina-embeddings-v2-base-en",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         embeddings: list = SchemaField(description="List of embeddings")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/jina/fact_checker.py

@@ -8,13 +8,7 @@ from backend.blocks.jina._auth import (
     JinaCredentialsField,
     JinaCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 
@@ -26,13 +20,13 @@ class Reference(TypedDict):
 
 
 class FactCheckerBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         statement: str = SchemaField(
             description="The statement to check for factuality"
         )
         credentials: JinaCredentialsInput = JinaCredentialsField()
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         factuality: float = SchemaField(
             description="The factuality score of the statement"
         )
@@ -42,6 +36,7 @@ class FactCheckerBlock(Block):
             description="List of references supporting or contradicting the statement",
             default=[],
         )
+        error: str = SchemaField(description="Error message if the check fails")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/jina/search.py

@@ -8,25 +8,20 @@ from backend.blocks.jina._auth import (
     JinaCredentialsInput,
 )
 from backend.blocks.search import GetRequest
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
 class SearchTheWebBlock(Block, GetRequest):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: JinaCredentialsInput = JinaCredentialsField()
         query: str = SchemaField(description="The search query to search the web for")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: str = SchemaField(
             description="The search results including content from top 5 URLs"
         )
+        error: str = SchemaField(description="Error message if the search fails")
 
     def __init__(self):
         super().__init__(
@@ -63,7 +58,7 @@ class SearchTheWebBlock(Block, GetRequest):
 
 
 class ExtractWebsiteContentBlock(Block, GetRequest):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: JinaCredentialsInput = JinaCredentialsField()
         url: str = SchemaField(description="The URL to scrape the content from")
         raw_content: bool = SchemaField(
@@ -73,7 +68,7 @@ class ExtractWebsiteContentBlock(Block, GetRequest):
             advanced=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         content: str = SchemaField(description="The scraped content from the given URL")
         error: str = SchemaField(
             description="Error message if the content cannot be retrieved"

autogpt_platform/backend/backend/blocks/linear/comment.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     OAuth2Credentials,
     SchemaField,
@@ -23,7 +22,7 @@ from .models import CreateCommentResponse
 class LinearCreateCommentBlock(Block):
     """Block for creating comments on Linear issues"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = linear.credentials_field(
             description="Linear credentials with comment creation permissions",
             required_scopes={LinearScope.COMMENTS_CREATE},
@@ -31,11 +30,12 @@ class LinearCreateCommentBlock(Block):
         issue_id: str = SchemaField(description="ID of the issue to comment on")
         comment: str = SchemaField(description="Comment text to add to the issue")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         comment_id: str = SchemaField(description="ID of the created comment")
         comment_body: str = SchemaField(
             description="Text content of the created comment"
         )
+        error: str = SchemaField(description="Error message if comment creation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/linear/issues.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     OAuth2Credentials,
     SchemaField,
@@ -23,7 +22,7 @@ from .models import CreateIssueResponse, Issue
 class LinearCreateIssueBlock(Block):
     """Block for creating issues on Linear"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = linear.credentials_field(
             description="Linear credentials with issue creation permissions",
             required_scopes={LinearScope.ISSUES_CREATE},
@@ -44,9 +43,10 @@ class LinearCreateIssueBlock(Block):
             default=None,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         issue_id: str = SchemaField(description="ID of the created issue")
         issue_title: str = SchemaField(description="Title of the created issue")
+        error: str = SchemaField(description="Error message if issue creation failed")
 
     def __init__(self):
         super().__init__(
@@ -129,14 +129,14 @@ class LinearCreateIssueBlock(Block):
 class LinearSearchIssuesBlock(Block):
     """Block for searching issues on Linear"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         term: str = SchemaField(description="Term to search for issues")
         credentials: CredentialsMetaInput = linear.credentials_field(
             description="Linear credentials with read permissions",
             required_scopes={LinearScope.READ},
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         issues: list[Issue] = SchemaField(description="List of issues")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/linear/projects.py

@@ -3,8 +3,7 @@ from backend.sdk import (
     Block,
     BlockCategory,
     BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
+    BlockSchema,
     CredentialsMetaInput,
     OAuth2Credentials,
     SchemaField,
@@ -23,15 +22,16 @@ from .models import Project
 class LinearSearchProjectsBlock(Block):
     """Block for searching projects on Linear"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput = linear.credentials_field(
             description="Linear credentials with read permissions",
             required_scopes={LinearScope.READ},
         )
         term: str = SchemaField(description="Term to search for projects")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         projects: list[Project] = SchemaField(description="List of projects")
+        error: str = SchemaField(description="Error message if issue creation failed")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/llm.py

@@ -16,13 +16,7 @@ from anthropic.types import ToolParam
 from groq import AsyncGroq
 from pydantic import BaseModel, SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -772,7 +766,7 @@ class AIBlockBase(Block, ABC):
 
 
 class AIStructuredResponseGeneratorBlock(AIBlockBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         prompt: str = SchemaField(
             description="The prompt to send to the language model.",
             placeholder="Enter your prompt here...",
@@ -839,11 +833,12 @@ class AIStructuredResponseGeneratorBlock(AIBlockBase):
             description="Ollama host for local  models",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         response: dict[str, Any] | list[dict[str, Any]] = SchemaField(
             description="The response object generated by the language model."
         )
         prompt: list = SchemaField(description="The prompt sent to the language model.")
+        error: str = SchemaField(description="Error message if the API call failed.")
 
     def __init__(self):
         super().__init__(
@@ -1212,7 +1207,7 @@ def trim_prompt(s: str) -> str:
 
 
 class AITextGeneratorBlock(AIBlockBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         prompt: str = SchemaField(
             description="The prompt to send to the language model. You can use any of the {keys} from Prompt Values to fill in the prompt with values from the prompt values dictionary by putting them in curly braces.",
             placeholder="Enter your prompt here...",
@@ -1250,11 +1245,12 @@ class AITextGeneratorBlock(AIBlockBase):
             description="The maximum number of tokens to generate in the chat completion.",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         response: str = SchemaField(
             description="The response generated by the language model."
         )
         prompt: list = SchemaField(description="The prompt sent to the language model.")
+        error: str = SchemaField(description="Error message if the API call failed.")
 
     def __init__(self):
         super().__init__(
@@ -1308,7 +1304,7 @@ class SummaryStyle(Enum):
 
 
 class AITextSummarizerBlock(AIBlockBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         text: str = SchemaField(
             description="The text to summarize.",
             placeholder="Enter the text to summarize here...",
@@ -1348,9 +1344,10 @@ class AITextSummarizerBlock(AIBlockBase):
             description="Ollama host for local  models",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         summary: str = SchemaField(description="The final summary of the text.")
         prompt: list = SchemaField(description="The prompt sent to the language model.")
+        error: str = SchemaField(description="Error message if the API call failed.")
 
     def __init__(self):
         super().__init__(
@@ -1454,20 +1451,7 @@ class AITextSummarizerBlock(AIBlockBase):
             credentials=credentials,
         )
 
-        summary = llm_response["summary"]
-
-        # Validate that the LLM returned a string and not a list or other type
-        if not isinstance(summary, str):
-            from backend.util.truncate import truncate
-
-            truncated_summary = truncate(summary, 500)
-            raise ValueError(
-                f"LLM generation failed: Expected a string summary, but received {type(summary).__name__}. "
-                f"The language model incorrectly formatted its response. "
-                f"Received value: {json.dumps(truncated_summary)}"
-            )
-
-        return summary
+        return llm_response["summary"]
 
     async def _combine_summaries(
         self, summaries: list[str], input_data: Input, credentials: APIKeyCredentials
@@ -1489,20 +1473,7 @@ class AITextSummarizerBlock(AIBlockBase):
                 credentials=credentials,
             )
 
-            final_summary = llm_response["final_summary"]
-
-            # Validate that the LLM returned a string and not a list or other type
-            if not isinstance(final_summary, str):
-                from backend.util.truncate import truncate
-
-                truncated_final_summary = truncate(final_summary, 500)
-                raise ValueError(
-                    f"LLM generation failed: Expected a string final summary, but received {type(final_summary).__name__}. "
-                    f"The language model incorrectly formatted its response. "
-                    f"Received value: {json.dumps(truncated_final_summary)}"
-                )
-
-            return final_summary
+            return llm_response["final_summary"]
         else:
             # If combined summaries are still too long, recursively summarize
             block = AITextSummarizerBlock()
@@ -1520,7 +1491,7 @@ class AITextSummarizerBlock(AIBlockBase):
 
 
 class AIConversationBlock(AIBlockBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         prompt: str = SchemaField(
             description="The prompt to send to the language model.",
             placeholder="Enter your prompt here...",
@@ -1547,11 +1518,12 @@ class AIConversationBlock(AIBlockBase):
             description="Ollama host for local  models",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         response: str = SchemaField(
             description="The model's response to the conversation."
         )
         prompt: list = SchemaField(description="The prompt sent to the language model.")
+        error: str = SchemaField(description="Error message if the API call failed.")
 
     def __init__(self):
         super().__init__(
@@ -1618,7 +1590,7 @@ class AIConversationBlock(AIBlockBase):
 
 
 class AIListGeneratorBlock(AIBlockBase):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         focus: str | None = SchemaField(
             description="The focus of the list to generate.",
             placeholder="The top 5 most interesting news stories in the data.",
@@ -1655,12 +1627,15 @@ class AIListGeneratorBlock(AIBlockBase):
             description="Ollama host for local  models",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         generated_list: List[str] = SchemaField(description="The generated list.")
         list_item: str = SchemaField(
             description="Each individual item in the list.",
         )
         prompt: list = SchemaField(description="The prompt sent to the language model.")
+        error: str = SchemaField(
+            description="Error message if the list generation failed."
+        )
 
     def __init__(self):
         super().__init__(
@@ -1778,7 +1753,6 @@ class AIListGeneratorBlock(AIBlockBase):
             |```
 
             |Do not include any explanations or additional text, just respond with the list in the format specified above.
-            |Do not include code fences or any other formatting, just the raw list.
             """
         # If a focus is provided, add it to the prompt
         if input_data.focus:

autogpt_platform/backend/backend/blocks/maths.py

@@ -2,13 +2,7 @@ import operator
 from enum import Enum
 from typing import Any
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 
 
@@ -21,7 +15,7 @@ class Operation(Enum):
 
 
 class CalculatorBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         operation: Operation = SchemaField(
             description="Choose the math operation you want to perform",
             placeholder="Select an operation",
@@ -37,7 +31,7 @@ class CalculatorBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: float = SchemaField(description="The result of your calculation")
 
     def __init__(self):
@@ -91,13 +85,13 @@ class CalculatorBlock(Block):
 
 
 class CountItemsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         collection: Any = SchemaField(
             description="Enter the collection you want to count. This can be a list, dictionary, string, or any other iterable.",
             placeholder="For example: [1, 2, 3] or {'a': 1, 'b': 2} or 'hello'",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         count: int = SchemaField(description="The number of items in the collection")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/media.py

@@ -6,20 +6,14 @@ from moviepy.audio.io.AudioFileClip import AudioFileClip
 from moviepy.video.fx.Loop import Loop
 from moviepy.video.io.VideoFileClip import VideoFileClip
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.file import MediaFileType, get_exec_file_path, store_media_file
 
 
 class MediaDurationBlock(Block):
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         media_in: MediaFileType = SchemaField(
             description="Media input (URL, data URI, or local path)."
         )
@@ -28,10 +22,13 @@ class MediaDurationBlock(Block):
             default=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         duration: float = SchemaField(
             description="Duration of the media file (in seconds)."
         )
+        error: str = SchemaField(
+            description="Error message if something fails.", default=""
+        )
 
     def __init__(self):
         super().__init__(
@@ -73,7 +70,7 @@ class LoopVideoBlock(Block):
     Block for looping (repeating) a video clip until a given duration or number of loops.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         video_in: MediaFileType = SchemaField(
             description="The input video (can be a URL, data URI, or local path)."
         )
@@ -93,10 +90,13 @@ class LoopVideoBlock(Block):
             default="file_path",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_out: str = SchemaField(
             description="Looped video returned either as a relative path or a data URI."
         )
+        error: str = SchemaField(
+            description="Error message if something fails.", default=""
+        )
 
     def __init__(self):
         super().__init__(
@@ -166,7 +166,7 @@ class AddAudioToVideoBlock(Block):
     Optionally scale the volume of the new track.
     """
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         video_in: MediaFileType = SchemaField(
             description="Video input (URL, data URI, or local path)."
         )
@@ -182,10 +182,13 @@ class AddAudioToVideoBlock(Block):
             default="file_path",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         video_out: MediaFileType = SchemaField(
             description="Final video (with attached audio), as a path or data URI."
         )
+        error: str = SchemaField(
+            description="Error message if something fails.", default=""
+        )
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/medium.py

@@ -3,13 +3,7 @@ from typing import List, Literal
 
 from pydantic import SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     BlockSecret,
@@ -43,7 +37,7 @@ class PublishToMediumStatus(str, Enum):
 
 
 class PublishToMediumBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         author_id: BlockSecret = SecretField(
             key="medium_author_id",
             description="""The Medium AuthorID of the user. You can get this by calling the /me endpoint of the Medium API.\n\ncurl -H "Authorization: Bearer YOUR_ACCESS_TOKEN" https://api.medium.com/v1/me" the response will contain the authorId field.""",
@@ -90,7 +84,7 @@ class PublishToMediumBlock(Block):
             description="The Medium integration can be used with any API key with sufficient permissions for the blocks it is used on.",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post_id: str = SchemaField(description="The ID of the created Medium post")
         post_url: str = SchemaField(description="The URL of the created Medium post")
         published_at: int = SchemaField(

autogpt_platform/backend/backend/blocks/mem0.py

@@ -3,7 +3,7 @@ from typing import Any, Literal, Optional, Union
 from mem0 import MemoryClient
 from pydantic import BaseModel, SecretStr
 
-from backend.data.block import Block, BlockOutput, BlockSchemaInput, BlockSchemaOutput
+from backend.data.block import Block, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -55,7 +55,7 @@ class AddMemoryBlock(Block, Mem0Base):
 
     Always limited by user_id and optional graph_id and graph_exec_id"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.MEM0], Literal["api_key"]
         ] = CredentialsField(description="Mem0 API key credentials")
@@ -74,12 +74,13 @@ class AddMemoryBlock(Block, Mem0Base):
             description="Limit the memory to the agent", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         action: str = SchemaField(description="Action of the operation")
         memory: str = SchemaField(description="Memory created")
         results: list[dict[str, str]] = SchemaField(
             description="List of all results from the operation"
         )
+        error: str = SchemaField(description="Error message if operation fails")
 
     def __init__(self):
         super().__init__(
@@ -171,7 +172,7 @@ class AddMemoryBlock(Block, Mem0Base):
 class SearchMemoryBlock(Block, Mem0Base):
     """Block for searching memories in Mem0"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.MEM0], Literal["api_key"]
         ] = CredentialsField(description="Mem0 API key credentials")
@@ -200,8 +201,9 @@ class SearchMemoryBlock(Block, Mem0Base):
             description="Limit the memory to the agent", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         memories: Any = SchemaField(description="List of matching memories")
+        error: str = SchemaField(description="Error message if operation fails")
 
     def __init__(self):
         super().__init__(
@@ -264,7 +266,7 @@ class SearchMemoryBlock(Block, Mem0Base):
 class GetAllMemoriesBlock(Block, Mem0Base):
     """Block for retrieving all memories from Mem0"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.MEM0], Literal["api_key"]
         ] = CredentialsField(description="Mem0 API key credentials")
@@ -287,8 +289,9 @@ class GetAllMemoriesBlock(Block, Mem0Base):
             description="Limit the memory to the agent", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         memories: Any = SchemaField(description="List of memories")
+        error: str = SchemaField(description="Error message if operation fails")
 
     def __init__(self):
         super().__init__(
@@ -350,7 +353,7 @@ class GetAllMemoriesBlock(Block, Mem0Base):
 class GetLatestMemoryBlock(Block, Mem0Base):
     """Block for retrieving the latest memory from Mem0"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: CredentialsMetaInput[
             Literal[ProviderName.MEM0], Literal["api_key"]
         ] = CredentialsField(description="Mem0 API key credentials")
@@ -377,11 +380,12 @@ class GetLatestMemoryBlock(Block, Mem0Base):
             description="Limit the memory to the agent", default=True
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         memory: Optional[dict[str, Any]] = SchemaField(
             description="Latest memory if found"
         )
         found: bool = SchemaField(description="Whether a memory was found")
+        error: str = SchemaField(description="Error message if operation fails")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/notion/create_page.py

@@ -4,13 +4,7 @@ from typing import Any, Dict, List, Optional
 
 from pydantic import model_validator
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import NotionClient
@@ -26,7 +20,7 @@ from ._auth import (
 class NotionCreatePageBlock(Block):
     """Create a new page in Notion with content."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NotionCredentialsInput = NotionCredentialsField()
         parent_page_id: Optional[str] = SchemaField(
             description="Parent page ID to create the page under. Either this OR parent_database_id is required.",
@@ -64,9 +58,10 @@ class NotionCreatePageBlock(Block):
                 )
             return self
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         page_id: str = SchemaField(description="ID of the created page.")
         page_url: str = SchemaField(description="URL of the created page.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/notion/read_database.py

@@ -2,13 +2,7 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import NotionClient, parse_rich_text
@@ -24,7 +18,7 @@ from ._auth import (
 class NotionReadDatabaseBlock(Block):
     """Query a Notion database and retrieve entries with their properties."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NotionCredentialsInput = NotionCredentialsField()
         database_id: str = SchemaField(
             description="Notion database ID. Must be accessible by the connected integration.",
@@ -50,7 +44,7 @@ class NotionReadDatabaseBlock(Block):
             le=100,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         entries: List[Dict[str, Any]] = SchemaField(
             description="List of database entries with their properties."
         )
@@ -65,6 +59,7 @@ class NotionReadDatabaseBlock(Block):
         )
         count: int = SchemaField(description="Number of entries retrieved.")
         database_title: str = SchemaField(description="Title of the database.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/notion/read_page.py

@@ -1,12 +1,6 @@
 from __future__ import annotations
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import NotionClient
@@ -22,14 +16,15 @@ from ._auth import (
 class NotionReadPageBlock(Block):
     """Read a Notion page by ID and return its raw JSON."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NotionCredentialsInput = NotionCredentialsField()
         page_id: str = SchemaField(
             description="Notion page ID. Must be accessible by the connected integration. You can get this from the page URL notion.so/A-Page-586edd711467478da59fe3ce29a1ffab would be 586edd711467478da59fe35e29a1ffab",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         page: dict = SchemaField(description="Raw Notion page JSON.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/notion/read_page_markdown.py

@@ -1,12 +1,6 @@
 from __future__ import annotations
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import NotionClient, blocks_to_markdown, extract_page_title
@@ -22,7 +16,7 @@ from ._auth import (
 class NotionReadPageMarkdownBlock(Block):
     """Read a Notion page and convert it to clean Markdown format."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NotionCredentialsInput = NotionCredentialsField()
         page_id: str = SchemaField(
             description="Notion page ID. Must be accessible by the connected integration. You can get this from the page URL notion.so/A-Page-586edd711467478da59fe35e29a1ffab would be 586edd711467478da59fe35e29a1ffab",
@@ -32,9 +26,10 @@ class NotionReadPageMarkdownBlock(Block):
             default=True,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         markdown: str = SchemaField(description="Page content in Markdown format.")
         title: str = SchemaField(description="Page title.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/notion/search.py

@@ -4,13 +4,7 @@ from typing import List, Optional
 
 from pydantic import BaseModel
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import OAuth2Credentials, SchemaField
 
 from ._api import NotionClient, extract_page_title, parse_rich_text
@@ -41,7 +35,7 @@ class NotionSearchResult(BaseModel):
 class NotionSearchBlock(Block):
     """Search across your Notion workspace for pages and databases."""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NotionCredentialsInput = NotionCredentialsField()
         query: str = SchemaField(
             description="Search query text. Leave empty to get all accessible pages/databases.",
@@ -55,7 +49,7 @@ class NotionSearchBlock(Block):
             description="Maximum number of results to return", default=20, ge=1, le=100
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: List[NotionSearchResult] = SchemaField(
             description="List of search results with title, type, URL, and metadata."
         )
@@ -66,6 +60,7 @@ class NotionSearchBlock(Block):
             description="List of IDs from search results for batch operations."
         )
         count: int = SchemaField(description="Number of results found.")
+        error: str = SchemaField(description="Error message if the operation failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/nvidia/deepfake.py

@@ -3,20 +3,14 @@ from backend.blocks.nvidia._auth import (
     NvidiaCredentialsField,
     NvidiaCredentialsInput,
 )
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.request import Requests
 from backend.util.type import MediaFileType
 
 
 class NvidiaDeepfakeDetectBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: NvidiaCredentialsInput = NvidiaCredentialsField()
         image_base64: MediaFileType = SchemaField(
             description="Image to analyze for deepfakes",
@@ -26,7 +20,7 @@ class NvidiaDeepfakeDetectBlock(Block):
             default=False,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         status: str = SchemaField(
             description="Detection status (SUCCESS, ERROR, CONTENT_FILTERED)",
         )

autogpt_platform/backend/backend/blocks/perplexity.py

@@ -6,13 +6,7 @@ from typing import Any, Literal
 import openai
 from pydantic import SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -60,7 +54,7 @@ def PerplexityCredentialsField() -> PerplexityCredentials:
 
 
 class PerplexityBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         prompt: str = SchemaField(
             description="The query to send to the Perplexity model.",
             placeholder="Enter your query here...",
@@ -84,13 +78,14 @@ class PerplexityBlock(Block):
             description="The maximum number of tokens to generate.",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         response: str = SchemaField(
             description="The response from the Perplexity model."
         )
         annotations: list[dict[str, Any]] = SchemaField(
             description="List of URL citations and annotations from the response."
         )
+        error: str = SchemaField(description="Error message if the API call failed.")
 
     def __init__(self):
         super().__init__(

autogpt_platform/backend/backend/blocks/persistence.py

@@ -1,13 +1,7 @@
 import logging
 from typing import Any, Literal
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import SchemaField
 from backend.util.clients import get_database_manager_async_client
 
@@ -28,7 +22,7 @@ def get_storage_key(key: str, scope: StorageScope, graph_id: str) -> str:
 class PersistInformationBlock(Block):
     """Block for persisting key-value data for the current user with configurable scope"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         key: str = SchemaField(description="Key to store the information under")
         value: Any = SchemaField(description="Value to store")
         scope: StorageScope = SchemaField(
@@ -36,7 +30,7 @@ class PersistInformationBlock(Block):
             default="within_agent",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         value: Any = SchemaField(description="Value that was stored")
 
     def __init__(self):
@@ -96,7 +90,7 @@ class PersistInformationBlock(Block):
 class RetrieveInformationBlock(Block):
     """Block for retrieving key-value data for the current user with configurable scope"""
 
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         key: str = SchemaField(description="Key to retrieve the information for")
         scope: StorageScope = SchemaField(
             description="Scope of persistence: within_agent (shared across all runs of this agent) or across_agents (shared across all agents for this user)",
@@ -106,7 +100,7 @@ class RetrieveInformationBlock(Block):
             description="Default value to return if key is not found", default=None
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         value: Any = SchemaField(description="Retrieved value or default value")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/pinecone.py

@@ -3,13 +3,7 @@ from typing import Any, Literal
 
 from pinecone import Pinecone, ServerlessSpec
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     APIKeyCredentials,
     CredentialsField,
@@ -33,7 +27,7 @@ def PineconeCredentialsField() -> PineconeCredentialsInput:
 
 
 class PineconeInitBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: PineconeCredentialsInput = PineconeCredentialsField()
         index_name: str = SchemaField(description="Name of the Pinecone index")
         dimension: int = SchemaField(
@@ -49,7 +43,7 @@ class PineconeInitBlock(Block):
             description="Region for serverless", default="us-east-1"
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         index: str = SchemaField(description="Name of the initialized Pinecone index")
         message: str = SchemaField(description="Status message")
 
@@ -89,7 +83,7 @@ class PineconeInitBlock(Block):
 
 
 class PineconeQueryBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: PineconeCredentialsInput = PineconeCredentialsField()
         query_vector: list = SchemaField(description="Query vector")
         namespace: str = SchemaField(
@@ -108,7 +102,7 @@ class PineconeQueryBlock(Block):
         host: str = SchemaField(description="Host for pinecone", default="")
         idx_name: str = SchemaField(description="Index name for pinecone")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         results: Any = SchemaField(description="Query results from Pinecone")
         combined_results: Any = SchemaField(
             description="Combined results from Pinecone"
@@ -172,7 +166,7 @@ class PineconeQueryBlock(Block):
 
 
 class PineconeInsertBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: PineconeCredentialsInput = PineconeCredentialsField()
         index: str = SchemaField(description="Initialized Pinecone index")
         chunks: list = SchemaField(description="List of text chunks to ingest")
@@ -187,7 +181,7 @@ class PineconeInsertBlock(Block):
             default_factory=dict,
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         upsert_response: str = SchemaField(
             description="Response from Pinecone upsert operation"
         )

autogpt_platform/backend/backend/blocks/reddit.py

@@ -4,13 +4,7 @@ from typing import Iterator, Literal
 import praw
 from pydantic import BaseModel, SecretStr
 
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import (
     CredentialsField,
     CredentialsMetaInput,
@@ -82,7 +76,7 @@ def get_praw(creds: RedditCredentials) -> praw.Reddit:
 
 
 class GetRedditPostsBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         subreddit: str = SchemaField(
             description="Subreddit name, excluding the /r/ prefix",
             default="writingprompts",
@@ -100,7 +94,7 @@ class GetRedditPostsBlock(Block):
             description="Number of posts to fetch", default=10
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         post: RedditPost = SchemaField(description="Reddit post")
         posts: list[RedditPost] = SchemaField(description="List of all Reddit posts")
 
@@ -200,11 +194,11 @@ class GetRedditPostsBlock(Block):
 
 
 class PostRedditCommentBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: RedditCredentialsInput = RedditCredentialsField()
         data: RedditComment = SchemaField(description="Reddit comment")
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         comment_id: str = SchemaField(description="Posted comment ID")
 
     def __init__(self):

autogpt_platform/backend/backend/blocks/replicate/flux_advanced.py

@@ -10,13 +10,7 @@ from backend.blocks.replicate._auth import (
     ReplicateCredentialsInput,
 )
 from backend.blocks.replicate._helper import ReplicateOutputs, extract_result
-from backend.data.block import (
-    Block,
-    BlockCategory,
-    BlockOutput,
-    BlockSchemaInput,
-    BlockSchemaOutput,
-)
+from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
 from backend.data.model import APIKeyCredentials, CredentialsField, SchemaField
 
 
@@ -44,7 +38,7 @@ class ImageType(str, Enum):
 
 
 class ReplicateFluxAdvancedModelBlock(Block):
-    class Input(BlockSchemaInput):
+    class Input(BlockSchema):
         credentials: ReplicateCredentialsInput = CredentialsField(
             description="The Replicate integration can be used with "
             "any API key with sufficient permissions for the blocks it is used on.",
@@ -111,8 +105,9 @@ class ReplicateFluxAdvancedModelBlock(Block):
             title="Safety Tolerance",
         )
 
-    class Output(BlockSchemaOutput):
+    class Output(BlockSchema):
         result: str = SchemaField(description="Generated output")
+        error: str = SchemaField(description="Error message if the model run failed")
 
     def __init__(self):
         super().__init__(