@dxos/plugin-assistant 0.8.4-main.72ec0f3 → 0.8.4-main.74a063c4e0

package/src/meta.ts

@@ -2,11 +2,11 @@
 // Copyright 2023 DXOS.org
 //
 
-import { type PluginMeta } from '@dxos/app-framework';
+import { type Plugin } from '@dxos/app-framework';
 import { trim } from '@dxos/util';
 
-export const meta: PluginMeta = {
-  id: 'dxos.org/plugin/assistant',
+export const meta: Plugin.Meta = {
+  id: 'org.dxos.plugin.assistant',
   name: 'Assistant',
   description: trim`
     Intelligent AI assistant that can analyze and interact with objects across your spaces.
@@ -18,4 +18,7 @@ export const meta: PluginMeta = {
   tags: ['labs'],
 };
 
-export const ASSISTANT_DIALOG = `${meta.id}/assistant/dialog`;
+export const ASSISTANT_DIALOG = `${meta.id}.assistant.dialog`;
+
+/** Companion variant identifier for the assistant chat panel. */
+export const ASSISTANT_COMPANION_VARIANT = 'assistant-chat';

package/src/operations/create-chat.ts

@@ -0,0 +1,80 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Capabilities, Capability } from '@dxos/app-framework';
+import { AiContextBinder } from '@dxos/assistant';
+import { Chat, DatabaseBlueprint, ProjectWizardBlueprint } from '@dxos/assistant-toolkit';
+import { Blueprint } from '@dxos/blueprints';
+import { Feed, Filter, Obj, Ref } from '@dxos/echo';
+import { createFeedServiceLayer } from '@dxos/echo-db';
+import { invariant } from '@dxos/invariant';
+import { Operation } from '@dxos/operation';
+import { ClientCapabilities } from '@dxos/plugin-client/types';
+
+import { AssistantBlueprint } from '#blueprints';
+
+import { CreateChat } from './definitions';
+
+const handler: Operation.WithHandler<typeof CreateChat> = CreateChat.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ db, name, addToSpace = true }) {
+      const registry = yield* Capability.get(Capabilities.AtomRegistry);
+      const client = yield* Capability.get(ClientCapabilities.Client);
+      const space = client.spaces.get(db.spaceId);
+      invariant(space, 'Space not found');
+      const feed = space.db.add(Feed.make());
+      const chat = Chat.make({ name, feed: Ref.make(feed) });
+      Obj.setParent(feed, chat);
+      if (addToSpace) {
+        space.db.add(chat);
+      }
+
+      // TODO(wittjosiah): This should be a space-level setting.
+      // TODO(burdon): Clone when activated. Copy-on-write for template.
+      const blueprints = yield* Effect.promise(() => db.query(Filter.type(Blueprint.Blueprint)).run());
+      let defaultAssistantBlueprint = blueprints.find((blueprint) => blueprint.key === AssistantBlueprint.key);
+      if (!defaultAssistantBlueprint) {
+        defaultAssistantBlueprint = db.add(AssistantBlueprint.make());
+      }
+      let defaultDatabaseBlueprint = blueprints.find((blueprint) => blueprint.key === DatabaseBlueprint.key);
+      if (!defaultDatabaseBlueprint) {
+        defaultDatabaseBlueprint = db.add(DatabaseBlueprint.make());
+      }
+      let defaultProjectWizardBlueprint = blueprints.find((blueprint) => blueprint.key === ProjectWizardBlueprint.key);
+      if (!defaultProjectWizardBlueprint) {
+        defaultProjectWizardBlueprint = db.add(ProjectWizardBlueprint.make());
+      }
+      // Dynamic import to avoid circular dependency with the barrel that also exports BlueprintManagerHandlers.
+      const { BlueprintManagerBlueprint } = yield* Effect.promise(() => import('@dxos/assistant-toolkit'));
+      let defaultBlueprintManagerBlueprint = blueprints.find(
+        (blueprint) => blueprint.key === BlueprintManagerBlueprint.key,
+      );
+      if (!defaultBlueprintManagerBlueprint) {
+        defaultBlueprintManagerBlueprint = db.add(BlueprintManagerBlueprint.make());
+      }
+
+      const feedServiceLayer = createFeedServiceLayer(space.queues);
+      const runtime = yield* Effect.runtime<Feed.FeedService>().pipe(Effect.provide(feedServiceLayer));
+      const binder = new AiContextBinder({ feed, runtime, registry });
+      yield* Effect.promise(() =>
+        binder.use((b: AiContextBinder) =>
+          b.bind({
+            blueprints: [
+              Ref.make(defaultAssistantBlueprint!),
+              Ref.make(defaultDatabaseBlueprint!),
+              Ref.make(defaultProjectWizardBlueprint!),
+              Ref.make(defaultBlueprintManagerBlueprint!),
+            ],
+          }),
+        ),
+      );
+
+      return { object: chat };
+    }),
+  ),
+);
+
+export default handler;

package/src/operations/definitions.ts

@@ -0,0 +1,127 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Schema from 'effect/Schema';
+
+import { AiService } from '@dxos/ai';
+import { Capability } from '@dxos/app-framework';
+import { Chat } from '@dxos/assistant-toolkit';
+import { Prompt } from '@dxos/blueprints';
+import { SpaceSchema } from '@dxos/client/echo';
+import { Collection, Database, DXN, Feed, Obj, Ref } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { meta } from '#meta';
+
+const ASSISTANT_OPERATION = `${meta.id}.operation`;
+
+export const OnCreateSpace = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.on-create-space`, name: 'On Create Space' },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    space: SpaceSchema,
+    rootCollection: Collection.Collection,
+  }),
+  output: Schema.Void,
+});
+
+export const CreateChat = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.create-chat`, name: 'Create Chat' },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    db: Database.Database,
+    name: Schema.optional(Schema.String),
+    /** If false, chat is created in-memory only and not added to space. Defaults to true. */
+    addToSpace: Schema.optional(Schema.Boolean),
+  }),
+  output: Schema.Struct({
+    object: Chat.Chat,
+  }),
+});
+
+export const UpdateChatName = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.update-chat-name`, name: 'Update Chat Name' },
+  services: [Database.Service, Feed.FeedService, AiService.AiService],
+  input: Schema.Struct({
+    chat: Chat.Chat,
+  }),
+  output: Schema.Void,
+});
+
+export const SetCurrentChat = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.set-current-chat`, name: 'Set Current Chat' },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    companionTo: Obj.Unknown,
+    chat: Chat.Chat.pipe(Schema.optional),
+  }),
+  output: Schema.Void,
+});
+
+export const RunPromptInNewChat = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.run-prompt-in-new-chat`, name: 'Run Prompt In New Chat' },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    db: Database.Database,
+    /** Context objects to bind to the new chat. */
+    objects: Schema.optional(Schema.Array(Obj.Unknown)),
+    /** Blueprint keys to look up and bind to the new chat. */
+    blueprints: Schema.optional(Schema.Array(Schema.String)),
+    /** Raw instructions or an existing Prompt object reference (e.g. from the prompts surface). */
+    prompt: Schema.Union(Schema.String, Ref.Ref(Prompt.Prompt)),
+    /**
+     * When true, skips opening the chat: runs the Agent prompt operation against the new chat via the compute runtime (traced).
+     */
+    background: Schema.optional(Schema.Boolean),
+  }),
+  output: Schema.Struct({
+    object: Chat.Chat,
+  }),
+});
+
+const NavigationTargetSchema = Schema.Struct({
+  path: Schema.String.annotations({ description: 'Navigation path to use with the Open operation.' }),
+  label: Schema.String.annotations({ description: 'Human-readable label.' }),
+  type: Schema.String.annotations({ description: 'Object type.' }),
+});
+
+export const ResolveNavigationTargets = Operation.make({
+  meta: {
+    key: `${ASSISTANT_OPERATION}.resolve-navigation-targets`,
+    name: 'Resolve navigation targets',
+    description:
+      'Resolve navigation targets within the application. The returned paths can be used with the Open operation. Without a query, returns pages that can be navigated to.',
+  },
+  input: Schema.Struct({
+    query: Schema.optional(
+      Schema.Struct({
+        dxn: DXN.Schema.pipe(Schema.optional),
+      }),
+    ),
+  }),
+  output: Schema.Struct({
+    targets: Schema.Array(NavigationTargetSchema),
+  }),
+  services: [Capability.Service],
+});
+
+export const EnsureCompanionChat = Operation.make({
+  meta: { key: `${ASSISTANT_OPERATION}.ensure-companion-chat`, name: 'Ensure Companion Chat' },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    db: Database.Database,
+    companionTo: Obj.Unknown,
+  }),
+  output: Schema.Struct({
+    chat: Chat.Chat,
+    /** Whether the returned chat was already persisted in the space. */
+    persisted: Schema.Boolean,
+  }),
+});
+
+export const BlueprintForm = Schema.Struct({
+  key: Schema.String,
+  name: Schema.String,
+  description: Schema.optional(Schema.String),
+});

