Notebook support

.devcontainer/devcontainer.json

@@ -19,6 +19,7 @@
     },
     "features": {
         "ghcr.io/devcontainers/features/docker-in-docker:2": {},
-        "ghcr.io/devcontainers/features/azure-cli:1.2.5": {}
+        "ghcr.io/devcontainers/features/azure-cli:1.2.5": {},
+        "ghcr.io/devcontainers/features/python:1.6.1": {}
     }
 }

demo/genaisrc/poem.genaibook

@@ -0,0 +1,23 @@
+{
+  "cells": [
+    {
+      "type": "code",
+      "source": "$`Write a 1 sentence poem`",
+      "outputs": [
+        {
+          "items": [
+            {
+              "mime": "text/markdown",
+              "data": "496e20746865207175696574206d6561646f772c207768697370657273206f66207468652077696e642064616e636520776974682074686520646169736965732e"
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "type": "code",
+      "source": "def(\"FILE\", env.vars.output.text)\n$`Summarize FILE`\n",
+      "outputs": []
+    }
+  ]
+}

packages/core/src/ast.ts

@@ -78,10 +78,6 @@ export class Fragment {
         Object.assign(this, init)
         if (!this.fullId) this.fullId = this.id
     }
-
-    get project() {
-        return this.file.project
-    }
 }
 
 export function templateGroup(template: PromptScript) {

packages/core/src/constants.ts

@@ -163,6 +163,8 @@ export const XLSX_MIME_TYPE =
 export const JSON_MIME_TYPE = "application/json"
 export const JSON_SCHEMA_MIME_TYPE = "application/schema+json"
 export const JAVASCRIPT_MIME_TYPE = "application/javascript"
+export const MARKDOWN_MIME_TYPE = "text/markdown"
+export const YAML_MIME_TYPE = "application/yaml"
 
 export const JSON_META_SCHEMA_URI =
     "https://json-schema.org/draft/2020-12/schema"

packages/core/src/markdown.ts

@@ -18,3 +18,11 @@ export function fenceMD(t: string, contentType?: string) {
 export function link(text: string, href: string) {
     return href ? `[${text}](${href})` : text
 }
+
+export function details(summary: string, body: string) {
+    return `\n<details><summary>${summary}</summary>
+
+${body}
+
+</summary></details>\n`
+}

packages/sample/src/jupytest.ipynb

@@ -0,0 +1,38 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "",
+     "evalue": "",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[1;31mRunning cells with 'Python 3.11.2' requires the ipykernel package.\n",
+      "\u001b[1;31mRun the following command to install 'ipykernel' into the Python environment. \n",
+      "\u001b[1;31mCommand: '/bin/python3 -m pip install ipykernel -U --user --force-reinstall'"
+     ]
+    }
+   ],
+   "source": [
+    "x = 1\n",
+    "\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "name": "python",
+   "version": "3.11.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

packages/sample/src/node.nnb

@@ -0,0 +1,41 @@
+{
+    "cells": [
+        {
+            "language": "typescript",
+            "source": [
+                "var x = 2\ny =1"
+            ],
+            "outputs": [
+                {
+                    "items": [
+                        {
+                            "mime": "text/plain",
+                            "value": [
+                                "\u001b[33m1\u001b[39m"
+                            ]
+                        }
+                    ]
+                }
+            ]
+        },
+        {
+            "language": "typescript",
+            "source": [
+                "// <reference path=\"../genaisrc/genaiscript.d.ts\" />\nconsole.log({ x,y})\n"
+            ],
+            "outputs": [
+                {
+                    "items": [
+                        {
+                            "mime": "application/vnd.code.notebook.stdout",
+                            "value": [
+                                "{ x: 2, y: 1 }",
+                                ""
+                            ]
+                        }
+                    ]
+                }
+            ]
+        }
+    ]
+}

packages/vscode/package.json

@@ -41,7 +41,8 @@
     },
     "activationEvents": [
         "workspaceContains:**/*.gpspec.md",
-        "workspaceContains:**/*.genai.js"
+        "workspaceContains:**/*.genai.js",
+        "onNotebook:genaiscript"
     ],
     "main": "./built/extension.js",
     "bugs": {
@@ -56,6 +57,17 @@
         "./icon-dark.svg"
     ],
     "contributes": {
+        "notebooks": [
+            {
+                "type": "genaiscript",
+                "displayName": "GenAIScript",
+                "selector": [
+                    {
+                        "filenamePattern": "*.genaibook"
+                    }
+                ]
+            }
+        ],
         "markdown.previewStyles": [
             "./markdown.css"
         ],
@@ -213,6 +225,11 @@
             }
         ],
         "commands": [
+            {
+                "command": "genaiscript.notebook.new",
+                "title": "New Notebook",
+                "category": "GenAIScript"
+            },
             {
                 "command": "genaiscript.request.abort",
                 "title": "Abort OpenAI request",
@@ -347,4 +364,4 @@
     "enabledApiProposals": [
         "languageModels"
     ]
-}
+}

packages/vscode/src/extension.ts

@@ -11,6 +11,7 @@ import { activateOpenAIRequestTreeDataProvider } from "./openairequesttree"
 import { activateAIRequestTreeDataProvider } from "./airequesttree"
 import { activateTestController } from "./testcontroller"
 import { activateModelCompletionProvider } from "./modelcompletionprovider"
+import { activateNotebook } from "./notebook"
 
 export async function activate(context: ExtensionContext) {
     const state = new ExtensionState(context)
@@ -22,6 +23,7 @@ export async function activate(context: ExtensionContext) {
     activateOpenAIRequestTreeDataProvider(state)
     activateStatusBar(state)
     activateModelCompletionProvider(state)
+    activateNotebook(state)
     // activateChatParticipant(state)
 
     context.subscriptions.push(