@dxos/plugin-assistant 0.8.4-main.3eb6e50203 → 0.8.4-main.3fbcb4aa9b

package/src/components/Typewriter/Typewriter.stories.tsx

@@ -1,83 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import type * as LanguageModel from '@effect/ai/LanguageModel';
-import type { Meta, StoryObj } from '@storybook/react-vite';
-import * as Layer from 'effect/Layer';
-import * as ManagedRuntime from 'effect/ManagedRuntime';
-import type * as Runtime from 'effect/Runtime';
-import React, { useEffect, useState } from 'react';
-
-import { AiService } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-
-import { Typewriter } from './Typewriter';
-
-const meta: Meta<typeof Typewriter> = {
-  title: 'plugins/plugin-assistant/Typewriter',
-  component: Typewriter,
-  parameters: {
-    layout: 'centered',
-  },
-};
-
-export default meta;
-
-type Story = StoryObj<typeof Typewriter>;
-
-const sampleText = `# The Future of Artificial Intelligence
-
-Artificial Intelligence (AI) has rapidly evolved from a theoretical concept to a transformative force reshaping industries, economies, and daily life. At its core, AI refers to the development of computer systems capable of performing tasks that typically require human intelligence, such as visual perception, speech recognition, decision-making, and language translation.
-
-## Machine Learning and Neural Networks
-
-The resurgence of AI in recent years is largely driven by advancements in machine learning (ML) and deep learning. Neural networks, inspired by the human brain's architecture, enable computers to learn from vast amounts of data. This has led to breakthroughs in areas like natural language processing (NLP), allowing models like GPT-4 to generate human-like text and code.
-
-## Ethical Considerations
-
-As AI becomes more autonomous, ethical concerns have moved to the forefront. Issues such as algorithmic bias, data privacy, and the potential displacement of jobs are critical topics of debate. Ensuring that AI systems are fair, transparent, and aligned with human values is essential for their sustainable development.
-
-## Conclusion
-
-The journey of AI is just beginning. With responsible innovation and global collaboration, AI has the potential to solve some of humanity's most pressing challenges, from climate change to personalized medicine.
-
-(try selecting some text to see the AI assistant, or wait for the proofreader to find the typo below)
-
-The quick brown fox jumps over teh lazy dog. i think this is cool.
-`;
-
-export const Default: Story = {
-  render: () => {
-    const [rtEffect, setRtEffect] = useState<Runtime.Runtime<LanguageModel.LanguageModel>>();
-    useEffect(() => {
-      let disposed = false;
-      const rt = ManagedRuntime.make(
-        AiService.model('@anthropic/claude-haiku-4-5').pipe(
-          Layer.provide(AiServiceTestingPreset('edge-remote')),
-          Layer.orDie,
-        ),
-      );
-      queueMicrotask(async () => {
-        if (disposed) return;
-        const x = await rt.runtime();
-        if (disposed) return;
-        setRtEffect(x);
-      });
-      return () => {
-        disposed = true;
-        void rt.dispose();
-      };
-    }, []);
-
-    if (!rtEffect) {
-      return <div>...Loading</div>;
-    }
-
-    return (
-      <div className='w-[800px] p-4'>
-        <Typewriter initialContent={sampleText} runtime={rtEffect} />
-      </div>
-    );
-  },
-};

package/src/components/Typewriter/Typewriter.tsx

@@ -1,54 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import { type Extension } from '@codemirror/state';
-import type * as LanguageModel from '@effect/ai/LanguageModel';
-import type * as Runtime from 'effect/Runtime';
-import React, { useMemo, useRef } from 'react';
-
-import { DensityProvider, ThemeProvider } from '@dxos/react-ui';
-import { Editor, type EditorController } from '@dxos/react-ui-editor';
-import {
-  createBasicExtensions,
-  createMarkdownExtensions,
-  createThemeExtensions,
-  decorateMarkdown,
-} from '@dxos/ui-editor';
-
-import { assistant } from './assistant-extension';
-
-export type TypewriterProps = {
-  initialContent?: string;
-  extensions?: Extension[];
-  runtime: Runtime.Runtime<LanguageModel.LanguageModel>;
-};
-
-export const Typewriter = ({ initialContent = '', extensions = [], runtime }: TypewriterProps) => {
-  const editorRef = useRef<EditorController>(null);
-  const extension = useMemo(
-    () => [
-      createBasicExtensions({ scrollPastEnd: true, search: true }),
-      createThemeExtensions({ syntaxHighlighting: true }),
-      createMarkdownExtensions(),
-      decorateMarkdown(),
-      assistant(runtime),
-      ...extensions,
-    ],
-    [runtime, extensions],
-  );
-
-  return (
-    <ThemeProvider>
-      <DensityProvider density='fine'>
-        <div className='is-full bs-full grid overflow-hidden'>
-          <Editor.Root ref={editorRef}>
-            <Editor.Viewport>
-              <Editor.Content initialValue={initialContent} extensions={extension} />
-            </Editor.Viewport>
-          </Editor.Root>
-        </div>
-      </DensityProvider>
-    </ThemeProvider>
-  );
-};