package/src/operations/ensure-companion-chat.ts

@@ -0,0 +1,61 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Capabilities, Capability } from '@dxos/app-framework';
+import { Chat } from '@dxos/assistant-toolkit';
+import { Filter, Obj, Query } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { AssistantCapabilities } from '../types';
+import { CreateChat, EnsureCompanionChat, SetCurrentChat } from './definitions';
+
+const handler: Operation.WithHandler<typeof EnsureCompanionChat> = EnsureCompanionChat.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ db, companionTo }) {
+      const operationInvoker = yield* Capability.get(Capabilities.OperationInvoker);
+      const companionDxn = Obj.getDXN(companionTo).toString();
+
+      // 1. Look for an existing persisted companion chat in the space.
+      const existingChats = yield* Effect.promise(() =>
+        db.query(Query.select(Filter.id(companionTo.id)).targetOf(Chat.CompanionTo).source()).run(),
+      );
+      if (existingChats.length > 0) {
+        const chat = existingChats.at(-1) as Chat.Chat;
+        // Cache the persisted chat so the graph connector can resolve it immediately
+        // via the cache fallback, without waiting for AtomObj.make(ref) to hydrate.
+        yield* Capabilities.updateAtomValue(AssistantCapabilities.CompanionChatCache, (current) => ({
+          ...current,
+          [companionDxn]: chat,
+        }));
+        yield* Effect.promise(() => operationInvoker.invokePromise(SetCurrentChat, { companionTo, chat }));
+        return { chat, persisted: true };
+      }
+
+      // 2. Return cached transient chat for this companion if present.
+      const cache = yield* Capabilities.getAtomValue(AssistantCapabilities.CompanionChatCache);
+      const cached = cache[companionDxn] as Chat.Chat | undefined;
+      if (cached) {
+        return { chat: cached, persisted: false };
+      }
+
+      // 3. Create a new transient chat, cache it, and return it without persisting.
+      const { data } = yield* Effect.promise(() =>
+        operationInvoker.invokePromise(CreateChat, { db, addToSpace: false }),
+      );
+      if (!data?.object) {
+        return yield* Effect.fail(new Error('CreateChat did not return a chat object'));
+      }
+      const chat = data.object;
+      yield* Capabilities.updateAtomValue(AssistantCapabilities.CompanionChatCache, (current) => ({
+        ...current,
+        [companionDxn]: chat,
+      }));
+      return { chat, persisted: false };
+    }),
+  ),
+);
+
+export default handler;

