@deepnote/blocks 4.3.0 → 4.5.0

package/dist/index.cjs

@@ -85,6 +85,17 @@ var InvalidValueError = class extends DeepnoteError {
 	}
 };
 
+//#endregion
+//#region src/blocks/agent-blocks.ts
+function createPythonCodeForAgentBlock(block) {
+	const prompt = block.content ?? "";
+	if (!prompt.trim()) return "# [agent block] (empty system prompt)";
+	return `# [agent block] System prompt:\n${prompt.split("\n").map((line) => `# ${line}`).join("\n")}`;
+}
+function isAgentBlock(block) {
+	return block.type === "agent";
+}
+
 //#endregion
 //#region src/blocks/executable-blocks.ts
 /**
@@ -102,6 +113,7 @@ const INPUT_BLOCK_TYPES = new Set([
 	"input-file"
 ]);
 const executableBlockTypes = new Set([
+	"agent",
 	"code",
 	"sql",
 	"notebook-function",
@@ -340,6 +352,21 @@ const bigNumberBlockSchema = zod.z.object({
 		deepnote_big_number_comparison_format: zod.z.string().optional()
 	}).default({})
 });
+const mcpServerSchema = zod.z.object({
+	name: zod.z.string(),
+	command: zod.z.string(),
+	args: zod.z.array(zod.z.string()).optional(),
+	env: zod.z.record(zod.z.string()).optional()
+});
+const agentBlockSchema = zod.z.object({
+	...executableBlockFields,
+	type: zod.z.literal("agent"),
+	content: zod.z.string().optional(),
+	metadata: executableBlockMetadataSchema.extend({
+		deepnote_agent_model: zod.z.string().default("auto"),
+		deepnote_mcp_servers: zod.z.array(mcpServerSchema).optional()
+	}).default({})
+});
 const inputTextBlockSchema = zod.z.object({
 	...executableBlockFields,
 	type: zod.z.literal("input-text"),
@@ -435,6 +462,7 @@ const deepnoteBlockSchema = zod.z.discriminatedUnion("type", [
 	textCellBulletBlockSchema,
 	textCellTodoBlockSchema,
 	textCellCalloutBlockSchema,
+	agentBlockSchema,
 	codeBlockSchema,
 	sqlBlockSchema,
 	notebookFunctionBlockSchema,
@@ -521,6 +549,7 @@ const deepnoteFileSchema = zod.z.object({
 				customImage: zod.z.string().optional(),
 				pythonVersion: zod.z.string().optional()
 			}).optional(),
+			mcpServers: zod.z.array(mcpServerSchema).optional(),
 			requirements: zod.z.array(zod.z.string()).optional(),
 			sqlCacheMaxAge: zod.z.number().optional()
 		}).optional()
@@ -769,6 +798,45 @@ function stripMarkdown(block) {
 	throw new UnsupportedBlockTypeError(`Stripping markdown from block type ${block.type} is not supported yet.`);
 }
 
+//#endregion
+//#region src/output-text.ts
+const unknownObjectSchema = zod.z.object({}).passthrough();
+/**
+* Extract a human-readable text string from a single Jupyter-style output object.
+* Returns null if the output type is unrecognized or has no textual representation.
+*/
+function extractOutputText(output, options) {
+	const outResult = unknownObjectSchema.safeParse(output);
+	if (!outResult.success) return null;
+	const out = outResult.data;
+	if (out.output_type === "stream" && typeof out.text === "string") return out.text;
+	if (out.output_type === "execute_result" || out.output_type === "display_data") {
+		const dataResult = unknownObjectSchema.safeParse(out.data);
+		const data = dataResult.success ? dataResult.data : void 0;
+		if (data?.["text/plain"]) return String(data["text/plain"]);
+		if (data?.["text/html"]) return "[HTML output]";
+		if (data?.["image/png"] || data?.["image/jpeg"]) return "[Image output]";
+	}
+	if (out.output_type === "error") {
+		let text = `Error: ${String(out.ename || "Error")}: ${String(out.evalue || "")}`;
+		if (options?.includeTraceback && Array.isArray(out.traceback)) text += "\n" + out.traceback.map((line) => line.replace(/\x1b\[[0-9;]*m/g, "")).join("\n");
+		return text;
+	}
+	return null;
+}
+/**
+* Extract human-readable text from an array of Jupyter-style output objects.
+* Joins non-null results with newlines.
+*/
+function extractOutputsText(outputs, options) {
+	const texts = [];
+	for (const output of outputs) {
+		const text = extractOutputText(output, options);
+		if (text != null) texts.push(text);
+	}
+	return texts.join("\n");
+}
+
 //#endregion
 //#region src/blocks/python-utils.ts
 function escapePythonString(value) {
@@ -1188,6 +1256,7 @@ function isVisualizationBlock(block) {
 //#endregion
 //#region src/python-code.ts
 function createPythonCode(block, executionContext) {
+	if (isAgentBlock(block)) return createPythonCodeForAgentBlock(block);
 	if (isCodeBlock(block)) return createPythonCodeForCodeBlock(block);
 	if (isSqlBlock(block)) return createPythonCodeForSqlBlock(block);
 	if (isInputTextBlock(block)) return createPythonCodeForInputTextBlock(block);
@@ -1227,10 +1296,14 @@ exports.environmentSchema = environmentSchema;
 exports.executionErrorSchema = executionErrorSchema;
 exports.executionSchema = executionSchema;
 exports.executionSummarySchema = executionSummarySchema;
+exports.extractOutputText = extractOutputText;
+exports.extractOutputsText = extractOutputsText;
 exports.generateSortingKey = generateSortingKey;
 exports.getSqlEnvVarName = getSqlEnvVarName;
+exports.isAgentBlock = isAgentBlock;
 exports.isExecutableBlock = isExecutableBlock;
 exports.isExecutableBlockType = isExecutableBlockType;
+exports.mcpServerSchema = mcpServerSchema;
 exports.parseYaml = parseYaml;
 exports.serializeDeepnoteFile = serializeDeepnoteFile;
 exports.serializeDeepnoteSnapshot = serializeDeepnoteSnapshot;