@deepnote/blocks 4.2.0 → 4.4.0

package/dist/index.cjs

@@ -102,6 +102,7 @@ const INPUT_BLOCK_TYPES = new Set([
 	"input-file"
 ]);
 const executableBlockTypes = new Set([
+	"agent",
 	"code",
 	"sql",
 	"notebook-function",
@@ -325,6 +326,15 @@ 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"),
@@ -435,6 +445,7 @@ const deepnoteBlockSchema = zod.z.discriminatedUnion("type", [
 	textCellBulletBlockSchema,
 	textCellTodoBlockSchema,
 	textCellCalloutBlockSchema,
+	agentBlockSchema,
 	codeBlockSchema,
 	sqlBlockSchema,
 	notebookFunctionBlockSchema,
@@ -655,17 +666,37 @@ const yamlOptions = {
 	defaultStringType: "PLAIN",
 	defaultKeyType: "PLAIN"
 };
+function generateSortingKey(index) {
+	return String(index).padStart(6, "0");
+}
+function normalizeSortingKeys(data) {
+	return {
+		...data,
+		project: {
+			...data.project,
+			notebooks: data.project.notebooks.map((notebook) => ({
+				...notebook,
+				blocks: notebook.blocks.map((block, index) => ({
+					...block,
+					sortingKey: generateSortingKey(index)
+				}))
+			}))
+		}
+	};
+}
 /**
 * Serialize a DeepnoteFile to a YAML string.
 */
 function serializeDeepnoteFile(file) {
-	return (0, yaml.stringify)(deepnoteFileSchema.parse(file), yamlOptions);
+	const withNormalizedSortingKeys = normalizeSortingKeys(file);
+	return (0, yaml.stringify)(deepnoteFileSchema.parse(withNormalizedSortingKeys), yamlOptions);
 }
 /**
 * Serialize a DeepnoteSnapshot to a YAML string.
 */
 function serializeDeepnoteSnapshot(snapshot) {
-	return (0, yaml.stringify)(deepnoteSnapshotSchema.parse(snapshot), yamlOptions);
+	const withNormalizedSortingKeys = normalizeSortingKeys(snapshot);
+	return (0, yaml.stringify)(deepnoteSnapshotSchema.parse(withNormalizedSortingKeys), yamlOptions);
 }
 
 //#endregion
@@ -749,6 +780,17 @@ function stripMarkdown(block) {
 	throw new UnsupportedBlockTypeError(`Stripping markdown from block type ${block.type} is not supported yet.`);
 }
 
+//#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/python-utils.ts
 function escapePythonString(value) {
@@ -1168,6 +1210,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);
@@ -1207,6 +1250,7 @@ exports.environmentSchema = environmentSchema;
 exports.executionErrorSchema = executionErrorSchema;
 exports.executionSchema = executionSchema;
 exports.executionSummarySchema = executionSummarySchema;
+exports.generateSortingKey = generateSortingKey;
 exports.getSqlEnvVarName = getSqlEnvVarName;
 exports.isExecutableBlock = isExecutableBlock;
 exports.isExecutableBlockType = isExecutableBlockType;