package/src/operations/index.ts

@@ -0,0 +1,17 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { OperationHandlerSet } from '@dxos/operation';
+
+export * as AssistantOperation from './definitions';
+
+export const AssistantOperationHandlerSet = OperationHandlerSet.lazy(
+  () => import('./create-chat'),
+  () => import('./ensure-companion-chat'),
+  () => import('./on-create-space'),
+  () => import('./resolve-navigation-targets'),
+  () => import('./run-prompt-in-new-chat'),
+  () => import('./set-current-chat'),
+  () => import('./update-chat-name'),
+);

package/src/operations/on-create-space.ts

@@ -0,0 +1,24 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { AgentPrompt } from '@dxos/assistant-toolkit';
+import { Operation } from '@dxos/operation';
+
+import { CreateChat, OnCreateSpace } from './definitions';
+
+const handler: Operation.WithHandler<typeof OnCreateSpace> = OnCreateSpace.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ space }) {
+      // TODO(wittjosiah): Remove once function registry is avaiable.
+      space.db.add(Operation.serialize(AgentPrompt));
+
+      const { object: chat } = yield* Operation.invoke(CreateChat, { db: space.db });
+      space.db.add(chat);
+    }),
+  ),
+);
+
+export default handler;

package/src/operations/resolve-navigation-targets.ts

@@ -0,0 +1,26 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Capability } from '@dxos/app-framework';
+import { AppCapabilities } from '@dxos/app-toolkit';
+import { Operation } from '@dxos/operation';
+
+import { ResolveNavigationTargets } from './definitions';
+
+export default ResolveNavigationTargets.pipe(
+  Operation.withHandler(
+    Effect.fn(function* ({ query }) {
+      const capabilities = yield* Capability.Service;
+
+      // Delegate to contributed resolvers.
+      const resolvers = capabilities.getAll(AppCapabilities.NavigationTargetResolver);
+      const results = yield* Effect.forEach(resolvers, (resolver) =>
+        resolver(query as AppCapabilities.NavigationQuery).pipe(Effect.catchAll(() => Effect.succeed([]))),
+      );
+      return { targets: results.flat() };
+    }),
+  ),
+);