package/src/components/Typewriter/assistant-extension.tsx

@@ -1,144 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import { type Diagnostic, linter } from '@codemirror/lint';
-import { type Extension, StateField } from '@codemirror/state';
-import { EditorView, type Tooltip, showTooltip } from '@codemirror/view';
-import * as LanguageModel from '@effect/ai/LanguageModel';
-import * as Effect from 'effect/Effect';
-import * as Runtime from 'effect/Runtime';
-import React from 'react';
-import { createRoot } from 'react-dom/client';
-
-import { unwrapExit } from '@dxos/effect';
-import { log } from '@dxos/log';
-
-import { AssistantToolbar } from './AssistantToolbar';
-
-export const assistant = (runtime: Runtime.Runtime<LanguageModel.LanguageModel>): Extension[] => {
-  return [
-    assistantState(runtime),
-    assistantLinter(runtime),
-    EditorView.baseTheme({
-      '.cm-tooltip-assistant': {
-        backgroundColor: 'transparent !important',
-        border: 'none !important',
-        padding: '0 !important',
-      },
-    }),
-  ];
-};
-
-//
-// Toolbar State & Tooltip
-//
-
-const assistantState = (runtime: Runtime.Runtime<LanguageModel.LanguageModel>) =>
-  StateField.define<Tooltip | null>({
-    create: () => null,
-    update: (tooltip, tr): Tooltip | null => {
-      if (tr.selection || tr.docChanged) {
-        const { from, to } = tr.state.selection.main;
-        if (from === to) {
-          return null;
-        } // No selection
-
-        // Check if we should show tooltip
-        // We return a tooltip object
-        return {
-          pos: from,
-          above: false,
-          strictSide: true,
-          create: (view) => {
-            const dom = document.createElement('div');
-            dom.classList.add('cm-tooltip-assistant');
-            const root = createRoot(dom);
-
-            root.render(React.createElement(AssistantToolbar, { view, runtime, from, to }));
-
-            return {
-              dom,
-              destroy: () => {
-                // Delay unmount to allow animations if needed, but for now just cleanup
-                setTimeout(() => root.unmount(), 0);
-              },
-            };
-          },
-        };
-      }
-      return tooltip;
-    },
-    provide: (f) => showTooltip.from(f),
-  });
-
-//
-// Linter
-//
-
-const assistantLinter = (runtime: Runtime.Runtime<LanguageModel.LanguageModel>) =>
-  linter(async (view) => {
-    const text = view.state.doc.toString();
-    if (text.length < 10) {
-      return [];
-    }
-
-    // Run proofreading effect
-    const prompt = `Proofread the following text. Identify typos and simple grammar mistakes. Return ONLY a valid JSON array of objects with fields: "original" (string), "replacement" (string), "context" (string, 3-5 words around match). Text: \n\n${text}`;
-
-    // We mocked the model behavior in the story, but here we assume the model returns a string.
-    // In a real implementation we would parse the JSON.
-    // For this refactor, we just wire it up.
-
-    // NOTE: This runs on every doc view update that triggers linter. Linter is debounced by default (750ms).
-
-    try {
-      const result = unwrapExit(
-        await Effect.gen(function* () {
-          const generated = yield* LanguageModel.generateText({ prompt });
-          return generated.text;
-        }).pipe(Runtime.runPromiseExit(runtime)),
-      );
-
-      // Parse result - assuming simpler mock response format for now or we build a robust parser
-      // For the purpose of this task (UX demo), we can assume the mock returns a specific format or we parse it.
-      // Let's assume the mock returns a JSON string.
-
-      let suggestions: any[] = [];
-      try {
-        // Find JSON in response
-        const match = result.match(/\[.*\]/s);
-        if (match) {
-          suggestions = JSON.parse(match[0]);
-        }
-      } catch (e) {
-        // failed to parse
-      }
-
-      const diagnostics: Diagnostic[] = [];
-      for (const s of suggestions) {
-        // Naive finding of position
-        const idx = text.indexOf(s.original);
-        if (idx !== -1) {
-          diagnostics.push({
-            from: idx,
-            to: idx + s.original.length,
-            severity: 'info',
-            message: `Suggestion: ${s.replacement}`,
-            actions: [
-              {
-                name: 'Apply',
-                apply: (view, from, to) => {
-                  view.dispatch({ changes: { from, to, insert: s.replacement } });
-                },
-              },
-            ],
-          });
-        }
-      }
-      return diagnostics;
-    } catch (e) {
-      log.error('Proofreading failed', { error: e });
-      return [];
-    }
-  });

