diff --git a/cli/src/commands/schema/types.ts b/cli/src/commands/schema/types.ts
new file mode 100644
index 000000000..aae59a8b7
--- /dev/null
+++ b/cli/src/commands/schema/types.ts
@@ -0,0 +1,99 @@
+import { Schemas } from '@xata.io/client';
+
+export type BranchSchemaFormatted =
+  | {
+      schema: {
+        tables: {
+          name: string;
+          uniqueConstraints: Schemas.BranchSchema['tables'][number]['uniqueConstraints'];
+          checkConstraints: Schemas.BranchSchema['tables'][number]['checkConstraints'];
+          foreignKeys: Schemas.BranchSchema['tables'][number]['foreignKeys'];
+          columns: {
+            name: string;
+            type: string;
+            unique: boolean;
+            notNull: boolean;
+            defaultValue: any;
+            comment: string;
+          }[];
+        }[];
+      };
+    }
+  | undefined;
+
+export type ColumnData = {
+  name: string;
+  type: string;
+  unique: boolean;
+  nullable: boolean;
+  defaultValue?: string;
+  vector?: {
+    dimension: number;
+  };
+  originalName: string;
+  tableName: string;
+  link?: {
+    table: string;
+  };
+  file?: {
+    defaultPublicAccess: boolean;
+  };
+  'file[]'?: {
+    defaultPublicAccess: boolean;
+  };
+};
+
+export type AddTablePayload = {
+  type: 'add-table';
+  table: {
+    name: string;
+  };
+};
+
+export type EditTablePayload = {
+  type: 'edit-table';
+  table: {
+    name: string;
+    newName: string;
+  };
+};
+
+export type DeleteTablePayload = {
+  name: string;
+};
+
+export type AddColumnPayload = {
+  type: 'add-column';
+  tableName: string;
+  column: ColumnData;
+};
+
+export type EditColumnPayload = {
+  type: 'edit-column';
+  column: ColumnData;
+};
+
+export type DeleteColumnPayload = { [tableName: string]: string[] };
+
+export type FormatPayload = {
+  type: 'space' | 'migrate' | 'schema';
+};
+
+export type SelectChoice = {
+  name: FormatPayload | AddTablePayload | EditTablePayload | AddColumnPayload | EditColumnPayload;
+  message: string;
+  role?: string;
+  choices?: SelectChoice[];
+  disabled?: boolean;
+  hint?: string;
+};
+
+export type ValidationState = {
+  values: { name: string };
+  items: { name: string; input: string }[];
+  fields: { name: string; initial: string }[];
+};
+
+export type ColumnAdditions = { [tableName: string]: { [columnName: string]: AddColumnPayload['column'] } };
+
+export type ColumnEdits = { [tableName: string]: { [columnName: string]: AddColumnPayload['column'] } };