@dxos/assistant-toolkit 0.8.4-main.ae835ea → 0.8.4-main.bc674ce

package/package.json

@@ -1,9 +1,13 @@
 {
   "name": "@dxos/assistant-toolkit",
-  "version": "0.8.4-main.ae835ea",
+  "version": "0.8.4-main.bc674ce",
   "description": "Assistant toolkit, definitions, and utilities.",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/dxos/dxos"
+  },
   "license": "MIT",
   "author": "DXOS.org",
   "type": "module",
@@ -13,53 +17,65 @@
       "types": "./dist/types/src/index.d.ts",
       "browser": "./dist/lib/browser/index.mjs",
       "node": "./dist/lib/node-esm/index.mjs"
+    },
+    "./testing": {
+      "source": "./src/testing/index.ts",
+      "types": "./dist/types/src/testing/index.d.ts",
+      "browser": "./dist/lib/browser/testing/index.mjs",
+      "node": "./dist/lib/node-esm/testing/index.mjs"
     }
   },
   "types": "dist/types/src/index.d.ts",
   "typesVersions": {
-    "*": {}
+    "*": {
+      "testing": [
+        "./dist/types/src/testing/index.d.ts"
+      ]
+    }
   },
   "files": [
     "dist",
     "src"
   ],
   "dependencies": {
-    "@effect/ai": "0.29.1",
-    "@effect/ai-anthropic": "0.19.2",
-    "@effect/experimental": "0.56.0",
-    "@effect/platform": "^0.92.1",
-    "dfx": "^0.126.0",
+    "@effect/ai": "0.32.1",
+    "@effect/ai-anthropic": "0.22.0",
+    "@effect/experimental": "0.57.11",
+    "@effect/platform": "0.93.6",
+    "dfx": "^0.127.0",
     "exa-js": "^1.5.12",
-    "@dxos/ai": "0.8.4-main.ae835ea",
-    "@dxos/app-framework": "0.8.4-main.ae835ea",
-    "@dxos/assistant": "0.8.4-main.ae835ea",
-    "@dxos/blueprints": "0.8.4-main.ae835ea",
-    "@dxos/context": "0.8.4-main.ae835ea",
-    "@dxos/conductor": "0.8.4-main.ae835ea",
-    "@dxos/echo-db": "0.8.4-main.ae835ea",
-    "@dxos/echo": "0.8.4-main.ae835ea",
-    "@dxos/echo-protocol": "0.8.4-main.ae835ea",
-    "@dxos/edge-client": "0.8.4-main.ae835ea",
-    "@dxos/functions": "0.8.4-main.ae835ea",
-    "@dxos/errors": "0.8.4-main.ae835ea",
-    "@dxos/invariant": "0.8.4-main.ae835ea",
-    "@dxos/keys": "0.8.4-main.ae835ea",
-    "@dxos/log": "0.8.4-main.ae835ea",
-    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.ae835ea",
-    "@dxos/util": "0.8.4-main.ae835ea",
-    "@dxos/effect": "0.8.4-main.ae835ea",
-    "@dxos/plugin-markdown": "0.8.4-main.ae835ea",
-    "@dxos/schema": "0.8.4-main.ae835ea"
+    "@dxos/ai": "0.8.4-main.bc674ce",
+    "@dxos/blueprints": "0.8.4-main.bc674ce",
+    "@dxos/app-framework": "0.8.4-main.bc674ce",
+    "@dxos/assistant": "0.8.4-main.bc674ce",
+    "@dxos/echo-db": "0.8.4-main.bc674ce",
+    "@dxos/context": "0.8.4-main.bc674ce",
+    "@dxos/echo": "0.8.4-main.bc674ce",
+    "@dxos/echo-protocol": "0.8.4-main.bc674ce",
+    "@dxos/edge-client": "0.8.4-main.bc674ce",
+    "@dxos/effect": "0.8.4-main.bc674ce",
+    "@dxos/errors": "0.8.4-main.bc674ce",
+    "@dxos/functions": "0.8.4-main.bc674ce",
+    "@dxos/functions-runtime": "0.8.4-main.bc674ce",
+    "@dxos/invariant": "0.8.4-main.bc674ce",
+    "@dxos/log": "0.8.4-main.bc674ce",
+    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.bc674ce",
+    "@dxos/plugin-markdown": "0.8.4-main.bc674ce",
+    "@dxos/keys": "0.8.4-main.bc674ce",
+    "@dxos/types": "0.8.4-main.bc674ce",
+    "@dxos/util": "0.8.4-main.bc674ce",
+    "@dxos/schema": "0.8.4-main.bc674ce",
+    "@dxos/conductor": "0.8.4-main.bc674ce"
   },
   "devDependencies": {
-    "effect": "3.18.3",
-    "react": "~19.2.0",
-    "react-dom": "~19.2.0"
+    "effect": "3.19.11",
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3"
   },
   "peerDependencies": {
-    "effect": "^3.13.3",
-    "react": "^19.0.0",
-    "react-dom": "^19.0.0"
+    "effect": "3.19.11",
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3"
   },
   "publishConfig": {
     "access": "public"

package/src/blueprints/design/design-blueprint.test.ts

@@ -17,24 +17,21 @@ import {
 } from '@dxos/assistant';
 import { Blueprint } from '@dxos/blueprints';
 import { Obj, Ref } from '@dxos/echo';
-import { TestHelpers, acquireReleaseResource } from '@dxos/effect';
-import {
-  ComputeEventLogger,
-  DatabaseService,
-  FunctionInvocationService,
-  QueueService,
-  TracingService,
-} from '@dxos/functions';
-import { TestDatabaseLayer } from '@dxos/functions/testing';
+import { Database } from '@dxos/echo';
+import { acquireReleaseResource } from '@dxos/effect';
+import { TestHelpers } from '@dxos/effect/testing';
+import { QueueService, TracingService } from '@dxos/functions';
+import { FunctionInvocationServiceLayerTestMocked, TestDatabaseLayer } from '@dxos/functions-runtime/testing';
 import { log } from '@dxos/log';
 import { Markdown } from '@dxos/plugin-markdown/types';
-import { DataType } from '@dxos/schema';
+import { Text } from '@dxos/schema';
+import { type Message } from '@dxos/types';
 import { trim } from '@dxos/util';
 
 import { Document } from '../../functions';
 import { testToolkit } from '../testing';
 
-import blueprint from './design-blueprint';
+import { blueprint } from './design-blueprint';
 
 describe('Design Blueprint', { timeout: 120_000 }, () => {
   it.scoped(
@@ -42,22 +39,22 @@ describe('Design Blueprint', { timeout: 120_000 }, () => {
     Effect.fn(
       function* ({ expect }) {
         const observer = GenerationObserver.fromPrinter(new ConsolePrinter());
-        const queue = yield* QueueService.createQueue<DataType.Message | ContextBinding>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation(queue));
+        const queue = yield* QueueService.createQueue<Message.Message | ContextBinding>();
+        const conversation = yield* acquireReleaseResource(() => new AiConversation({ queue }));
 
-        yield* DatabaseService.add(blueprint);
+        yield* Database.Service.add(blueprint);
         yield* Effect.promise(() =>
           conversation.context.bind({
             blueprints: [Ref.make(blueprint)],
           }),
         );
 
-        const artifact = yield* DatabaseService.add(Markdown.makeDocument({ content: 'Hello, world!' }));
+        const artifact = yield* Database.Service.add(Markdown.make({ content: 'Hello, world!' }));
         let prevContent = artifact.content;
 
         {
           const prompt = trim`
-            I want to design a new feature for our product. 
+            I want to design a new feature for our product.
 
             We need to add a user profile system with the following requirements:
             1. Users should be able to create and edit their profiles
@@ -77,7 +74,7 @@ describe('Design Blueprint', { timeout: 120_000 }, () => {
 
         {
           const prompt = trim`
-            I want this to be built on top of Durable Objects and SQLite database. 
+            I want this to be built on top of Durable Objects and SQLite database.
             Adjust the spec to reflect this.
           `;
 
@@ -92,16 +89,16 @@ describe('Design Blueprint', { timeout: 120_000 }, () => {
           makeToolExecutionServiceFromFunctions(testToolkit, testToolkit.toLayer({}) as any),
           AiService.model('@anthropic/claude-3-5-sonnet-20241022'),
         ).pipe(
-          Layer.provideMerge(TestDatabaseLayer({ types: [DataType.Text, Markdown.Document, Blueprint.Blueprint] })),
+          Layer.provideMerge(TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] })),
           Layer.provideMerge(AiServiceTestingPreset('direct')),
           Layer.provideMerge(
-            FunctionInvocationService.layerTestMocked({ functions: [Document.read, Document.update] }).pipe(
-              Layer.provideMerge(ComputeEventLogger.layerFromTracing),
+            FunctionInvocationServiceLayerTestMocked({ functions: [Document.read, Document.update] }).pipe(
               Layer.provideMerge(TracingService.layerNoop),
             ),
           ),
         ),
       ),
+      TestHelpers.provideTestContext,
       TestHelpers.taggedTest('llm'),
     ),
   );

package/src/blueprints/design/design-blueprint.ts

@@ -4,8 +4,8 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 import { trim } from '@dxos/util';
 
 import { Document } from '../../functions';
@@ -20,14 +20,12 @@ const instructions = trim`
   Do not announce when you read or write the design spec document.
 `;
 
-const blueprint: Blueprint.Blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/design',
   name: 'Design Spec',
   description: 'Preserve the conversation in the design spec.',
   instructions: {
-    source: Ref.make(DataType.makeText(instructions)),
+    source: Ref.make(Text.make(instructions)),
   },
   tools: [Document.read, Document.update].map((fn) => ToolId.make(fn.key)),
 });
-
-export default blueprint;

package/src/blueprints/design/index.ts

@@ -2,6 +2,6 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './design-blueprint';
+import { blueprint } from './design-blueprint';
 
 export default blueprint;

package/src/blueprints/discord/discord-blueprint.ts

@@ -4,8 +4,8 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 import { trim } from '@dxos/util';
 
 import { Discord } from '../../functions';
@@ -21,14 +21,12 @@ const instructions = trim`
   DXOS serverId: 837138313172353095
 `;
 
-export const blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/discord',
   name: 'Discord',
   description: 'Discord integration.',
   instructions: {
-    source: Ref.make(DataType.makeText(instructions)),
+    source: Ref.make(Text.make(instructions)),
   },
   tools: [ToolId.make(Discord.fetch.key)],
 });
-
-export default blueprint;

package/src/blueprints/discord/index.ts

@@ -2,6 +2,6 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './discord-blueprint';
+import { blueprint } from './discord-blueprint';
 
 export default blueprint;

package/src/blueprints/linear/index.ts

@@ -2,6 +2,6 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './linear-blueprint';
+import { blueprint } from './linear-blueprint';
 
 export default blueprint;

package/src/blueprints/linear/linear-blueprint.ts

@@ -4,8 +4,8 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 import { trim } from '@dxos/util';
 
 import { Linear } from '../../functions';
@@ -22,14 +22,12 @@ const instructions = trim`
   DXOS teamId: 1127c63a-6f77-4725-9229-50f6cd47321c
 `;
 
-export const blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/linear',
   name: 'Linear',
   description: 'Syncs Linear workspaces.',
   instructions: {
-    source: Ref.make(DataType.makeText(instructions)),
+    source: Ref.make(Text.make(instructions)),
   },
   tools: [Linear.sync].map((tool) => ToolId.make(tool.key)),
 });
-
-export default blueprint;

package/src/blueprints/planning/index.ts

@@ -2,6 +2,6 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './planning-blueprint';
+import { blueprint } from './planning-blueprint';
 
 export default blueprint;

package/src/blueprints/planning/planning-blueprint.test.ts

@@ -16,42 +16,39 @@ import {
 } from '@dxos/assistant';
 import { Blueprint } from '@dxos/blueprints';
 import { Obj, Ref } from '@dxos/echo';
-import { TestHelpers, acquireReleaseResource } from '@dxos/effect';
-import {
-  ComputeEventLogger,
-  DatabaseService,
-  FunctionImplementationResolver,
-  FunctionInvocationService,
-  QueueService,
-  TracingService,
-} from '@dxos/functions';
-import { TestDatabaseLayer } from '@dxos/functions/testing';
+import { Database } from '@dxos/echo';
+import { acquireReleaseResource } from '@dxos/effect';
+import { TestHelpers } from '@dxos/effect/testing';
+import { QueueService, TracingService } from '@dxos/functions';
+import { FunctionImplementationResolver } from '@dxos/functions-runtime';
+import { FunctionInvocationServiceLayerTestMocked, TestDatabaseLayer } from '@dxos/functions-runtime/testing';
 import { log } from '@dxos/log';
 import { Markdown } from '@dxos/plugin-markdown/types';
-import { DataType } from '@dxos/schema';
+import { Text } from '@dxos/schema';
+import { type Message } from '@dxos/types';
 import { trim } from '@dxos/util';
 
 import { Tasks } from '../../functions';
 import { type TestStep, runSteps, testToolkit } from '../testing';
 
-import blueprint from './planning-blueprint';
+import { blueprint } from './planning-blueprint';
 
 describe('Planning Blueprint', { timeout: 120_000 }, () => {
   it.scoped(
     'planning blueprint',
     Effect.fn(
       function* ({ expect }) {
-        const queue = yield* QueueService.createQueue<DataType.Message | ContextBinding>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation(queue));
+        const queue = yield* QueueService.createQueue<Message.Message | ContextBinding>();
+        const conversation = yield* acquireReleaseResource(() => new AiConversation({ queue }));
 
-        yield* DatabaseService.add(blueprint);
+        yield* Database.Service.add(blueprint);
         yield* Effect.promise(() =>
           conversation.context.bind({
             blueprints: [Ref.make(blueprint)],
           }),
         );
 
-        const artifact = yield* DatabaseService.add(Markdown.makeDocument());
+        const artifact = yield* Database.Service.add(Markdown.make());
         let prevContent = artifact.content;
         const matchList =
           ({ includes = [], excludes = [] }: { includes: RegExp[]; excludes?: RegExp[] }) =>
@@ -107,22 +104,22 @@ describe('Planning Blueprint', { timeout: 120_000 }, () => {
       },
       Effect.provide(
         Layer.mergeAll(
-          TestDatabaseLayer({ types: [DataType.Text, Markdown.Document, Blueprint.Blueprint] }),
+          TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] }),
           makeToolResolverFromFunctions([Tasks.read, Tasks.update], testToolkit),
           makeToolExecutionServiceFromFunctions(testToolkit, testToolkit.toLayer({}) as any),
           AiService.model('@anthropic/claude-3-5-sonnet-20241022'),
         ).pipe(
           Layer.provideMerge(
-            FunctionInvocationService.layerTestMocked({ functions: [Tasks.read, Tasks.update] }).pipe(
-              Layer.provideMerge(ComputeEventLogger.layerFromTracing),
+            FunctionInvocationServiceLayerTestMocked({ functions: [Tasks.read, Tasks.update] }).pipe(
               Layer.provideMerge(TracingService.layerNoop),
             ),
           ),
           Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: [Tasks.read, Tasks.update] })),
-          Layer.provideMerge(TestDatabaseLayer({ types: [DataType.Text, Markdown.Document, Blueprint.Blueprint] })),
+          Layer.provideMerge(TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] })),
           Layer.provideMerge(AiServiceTestingPreset('direct')),
         ),
       ),
+      TestHelpers.provideTestContext,
       TestHelpers.taggedTest('llm'),
     ),
   );

package/src/blueprints/planning/planning-blueprint.ts

@@ -4,8 +4,8 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 import { trim } from '@dxos/util';
 
 import { Tasks } from '../../functions';
@@ -85,14 +85,12 @@ const instructions = trim`
   - Be precise with task descriptions and hierarchy levels
 `;
 
-export const blueprint: Blueprint.Blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/planning',
   name: 'Planning',
   description: 'Plans and tracks complex tasks with artifact management.',
   instructions: {
-    source: Ref.make(DataType.makeText(instructions)),
+    source: Ref.make(Text.make(instructions)),
   },
   tools: [Tasks.read, Tasks.update].map((fn) => ToolId.make(fn.key)),
 });
-
-export default blueprint;

package/src/blueprints/research/index.ts

@@ -2,6 +2,6 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './research-blueprint';
+import { blueprint } from './research-blueprint';
 
 export default blueprint;

package/src/blueprints/research/research-blueprint.ts

@@ -4,8 +4,8 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 import { trim } from '@dxos/util';
 
 import { Research } from '../../functions';
@@ -17,29 +17,35 @@ const instructions = trim`
   {{! Research }}
 
   You are an analyst that does research tasks using tools that scrape the web and create structured data.
-  The result of the research is a set of structured entities forming an interconnected graph.
-  When you are done, reply with the created objects.
-  Do not print the data, instead reply with inline references to the created objects.
-  Those will be later substituted with the pills representing the created objects.
-  Print the rest of the created objects as block references after the main note.
-
-  <example>
-    Based on my research, Google was founded by @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M and @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ
-
-    <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M</dxn></object>
-    <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ</dxn></object>
-    <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW92333</dxn></object>
-  </example>
+  Structured data extraction is an experimental feature -- only enable it if the user explicitly requests it in the prompt.
+  Prefer updating existing notes instead of creating new ones.
+
+  <strcutured_mode>
+    When you are done, reply with the created objects.
+    Do not print the data, instead reply with inline references to the created objects.
+    Those will be later substituted with the pills representing the created objects.
+    Print the rest of the created objects as block references after the main note.
+
+    <example>
+      Based on my research, Google was founded by @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M and @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ
+
+      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M</dxn></object>
+      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ</dxn></object>
+      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW92333</dxn></object>
+    </example>
+  </structured_mode>
+
+  <unstructured_mode>
+    Reply normally with the text mode of the result of your research.
+  </unstructured_mode>
 `;
 
-export const blueprint: Blueprint.Blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/research',
   name: 'Research',
   description: 'Researches the web and creates structured data.',
   instructions: {
-    source: Ref.make(DataType.makeText(instructions)),
+    source: Ref.make(Text.make(instructions)),
   },
   tools: [Research.create, Research.research].map((fn) => ToolId.make(fn.key)),
 });
-
-export default blueprint;

package/src/blueprints/testing.ts

@@ -6,10 +6,10 @@ import * as Toolkit from '@effect/ai/Toolkit';
 import * as Effect from 'effect/Effect';
 
 import { ConsolePrinter } from '@dxos/ai';
-import { type AiConversation, type AiConversationRunParams, GenerationObserver } from '@dxos/assistant';
+import { type AiConversation, type AiConversationRunProps, GenerationObserver } from '@dxos/assistant';
 import { log } from '@dxos/log';
 
-export type TestStep = Pick<AiConversationRunParams, 'prompt' | 'system'> & {
+export type TestStep = Pick<AiConversationRunProps, 'prompt' | 'system'> & {
   test?: () => Promise<void>;
 };
 

package/src/blueprints/websearch/index.ts

@@ -2,7 +2,8 @@
 // Copyright 2025 DXOS.org
 //
 
-import blueprint from './websearch-blueprint';
+import { blueprint } from './websearch-blueprint';
 
 export { WebSearchToolkit } from './websearch-toolkit';
+
 export default blueprint;

package/src/blueprints/websearch/websearch-blueprint.ts

@@ -4,17 +4,15 @@
 
 import { ToolId } from '@dxos/ai';
 import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
+import { Ref } from '@dxos/echo';
+import { Text } from '@dxos/schema';
 
-const blueprint: Blueprint.Blueprint = Obj.make(Blueprint.Blueprint, {
+export const blueprint = Blueprint.make({
   key: 'dxos.org/blueprint/web-search',
   name: 'Web Search',
   description: 'Search the web.',
   instructions: {
-    source: Ref.make(DataType.makeText('')), // No instructions required.
+    source: Ref.make(Text.make()),
   },
   tools: [ToolId.make('AnthropicWebSearch')],
 });
-
-export default blueprint;

package/src/crud/graph.test.ts

@@ -0,0 +1,46 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { describe, it } from '@effect/vitest';
+import * as Effect from 'effect/Effect';
+import * as Layer from 'effect/Layer';
+
+import { AiSession } from '@dxos/assistant';
+import { AssistantTestLayer } from '@dxos/assistant/testing';
+import { TestHelpers } from '@dxos/effect/testing';
+import { log } from '@dxos/log';
+import { Project } from '@dxos/types';
+
+import { makeGraphWriterHandler, makeGraphWriterToolkit } from './graph';
+
+const TestLayer = AssistantTestLayer({});
+
+describe('graph', () => {
+  // let builder: EchoTestBuilder;
+  // test('findRelatedSchema', async () => {
+  //   const db = await createEchoDatabase();
+  //   const relatedSchemas = await findRelatedSchema(db, Schema.Struct({}));
+  // });
+
+  const Toolkit = makeGraphWriterToolkit({ schema: [Project.Project] });
+  const ToolkitLayer = makeGraphWriterHandler(Toolkit);
+
+  it.effect.skip(
+    'calculator',
+    Effect.fn(
+      function* (_) {
+        const session = new AiSession();
+        const toolkit = yield* Toolkit;
+        const response = yield* session.run({
+          toolkit,
+          prompt: 'What is 10 + 20?',
+        });
+
+        log.info('response', { response });
+      },
+      Effect.provide(Layer.mergeAll(TestLayer, ToolkitLayer)),
+      TestHelpers.provideTestContext,
+    ),
+  );
+});