package/src/components/Typewriter/index.ts

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

package/src/functions/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { default as list } from './object-list';
-export { default as load } from './object-load';
-export { default as search } from './object-search';

package/src/functions/object-list.ts

@@ -1,49 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { Filter, Obj, Query } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-import { trim } from '@dxos/util';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-list',
-  name: 'List objects',
-  description:
-    // TODO(wittjosiah): Find a better way to prompt for looking up typenames before querying with them.
-    trim`
-      Lists the objects of the given type.
-      Check the get-schemas tool for available types before calling this function.
-    `,
-  inputSchema: Schema.Struct({
-    typename: Schema.String.annotations({
-      description: 'The typename of the objects to list.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    results: Schema.Array(
-      Schema.Struct({
-        dxn: Schema.String.annotations({ description: 'The DXN of the object.' }),
-        label: Schema.optional(Schema.String.annotations({ description: 'The label of the object.' })),
-      }),
-    ),
-  }),
-  handler: Effect.fn(function* ({ data: { typename } }) {
-    // TODO(wittjosiah): Typename query is not working for dynamic schemas.
-    const [schema] = yield* Database.runSchemaQuery({ typename });
-    const filter = schema ? Filter.type(schema) : Filter.typename(typename);
-
-    const objects = yield* Database.runQuery(Query.select(filter));
-    const results = objects.map((object) => ({
-      dxn: Obj.getDXN(object).toString(),
-      label: Obj.getLabel(object),
-    }));
-
-    return { results };
-  }),
-});

package/src/functions/object-load.ts

@@ -1,44 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-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 { Database, Entity, Type } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-load',
-  name: 'Load object',
-  description: 'Loads the object.',
-  inputSchema: Schema.Struct({
-    object: Type.Ref(Type.Obj).annotations({
-      description: 'The ID of the object.',
-    }),
-    typename: Schema.String.annotations({
-      description: 'The typename of the object.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    // TODO(wittjosiah): Type.Entity.Any
-    object: Schema.Any,
-  }),
-  handler: Effect.fn(function* ({ data: { object, typename } }) {
-    const loadedObject = yield* Database.load(object);
-    return yield* Function.pipe(
-      Option.fromNullable(loadedObject),
-      Option.flatMap((obj) => (Entity.getTypename(obj) === typename ? Option.some(obj) : Option.none())),
-      Option.match({
-        onNone: () => Effect.fail('Object not found.'),
-        onSome: (obj) => {
-          // log.info('object', { object: obj });
-          return Effect.succeed({ object: obj });
-        },
-      }),
-    );
-  }),
-});

package/src/functions/object-search.ts

@@ -1,46 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { Filter, Obj, Query } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-import { trim } from '@dxos/util';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-search',
-  name: 'Search objects',
-  description:
-    // TODO(wittjosiah): Find a better way to prompt for looking up typenames before querying with them.
-    trim`
-      Searches the objects using full-text search.
-    `,
-  inputSchema: Schema.Struct({
-    query: Schema.String.annotations({
-      description: 'The query to search for.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    results: Schema.Array(
-      Schema.Struct({
-        dxn: Schema.String.annotations({ description: 'The DXN of the object.' }),
-        label: Schema.optional(Schema.String.annotations({ description: 'The label of the object.' })),
-      }),
-    ),
-  }),
-  handler: Effect.fn(function* ({ data: { query } }) {
-    const objects = yield* Database.runQuery(
-      Query.select(Filter.text(query, { type: 'full-text' })).options({ allQueuesFromSpaces: true }),
-    );
-    const results = objects.map((object) => ({
-      dxn: Obj.getDXN(object).toString(),
-      label: Obj.getLabel(object),
-    }));
-
-    return { results };
-  }),
-});

package/src/processor/update-name.ts

