@dxos/plugin-assistant 0.8.4-main.f5c0578 → 0.8.4-main.fcfe5033a5

package/src/stories/components/GraphContainer.tsx

@@ -1,118 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Match } from 'effect';
-import React, { useCallback, useEffect, useRef, useState } from 'react';
-
-import { ResearchGraph } from '@dxos/assistant-testing';
-import { Filter, Query } from '@dxos/echo';
-import { D3ForceGraph, useGraphModel } from '@dxos/plugin-explorer';
-import { useQuery } from '@dxos/react-client/echo';
-import { IconButton, Toolbar } from '@dxos/react-ui';
-import { ChatEditor, type ChatEditorController, type ChatEditorProps } from '@dxos/react-ui-chat';
-import { SyntaxHighlighter } from '@dxos/react-ui-syntax-highlighter';
-import { mx } from '@dxos/react-ui-theme';
-
-import { type Expression, QueryParser, createFilter } from '../../parser';
-import { useFlush, useMatcherExtension } from '../hooks';
-
-import { type ComponentProps } from './types';
-
-export const GraphContainer = ({ space }: ComponentProps) => {
-  const [ast, setAst] = useState<Expression | undefined>();
-  const [filter, setFilter] = useState<Filter.Any>();
-  const [open, setOpen] = useState(false);
-
-  const [researchGraph] = useQuery(space, Query.type(ResearchGraph));
-  const queue = researchGraph?.queue.target;
-
-  const model = useGraphModel(space, undefined, undefined, queue);
-  useEffect(() => {
-    model?.setFilter(filter ?? Filter.everything());
-  }, [model, filter]);
-
-  const handleSubmit = useCallback<NonNullable<ChatEditorProps['onSubmit']>>(
-    (text) => {
-      try {
-        const parser = new QueryParser(text);
-        const ast = parser.parse();
-        setAst(ast);
-        const filter = createFilter(ast);
-        setFilter(filter);
-        setOpen(true);
-      } catch {
-        // TODO(mykola): Make hybrid search.
-        const filter = Filter.text(text, { type: 'vector' });
-        setFilter(filter);
-      }
-    },
-    [space],
-  );
-
-  return (
-    <div className={mx('relative grid h-full', open && 'grid-rows-[min-content_1fr]')}>
-      <SearchBar space={space} onSubmit={handleSubmit} />
-      <D3ForceGraph model={model} />
-
-      {/* TODO(burdon): Create component with context state for story. */}
-      {(open && (
-        <div className='absolute left-2 right-2 bottom-2 h-[8rem] flex overflow-hidden bg-baseSurface border border-subduedSeparator'>
-          <SyntaxHighlighter language='json' classNames='text-sm'>
-            {JSON.stringify({ ast, filter }, null, 2)}
-          </SyntaxHighlighter>
-          <div className='absolute bottom-1 right-1'>
-            <IconButton variant='ghost' icon='ph--x--regular' iconOnly label='Close' onClick={() => setOpen(false)} />
-          </div>
-        </div>
-      )) || (
-        <div className='absolute bottom-3 right-3'>
-          <IconButton
-            variant='ghost'
-            icon='ph--arrow-line-up--regular'
-            iconOnly
-            label='Open'
-            onClick={() => setOpen(true)}
-          />
-        </div>
-      )}
-    </div>
-  );
-};
-
-export const SearchBar = ({ space, onSubmit }: ComponentProps & Pick<ChatEditorProps, 'onSubmit'>) => {
-  const { state: flushState, handleFlush } = useFlush(space);
-  const extensions = useMatcherExtension(space);
-  const editorRef = useRef<ChatEditorController>(null);
-
-  return (
-    <Toolbar.Root classNames='density-coarse border-b border-subduedSeparator'>
-      <ChatEditor
-        ref={editorRef}
-        autoFocus
-        placeholder='Search'
-        extensions={extensions}
-        fireIfEmpty
-        onSubmit={onSubmit}
-      />
-      <Toolbar.IconButton
-        icon='ph--magnifying-glass--regular'
-        iconOnly
-        label='Search'
-        onClick={() => onSubmit?.(editorRef.current?.getText() ?? '')}
-      />
-      <Toolbar.IconButton
-        disabled={flushState === 'flushing'}
-        icon={Match.value(flushState).pipe(
-          Match.when('idle', () => 'ph--floppy-disk--regular'),
-          Match.when('flushing', () => 'ph--spinner--regular'),
-          Match.when('flushed', () => 'ph--check--regular'),
-          Match.exhaustive,
-        )}
-        iconOnly
-        label='flush'
-        onClick={handleFlush}
-      />
-    </Toolbar.Root>
-  );
-};