package/src/operations/run-prompt-in-new-chat.ts

@@ -0,0 +1,113 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Capabilities, Capability } from '@dxos/app-framework';
+import { getObjectPathFromObject, LayoutOperation } from '@dxos/app-toolkit';
+import { AiContextBinder } from '@dxos/assistant';
+import { AgentPrompt } from '@dxos/assistant-toolkit';
+import { Blueprint, Prompt, Template } from '@dxos/blueprints';
+import { Database, Feed, Filter, Obj, Ref } from '@dxos/echo';
+import { createFeedServiceLayer } from '@dxos/echo-db';
+import { invariant } from '@dxos/invariant';
+import { log } from '@dxos/log';
+import { Operation } from '@dxos/operation';
+import { AutomationCapabilities } from '@dxos/plugin-automation/types';
+import { ClientCapabilities } from '@dxos/plugin-client/types';
+import { Text } from '@dxos/schema';
+
+import { AssistantCapabilities } from '../types';
+import { CreateChat, RunPromptInNewChat } from './definitions';
+
+const handler: Operation.WithHandler<typeof RunPromptInNewChat> = RunPromptInNewChat.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(
+      function* ({ db, prompt, objects, blueprints, background }) {
+        const registry = yield* Capability.get(Capabilities.AtomRegistry);
+        const { object: chat } = yield* Operation.invoke(CreateChat, { db });
+
+        if ((objects && objects.length > 0) || (blueprints && blueprints.length > 0)) {
+          const feedTarget = chat.feed.target;
+          invariant(feedTarget, 'Chat feed not found.');
+          const client = yield* Capability.get(ClientCapabilities.Client);
+          const space = client.spaces.get(db.spaceId);
+          invariant(space, 'Space not found.');
+          const feedServiceLayer = createFeedServiceLayer(space.queues);
+          const runtime = yield* Effect.runtime<Feed.FeedService>().pipe(Effect.provide(feedServiceLayer));
+          const binder = new AiContextBinder({ feed: feedTarget, runtime, registry });
+          yield* Effect.promise(() =>
+            binder.use(async (b: AiContextBinder) => {
+              const bindingProps: Parameters<AiContextBinder['bind']>[0] = {};
+
+              if (objects && objects.length > 0) {
+                bindingProps.objects = objects.map((obj) => Ref.make(obj));
+              }
+
+              if (blueprints && blueprints.length > 0) {
+                const allBlueprints = await db.query(Filter.type(Blueprint.Blueprint)).run();
+                const matchedBlueprints = allBlueprints.filter(
+                  (blueprint) => blueprint.key && blueprints.includes(blueprint.key),
+                );
+                if (matchedBlueprints.length > 0) {
+                  bindingProps.blueprints = matchedBlueprints.map((blueprint) => Ref.make(blueprint));
+                }
+              }
+
+              await b.bind(bindingProps);
+            }),
+          );
+        }
+
+        if (background) {
+          const promptRef =
+            typeof prompt === 'string'
+              ? Ref.make(
+                  Prompt.make({
+                    instructions: prompt,
+                    blueprints: [],
+                    context: [],
+                  }),
+                )
+              : prompt;
+          yield* Database.flush();
+          const computeRuntime = yield* Capability.get(AutomationCapabilities.ComputeRuntime);
+          const runtime = yield* computeRuntime.getRuntime(db.spaceId).runtimeEffect;
+          yield* Operation.invoke(AgentPrompt, {
+            prompt: promptRef,
+            input: {},
+            chat: Ref.make(chat),
+          }).pipe(
+            Effect.provide(runtime),
+            Effect.catchAll((error) => {
+              log.catch(error);
+              return Effect.void;
+            }),
+          );
+          return { object: chat };
+        }
+
+        const chatPath = getObjectPathFromObject(chat);
+        const pendingPromptText =
+          typeof prompt === 'string'
+            ? prompt
+            : yield* Effect.gen(function* () {
+                const promptObj = yield* Effect.promise(() => prompt.load());
+                const source = yield* Effect.promise(() => promptObj.instructions.source.load());
+                invariant(Obj.instanceOf(Text.Text, source), 'Prompt template source must be Text.');
+                return Template.process(source.content ?? '');
+              });
+        yield* Capabilities.updateAtomValue(AssistantCapabilities.State, (current) => ({
+          ...current,
+          pendingPrompts: { ...current.pendingPrompts, [chatPath]: pendingPromptText },
+        }));
+        yield* Operation.invoke(LayoutOperation.Open, { subject: [chatPath] });
+        return { object: chat };
+      },
+      (effect, { db }) => effect.pipe(Effect.provide(Database.layer(db))),
+    ),
+  ),
+);
+
+export default handler;

