@deepnote/blocks 4.3.0 → 4.5.0

package/dist/index.js

@@ -59,6 +59,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
 /**
@@ -76,6 +87,7 @@ const INPUT_BLOCK_TYPES = new Set([
 	"input-file"
 ]);
 const executableBlockTypes = new Set([
+	"agent",
 	"code",
 	"sql",
 	"notebook-function",
@@ -314,6 +326,21 @@ const bigNumberBlockSchema = z.object({
 		deepnote_big_number_comparison_format: z.string().optional()
 	}).default({})
 });
+const mcpServerSchema = z.object({
+	name: z.string(),
+	command: z.string(),
+	args: z.array(z.string()).optional(),
+	env: z.record(z.string()).optional()
+});
+const agentBlockSchema = z.object({
+	...executableBlockFields,
+	type: z.literal("agent"),
+	content: z.string().optional(),
+	metadata: executableBlockMetadataSchema.extend({
+		deepnote_agent_model: z.string().default("auto"),
+		deepnote_mcp_servers: z.array(mcpServerSchema).optional()
+	}).default({})
+});
 const inputTextBlockSchema = z.object({
 	...executableBlockFields,
 	type: z.literal("input-text"),
@@ -409,6 +436,7 @@ const deepnoteBlockSchema = z.discriminatedUnion("type", [
 	textCellBulletBlockSchema,
 	textCellTodoBlockSchema,
 	textCellCalloutBlockSchema,
+	agentBlockSchema,
 	codeBlockSchema,
 	sqlBlockSchema,
 	notebookFunctionBlockSchema,
@@ -495,6 +523,7 @@ const deepnoteFileSchema = z.object({
 				customImage: z.string().optional(),
 				pythonVersion: z.string().optional()
 			}).optional(),
+			mcpServers: z.array(mcpServerSchema).optional(),
 			requirements: z.array(z.string()).optional(),
 			sqlCacheMaxAge: z.number().optional()
 		}).optional()
@@ -743,6 +772,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 = 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) {
@@ -1162,6 +1230,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);
@@ -1180,4 +1249,4 @@ function createPythonCode(block, executionContext) {
 }
 
 //#endregion
-export { DeepnoteError, EncodingError, INPUT_BLOCK_TYPES, InvalidValueError, ParseError, ProhibitedYamlFeatureError, SchemaValidationError, UnsupportedBlockTypeError, YamlParseError, convertToEnvironmentVariableName, createMarkdown, createPythonCode, decodeUtf8NoBom, deepnoteBlockSchema, deepnoteFileSchema, deepnoteSnapshotSchema, deserializeDeepnoteFile, environmentSchema, executionErrorSchema, executionSchema, executionSummarySchema, generateSortingKey, getSqlEnvVarName, isExecutableBlock, isExecutableBlockType, parseYaml, serializeDeepnoteFile, serializeDeepnoteSnapshot, stripMarkdown };
+export { DeepnoteError, EncodingError, INPUT_BLOCK_TYPES, InvalidValueError, ParseError, ProhibitedYamlFeatureError, SchemaValidationError, UnsupportedBlockTypeError, YamlParseError, convertToEnvironmentVariableName, createMarkdown, createPythonCode, decodeUtf8NoBom, deepnoteBlockSchema, deepnoteFileSchema, deepnoteSnapshotSchema, deserializeDeepnoteFile, environmentSchema, executionErrorSchema, executionSchema, executionSummarySchema, extractOutputText, extractOutputsText, generateSortingKey, getSqlEnvVarName, isAgentBlock, isExecutableBlock, isExecutableBlockType, mcpServerSchema, parseYaml, serializeDeepnoteFile, serializeDeepnoteSnapshot, stripMarkdown };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deepnote/blocks",
-  "version": "4.3.0",
+  "version": "4.5.0",
   "description": "",
   "keywords": [],
   "repository": {