package/src/stories/components/LoggingContainer.tsx

@@ -1,29 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import React from 'react';
-
-import { Filter } from '@dxos/echo';
-import { useQuery } from '@dxos/react-client/echo';
-import { Toolbar } from '@dxos/react-ui';
-
-import { Timeline } from '../../components';
-import { useExecutionGraph } from '../../hooks';
-import { Assistant } from '../../types';
-
-import { type ComponentProps } from './types';
-
-export const LoggingContainer = ({ space }: ComponentProps) => {
-  const [chat] = useQuery(space, Filter.type(Assistant.Chat));
-  const { branches, commits } = useExecutionGraph(chat?.traceQueue);
-
-  return (
-    <div className='flex flex-col h-full'>
-      <Toolbar.Root classNames='density-coarse border-b border-subduedSeparator'>Timeline</Toolbar.Root>
-      <Timeline branches={branches} commits={commits} />
-    </div>
-  );
-};

package/src/stories/components/SurfaceContainer.tsx

@@ -1,62 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import React from 'react';
-
-import { Surface } from '@dxos/app-framework';
-import { AiContextBinder } from '@dxos/assistant';
-import { Filter, Obj, Ref, type Type } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { useQuery } from '@dxos/react-client/echo';
-import { useAsyncState } from '@dxos/react-ui';
-import { mx } from '@dxos/react-ui-theme';
-
-import { Assistant } from '../../types';
-
-import { type ComponentProps } from './types';
-
-/**
- * Shows the surface relating to the first bound object to the curent chat.
- */
-export const SurfaceContainer = ({ space, debug }: ComponentProps) => {
-  const chats = useQuery(space, Filter.type(Assistant.Chat));
-  const [objects] = useAsyncState<Type.Expando[]>(async () => {
-    if (!chats.length) {
-      return [];
-    }
-
-    // Get the latest chat (is this deterministic?)
-    const chat = chats[chats.length - 1];
-    const binder = new AiContextBinder(chat.queue.target!);
-    const refs = binder.objects.value;
-    const objects = await Ref.Array.loadAll(refs);
-    // TODO(burdon): Auto log meta for ECHO objects?
-    log('loaded', { objects: objects.map((obj) => ({ typename: Obj.getTypename(obj), id: obj.id })) });
-    return objects;
-  }, [chats]);
-
-  // TODO(burdon): Specify role hint to hide toolbar.
-  return (
-    <div className='flex flex-col bs-full overflow-y-auto divide-y divide-separator'>
-      {objects?.map((object) => (
-        <div key={object.id} className='group contents'>
-          {debug && (
-            <div
-              className={mx(
-                'flex gap-2 items-center text-xs justify-center',
-                'text-subdued group-first:border-none border-t border-subduedSeparator',
-              )}
-            >
-              <span>{Obj.getTypename(object)}</span>
-              <span>{object.id}</span>
-            </div>
-          )}
-          <Surface role='section' limit={1} data={{ subject: object }} />
-        </div>
-      ))}
-    </div>
-  );
-};

package/src/stories/components/TasksContainer.tsx

