scripts[patch]: Add notebook check for unexpected rebuild error (lang…

…chain-ai#6602)

libs/langchain-scripts/bin/validate_notebook.js

@@ -1 +1 @@
-import "../dist/validate_notebook.js";
+import "../dist/notebooks/index.js";

libs/langchain-scripts/src/validate_notebook.ts → libs/langchain-scripts/src/notebooks/check_notebook_type_errors.ts

@@ -69,7 +69,7 @@ if (!pathname) {
   throw new Error("No pathname provided.");
 }
 
-const run = async () => {
+export async function checkNotebookTypeErrors() {
   if (!pathname.endsWith(".ipynb")) {
     throw new Error("Only .ipynb files are supported.");
   }
@@ -130,10 +130,4 @@ const run = async () => {
       // Do nothing
     }
   }
-};
-
-try {
-  void run();
-} catch {
-  process.exit(1);
 }

libs/langchain-scripts/src/notebooks/check_unexpected_rebuild_timer.ts

@@ -0,0 +1,26 @@
+import fs from "fs";
+
+const [pathname] = process.argv.slice(2);
+
+if (!pathname) {
+  throw new Error("No pathname provided.");
+}
+
+/**
+ * tslab will sometimes throw an error inside the output cells of a notebook
+ * if the notebook is being rebuilt. This function checks for that error,
+ * because we do not want to commit that to our docs.
+ */
+export async function checkUnexpectedRebuildError() {
+  if (!pathname.endsWith(".ipynb")) {
+    throw new Error("Only .ipynb files are supported.");
+  }
+  const notebookContents = await fs.promises.readFile(pathname, "utf-8");
+  if (
+    notebookContents.includes(
+      "UncaughtException: Error: Unexpected pending rebuildTimer"
+    )
+  ) {
+    throw new Error(`Found unexpected pending rebuildTimer in ${pathname}`);
+  }
+}

libs/langchain-scripts/src/notebooks/index.ts

@@ -0,0 +1,12 @@
+import { checkNotebookTypeErrors } from "./check_notebook_type_errors.js";
+import { checkUnexpectedRebuildError } from "./check_unexpected_rebuild_timer.js";
+
+async function main() {
+  await Promise.all([checkNotebookTypeErrors(), checkUnexpectedRebuildError()]);
+}
+
+try {
+  void main();
+} catch {
+  process.exit(1);
+}