diff --git a/libs/sdk-js/src/schema.ts b/libs/sdk-js/src/schema.ts
index ac68fa844..69cb069f7 100644
--- a/libs/sdk-js/src/schema.ts
+++ b/libs/sdk-js/src/schema.ts
@@ -39,11 +39,17 @@ export interface GraphSchema {
   graph_id: string;
 
   /**
-   * The schema for the graph state.
+   * The schema for the input state.
    * Missing if unable to generate JSON schema from graph.
    */
   input_schema?: JSONSchema7;
 
+  /**
+   * The schema for the output state.
+   * Missing if unable to generate JSON schema from graph.
+   */
+  output_schema?: JSONSchema7;
+
   /**
    * The schema for the graph state.
    * Missing if unable to generate JSON schema from graph.