@@ -1,51 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import React from 'react';
-
-import { Filter, Obj } from '@dxos/echo';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { createDocAccessor, useQuery } from '@dxos/react-client/echo';
-import { Toolbar, useThemeContext } from '@dxos/react-ui';
-import {
-  Editor,
-  createBasicExtensions,
-  createDataExtensions,
-  createMarkdownExtensions,
-  createThemeExtensions,
-  outliner,
-} from '@dxos/react-ui-editor';
-
-import { type ComponentProps } from './types';
-
-export const TasksContainer = ({ space }: ComponentProps) => {
-  const { themeMode } = useThemeContext();
-  const [document] = useQuery(space, Filter.type(Markdown.Document));
-  if (!document?.content.target) {
-    return null;
-  }
-
-  return (
-    <>
-      <Toolbar.Root classNames='border-b border-subduedSeparator'>
-        <h2>{Obj.getLabel(document)}</h2>
-      </Toolbar.Root>
-      <Editor
-        id={document.id}
-        text={document.content.target}
-        classNames='h-full p-2 overflow-hidden'
-        extensions={[
-          // TODO(burdon): Create util.
-          createDataExtensions({ id: document.id, text: createDocAccessor(document.content.target, ['content']) }),
-          createBasicExtensions({ readOnly: false }),
-          createMarkdownExtensions({ themeMode }),
-          createThemeExtensions({ themeMode }),
-          outliner(),
-        ]}
-      />
-    </>
-  );
-};

package/src/stories/components/index.ts

@@ -1,12 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './types';
-
-export * from './BlueprintContainer';
-export * from './ChatContainer';
-export * from './GraphContainer';
-export * from './LoggingContainer';
-export * from './SurfaceContainer';
-export * from './TasksContainer';

package/src/stories/components/types.ts

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Space } from '@dxos/client/echo';
-
-export type ComponentProps = {
-  space: Space;
-  debug?: boolean;
-  onEvent?: (event: string) => void;
-};

package/src/stories/hooks/index.ts

@@ -1,6 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './useFlush';
-export * from './useMatcherExtension';

package/src/stories/hooks/useMatcherExtension.ts

@@ -1,45 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-// TODO(burdon): Move def to @dxos/echo.
-import { useMemo } from 'react';
-
-import { type Space } from '@dxos/client/echo';
-import { Type } from '@dxos/echo';
-import { type TypeaheadContext, matchCompletion, staticCompletion, typeahead } from '@dxos/react-ui-editor';
-
-export const useMatcherExtension = (space: Space) => {
-  return useMemo(() => {
-    const handleComplete = ({ line }: TypeaheadContext) => {
-      const words = line.split(/\s+/).filter(Boolean);
-      if (words.length > 0) {
-        const word = words.at(-1)!;
-
-        // Match type.
-        const match = word.match(/^type:(.+)/);
-        if (match) {
-          const part = match[1];
-          for (const schema of space?.db.graph.schemaRegistry.schemas ?? []) {
-            const typename = Type.getTypename(schema);
-            if (typename) {
-              const completion = matchCompletion(typename, part);
-              if (completion) {
-                return completion;
-              }
-            }
-          }
-        }
-
-        // Match static.
-        return staticCompletion(['type:', 'AND', 'OR', 'NOT'])({ line });
-      }
-    };
-
-    return [
-      typeahead({
-        onComplete: handleComplete,
-      }),
-    ];
-  }, [space]);
-};

package/src/stories/testing/data.ts

