@dxos/plugin-assistant 0.8.3-staging.0fa589b → 0.8.4-main.c1de068

package/src/testing/test-blueprint.ts

@@ -1,58 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolRegistry } from '@dxos/ai';
-import { EXA_API_KEY } from '@dxos/ai/testing';
-import {
-  type BlueprintDefinition,
-  BlueprintParser,
-  createExaTool,
-  createGraphWriterTool,
-  createLocalSearchTool,
-} from '@dxos/assistant';
-import { type Space } from '@dxos/client/echo';
-import { type DXN } from '@dxos/keys';
-import { DataTypes } from '@dxos/schema';
-import { isNonNullable } from '@dxos/util';
-
-// TODO(dmaretskyi): make db available through services (same as function executor).
-// TODO(burdon): Can tools implement "aspects" so that variants can be used rather than an explicit reference?
-export const createToolRegistry = (space: Space, queueDxn?: DXN): ToolRegistry => {
-  return new ToolRegistry(
-    [
-      createExaTool({ apiKey: EXA_API_KEY }),
-      createLocalSearchTool(space.db, queueDxn && space.queues.get(queueDxn)),
-      queueDxn &&
-        createGraphWriterTool({
-          db: space.db,
-          queue: space.queues.get(queueDxn),
-          schema: DataTypes,
-          onDone: async (objects) => {
-            const queue = space.queues.get(queueDxn);
-            await queue.append(objects);
-          },
-        }),
-    ].filter(isNonNullable),
-  );
-};
-
-export const RESEARCH_BLUEPRINT_DEFINITION: BlueprintDefinition = {
-  steps: [
-    {
-      instructions: 'Research information and entities related to the selected objects.',
-      tools: ['search/web_search'],
-    },
-    {
-      instructions:
-        'Based on your research find matching entires that are already in the graph. Do exaustive research.',
-      tools: ['search/local_search'],
-    },
-    {
-      instructions: 'Add researched data to the graph. Make connections to existing objects.',
-      tools: ['search/local_search', 'graph/writer'],
-    },
-  ],
-};
-
-export const RESEARCH_BLUEPRINT = BlueprintParser.create().parse(RESEARCH_BLUEPRINT_DEFINITION);

package/src/types/chat.ts

@@ -1,32 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { Queue } from '@dxos/client/echo';
-import { Type } from '@dxos/echo';
-
-export const AIChatType = Schema.Struct({
-  id: Type.ObjectId,
-  name: Schema.optional(Schema.String),
-  queue: Type.Ref(Queue),
-}).pipe(
-  Type.Obj({
-    typename: 'dxos.org/type/AIChat',
-    version: '0.2.0',
-  }),
-);
-export interface AIChatType extends Schema.Schema.Type<typeof AIChatType> {}
-
-export const CompanionTo = Schema.Struct({
-  id: Type.ObjectId,
-}).pipe(
-  Type.Relation({
-    typename: 'dxos.org/relation/CompanionTo',
-    version: '0.1.0',
-    source: AIChatType,
-    target: Type.Expando,
-  }),
-);
-export interface CompanionTo extends Schema.Schema.Type<typeof CompanionTo> {}

package/src/types/template.ts

@@ -1,66 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { Type } from '@dxos/echo';
-import { LabelAnnotation } from '@dxos/echo-schema';
-
-// TODO(burdon): Change to Schema.Literal (and discriminated union).
-export enum TemplateInputType {
-  VALUE = 0,
-  PASS_THROUGH = 1,
-  RETRIEVER = 2,
-  FUNCTION = 3,
-  QUERY = 4,
-  RESOLVER = 5,
-  CONTEXT = 6,
-  SCHEMA = 7,
-}
-
-export const TemplateInputSchema = Schema.mutable(
-  Schema.Struct({
-    name: Schema.String,
-    type: Schema.optional(Schema.Enums(TemplateInputType)),
-    value: Schema.optional(Schema.String),
-  }),
-);
-
-export type TemplateInput = Schema.Schema.Type<typeof TemplateInputSchema>;
-
-export const TemplateKinds = ['always', 'schema-matching', 'automatically', 'manual'] as const;
-export type TemplateKind = (typeof TemplateKinds)[number];
-
-export const TemplateKindSchema = Schema.Union(
-  Schema.Struct({
-    include: Schema.Literal('always'),
-  }),
-  Schema.Struct({
-    include: Schema.Literal('schema-matching'),
-    typename: Schema.String,
-  }),
-  Schema.Struct({
-    include: Schema.Literal('automatically'),
-    description: Schema.String,
-  }),
-  Schema.Struct({
-    include: Schema.Literal('manual'),
-  }),
-);
-
-export type TemplateKindType = Schema.Schema.Type<typeof TemplateKindSchema>;
-export const TemplateType = Schema.Struct({
-  name: Schema.optional(Schema.String),
-  kind: Schema.mutable(TemplateKindSchema),
-  source: Schema.String,
-  inputs: Schema.optional(Schema.mutable(Schema.Array(TemplateInputSchema))),
-  command: Schema.optional(Schema.String),
-}).pipe(
-  LabelAnnotation.set(['name']),
-  Type.Obj({
-    typename: 'dxos.org/type/Template',
-    version: '0.1.0',
-  }),
-);
-export interface TemplateType extends Schema.Schema.Type<typeof TemplateType> {}

package/src/types/types.ts

@@ -1,48 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { Blueprint } from '@dxos/assistant';
-import { SpaceSchema } from '@dxos/react-client/echo';
-
-import { AIChatType } from './chat';
-import { ASSISTANT_PLUGIN } from '../meta';
-
-export namespace AssistantAction {
-  const ASSISTANT_ACTION = `${ASSISTANT_PLUGIN}/action`;
-
-  export class CreateChat extends Schema.TaggedClass<CreateChat>()(`${ASSISTANT_ACTION}/create-chat`, {
-    input: Schema.Struct({
-      space: SpaceSchema,
-      name: Schema.optional(Schema.String),
-    }),
-    output: Schema.Struct({
-      object: AIChatType,
-    }),
-  }) {}
-
-  export class CreateBlueprint extends Schema.TaggedClass<CreateBlueprint>()(`${ASSISTANT_ACTION}/create-blueprint`, {
-    input: Schema.Struct({
-      name: Schema.optional(Schema.String),
-    }),
-    output: Schema.Struct({
-      object: Blueprint,
-    }),
-  }) {}
-}
-
-export const LLM_PROVIDERS = ['edge', 'ollama', 'lmstudio'] as const;
-
-export const AssistantSettingsSchema = Schema.mutable(
-  Schema.Struct({
-    llmProvider: Schema.optional(Schema.Literal(...LLM_PROVIDERS)),
-    edgeModel: Schema.optional(Schema.String),
-    ollamaModel: Schema.optional(Schema.String),
-    lmstudioModel: Schema.optional(Schema.String),
-    customPrompts: Schema.optional(Schema.Boolean),
-  }),
-);
-
-export type AssistantSettingsProps = Schema.Schema.Type<typeof AssistantSettingsSchema>;