@dxos/plugin-assistant 0.8.4-main.c85a9c8dae → 0.8.4-main.d9fc60f731

package/src/processor/processor.test.ts

@@ -1,73 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Tool from '@effect/ai/Tool';
-import * as Toolkit from '@effect/ai/Toolkit';
-import { describe, it } from '@effect/vitest';
-import * as Effect from 'effect/Effect';
-import * as Layer from 'effect/Layer';
-import * as Schema from 'effect/Schema';
-
-import { AiService } from '@dxos/ai';
-import { GenericToolkit } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-import { AiConversation, ToolExecutionServices } from '@dxos/assistant';
-import { acquireReleaseResource } from '@dxos/effect';
-import { TestHelpers } from '@dxos/effect/testing';
-import { CredentialsService, QueueService, TracingService } from '@dxos/functions';
-import { FunctionInvocationServiceLayerTestMocked, TestDatabaseLayer } from '@dxos/functions-runtime/testing';
-import { type Message } from '@dxos/types';
-
-import { AiChatProcessor, type AiChatServices } from './processor';
-
-const TestToolkit = Toolkit.make(
-  Tool.make('random', {
-    description: 'Random number generator',
-    parameters: {},
-    success: Schema.Number,
-  }),
-);
-
-// TODO(burdon): Explain structure.
-const TestServicesLayer = Layer.mergeAll(
-  TracingService.layerNoop,
-  AiServiceTestingPreset('direct'),
-  TestDatabaseLayer({
-    // indexing: { vector: true },
-    // types: [],
-  }),
-  // CredentialsService.configuredLayer([{ service: 'exa.ai', apiKey: EXA_API_KEY }]),
-  FunctionInvocationServiceLayerTestMocked({ functions: [] }).pipe(Layer.provideMerge(TracingService.layerNoop)),
-);
-
-const TestLayer: Layer.Layer<AiChatServices, never, never> = Layer.mergeAll(
-  AiService.model('@anthropic/claude-opus-4-0'),
-  ToolExecutionServices,
-  CredentialsService.layerFromDatabase(),
-).pipe(
-  Layer.provideMerge(GenericToolkit.providerLayer(GenericToolkit.make(TestToolkit, TestToolkit.toLayer({} as any)))),
-  Layer.provideMerge(TestServicesLayer),
-  Layer.orDie,
-);
-
-// TODO(burdon): Create actual test with mock LLM.
-describe('Chat processor', () => {
-  it.scoped(
-    'basic',
-    Effect.fn(
-      function* ({ expect }) {
-        const services = yield* Effect.runtime<AiChatServices>();
-        const queue = yield* QueueService.createQueue<Message.Message>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation({ queue }));
-        const processor = new AiChatProcessor(conversation, async () => services);
-        const result = yield* Effect.promise(() => processor.request({ message: 'Hello' }));
-        void processor.cancel();
-        expect(processor.active).to.be.false;
-        expect(result).to.exist;
-      },
-      Effect.provide(TestLayer),
-      TestHelpers.taggedTest('llm'),
-    ),
-  );
-});

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, Obj } from '@dxos/echo';
-import { Collection } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-
-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: Obj.Unknown,
-        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`);
-}