@@ -1,130 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Live, type Space } from '@dxos/client/echo';
-import { Obj, Relation, type Type } from '@dxos/echo';
-import { invariant } from '@dxos/invariant';
-import { DataType } from '@dxos/schema';
-
-export const testTypes = [DataType.Organization, DataType.Person, DataType.Employer, DataType.HasConnection];
-
-const organizations: (Type.Properties<DataType.Organization> & { id: string })[] = [
-  { id: 'dxos', name: 'DXOS', website: 'https://dxos.org' },
-  { id: 'socket_supply', name: 'Socket Supply', website: 'https://socketsupply.com' },
-  { id: 'ink_and_switch', name: 'Ink & Switch', website: 'https://inkandswitch.com' },
-  { id: 'effectful', name: 'Effectful', website: 'https://effect.co' },
-  { id: 'blueyard', name: 'Blue Yard', website: 'https://blueyard.com' },
-  { id: 'backed', name: 'Backed', website: 'https://backed.vc' },
-  { id: 'protocol_labs', name: 'Protocol Labs', website: 'https://protocol.ai' },
-  { id: 'newlab', name: 'Newlab', website: 'https://newlab.com' },
-  { id: 'google', name: 'Google', website: 'https://google.com' },
-  { id: 'microsoft', name: 'Microsoft', website: 'https://microsoft.com' },
-  { id: 'openai', name: 'OpenAI', website: 'https://openai.com' },
-  { id: 'anthropic', name: 'Anthropic', website: 'https://anthropic.com' },
-  { id: 'amazon', name: 'Amazon', website: 'https://amazon.com' },
-  { id: 'deshaw', name: 'D. E. Shaw & Co.', website: 'https://deshaw.com' },
-];
-
-const people: (Type.Properties<DataType.Person> & { id: string })[] = [
-  { id: 'rich_burdon', fullName: 'Rich Burdon' },
-  { id: 'josiah_witt', fullName: 'Josiah Witt' },
-  { id: 'dima_dmaretskyi', fullName: 'Dima Maretskyi' },
-  { id: 'chad_fowler', fullName: 'Chad Fowler' },
-  { id: 'ciaran_oleary', fullName: "Ciarán O'Leary" },
-  { id: 'jason_whitmire', fullName: 'Jason Whitmire' },
-  { id: 'alex_brunicki', fullName: 'Alex Brunicki' },
-  { id: 'andre_de_haes', fullName: 'Andre de Haes' },
-  { id: 'scott_cohen', fullName: 'Scott Cohen' },
-  { id: 'juan_benet', fullName: 'Juan Benet' },
-  { id: 'satya_nadella', fullName: 'Satya Nadella' },
-  { id: 'kevin_scott', fullName: 'Kevin Scott' },
-  { id: 'jeff_bezos', fullName: 'Jeff Bezos' },
-];
-
-const testObjects: Record<string, any[]> = {
-  [DataType.Organization.typename]: organizations,
-  [DataType.Person.typename]: people,
-};
-
-const testRelationships: Record<
-  string,
-  ({
-    source: string;
-    target: string;
-  } & Record<string, any>)[]
-> = {
-  [DataType.Employer.typename]: [
-    // prettier-ignore
-    { source: 'rich_burdon', target: 'dxos' },
-    { source: 'rich_burdon', target: 'google', active: false }, // TODO(burdon): Should not contribute to force.
-    { source: 'rich_burdon', target: 'deshaw', active: false },
-    { source: 'josiah_witt', target: 'dxos' },
-    { source: 'dima_dmaretskyi', target: 'dxos' },
-    { source: 'chad_fowler', target: 'blueyard' },
-    { source: 'chad_fowler', target: 'microsoft', active: false },
-    { source: 'ciaran_oleary', target: 'blueyard' },
-    { source: 'jason_whitmire', target: 'blueyard' },
-    { source: 'juan_benet', target: 'protocol_labs' },
-    { source: 'alex_brunicki', target: 'backed' },
-    { source: 'andre_de_haes', target: 'backed' },
-    { source: 'scott_cohen', target: 'newlab' },
-    { source: 'satya_nadella', target: 'microsoft' },
-    { source: 'kevin_scott', target: 'microsoft' },
-    { source: 'kevin_scott', target: 'google', active: false },
-    { source: 'jeff_bezos', target: 'amazon' },
-    { source: 'jeff_bezos', target: 'deshaw', active: false },
-  ],
-
-  // TODO(burdon): Limit graph view to selected relationship types.
-  [DataType.HasConnection.typename]: [
-    // prettier-ignore
-    { kind: 'partner', source: 'dxos', target: 'ink_and_switch' },
-    { kind: 'partner', source: 'dxos', target: 'effectful' },
-    { kind: 'partner', source: 'dxos', target: 'socket_supply' },
-
-    // prettier-ignore
-    { kind: 'investor', source: 'blueyard', target: 'dxos' },
-    { kind: 'investor', source: 'blueyard', target: 'protocol_labs' },
-    { kind: 'investor', source: 'protocol_labs', target: 'dxos' },
-    { kind: 'investor', source: 'backed', target: 'dxos' },
-    { kind: 'investor', source: 'newlab', target: 'dxos' },
-    { kind: 'investor', source: 'microsoft', target: 'openai' },
-    { kind: 'investor', source: 'google', target: 'anthropic' },
-    { kind: 'investor', source: 'amazon', target: 'anthropic' },
-  ],
-};
-
-export const addTestData = async (space: Space): Promise<void> => {
-  const objectMap = new Map<string, Live<any>>();
-
-  for (const [typename, objects] of Object.entries(testObjects)) {
-    const schema = space.db.graph.schemaRegistry.getSchema(typename);
-    invariant(schema, `Schema not found: ${typename}`);
-    for (const { id, ...data } of objects) {
-      const object = space.db.add(Obj.make(schema, data));
-      objectMap.set(id, object);
-    }
-  }
-
-  for (const [typename, relationships] of Object.entries(testRelationships)) {
-    const schema = space.db.graph.schemaRegistry.getSchema(typename);
-    invariant(schema, `Schema not found: ${typename}`);
-
-    for (const { source, target, ...data } of relationships) {
-      const sourceObject = objectMap.get(source);
-      const targetObject = objectMap.get(target);
-      invariant(sourceObject, `Source object not found: ${source}`);
-      invariant(targetObject, `Target object not found: ${target}`);
-
-      space.db.add(
-        Relation.make(schema, {
-          // TODO(burdon): Test source/target types match.
-          [Relation.Source]: sourceObject,
-          [Relation.Target]: targetObject,
-          ...data,
-        }),
-      );
-    }
-  }
-};

