@deepnote/blocks 4.4.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
 /**
@@ -326,15 +337,6 @@ const buttonBlockSchema = zod.z.object({
 		deepnote_variable_name: zod.z.string().optional()
 	}).default({})
 });
-const agentBlockSchema = zod.z.object({
-	...executableBlockFields,
-	type: zod.z.literal("agent"),
-	content: zod.z.string().optional(),
-	metadata: executableBlockMetadataSchema.extend({
-		deepnote_variable_name: zod.z.string().optional(),
-		deepnote_agent_model: zod.z.string().optional()
-	}).default({})
-});
 const bigNumberBlockSchema = zod.z.object({
 	...executableBlockFields,
 	type: zod.z.literal("big-number"),
@@ -350,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"),
@@ -532,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()
@@ -781,14 +799,42 @@ function stripMarkdown(block) {
 }
 
 //#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")}`;
+//#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;
 }
-function isAgentBlock(block) {
-	return block.type === "agent";
+/**
+* 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
@@ -1250,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;