package/src/operations/set-current-chat.ts

@@ -0,0 +1,50 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Capabilities, Capability } from '@dxos/app-framework';
+import { Obj } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { AssistantCapabilities } from '../types';
+import { CreateChat, SetCurrentChat } from './definitions';
+
+const handler: Operation.WithHandler<typeof SetCurrentChat> = SetCurrentChat.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ companionTo, chat }) {
+      const companionToId = Obj.getDXN(companionTo).toString();
+
+      if (chat) {
+        const chatId = Obj.getDXN(chat).toString();
+        yield* Capabilities.updateAtomValue(AssistantCapabilities.State, (current) => ({
+          ...current,
+          currentChat: { ...current.currentChat, [companionToId]: chatId },
+        }));
+      } else {
+        // New thread: create a transient chat and cache it so the graph connector
+        // can resolve it immediately via the cache fallback tier.
+        const operationInvoker = yield* Capability.get(Capabilities.OperationInvoker);
+        const db = Obj.getDatabase(companionTo);
+        if (db) {
+          const { data } = yield* Effect.promise(() =>
+            operationInvoker.invokePromise(CreateChat, { db, addToSpace: false }),
+          );
+          if (data?.object) {
+            yield* Capabilities.updateAtomValue(AssistantCapabilities.CompanionChatCache, (current) => ({
+              ...current,
+              [companionToId]: data.object,
+            }));
+          }
+        }
+        yield* Capabilities.updateAtomValue(AssistantCapabilities.State, (current) => ({
+          ...current,
+          currentChat: { ...current.currentChat, [companionToId]: undefined },
+        }));
+      }
+    }),
+  ),
+);
+
+export default handler;

package/src/operations/update-chat-name.ts

@@ -0,0 +1,61 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as LanguageModel from '@effect/ai/LanguageModel';
+import * as Prompt from '@effect/ai/Prompt';
+import * as Effect from 'effect/Effect';
+
+import { AiPreprocessor, AiService } from '@dxos/ai';
+import { Database, Feed, Filter, Obj } from '@dxos/echo';
+import { log } from '@dxos/log';
+import { Operation } from '@dxos/operation';
+import { Message } from '@dxos/types';
+import { trim } from '@dxos/util';
+
+import { UpdateChatName } from './definitions';
+
+const handler: Operation.WithHandler<typeof UpdateChatName> = UpdateChatName.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(
+      function* ({ chat }) {
+        log.info('updating chat name', { chat });
+
+        const feed = yield* Database.load(chat.feed);
+        const history = yield* Feed.runQuery(feed, Filter.type(Message.Message));
+
+        log.info('history', { history: history.length });
+
+        const system = trim`
+          It is extremely important that you respond only with the title and nothing else.
+          Do not use markdown or other formatting.
+          If you cannot do this effectively respond with "New Chat".
+
+          <example_reply>
+          Fishing Trip
+          </example_reply>
+        `;
+        const prompt = 'Suggest a name for this chat';
+
+        const historyPrompt = yield* AiPreprocessor.preprocessPrompt(history, {
+          system,
+          cacheControl: 'ephemeral',
+        });
+
+        const response = yield* LanguageModel.generateText({
+          prompt: Prompt.merge(historyPrompt, prompt),
+        });
+
+        const newName = response.text.replaceAll(/[^a-zA-Z0-9\s]/g, '').trim();
+
+        Obj.change(chat, (chat) => {
+          chat.name = newName;
+        });
+        log.info('chat name updated', { chat, newName: chat.name });
+      },
+      Effect.provide(AiService.model('@anthropic/claude-haiku-4-5')),
+    ),
+  ),
+);
+
+export default handler;

package/src/processor/index.ts

@@ -4,4 +4,3 @@
 
 export * from './presets';
 export * from './processor';
-export * from './update-name';