package/src/stories/testing/index.ts

@@ -1,6 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './data';
-export * from './testing';

package/src/stories/testing/testing.tsx

@@ -1,163 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import {
-  Capabilities,
-  Events,
-  IntentPlugin,
-  type Plugin,
-  SettingsPlugin,
-  contributes,
-  defineModule,
-  definePlugin,
-} from '@dxos/app-framework';
-import { withPluginManager } from '@dxos/app-framework/testing';
-import { AiContextBinder } from '@dxos/assistant';
-import {
-  DESIGN_BLUEPRINT,
-  PLANNING_BLUEPRINT,
-  RESEARCH_BLUEPRINT,
-  readDocument,
-  readTasks,
-  remoteServiceEndpoints,
-  research,
-  updateDocument,
-  updateTasks,
-} from '@dxos/assistant-testing';
-import { Blueprint } from '@dxos/blueprints';
-import { type Space } from '@dxos/client/echo';
-import { Obj, Ref } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { AttentionPlugin } from '@dxos/plugin-attention';
-import { ClientPlugin } from '@dxos/plugin-client';
-import { type ClientPluginOptions } from '@dxos/plugin-client/types';
-import { GraphPlugin } from '@dxos/plugin-graph';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { SpacePlugin } from '@dxos/plugin-space';
-import { Config } from '@dxos/react-client';
-import { type DataType } from '@dxos/schema';
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { AssistantPlugin } from '../../AssistantPlugin';
-import { Assistant } from '../../types';
-
-// TODO(burdon): Factor out.
-export const config = {
-  remote: new Config({
-    runtime: {
-      services: {
-        ai: {
-          // TODO(burdon): Normalize props ('url'?)
-          server: remoteServiceEndpoints.ai,
-        },
-        edge: {
-          url: remoteServiceEndpoints.edge,
-        },
-      },
-    },
-  }),
-  persistent: new Config({
-    runtime: {
-      client: {
-        storage: {
-          persistent: true,
-        },
-      },
-      services: {
-        ai: {
-          server: remoteServiceEndpoints.ai,
-        },
-        edge: {
-          url: remoteServiceEndpoints.edge,
-        },
-      },
-    },
-  }),
-};
-
-type DecoratorsProps = Omit<ClientPluginOptions, 'onClientInitialized' | 'onSpacesReady'> & {
-  plugins?: Plugin[];
-  accessTokens?: DataType.AccessToken[];
-  onInit?: (props: { space: Space; chat: Assistant.Chat; binder: AiContextBinder }) => Promise<void>;
-};
-
-/**
- * Create storybook decorators.
- */
-export const getDecorators = ({ types = [], plugins = [], accessTokens = [], onInit, ...props }: DecoratorsProps) => [
-  withPluginManager({
-    plugins: [
-      // System plugins.
-      AttentionPlugin(),
-      GraphPlugin(),
-      IntentPlugin(),
-      SettingsPlugin(),
-      SpacePlugin(),
-      ClientPlugin({
-        types: [Markdown.Document, Assistant.Chat, Blueprint.Blueprint, ...types],
-        onClientInitialized: async ({ client }) => {
-          log.info('onClientInitialized', { identity: client.halo.identity.get()?.did });
-          // Abort if already initialized.
-          if (client.halo.identity.get()) {
-            return;
-          }
-
-          await client.halo.createIdentity();
-          await client.spaces.waitUntilReady();
-
-          const space = client.spaces.default;
-          // TODO(burdon): Should not require this.
-          //  ERROR: invariant violation: Database was not initialized with root object.
-          // TODO(burdon): onSpacesReady is never called.
-          await space.waitUntilReady();
-
-          // Add tokens.
-          for (const accessToken of accessTokens) {
-            space.db.add(Obj.clone(accessToken));
-          }
-
-          // Create chat and queue.
-          const chat = space.db.add(
-            Obj.make(Assistant.Chat, {
-              queue: Ref.fromDXN(space.queues.create().dxn),
-              traceQueue: Ref.fromDXN(space.queues.create().dxn),
-            }),
-          );
-
-          await space.db.flush({ indexes: true });
-          await onInit?.({ space, chat, binder: new AiContextBinder(await chat.queue.load()) });
-          await space.db.flush({ indexes: true });
-        },
-        ...props,
-      }),
-
-      // User plugins.
-      AssistantPlugin(),
-
-      // Custom.
-      definePlugin({ id: 'example.com/plugin/testing', name: 'Testing' }, [
-        defineModule({
-          id: 'example.com/plugin/testing/module/testing',
-          activatesOn: Events.SetupArtifactDefinition,
-          activate: () => [
-            contributes(Capabilities.BlueprintDefinition, DESIGN_BLUEPRINT),
-            contributes(Capabilities.BlueprintDefinition, PLANNING_BLUEPRINT),
-            contributes(Capabilities.BlueprintDefinition, RESEARCH_BLUEPRINT),
-            contributes(Capabilities.Functions, [readDocument, updateDocument]),
-            contributes(Capabilities.Functions, [readTasks, updateTasks]),
-            contributes(Capabilities.Functions, [research]),
-          ],
-        }),
-      ]),
-
-      // Test-specific.
-      ...plugins,
-    ],
-  }),
-  withLayout({
-    fullscreen: true,
-    classNames: 'justify-center bg-deckSurface',
-  }),
-  withTheme,
-];

package/src/testing/test-functions.ts

@@ -1,16 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { DEFAULT_EDGE_MODEL } from '@dxos/ai';
-import { createSystemPrompt } from '@dxos/assistant';
-
-import { type AiChatProcessorOptions } from '../hooks';
-
-/**
- * @deprecated
- */
-export const createProcessorOptions = (): Pick<AiChatProcessorOptions, 'model' | 'system'> => ({
-  model: DEFAULT_EDGE_MODEL,
-  system: createSystemPrompt({}),
-});