@dxos/assistant-toolkit 0.9.1-main.c7dcc2e112 → 0.9.1-staging.ee54ba693a

package/src/crud/graph.ts

@@ -1,364 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Tool from '@effect/ai/Tool';
-import * as Toolkit from '@effect/ai/Toolkit';
-import * as Context from 'effect/Context';
-import * as Effect from 'effect/Effect';
-import * as Function from 'effect/Function';
-import * as Option from 'effect/Option';
-import * as Schema from 'effect/Schema';
-import * as SchemaAST from 'effect/SchemaAST';
-
-import { Database, Entity, Feed, Filter, Obj, Query, Relation, Scope, Type } from '@dxos/echo';
-import { isEncodedReference } from '@dxos/echo-protocol';
-import { ReferenceAnnotationId, getTypeAnnotation, getTypeIdentifierAnnotation } from '@dxos/echo/Annotation';
-import { createObject } from '@dxos/echo/internal';
-import { SchemaEx } from '@dxos/effect';
-import { DXN, EID, EntityId, type URI } from '@dxos/keys';
-import { log } from '@dxos/log';
-import { deepMapValues, isNonNullable, trim } from '@dxos/util';
-
-// TODO(burdon): Unify with the graph schema.
-export const Subgraph = Schema.Struct({
-  /** Objects and relations. */
-  objects: Schema.Array(Schema.Any),
-});
-
-export type Subgraph = Schema.Schema.Type<typeof Subgraph>;
-export type RelatedSchema = {
-  schema: Type.AnyEntity;
-  kind: 'reference' | 'relation';
-};
-
-/**
- * Find all schemas that are related to the given schema.
- *
- * @param db
- * @param schema
- * @returns
- */
-export const findRelatedSchema = async (db: Database.Database, anchor: Type.AnyEntity): Promise<RelatedSchema[]> => {
-  const allSchemas = await db.query(Query.select(Filter.type(Type.Type)).from(Scope.space(), Scope.registry())).run();
-
-  // TODO(dmaretskyi): Also do references.
-  return allSchemas
-    .filter((schema) => {
-      const annotation = getTypeAnnotation(Type.getSchema(schema));
-      if (annotation?.kind !== Entity.Kind.Relation) {
-        return false;
-      }
-
-      return (
-        isSchemaAddressableByDXN(anchor, annotation.sourceSchema!) ||
-        isSchemaAddressableByDXN(anchor, annotation.targetSchema!)
-      );
-    })
-    .map(
-      (schema): RelatedSchema => ({
-        kind: 'relation',
-        schema,
-      }),
-    );
-};
-
-/**
- * Non-strict DXN comparison.
- * Returns true if the DXN could be resolved to the schema.
- */
-const isSchemaAddressableByDXN = (type: Type.AnyEntity, dxn: DXN.DXN): boolean => {
-  if (getTypeIdentifierAnnotation(Type.getSchema(type)) === dxn) {
-    return true;
-  }
-
-  return DXN.getName(dxn) === Type.getTypename(type);
-};
-
-/**
- * Perform vector search in the local database.
- */
-// TODO(dmaretskyi): Rename `GraphReadToolkit`.
-export const LocalSearchToolkit = Toolkit.make(
-  Tool.make('search_local_search', {
-    description: 'Search the local database for information using a vector index',
-    parameters: {
-      query: Schema.String.annotations({
-        description: 'The query to search for. Could be a question or a topic or a set of keywords.',
-      }),
-    },
-    success: Schema.Unknown,
-    failure: Schema.Never,
-    dependencies: [Database.Service, Feed.FeedService],
-  }),
-);
-
-export const LocalSearchHandler = LocalSearchToolkit.toLayer({
-  search_local_search: Effect.fn(function* ({ query }) {
-    const objects = yield* Database.query(Query.select(Filter.text(query, { type: 'vector' }))).run;
-    const results = [...objects];
-
-    const feedOption = yield* Effect.serviceOption(Feed.ContextFeedService);
-    if (Option.isSome(feedOption)) {
-      const feedObjects = yield* Feed.runQuery(feedOption.value.feed, Filter.everything());
-      // TODO(dmaretskyi): Text search on the feed.
-      results.push(...feedObjects);
-    }
-
-    return trim`
-      <local_context>
-        ${JSON.stringify(results, null, 2)}
-      </local_context>
-    `;
-  }),
-});
-
-/**
- * Attached as an annotation to the writer tool.
- */
-class GraphWriterSchema extends Context.Tag('@dxos/assistant/GraphWriterSchema')<
-  GraphWriterSchema,
-  {
-    schema: Type.AnyEntity[];
-  }
->() {}
-
-/**
- * Forms typed objects that can be written to the graph database.
- */
-export const makeGraphWriterToolkit = ({ schema }: { schema: Type.AnyEntity[] }) => {
-  return Toolkit.make(
-    Tool.make('graph_writer', {
-      description: 'Write to the local graph database',
-      parameters: createExtractionSchema(schema).fields,
-      success: Schema.Unknown,
-      failure: Schema.Never,
-      dependencies: [Database.Service, Feed.ContextFeedService, Feed.FeedService],
-    }).annotateContext(Context.make(GraphWriterSchema, { schema })),
-  );
-};
-
-export const makeGraphWriterHandler = (
-  toolkit: ReturnType<typeof makeGraphWriterToolkit>,
-  {
-    onAppend,
-  }: {
-    onAppend?: (object: URI.URI[]) => void;
-  } = {},
-) => {
-  const { schema } = Context.get(
-    toolkit.tools.graph_writer.annotations as Context.Context<GraphWriterSchema>,
-    GraphWriterSchema,
-  );
-
-  return toolkit.toLayer({
-    graph_writer: Effect.fn(function* (input) {
-      const { db } = yield* Database.Service;
-      const { feed } = yield* Feed.ContextFeedService;
-      const data = yield* sanitizeObjects(schema, input as any, db, feed);
-      yield* Feed.append(feed, data as Obj.Unknown[]);
-
-      const dxns = data.map((obj) => Entity.getURI(obj));
-      onAppend?.(dxns);
-      return dxns;
-    }),
-  });
-};
-
-/**
- * Create a schema for structured data extraction.
- */
-export const createExtractionSchema = (types: Type.AnyEntity[]) => {
-  return Schema.Struct({
-    ...Object.fromEntries(
-      types
-        .map((type) => preprocessSchema(Type.getSchema(type)))
-        .map((schema, index) => [
-          `objects_${getSanitizedSchemaName(types[index])}`,
-          Schema.optional(Schema.Array(schema)).annotations({
-            description: `The objects of type: ${DXN.getName(DXN.tryMake(Type.getURI(types[index])!)!)}. ${SchemaAST.getDescriptionAnnotation(Type.getSchema(types[index]).ast).pipe(Option.getOrElse(() => ''))}`,
-          }),
-        ]),
-    ),
-  });
-};
-
-export const getSanitizedSchemaName = (schema: Type.AnyEntity) => {
-  return DXN.getName(DXN.tryMake(Type.getURI(schema)!)!).replaceAll(/[^a-zA-Z0-9]+/g, '_');
-};
-
-export const sanitizeObjects = (
-  types: Type.AnyEntity[],
-  data: Record<string, readonly unknown[]>,
-  db: Database.Database,
-  feed?: Feed.Feed,
-): Effect.Effect<Entity.Unknown[], never, Feed.FeedService> =>
-  Effect.gen(function* () {
-    const entries = types
-      .map(
-        (type) =>
-          data[`objects_${getSanitizedSchemaName(type)}`]?.map((object: any) => ({
-            data: object,
-            schema: type,
-          })) ?? [],
-      )
-      .flat();
-
-    const idMap = new Map<string, string>();
-    const existingIds = new Set<EntityId>();
-    const enitties = new Map<EntityId, Entity.Unknown>();
-
-    const resolveId = (id: string): EID.EID | undefined => {
-      if (EntityId.isValid(id)) {
-        existingIds.add(id);
-        return EID.make({ entityId: id });
-      }
-
-      const mappedId = idMap.get(id);
-      if (mappedId && EntityId.isValid(mappedId)) {
-        return EID.make({ entityId: mappedId });
-      }
-
-      return undefined;
-    };
-
-    const res = entries
-      .map((entry) => {
-        // This entry mutates existing object.
-        if (EntityId.isValid(entry.data.id)) {
-          return entry;
-        }
-
-        idMap.set(entry.data.id, EntityId.random());
-        entry.data.id = idMap.get(entry.data.id);
-        return entry;
-      })
-      .map((entry) => {
-        const data = deepMapValues(entry.data, (value, recurse) => {
-          if (isEncodedReference(value)) {
-            const ref = value['/'];
-            const id = resolveId(ref);
-
-            if (id) {
-              // Link to an existing object.
-              return { '/': id.toString() };
-            } else {
-              // Search URIs?
-              return { '/': `search:?q=${encodeURIComponent(ref)}` };
-            }
-          }
-
-          return recurse(value);
-        });
-
-        let sourceUri: EID.EID | undefined;
-        let targetUri: EID.EID | undefined;
-        if (Entity.getKind(Type.getSchema(entry.schema)) === 'relation') {
-          sourceUri = resolveId(data.source);
-          if (!sourceUri) {
-            log.warn('source not found', { source: data.source });
-          }
-          targetUri = resolveId(data.target);
-          if (!targetUri) {
-            log.warn('target not found', { target: data.target });
-          }
-          delete data.source;
-          delete data.target;
-        }
-
-        return {
-          data,
-          schema: entry.schema,
-          sourceUri,
-          targetUri,
-        };
-      })
-      .filter((object) => !existingIds.has(object.data.id)); // TODO(dmaretskyi): This dissallows updating existing objects.
-
-    // TODO(dmaretskyi): Use ref resolver.
-    const dbObjects = yield* Effect.promise(() => db.query(Query.select(Filter.id(...existingIds))).run());
-    const feedObjects = feed && existingIds.size > 0 ? yield* Feed.runQuery(feed, Filter.id(...existingIds)) : [];
-    const objects = [...dbObjects, ...feedObjects].filter(isNonNullable);
-
-    // TODO(dmaretskyi): Returns everything if IDs are empty!
-    log.info('objects', { dbObjects, feedObjects, existingIds });
-    const missing = Array.from(existingIds).filter((id) => !objects.some((object) => object.id === id));
-    if (missing.length > 0) {
-      throw new Error(`Object IDs do not point to existing objects: ${missing.join(', ')}`);
-    }
-
-    return res.flatMap(({ data, schema, sourceUri, targetUri }) => {
-      let skip = false;
-      if (sourceUri) {
-        const id = EID.getEntityId(sourceUri);
-        const obj = objects.find((object) => object.id === id) ?? (id ? enitties.get(id) : undefined);
-        if (obj) {
-          data[Relation.Source] = obj;
-        } else {
-          skip = true;
-        }
-      }
-      if (targetUri) {
-        const id = EID.getEntityId(targetUri);
-        const obj = objects.find((object) => object.id === id) ?? (id ? enitties.get(id) : undefined);
-        if (obj) {
-          data[Relation.Target] = obj;
-        } else {
-          skip = true;
-        }
-      }
-      if (!skip) {
-        const obj = createObject(schema, data);
-        enitties.set(obj.id, obj);
-        return [obj];
-      }
-      return [];
-    });
-  });
-
-const SoftRef = Schema.Struct({
-  '/': Schema.String,
-}).annotations({
-  description: 'Reference to another object.',
-});
-
-const preprocessSchema = (schema: Schema.Schema.AnyNoContext) => {
-  const isRelationSchema = Entity.getKind(schema) === 'relation';
-
-  const go = (ast: SchemaAST.AST, visited = new Set<SchemaAST.AST>()): SchemaAST.AST => {
-    if (visited.has(ast)) {
-      // Already visited this node, prevent infinite recursion.
-      return ast;
-    }
-    visited.add(ast);
-
-    if (SchemaAST.getAnnotation(ast, ReferenceAnnotationId).pipe(Option.isSome)) {
-      return SoftRef.ast;
-    }
-
-    return SchemaEx.mapAst(ast, (child) => go(child, visited));
-  };
-
-  return Schema.make<any, any, never>(SchemaEx.mapAst(schema.ast, (ast) => go(ast))).pipe(
-    Schema.omit('id'),
-    Schema.extend(
-      Schema.Struct({
-        id: Schema.String.annotations({
-          description: 'The id of this object. Come up with a unique id based on your judgement.',
-        }),
-      }),
-    ),
-    isRelationSchema
-      ? Schema.extend(
-          Schema.Struct({
-            source: Schema.String.annotations({
-              description: 'The id of the source object for this relation.',
-            }),
-            target: Schema.String.annotations({
-              description: 'The id of the target object for this relation.',
-            }),
-          }),
-        )
-      : Function.identity<Schema.Schema.AnyNoContext>,
-  );
-};

