diff --git a/autogpt_platform/backend/backend/data/graph.py b/autogpt_platform/backend/backend/data/graph.py
index 55a98812c5eb..d97d246ea2a5 100644
--- a/autogpt_platform/backend/backend/data/graph.py
+++ b/autogpt_platform/backend/backend/data/graph.py
@@ -7,7 +7,6 @@
 
 from prisma.models import AgentGraph, AgentGraphExecution, AgentNode, AgentNodeLink
 from prisma.types import AgentGraphWhereInput
-from pydantic import BaseModel
 from pydantic.fields import computed_field
 
 from backend.blocks.agent import AgentExecutorBlock
@@ -21,19 +20,6 @@
 logger = logging.getLogger(__name__)
 
 
-class InputSchemaItem(BaseModel):
-    node_id: str
-    title: str | None = None
-    description: str | None = None
-    default: Any | None = None
-
-
-class OutputSchemaItem(BaseModel):
-    node_id: str
-    title: str | None = None
-    description: str | None = None
-
-
 class Link(BaseDbModel):
     source_id: str
     sink_id: str