@dxos/assistant 0.7.4-df37807 → 0.7.5-feature-compute.4d9d99a

package/src/ai-service/schema.ts

@@ -1,260 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema as S } from '@effect/schema';
-import { type Brand } from 'effect';
-import { ulid } from 'ulidx';
-
-import { JsonSchemaType } from '@dxos/echo-schema';
-import { SpaceId } from '@dxos/keys';
-
-// TODO(dmaretskyi): Extract IDs to protocols.
-export const SpaceIdSchema: S.Schema<SpaceId, string> = S.String.pipe(S.filter(SpaceId.isValid));
-export const ObjectIdBrand: unique symbol = Symbol('@dxos/echo/ObjectId');
-const ObjectIdSchema = S.ULID.pipe(S.brand(ObjectIdBrand));
-
-export type ObjectId = typeof ObjectIdSchema.Type;
-export const ObjectId: S.SchemaClass<string & Brand.Brand<typeof ObjectIdBrand>, string> & { random(): ObjectId } =
-  class extends ObjectIdSchema {
-    static random(): ObjectId {
-      return ulid() as ObjectId;
-    }
-  };
-
-// TODO(dmaretskyi): Dedup this schema with the one in dxos/edge.
-
-export const Space = S.Struct({
-  id: SpaceIdSchema,
-});
-export interface Space extends S.Schema.Type<typeof Space> {}
-
-export const Thread = S.Struct({
-  id: ObjectId,
-  spaceId: SpaceIdSchema,
-});
-export interface Thread extends S.Schema.Type<typeof Thread> {}
-
-export const MessageRole = S.String.pipe(S.filter((role) => role === 'user' || role === 'assistant'));
-export type MessageRole = S.Schema.Type<typeof MessageRole>;
-
-//
-// Message union
-//
-
-export const MessageTextContentBlock = S.Struct({
-  type: S.Literal('text'),
-  text: S.String,
-}).pipe(S.mutable);
-export type MessageTextContentBlock = S.Schema.Type<typeof MessageTextContentBlock>;
-
-export const ImageSource = S.Struct({
-  type: S.Literal('base64'),
-  media_type: S.Literal('image/jpeg', 'image/png', 'image/gif', 'image/webp'),
-  data: S.String,
-}).pipe(S.mutable);
-export type ImageSource = S.Schema.Type<typeof ImageSource>;
-
-export const MessageImageContentBlock = S.Struct({
-  type: S.Literal('image'),
-  id: S.optional(S.String),
-  source: S.optional(ImageSource),
-}).pipe(S.mutable);
-export type MessageImageContentBlock = S.Schema.Type<typeof MessageImageContentBlock>;
-
-export const MessageToolUseContentBlock = S.Struct({
-  type: S.Literal('tool_use'),
-
-  /**
-   * Opaque service-defined ID of the tool invocation.
-   * Used to match the tool result block.
-   */
-  id: S.String,
-  /**
-   * Tool name.
-   */
-  name: S.String,
-  input: S.Unknown,
-
-  /**
-   * Used to accumulate the partial tool input JSON in streaming mode.
-   */
-  inputJson: S.optional(S.String),
-}).pipe(S.mutable);
-
-export const MessageToolResultContentBlock = S.Struct({
-  type: S.Literal('tool_result'),
-  toolUseId: S.String,
-  content: S.String,
-  isError: S.optional(S.Boolean),
-});
-
-export const MessageContentBlock = S.Union(
-  MessageTextContentBlock,
-  MessageImageContentBlock,
-  MessageToolUseContentBlock,
-  MessageToolResultContentBlock,
-);
-export type MessageContentBlock = S.Schema.Type<typeof MessageContentBlock>;
-
-//
-//
-//
-
-export const Message = S.Struct({
-  id: ObjectId,
-  threadId: S.optional(ObjectId),
-  spaceId: S.optional(SpaceIdSchema),
-
-  role: MessageRole,
-
-  content: S.Array(MessageContentBlock).pipe(S.mutable),
-
-  /**
-   * ID of the message from the foreign provider.
-   */
-  // TODO(dmaretskyi): Should be in meta/keys.
-  foreignId: S.optional(S.String),
-
-  // TODO(dmaretskyi): Figure out how to deal with those.
-  // created: S.optional(S.DateFromString),
-  // updated: S.optional(S.DateFromString),
-});
-
-export interface Message extends S.Schema.Type<typeof Message> {}
-
-/**
- * Message transformed from the database row.
- */
-export const MessageFromDb = S.Struct({
-  ...Message.fields,
-  content: S.propertySignature(Message.fields.content.pipe((schema) => S.parseJson(schema))).pipe(
-    S.fromKey('contentJson'),
-  ),
-});
-
-//
-//
-//
-
-export const LLMModel = S.Literal(
-  '@hf/nousresearch/hermes-2-pro-mistral-7b',
-  '@anthropic/claude-3-5-sonnet-20241022',
-  '@anthropic/claude-3-5-haiku-20241022',
-  '@ollama/llama-3-2-3b',
-  '@ollama/llama-3-1-nemotron-70b-instruct',
-  '@ollama/llama-3-1-nemotron-mini-4b-instruct',
-);
-
-export type LLMModel = S.Schema.Type<typeof LLMModel>;
-
-export const ToolTypes = Object.freeze({
-  // TODO(dmaretskyi): Not implemented yet.
-  // DatabaseQuery: 'database_query',
-
-  TextToImage: 'text_to_image',
-});
-
-export const LLMTool = S.Struct({
-  name: S.String,
-
-  /**
-   * If the tool is implemented by the service a type should be provided.
-   * For user-implemented tools, this field should be omitted.
-   * See {@link ToolTypes} for the list of supported types.
-   */
-  type: S.optional(S.String),
-
-  /**
-   * Required for user-implemented tools.
-   */
-  description: S.optional(S.String),
-
-  /**
-   * Input schema for the tool in the JSON Schema format.
-   * Required for user-implemented tools.
-   */
-  parameters: S.optional(JsonSchemaType),
-
-  /**
-   * Tool-specific options.
-   */
-  options: S.optional(S.Any),
-
-  /**
-   * Javascript function to execute the tool.
-   */
-  execute: S.optional(S.Any),
-});
-
-export interface LLMTool extends S.Schema.Type<typeof LLMTool> {}
-
-//
-//
-//
-
-export const GenerateRequest = S.Struct({
-  model: LLMModel,
-
-  spaceId: SpaceIdSchema,
-  threadId: ObjectId,
-  /**
-   * System instructions to the LLM.
-   */
-  // TODO(dmaretskyi): Should this be part of the thread?.
-  systemPrompt: S.optional(S.String),
-
-  /**
-   * Tools available for the LLM.
-   */
-  tools: S.Array(LLMTool).pipe(S.mutable),
-});
-
-export interface GenerateRequest extends S.Schema.Type<typeof GenerateRequest> {}
-
-export type LLMStopReason = 'tool_use' | 'end_turn';
-
-//
-//
-//
-
-// TODO(dmaretskyi): Effect schema.
-// TODO(dmaretskyi): Rename GenerationStreamEvent.
-export type ResultStreamEvent =
-  | {
-      // TODO(dmaretskyi): Normalize types to our schema.
-      type: 'message_start';
-      message: Omit<Message, 'spaceId' | 'threadId'>;
-    }
-  | {
-      type: 'message_delta';
-      delta: {
-        stopReason: LLMStopReason;
-      };
-    }
-  | {
-      type: 'message_stop';
-    }
-  | {
-      type: 'content_block_start';
-      index: number;
-      content: MessageContentBlock;
-    }
-  | {
-      type: 'content_block_delta';
-      index: number;
-      delta:
-        | {
-            type: 'text_delta';
-            text: string;
-          }
-        | {
-            partial_json: string;
-
-            type: 'input_json_delta';
-          };
-    }
-  | {
-      type: 'content_block_stop';
-      index: number;
-    };

package/src/sanity.test.ts

@@ -1,9 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { test } from 'vitest';
-
-test('sanity', ({ expect }) => {
-  expect(true).toBe(true);
-});