package/src/crud/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './graph';

package/src/functions/READ_THIS.md

@@ -1,5 +0,0 @@
-###
-
-Can we move all functions into the corresponding blueprint folder?
-
-###

package/src/functions/agent/prompt.conversations.json

@@ -1 +0,0 @@
-{"conversations":[{"parameters":{"model":"ai.claude.model.claude-opus-4-6","stream":true,"tools":[{"name":"completeJob","inputSchema":{"type":"object","required":[],"properties":{"success":{"$id":"/schemas/any","title":"any"},"failure":{"type":"object","required":["message"],"properties":{"message":{"type":"string","description":"Short message describing the error."},"description":{"type":"string","description":"Optional longer message describing in detail what went wrong"}},"additionalProperties":false}},"additionalProperties":false}}]},"prompt":{"content":[{"role":"system","content":"You are an agent running in the non-interactive mode.\nThe user is unable to see what you are doing, and cannot answer any questions.\nDo not ask questions.\nComplete the task before you, and at the end call [completeJob] with the output.\nIf you are unable to complete the task, call [completeJob] with the failure reason.\nIf no output is required, call [completeJob] with an empty object: {}\nDo not stop until you call [completeJob].","options":{}},{"role":"user","content":"Reply with a single word: ack.\n<input>{}</input>","options":{"anthropic":{"cacheControl":{"ttl":"5m","type":"ephemeral"}}}}]},"response":[{"type":"response-metadata","id":"msg_011XBspJo8qpA74ASw5uEdDn","modelId":"claude-opus-4-6","timestamp":"1970-01-01T00:00:00.000Z","metadata":{}},{"type":"reasoning-start","id":"0","metadata":{}},{"type":"reasoning-delta","id":"0","delta":"The user wants me to reply with a single word \"ack\" and complete the job.","metadata":{}},{"type":"reasoning-delta","id":"0","delta":"","metadata":{"anthropic":{"type":"thinking","signature":"Eo8CCmMIDhgCKkDmvWuElJmvgQqKv/OM8h45Qm9vFiau8OtHTiKeFGzABfFt5k5Oc9XiWXg1jeCrSXFmAcI3XEaKXp6jF8PCj/O0Mg9jbGF1ZGUtb3B1cy00LTY4AEIIdGhpbmtpbmcSDJLLuLKNtiWzw7pLSxoMVS5+G0m2Rt90uT/BIjDHn93ko70QlooeTV4Dfqhzvh3iIOfFQSt2vGX3dLIl4vErRKSzNfAGaI0f3Se9POgqWmqghiUovsd6FCGPEP+fypxFkW0ljAzvkRZElLL1koC0efkHJXq7+BLOE5LqBCe+kl/Sd22oz6dpOJis0l06/ffWB/VmRLbouqvk9I8W3+bqLYd/MJanUuthpBgB"}}},{"type":"reasoning-end","id":"0","metadata":{}},{"type":"tool-params-start","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","name":"completeJob","providerExecuted":false,"metadata":{}},{"type":"tool-params-delta","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","delta":"","metadata":{}},{"type":"tool-params-delta","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","delta":"{\"success\"","metadata":{}},{"type":"tool-params-delta","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","delta":": \"ack\"}","metadata":{}},{"type":"tool-params-end","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","metadata":{}},{"type":"finish","reason":"tool-calls","usage":{"inputTokens":757,"outputTokens":89,"totalTokens":846},"metadata":{"anthropic":{"usage":{"cache_creation":{"ephemeral_1h_input_tokens":0,"ephemeral_5m_input_tokens":0},"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"input_tokens":757,"output_tokens":43,"server_tool_use":null,"service_tier":"standard"}}}}]},{"parameters":{"model":"ai.claude.model.claude-opus-4-6","stream":true,"tools":[{"name":"completeJob","inputSchema":{"type":"object","required":[],"properties":{"success":{"$id":"/schemas/any","title":"any"},"failure":{"type":"object","required":["message"],"properties":{"message":{"type":"string","description":"Short message describing the error."},"description":{"type":"string","description":"Optional longer message describing in detail what went wrong"}},"additionalProperties":false}},"additionalProperties":false}}]},"prompt":{"content":[{"role":"system","content":"You are an agent running in the non-interactive mode.\nThe user is unable to see what you are doing, and cannot answer any questions.\nDo not ask questions.\nComplete the task before you, and at the end call [completeJob] with the output.\nIf you are unable to complete the task, call [completeJob] with the failure reason.\nIf no output is required, call [completeJob] with an empty object: {}\nDo not stop until you call [completeJob].","options":{}},{"role":"user","content":"Reply with a single word: ack.\n<input>{}</input>","options":{}},{"role":"assistant","content":[{"type":"reasoning","text":"The user wants me to reply with a single word \"ack\" and complete the job.","options":{}},{"type":"tool-call","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","name":"completeJob","params":{"success":"ack"},"providerExecuted":false,"options":{}}],"options":{}},{"role":"tool","content":[{"type":"tool-result","id":"toolu_018phiRmk3xRDvMqXkqaVTo2","name":"completeJob","isFailure":false,"result":{},"providerExecuted":false,"options":{}}],"options":{"anthropic":{"cacheControl":{"ttl":"5m","type":"ephemeral"}}}}]},"response":[{"type":"response-metadata","id":"msg_0167q8hG371zH6naFSmJTKnL","modelId":"claude-opus-4-6","timestamp":"1970-01-01T00:00:00.000Z","metadata":{}},{"type":"finish","reason":"stop","usage":{"inputTokens":824,"outputTokens":2,"totalTokens":826},"metadata":{"anthropic":{"usage":{"cache_creation":{"ephemeral_1h_input_tokens":0,"ephemeral_5m_input_tokens":0},"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"input_tokens":824,"output_tokens":2,"server_tool_use":null,"service_tier":"standard"}}}}]},{"parameters":{"model":"ai.claude.model.claude-opus-4-8","stream":true,"tools":[{"name":"completeJob","inputSchema":{"type":"object","required":[],"properties":{"success":{"$id":"/schemas/any","title":"any"},"failure":{"type":"object","required":["message"],"properties":{"message":{"type":"string","description":"Short message describing the error."},"description":{"type":"string","description":"Optional longer message describing in detail what went wrong"}},"additionalProperties":false}},"additionalProperties":false}}]},"prompt":{"content":[{"role":"system","content":"You are an agent running in the non-interactive mode.\nThe user is unable to see what you are doing, and cannot answer any questions.\nDo not ask questions.\nComplete the task before you, and at the end call [completeJob] with the output.\nIf you are unable to complete the task, call [completeJob] with the failure reason.\nIf no output is required, call [completeJob] with an empty object: {}\nDo not stop until you call [completeJob].","options":{}},{"role":"user","content":"Reply with a single word: ack.\n<input>{}</input>","options":{"anthropic":{"cacheControl":{"ttl":"5m","type":"ephemeral"}}}}]},"response":[{"type":"response-metadata","id":"msg_015BQWUzrZFZMyJcjKfHsF5F","modelId":"claude-opus-4-8","timestamp":"1970-01-01T00:00:00.000Z","metadata":{}},{"type":"text-start","id":"0","metadata":{}},{"type":"text-delta","id":"0","delta":"ack","metadata":{}},{"type":"text-end","id":"0","metadata":{}},{"type":"tool-params-start","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","name":"completeJob","providerExecuted":false,"metadata":{}},{"type":"tool-params-delta","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","delta":"","metadata":{}},{"type":"tool-params-delta","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","delta":"{\"succ","metadata":{}},{"type":"tool-params-delta","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","delta":"ess\": \"ack\"}","metadata":{}},{"type":"tool-params-end","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","metadata":{}},{"type":"finish","reason":"tool-calls","usage":{"inputTokens":624,"outputTokens":52,"totalTokens":676},"metadata":{"anthropic":{"usage":{"cache_creation":{"ephemeral_1h_input_tokens":0,"ephemeral_5m_input_tokens":0},"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"input_tokens":624,"output_tokens":4,"server_tool_use":null,"service_tier":"standard"}}}}]},{"parameters":{"model":"ai.claude.model.claude-opus-4-8","stream":true,"tools":[{"name":"completeJob","inputSchema":{"type":"object","required":[],"properties":{"success":{"$id":"/schemas/any","title":"any"},"failure":{"type":"object","required":["message"],"properties":{"message":{"type":"string","description":"Short message describing the error."},"description":{"type":"string","description":"Optional longer message describing in detail what went wrong"}},"additionalProperties":false}},"additionalProperties":false}}]},"prompt":{"content":[{"role":"system","content":"You are an agent running in the non-interactive mode.\nThe user is unable to see what you are doing, and cannot answer any questions.\nDo not ask questions.\nComplete the task before you, and at the end call [completeJob] with the output.\nIf you are unable to complete the task, call [completeJob] with the failure reason.\nIf no output is required, call [completeJob] with an empty object: {}\nDo not stop until you call [completeJob].","options":{}},{"role":"user","content":"Reply with a single word: ack.\n<input>{}</input>","options":{}},{"role":"assistant","content":[{"type":"text","text":"ack","options":{}},{"type":"tool-call","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","name":"completeJob","params":{"success":"ack"},"providerExecuted":false,"options":{}}],"options":{}},{"role":"tool","content":[{"type":"tool-result","id":"toolu_0133KXrwo8bGDYrsoVKDsY1S","name":"completeJob","isFailure":false,"result":{},"providerExecuted":false,"options":{}}],"options":{"anthropic":{"cacheControl":{"ttl":"5m","type":"ephemeral"}}}}]},"response":[{"type":"response-metadata","id":"msg_01CriLe6LPTyJWJQvVN3WvMq","modelId":"claude-opus-4-8","timestamp":"1970-01-01T00:00:00.000Z","metadata":{}},{"type":"text-start","id":"0","metadata":{}},{"type":"text-delta","id":"0","delta":"ack","metadata":{}},{"type":"text-end","id":"0","metadata":{}},{"type":"finish","reason":"stop","usage":{"inputTokens":683,"outputTokens":4,"totalTokens":687},"metadata":{"anthropic":{"usage":{"cache_creation":{"ephemeral_1h_input_tokens":0,"ephemeral_5m_input_tokens":0},"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"input_tokens":683,"output_tokens":4,"server_tool_use":null,"service_tier":"standard"}}}}]}]}

package/src/functions/agent/prompt.test.ts

@@ -1,78 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import { describe, expect, it } from '@effect/vitest';
-import * as Effect from 'effect/Effect';
-
-import { AiContext } from '@dxos/assistant';
-import { Routine, Operation, OperationHandlerSet } from '@dxos/compute';
-import { Database, Feed, Filter, Obj, Ref } from '@dxos/echo';
-import { TestHelpers } from '@dxos/effect/testing';
-import { AssistantTestLayer } from '@dxos/functions-runtime/testing';
-import { EntityId } from '@dxos/keys';
-import { Text } from '@dxos/schema';
-import { Message } from '@dxos/types';
-
-import * as Chat from '../../types/Chat';
-import { AgentPrompt } from './definitions';
-import defaultAgentPrompt from './prompt';
-
-EntityId.dangerouslyDisableRandomness();
-
-const operationHandlerSet = OperationHandlerSet.make(defaultAgentPrompt);
-
-const TestLayer = AssistantTestLayer({
-  operationHandlers: operationHandlerSet,
-  types: [Chat.Chat, Message.Message, AiContext.Binding, Text.Text],
-  aiServicePreset: 'edge-remote',
-});
-
-const countFeedMessages = (feed: Feed.Feed) =>
-  Effect.gen(function* () {
-    const items = yield* Feed.runQuery(feed, Filter.everything());
-    return items.filter(Obj.instanceOf(Message.Message)).length;
-  });
-
-describe('Agent prompt', () => {
-  it.effect(
-    'chat mode appends assistant messages to the chat queue',
-    Effect.fnUntraced(
-      function* (_) {
-        const feed = yield* Database.add(Feed.make());
-        const messageCountBefore = yield* countFeedMessages(feed);
-
-        const chat = yield* Database.add(
-          Chat.make({
-            feed: Ref.make(feed),
-          }),
-        );
-
-        const prompt = yield* Database.add(
-          Routine.make({
-            name: 'chat-mode-test',
-            instructions: 'Reply with a single word: ack.',
-            blueprints: [],
-            context: [],
-          }),
-        );
-
-        yield* Database.flush();
-
-        const result = yield* Operation.invoke(AgentPrompt, {
-          prompt: Ref.make(prompt),
-          input: {},
-          chat: Ref.make(chat),
-        });
-
-        const messageCountAfter = yield* countFeedMessages(feed);
-
-        expect(messageCountAfter).toBeGreaterThan(messageCountBefore);
-        expect(result).toBe('ack');
-      },
-      Effect.provide(TestLayer),
-      TestHelpers.provideTestContext,
-    ),
-    { timeout: 60_000 },
-  );
-});

package/src/functions/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './agent';

package/src/supervisor/plan-completion-guard.ts

@@ -1,50 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Database, Feed, Filter } from '@dxos/echo';
-import { type CompletionGuard } from '@dxos/functions-runtime';
-
-import { Agent, Plan } from '../types';
-
-const findAgentForFeed = (feed: Feed.Feed): Effect.Effect<Agent.Agent | undefined, never, Database.Service> =>
-  Effect.gen(function* () {
-    const agents = yield* Database.query(Filter.type(Agent.Agent)).run;
-    for (const agent of agents) {
-      if (!agent.chat) {
-        continue;
-      }
-      const matches = yield* Effect.gen(function* () {
-        const chat = yield* Database.load(agent.chat!);
-        const chatFeed = yield* Database.load(chat.feed);
-        return chatFeed.id === feed.id;
-      }).pipe(Effect.orElseSucceed(() => false));
-      if (matches) {
-        return agent;
-      }
-    }
-    return undefined;
-  });
-
-const hasIncompleteTasks = (plan: Plan.Plan): boolean =>
-  plan.tasks.some((task) => task.status === 'todo' || task.status === 'in-progress');
-
-/**
- * Returns a markdown plan summary when the conversation's agent still has open plan tasks.
- */
-export const makePlanCompletionGuard = (): CompletionGuard => ({
-  getIncompletePlanSummary: (feed) =>
-    Effect.gen(function* () {
-      const agent = yield* findAgentForFeed(feed);
-      if (!agent?.plan) {
-        return undefined;
-      }
-      const plan = yield* Database.load(agent.plan).pipe(Effect.orElseSucceed(() => undefined));
-      if (!plan || !hasIncompleteTasks(plan)) {
-        return undefined;
-      }
-      return Plan.formatPlan(plan);
-    }),
-});

package/src/toolkits/WebToolkit.ts

@@ -1,44 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as AnthropicTool from '@effect/ai-anthropic/AnthropicTool';
-import * as Tool from '@effect/ai/Tool';
-import * as Toolkit from '@effect/ai/Toolkit';
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { proxyFetchLegacy } from '@dxos/edge-client/cors-proxy';
-
-// TODO(dmaretskyi): Testing only.
-export const WebToolkit = Toolkit.make(
-  AnthropicTool.WebSearch_20250305({}).pipe(
-    // TODO(dmaretskyi): Effect bug -- provider-defined tools don't support annotations.
-    // ToolFormatter.assign({
-    //   debugFormatResult: (result) =>
-    //     Array.isArray(result) ? result.map(Struct.pick('title', 'type', 'url', 'page_age')) : result,
-    // }),
-  ),
-  Tool.make('WebFetch', {
-    parameters: {
-      url: Schema.String,
-    },
-    success: Schema.String,
-  }),
-);
-
-export const layer = WebToolkit.toLayer({
-  WebFetch: Effect.fnUntraced(function* ({ url }) {
-    return yield* Effect.tryPromise({
-      try: async (): Promise<string> => {
-        const response = await proxyFetchLegacy(new URL(url));
-        if (!response.ok) {
-          const body = await response.text();
-          return `Fetch failed with status ${response.status}: ${body.slice(0, 256)}`;
-        }
-        return await response.text();
-      },
-      catch: (cause) => new Error(String(cause)),
-    }).pipe(Effect.catchAll((err) => Effect.succeed(err.message)));
-  }),
-});

package/src/toolkits/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * as WebToolkit from './WebToolkit';