@@ -1,58 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Exit from 'effect/Exit';
-import * as Fiber from 'effect/Fiber';
-import * as Runtime from 'effect/Runtime';
-
-import { AiService, DEFAULT_EDGE_MODEL, type ModelName } from '@dxos/ai';
-import { type AiConversation, AiSession } from '@dxos/assistant';
-import { type Chat } from '@dxos/assistant-toolkit';
-import { Obj } from '@dxos/echo';
-import { throwCause } from '@dxos/effect';
-import { trim } from '@dxos/util';
-
-import { type AiChatServices } from './processor';
-
-/**
- * Update the current chat's name.
- */
-// TODO(burdon): Convert this into a plugin tool.
-export const updateName = async (
-  runtime: Runtime.Runtime<AiChatServices>,
-  conversation: AiConversation,
-  chat: Chat.Chat,
-  model: ModelName = DEFAULT_EDGE_MODEL,
-): Promise<void> => {
-  const history = await conversation.getHistory();
-  const system = trim`
-    It is extremely important that you respond only with the title and nothing else.
-    If you cannot do this effectively respond with "New Chat".
-  `;
-  const prompt = 'Suggest a name for this chat';
-
-  const fiber = Effect.gen(this, function* () {
-    const session = new AiSession();
-    return yield* session.run({ system, prompt, history });
-  }).pipe(
-    Effect.provide(AiService.model(model)),
-    Effect.tap((messages) => {
-      // TODO(burdon): Parse response (should update via tool).
-      const message = messages.find((message) => message.sender.role === 'assistant');
-      const title = message?.blocks.find((block) => block._tag === 'text')?.text;
-      if (title) {
-        Obj.change(chat, (c) => {
-          c.name = title;
-        });
-      }
-    }),
-    Runtime.runFork(runtime), // Run in the background.
-  );
-
-  const response = await fiber.pipe(Fiber.join, Effect.runPromiseExit);
-  if (!Exit.isSuccess(response)) {
-    throwCause(response.cause);
-  }
-};

package/src/types/AssistantAction.ts

@@ -1,76 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Capability } from '@dxos/app-framework';
-import { Chat } from '@dxos/assistant-toolkit';
-import { SpaceSchema } from '@dxos/client/echo';
-import { Database, Type } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-import { Collection } from '@dxos/schema';
-
-import { meta } from '../meta';
-
-// Operations
-const ASSISTANT_OPERATION = `${meta.id}/operation`;
-
-export namespace AssistantOperation {
-  export const OnCreateSpace = Operation.make({
-    meta: { key: `${ASSISTANT_OPERATION}/on-create-space`, name: 'On Create Space' },
-    services: [Capability.Service],
-    schema: {
-      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],
-    schema: {
-      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: [Capability.Service],
-    schema: {
-      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],
-    schema: {
-      input: Schema.Struct({
-        companionTo: Type.Obj,
-        chat: Chat.Chat.pipe(Schema.optional),
-      }),
-      output: Schema.Void,
-    },
-  });
-
-  export const BlueprintForm = Schema.Struct({
-    key: Schema.String,
-    name: Schema.String,
-    description: Schema.optional(Schema.String),
-  });
-}

package/src/types/capabilities.ts

@@ -1,27 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Atom } from '@effect-atom/atom-react';
-import * as Schema from 'effect/Schema';
-
-import { Capability } from '@dxos/app-framework';
-
-import { meta } from '../meta';
-
-import { type Assistant } from './';
-
-export namespace AssistantCapabilities {
-  export const Settings = Capability.make<Atom.Writable<Assistant.Settings>>(`${meta.id}/capability/settings`);
-
-  export const StateSchema = Schema.mutable(
-    Schema.Struct({
-      /** Map of primary object dxn to current chat dxn. */
-      currentChat: Schema.Record({ key: Schema.String, value: Schema.UndefinedOr(Schema.String) }),
-    }),
-  );
-
-  export type AssistantState = Schema.Schema.Type<typeof StateSchema>;
-
-  export const State = Capability.make<Atom.Writable<AssistantState>>(`${meta.id}/capability/state`);
-}

package/src/types/defs.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export const LLM_PROVIDERS = ['edge', 'ollama', 'lmstudio'] as const;

package/src/types/events.ts

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ActivationEvent } from '@dxos/app-framework';
-
-import { meta } from '../meta';
-
-export namespace AssistantEvents {
-  export const SetupAiServiceProviders = ActivationEvent.make(`${meta.id}/event/setup-ai-service-providers`);
-}