@deepnote/blocks 4.2.0 → 4.4.0

package/dist/index.js

@@ -76,6 +76,7 @@ const INPUT_BLOCK_TYPES = new Set([
 	"input-file"
 ]);
 const executableBlockTypes = new Set([
+	"agent",
 	"code",
 	"sql",
 	"notebook-function",
@@ -299,6 +300,15 @@ const buttonBlockSchema = z.object({
 		deepnote_variable_name: z.string().optional()
 	}).default({})
 });
+const agentBlockSchema = z.object({
+	...executableBlockFields,
+	type: z.literal("agent"),
+	content: z.string().optional(),
+	metadata: executableBlockMetadataSchema.extend({
+		deepnote_variable_name: z.string().optional(),
+		deepnote_agent_model: z.string().optional()
+	}).default({})
+});
 const bigNumberBlockSchema = z.object({
 	...executableBlockFields,
 	type: z.literal("big-number"),
@@ -409,6 +419,7 @@ const deepnoteBlockSchema = z.discriminatedUnion("type", [
 	textCellBulletBlockSchema,
 	textCellTodoBlockSchema,
 	textCellCalloutBlockSchema,
+	agentBlockSchema,
 	codeBlockSchema,
 	sqlBlockSchema,
 	notebookFunctionBlockSchema,
@@ -629,17 +640,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 stringify(deepnoteFileSchema.parse(file), yamlOptions);
+	const withNormalizedSortingKeys = normalizeSortingKeys(file);
+	return stringify(deepnoteFileSchema.parse(withNormalizedSortingKeys), yamlOptions);
 }
 /**
 * Serialize a DeepnoteSnapshot to a YAML string.
 */
 function serializeDeepnoteSnapshot(snapshot) {
-	return stringify(deepnoteSnapshotSchema.parse(snapshot), yamlOptions);
+	const withNormalizedSortingKeys = normalizeSortingKeys(snapshot);
+	return stringify(deepnoteSnapshotSchema.parse(withNormalizedSortingKeys), yamlOptions);
 }
 
 //#endregion
@@ -723,6 +754,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) {
@@ -1142,6 +1184,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);
@@ -1160,4 +1203,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, 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, generateSortingKey, getSqlEnvVarName, isExecutableBlock, isExecutableBlockType, parseYaml, serializeDeepnoteFile, serializeDeepnoteSnapshot, stripMarkdown };

package/package.json

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