@dxos/assistant 0.8.4-main.b97322e → 0.8.4-main.bcb3aa67d6

package/dist/types/src/session/session.d.ts

@@ -1,19 +1,56 @@
-import { AiLanguageModel, AiToolkit, type AiError, type AiResponse, type AiTool } from '@effect/ai';
-import { Context, Effect, type Schema } from 'effect';
-import { ToolExecutionService, ToolResolverService, type AgentStatus, type AiInputPreprocessingError, type AiToolNotFoundError } from '@dxos/ai';
-import { Event } from '@dxos/async';
+import type * as AiError from '@effect/ai/AiError';
+import * as LanguageModel from '@effect/ai/LanguageModel';
+import type * as Tool from '@effect/ai/Tool';
+import type * as Toolkit from '@effect/ai/Toolkit';
+import * as Effect from 'effect/Effect';
+import { type AiToolNotFoundError, type PromptPreprocessingError, type ToolExecutionService, type ToolResolverService } from '@dxos/ai';
 import { type Blueprint } from '@dxos/blueprints';
-import { ObjectVersion } from '@dxos/echo-db';
-import { type ObjectId } from '@dxos/echo-schema';
-import { DataType } from '@dxos/schema';
-import { AiAssistantError } from '../errors';
-export type AiSessionOptions = {};
-export type SessionRunOptions<Tools extends AiTool.Any> = {
+import { Obj } from '@dxos/echo';
+import { type FunctionInvocationService, Trace, TracingService } from '@dxos/functions';
+import { Message } from '@dxos/types';
+import { type AiAssistantError } from '../errors';
+import { GenerationObserver } from './observer';
+export type AiSessionRunError = AiError.AiError | PromptPreprocessingError | AiToolNotFoundError | AiAssistantError;
+export type AiSessionRunRequirements = LanguageModel.LanguageModel | ToolExecutionService | ToolResolverService | FunctionInvocationService | Trace.TraceService
+/**
+ * @deprecated Retained for backward compatibility with tool handlers that use TracingService.emitStatus().
+ *   New code should use Trace.TraceService instead.
+ */
+ | TracingService;
+export type AiSessionOptions = {
+    /**
+     * Summarize before executing the prompt if the existing history exceeds this threshold.
+     */
+    summarizationThreshold?: number;
+    observer?: GenerationObserver;
+    /**
+     * Callback for when a message is received from the user, model, or tool.
+     * This is useful for streaming the output to a queue.
+     */
+    onOutput?: (message: Message.Message) => Effect.Effect<void, never, never>;
+};
+export type AiSessionRunProps<Tools extends Record<string, Tool.Any>> = {
     prompt: string;
-    history: DataType.Message[];
-    systemPrompt?: string;
-    toolkit?: AiToolkit.AiToolkit<Tools>;
-    blueprints?: Blueprint.Blueprint[];
+    system?: string;
+    history?: Message.Message[];
+    objects?: Obj.Unknown[];
+    blueprints?: readonly Blueprint.Blueprint[];
+    toolkit?: Toolkit.WithHandler<Tools>;
+};
+export type AiSessionBeginProps = {
+    prompt: string;
+    system?: string;
+    history?: Message.Message[];
+    objects?: Obj.Unknown[];
+    blueprints?: readonly Blueprint.Blueprint[];
+};
+export type AiSessionTurnProps<Tools extends Record<string, Tool.Any>> = {
+    system: string;
+    toolkit?: Toolkit.WithHandler<Tools>;
+};
+export type AiSessionTurnResult = {
+    messages: Message.Message[];
+    done: boolean;
 };
 /**
  * Contains message history, tools, current context.
@@ -27,234 +64,49 @@ export type SessionRunOptions<Tools extends AiTool.Any> = {
  */
 export declare class AiSession {
     private readonly _options;
+    /** Prevents concurrent execution of session. */
     private readonly _semaphore;
-    /** Pending messages (incl. the current user request). */
-    private _pending;
-    /** Current streaming response. */
-    private _stream;
-    /** Prior history from queue. */
+    private readonly _observer;
+    private readonly _onOutput;
+    /** Message history from queue. */
     private _history;
+    /** Pending messages for this session (incl. the current prompt). */
+    private _pending;
+    private _started;
+    private _ended;
+    private _toolCalls;
+    constructor(_options?: AiSessionOptions);
+    get duration(): number;
+    get toolCalls(): number;
+    get pending(): readonly Message.Message[];
+    private _submitMessage;
+    getToolCalls: () => {
+        block: {
+            readonly name: string;
+            readonly _tag: "toolCall";
+            readonly pending?: boolean | undefined;
+            readonly toolCallId: string;
+            readonly input: string;
+            readonly providerExecuted: boolean;
+        };
+        message: Message.Message;
+    }[];
     /**
-     * New message.
-     */
-    readonly message: Event<import("@dxos/schema").Message>;
-    /**
-     * Complete block added to Message.
-     */
-    readonly block: Event<{
-        version: unknown;
-        _tag: "anchor";
-        pending?: boolean | undefined;
-        objectId: string;
-    } | {
-        name?: string | undefined;
-        _tag: "file";
-        pending?: boolean | undefined;
-        url: string;
-        mediaType?: string | undefined;
-    } | {
-        id?: string | undefined;
-        source?: {
-            type: "base64";
-            mediaType: string;
-            data: string;
-        } | {
-            type: "http";
-            url: string;
-        } | undefined;
-        _tag: "image";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "json";
-        pending?: boolean | undefined;
-        data: string;
-        disposition?: string | undefined;
-    } | {
-        text: string;
-        _tag: "proposal";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "reasoning";
-        pending?: boolean | undefined;
-        reasoningText?: string | undefined;
-        redactedText?: string | undefined;
-        signature?: string | undefined;
-    } | {
-        reference: import("@dxos/echo-schema").Ref<import("@dxos/echo/Type").Expando>;
-        _tag: "reference";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "select";
-        pending?: boolean | undefined;
-        options: readonly string[];
-    } | {
-        _tag: "status";
-        pending?: boolean | undefined;
-        statusText: string;
-    } | {
-        text: string;
-        _tag: "suggest";
-        pending?: boolean | undefined;
-    } | {
-        text: string;
-        _tag: "text";
-        pending?: boolean | undefined;
-        disposition?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        name: string;
-        input: unknown;
-        _tag: "toolCall";
-        pending?: boolean | undefined;
-        toolCallId: string;
-    } | {
-        _tag: "toolList";
-        pending?: boolean | undefined;
-    } | {
-        name: string;
-        _tag: "toolResult";
-        pending?: boolean | undefined;
-        toolCallId: string;
-        result: unknown;
-    } | {
-        text: string;
-        _tag: "transcript";
-        pending?: boolean | undefined;
-        started: string;
-    }>;
-    /**
-     * Update partial block (while streaming).
-     */
-    readonly update: Event<{
-        version: unknown;
-        _tag: "anchor";
-        pending?: boolean | undefined;
-        objectId: string;
-    } | {
-        name?: string | undefined;
-        _tag: "file";
-        pending?: boolean | undefined;
-        url: string;
-        mediaType?: string | undefined;
-    } | {
-        id?: string | undefined;
-        source?: {
-            type: "base64";
-            mediaType: string;
-            data: string;
-        } | {
-            type: "http";
-            url: string;
-        } | undefined;
-        _tag: "image";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "json";
-        pending?: boolean | undefined;
-        data: string;
-        disposition?: string | undefined;
-    } | {
-        text: string;
-        _tag: "proposal";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "reasoning";
-        pending?: boolean | undefined;
-        reasoningText?: string | undefined;
-        redactedText?: string | undefined;
-        signature?: string | undefined;
-    } | {
-        reference: import("@dxos/echo-schema").Ref<import("@dxos/echo/Type").Expando>;
-        _tag: "reference";
-        pending?: boolean | undefined;
-    } | {
-        _tag: "select";
-        pending?: boolean | undefined;
-        options: readonly string[];
-    } | {
-        _tag: "status";
-        pending?: boolean | undefined;
-        statusText: string;
-    } | {
-        text: string;
-        _tag: "suggest";
-        pending?: boolean | undefined;
-    } | {
-        text: string;
-        _tag: "text";
-        pending?: boolean | undefined;
-        disposition?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        name: string;
-        input: unknown;
-        _tag: "toolCall";
-        pending?: boolean | undefined;
-        toolCallId: string;
-    } | {
-        _tag: "toolList";
-        pending?: boolean | undefined;
-    } | {
-        name: string;
-        _tag: "toolResult";
-        pending?: boolean | undefined;
-        toolCallId: string;
-        result: unknown;
-    } | {
-        text: string;
-        _tag: "transcript";
-        pending?: boolean | undefined;
-        started: string;
-    }>;
-    /**
-     * Unparsed events from the underlying generation stream.
-     */
-    readonly streamEvent: Event<AiResponse.TextPart | AiResponse.ReasoningPart | AiResponse.RedactedReasoningPart | AiResponse.ToolCallPart | AiResponse.MetadataPart | AiResponse.FinishPart>;
-    /**
-     * User prompt or tool result.
-     */
-    readonly userMessage: Event<import("@dxos/schema").Message>;
-    /**
-     * Agent self-reporting its status.
-     * Triggered by the model.
+     * Initialize a session: set up history, perform summarization if needed, and submit the user prompt.
+     * Must be called before `runTurn()`.
      */
-    readonly statusReport: Event<AgentStatus>;
+    begin: ({ prompt, system, history, blueprints, objects, }: AiSessionBeginProps) => Effect.Effect<void, AiSessionRunError, AiSessionRunRequirements>;
     /**
-     * Emits when the session is done.
+     * Execute a single turn: one LLM generation followed by tool execution.
+     * The toolkit and system prompt can be updated between turns to reflect context changes (e.g. dynamically enabled blueprints).
      */
-    readonly done: Event<void>;
-    constructor(_options?: AiSessionOptions);
+    runAgentTurn: <Tools extends Record<string, Tool.Any>>({ system, toolkit, }: AiSessionTurnProps<Tools>) => Effect.Effect<AiSessionTurnResult, AiSessionRunError, AiSessionRunRequirements>;
+    runTools: <Tools extends Record<string, Tool.Any>>({ toolkit }: {
+        toolkit?: Toolkit.WithHandler<Tools>;
+    }) => Effect.Effect<void, AiError.AiError, TracingService | Trace.TraceService>;
     /**
-     * Runs the AI model loop interacting with tools and artifacts.
-     * @param options - The session options.
-     * @returns The messages generated by the session, including the user's prompt.
+     * Run a full conversation turn loop. Equivalent to calling `begin()` then `runTurn()` in a loop.
      */
-    run: <Tools extends AiTool.Any>(options: SessionRunOptions<Tools>) => Effect.Effect<DataType.Message[], AiAssistantError | AiInputPreprocessingError | AiToolNotFoundError | AiError.AiError, AiLanguageModel.AiLanguageModel | ToolResolverService | ToolExecutionService | AiTool.ToHandler<Tools>>;
-    runStructured<S extends Schema.Schema.AnyNoContext>(schema: S, options: SessionRunOptions<AiTool.Any>): Promise<Schema.Schema.Type<S>>;
-    private _formatUserPrompt;
-    abort(): void;
-}
-declare const ArtifactDiffResolver_base: Context.TagClass<ArtifactDiffResolver, "ArtifactDiffResolver", ArtifactDiffResolver.Service>;
-/**
- * Resolves artifact ids to their versions.
- * Used to give the model a sense of the changes to the artifacts made by users during the conversation.
- * The artifacts versions are pinned in the history, and whenever the artifact changes in-between assistant's steps,
- * a diff is inserted into the conversation.
- *
- * Can be optionally provided to the session run call.
- */
-export declare class ArtifactDiffResolver extends ArtifactDiffResolver_base {
-}
-export declare namespace ArtifactDiffResolver {
-    type Service = {
-        resolve: (artifacts: {
-            id: ObjectId;
-            lastVersion: ObjectVersion;
-        }[]) => Promise<Map<ObjectId, {
-            version: ObjectVersion;
-            diff?: string;
-        }>>;
-    };
+    run: <Tools extends Record<string, Tool.Any>>({ prompt, system: systemTemplate, history, objects, blueprints, toolkit, }: AiSessionRunProps<Tools>) => Effect.Effect<Message.Message[], AiSessionRunError, AiSessionRunRequirements>;
 }
-export {};
 //# sourceMappingURL=session.d.ts.map

package/dist/types/src/session/session.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../../src/session/session.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,KAAK,OAAO,EAAE,KAAK,UAAU,EAAE,KAAK,MAAM,EAAE,MAAM,YAAY,CAAC;AACpG,OAAO,EAAS,OAAO,EAAE,MAAM,EAAwB,KAAK,MAAM,EAAE,MAAM,QAAQ,CAAC;AAGnF,OAAO,EAKL,oBAAoB,EACpB,mBAAmB,EACnB,KAAK,WAAW,EAChB,KAAK,yBAAyB,EAC9B,KAAK,mBAAmB,EAEzB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,KAAK,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAGlD,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,KAAK,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,EAAE,QAAQ,EAAqB,MAAM,cAAc,CAAC;AAE3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAE7C,MAAM,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAElC,MAAM,MAAM,iBAAiB,CAAC,KAAK,SAAS,MAAM,CAAC,GAAG,IAAI;IACxD,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACrC,UAAU,CAAC,EAAE,SAAS,CAAC,SAAS,EAAE,CAAC;CACpC,CAAC;AAEF;;;;;;;;;GASG;AACH,qBAAa,SAAS;IAiDR,OAAO,CAAC,QAAQ,CAAC,QAAQ;IA/CrC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA2C;IAEtE,yDAAyD;IACzD,OAAO,CAAC,QAAQ,CAA0B;IAE1C,kCAAkC;IAClC,OAAO,CAAC,OAAO,CAA+B;IAE9C,gCAAgC;IAChC,OAAO,CAAC,QAAQ,CAA0B;IAE1C;;OAEG;IACH,SAAgB,OAAO,wCAAiC;IAExD;;OAEG;IACH,SAAgB,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAiC;IAEtD;;OAEG;IACH,SAAgB,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAiC;IAEvD;;OAEG;IACH,SAAgB,WAAW,uKAAgC;IAE3D;;OAEG;IACH,SAAgB,WAAW,wCAAiC;IAE5D;;;OAGG;IACH,SAAgB,YAAY,qBAA4B;IAExD;;OAEG;IACH,SAAgB,IAAI,cAAqB;gBAEZ,QAAQ,GAAE,gBAAqB;IAE5D;;;;OAIG;IAEH,GAAG,GAAI,KAAK,SAAS,MAAM,CAAC,GAAG,EAC7B,SAAS,iBAAiB,CAAC,KAAK,CAAC,KAChC,MAAM,CAAC,MAAM,CACd,QAAQ,CAAC,OAAO,EAAE,EAClB,gBAAgB,GAAG,yBAAyB,GAAG,mBAAmB,GAAG,OAAO,CAAC,OAAO,EACpF,eAAe,CAAC,eAAe,GAAG,mBAAmB,GAAG,oBAAoB,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CACvG,CA4F2E;IAEtE,aAAa,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,CAAC,YAAY,EACtD,MAAM,EAAE,CAAC,EACT,OAAO,EAAE,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,GACrC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAUjC,OAAO,CAAC,iBAAiB,CAwCpB;IAEL,KAAK,IAAI,IAAI;CAGd;;AA+BD;;;;;;;GAOG;AAEH,qBAAa,oBAAqB,SAAQ,yBAGvC;CAAG;AAEN,yBAAiB,oBAAoB,CAAC;IACpC,KAAY,OAAO,GAAG;QACpB,OAAO,EAAE,CAAC,SAAS,EAAE;YAAE,EAAE,EAAE,QAAQ,CAAC;YAAC,WAAW,EAAE,aAAa,CAAA;SAAE,EAAE,KAAK,OAAO,CAC7E,GAAG,CACD,QAAQ,EACR;YACE,OAAO,EAAE,aAAa,CAAC;YACvB,IAAI,CAAC,EAAE,MAAM,CAAC;SACf,CACF,CACF,CAAC;KACH,CAAC;CACH"}
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../../src/session/session.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,KAAK,OAAO,MAAM,oBAAoB,CAAC;AACnD,OAAO,KAAK,aAAa,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,KAAK,IAAI,MAAM,iBAAiB,CAAC;AAC7C,OAAO,KAAK,KAAK,OAAO,MAAM,oBAAoB,CAAC;AAEnD,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAIxC,OAAO,EAIL,KAAK,mBAAmB,EAExB,KAAK,wBAAwB,EAC7B,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EAEzB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,KAAK,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AACjC,OAAO,EAAE,KAAK,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAExF,OAAO,EAAgB,OAAO,EAAE,MAAM,aAAa,CAAC;AAEpD,OAAO,EAAE,KAAK,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAGlD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAIhD,MAAM,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,GAAG,wBAAwB,GAAG,mBAAmB,GAAG,gBAAgB,CAAC;AAEpH,MAAM,MAAM,wBAAwB,GAChC,aAAa,CAAC,aAAa,GAC3B,oBAAoB,GACpB,mBAAmB,GACnB,yBAAyB,GACzB,KAAK,CAAC,YAAY;AACpB;;;GAGG;GACD,cAAc,CAAC;AAEnB,MAAM,MAAM,gBAAgB,GAAG;IAC7B;;OAEG;IACH,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAGhC,QAAQ,CAAC,EAAE,kBAAkB,CAAC;IAC9B;;;OAGG;IACH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,KAAK,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;CAC5E,CAAC;AAEF,MAAM,MAAM,iBAAiB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI;IACtE,MAAM,EAAE,MAAM,CAAC;IAEf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC5B,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;IACxB,UAAU,CAAC,EAAE,SAAS,SAAS,CAAC,SAAS,EAAE,CAAC;IAC5C,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;CACtC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC5B,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;IACxB,UAAU,CAAC,EAAE,SAAS,SAAS,CAAC,SAAS,EAAE,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,kBAAkB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI;IACvE,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;CACtC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC5B,IAAI,EAAE,OAAO,CAAC;CACf,CAAC;AAEF;;;;;;;;;GASG;AAEH,qBAAa,SAAS;IAkBR,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAjBrC,gDAAgD;IAChD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA2C;IAEtE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAqB;IAC/C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAkE;IAE5F,kCAAkC;IAElC,OAAO,CAAC,QAAQ,CAAyB;IAEzC,oEAAoE;IACpE,OAAO,CAAC,QAAQ,CAAyB;IAEzC,OAAO,CAAC,QAAQ,CAAK;IACrB,OAAO,CAAC,MAAM,CAAK;IACnB,OAAO,CAAC,UAAU,CAAK;gBAEM,QAAQ,GAAE,gBAAqB;IAK5D,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,OAAO,IAAI,SAAS,OAAO,CAAC,OAAO,EAAE,CAExC;IAED,OAAO,CAAC,cAAc,CAajB;IAEL,YAAY;;;;;;;;;;QAOR;IAEJ;;;OAGG;IACH,KAAK,GAAI,mDAMN,mBAAmB,KAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,iBAAiB,EAAE,wBAAwB,CAAC,CAqBpF;IAEL;;;OAGG;IACH,YAAY,GAAI,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,sBAGrD,kBAAkB,CAAC,KAAK,CAAC,KAAG,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,iBAAiB,EAAE,wBAAwB,CAAC,CAuEzG;IAEL,QAAQ,GAAI,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,aAAa;QAAE,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;KAAE,+EA8BpG;IACL;;OAEG;IACH,GAAG,GAAI,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,4EAO5C,iBAAiB,CAAC,KAAK,CAAC,KAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,iBAAiB,EAAE,wBAAwB,CAAC,CAgB/B;CAC7E"}

package/dist/types/src/session/toolkit.d.ts

@@ -0,0 +1,18 @@
+import type * as Tool from '@effect/ai/Tool';
+import * as Toolkit from '@effect/ai/Toolkit';
+import * as Effect from 'effect/Effect';
+import { type AiToolNotFoundError, GenericToolkit, ToolExecutionService, ToolResolverService } from '@dxos/ai';
+import { type Blueprint } from '@dxos/blueprints';
+export type CreateToolkitProps = {
+    toolkit?: Toolkit.Any;
+    blueprints?: readonly Blueprint.Blueprint[];
+    /**
+     * Self-contained with handlers toolkits.
+     */
+    genericToolkits?: readonly GenericToolkit.GenericToolkit[];
+};
+/**
+ * Build a combined toolkit from the blueprint tools and the provided toolkit.
+ */
+export declare const createToolkit: ({ toolkit: toolkitProp, blueprints, genericToolkits, }: CreateToolkitProps) => Effect.Effect<Toolkit.WithHandler<any>, AiToolNotFoundError, ToolResolverService | ToolExecutionService | Tool.HandlersFor<any>>;
+//# sourceMappingURL=toolkit.d.ts.map

package/dist/types/src/session/toolkit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolkit.d.ts","sourceRoot":"","sources":["../../../../src/session/toolkit.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,KAAK,IAAI,MAAM,iBAAiB,CAAC;AAC7C,OAAO,KAAK,OAAO,MAAM,oBAAoB,CAAC;AAC9C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,KAAK,mBAAmB,EAAE,cAAc,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAC/G,OAAO,EAAE,KAAK,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAGlD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC;IACtB,UAAU,CAAC,EAAE,SAAS,SAAS,CAAC,SAAS,EAAE,CAAC;IAC5C;;OAEG;IACH,eAAe,CAAC,EAAE,SAAS,cAAc,CAAC,cAAc,EAAE,CAAC;CAC5D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,aAAa,GAAI,wDAI3B,kBAAkB,KAAG,MAAM,CAAC,MAAM,CACnC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,EACxB,mBAAmB,EACnB,mBAAmB,GAAG,oBAAoB,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAYhE,CAAC"}

package/dist/types/src/session/version-pin.d.ts

@@ -1,7 +1,7 @@
-import { Schema } from 'effect';
+import * as Schema from 'effect/Schema';
+import { type Obj } from '@dxos/echo';
 import { type ObjectVersion } from '@dxos/echo-db';
-import { type BaseEchoObject } from '@dxos/echo-schema';
-import { type ContentBlock } from '@dxos/schema';
+import { type ContentBlock } from '@dxos/types';
 declare const VersionPinSchema: Schema.Struct<{
     objectId: import("@dxos/keys").ObjectIdClass;
     version: Schema.Schema<ObjectVersion, ObjectVersion, never>;
@@ -14,7 +14,7 @@ export interface VersionPin extends Schema.Schema.Type<typeof VersionPinSchema>
 }
 export declare const VersionPin: typeof VersionPinSchema & {
     DISPOSITION: 'version-pin';
-    fromObject: (object: BaseEchoObject) => VersionPin;
+    fromObject: (object: Obj.Unknown) => VersionPin;
     createBlock: (pin: VersionPin) => ContentBlock.Any;
 };
 export {};

package/dist/types/src/session/version-pin.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"version-pin.d.ts","sourceRoot":"","sources":["../../../../src/session/version-pin.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,eAAe,CAAC;AAEnD,OAAO,EAAY,KAAK,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,cAAc,CAAC;AAKjD,QAAA,MAAM,gBAAgB;;;EAKpB,CAAC;AAEH;;;GAGG;AACH,MAAM,WAAW,UAAW,SAAQ,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,gBAAgB,CAAC;CAAG;AAElF,eAAO,MAAM,UAAU,EAAE,OAAO,gBAAgB,GAAG;IACjD,WAAW,EAAE,aAAa,CAAC;IAC3B,UAAU,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,UAAU,CAAC;IACnD,WAAW,EAAE,CAAC,GAAG,EAAE,UAAU,KAAK,YAAY,CAAC,GAAG,CAAC;CAiBpD,CAAC"}
+{"version":3,"file":"version-pin.d.ts","sourceRoot":"","sources":["../../../../src/session/version-pin.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,KAAK,GAAG,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,KAAK,aAAa,EAAc,MAAM,eAAe,CAAC;AAE/D,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,aAAa,CAAC;AAKhD,QAAA,MAAM,gBAAgB;;;EAKpB,CAAC;AAEH;;;GAGG;AACH,MAAM,WAAW,UAAW,SAAQ,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,gBAAgB,CAAC;CAAG;AAElF,eAAO,MAAM,UAAU,EAAE,OAAO,gBAAgB,GAAG;IACjD,WAAW,EAAE,aAAa,CAAC;IAC3B,UAAU,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,KAAK,UAAU,CAAC;IAChD,WAAW,EAAE,CAAC,GAAG,EAAE,UAAU,KAAK,YAAY,CAAC,GAAG,CAAC;CAiBpD,CAAC"}

package/dist/types/src/templates/index.d.ts

@@ -1,2 +1,2 @@
-export * from './prompt';
+export * from './system';
 //# sourceMappingURL=index.d.ts.map

package/dist/types/src/templates/system.d.ts

@@ -0,0 +1,17 @@
+import { type ModelCapabilities } from '@dxos/ai';
+import { Template } from '@dxos/blueprints';
+/**
+ * System template variables.
+ */
+export type SystemPromptOptions = ModelCapabilities & {
+    DATETIME: string;
+};
+/**
+ * Base system prompt.
+ * NOTE: This contains protocol instructions that can not be dynamically edited.
+ */
+export declare const createSystemPrompt: (options?: Partial<SystemPromptOptions>) => string;
+export declare const templates: {
+    system: Template.Template;
+};
+//# sourceMappingURL=system.d.ts.map

package/dist/types/src/templates/system.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"system.d.ts","sourceRoot":"","sources":["../../../../src/templates/system.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAK5C;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG,iBAAiB,GAAG;IACpD,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,kBAAkB,GAAI,UAAS,OAAO,CAAC,mBAAmB,CAAM,WAM1E,CAAC;AAOJ,eAAO,MAAM,SAAS;;CAKrB,CAAC"}

package/dist/types/src/testing/index.d.ts

@@ -1,3 +1,2 @@
-import type { SearchResponse } from 'exa-js';
-export declare const SEARCH_RESULTS: SearchResponse<any>[];
+export * from './layer';
 //# sourceMappingURL=index.d.ts.map

package/dist/types/src/testing/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/testing/index.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,QAAQ,CAAC;AAM7C,eAAO,MAAM,cAAc,EAAE,cAAc,CAAC,GAAG,CAAC,EAAgC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/testing/index.ts"],"names":[],"mappings":"AAIA,cAAc,SAAS,CAAC"}

package/dist/types/src/testing/layer.d.ts

@@ -0,0 +1,35 @@
+import { Registry } from '@effect-atom/atom';
+import * as Layer from 'effect/Layer';
+import { AiService, GenericToolkit, type ModelName, type ToolExecutionService, type ToolResolverService } from '@dxos/ai';
+import { Database, Feed, Type } from '@dxos/echo';
+import { CredentialsService, FunctionInvocationService, QueueService, type ServiceCredential, Trace, TracingService } from '@dxos/functions';
+import { Process, ProcessManager, ServiceResolver, TriggerDispatcher, TriggerStateStore } from '@dxos/functions-runtime';
+import { Blueprint } from '@dxos/blueprints';
+import { OperationHandlerSet, OperationRegistry } from '@dxos/operation';
+import * as LanguageModel from '@effect/ai/LanguageModel';
+import * as KeyValueStore from '@effect/platform/KeyValueStore';
+import { AgentService } from '../service';
+import type { TestContextService } from '@dxos/effect/testing';
+interface TestLayerOptions {
+    aiServicePreset?: 'direct' | 'edge-local' | 'edge-remote';
+    model?: ModelName;
+    operationHandlers?: OperationHandlerSet.OperationHandlerSet | OperationHandlerSet.OperationHandlerSet[];
+    toolkits?: GenericToolkit.GenericToolkit[];
+    types?: Type.AnyEntity[];
+    blueprints?: Blueprint.Blueprint[];
+    credentials?: ServiceCredential[];
+    tracing?: 'noop' | 'console' | 'pretty';
+    disableLlmMemoization?: boolean;
+    /**
+     * Core system prompt for the agent.
+     */
+    systemPrompt?: string;
+}
+export type AssistantTestServices = LanguageModel.LanguageModel | Feed.Service | CredentialsService | AgentService.AgentService | AiService.AiService | Database.Service | QueueService | Blueprint.RegistryService | OperationRegistry.Service | GenericToolkit.GenericToolkitProvider | ProcessManager.ProcessManagerService | Process.ProcessMonitorService | Registry.AtomRegistry | OperationHandlerSet.OperationHandlerProvider | KeyValueStore.KeyValueStore | ServiceResolver.ServiceResolver | TracingService | Trace.TraceService | Trace.TraceSink | ToolExecutionService | ToolResolverService | FunctionInvocationService;
+export declare const AssistantTestLayer: ({ aiServicePreset, model, operationHandlers, toolkits, types, credentials, tracing, disableLlmMemoization, blueprints, systemPrompt, }?: TestLayerOptions) => Layer.Layer<AssistantTestServices, never, TestContextService>;
+interface TestLayerWithTriggersOptions extends TestLayerOptions {
+}
+export type AssistantTestServicesWithTriggers = AssistantTestServices | TriggerDispatcher | TriggerStateStore;
+export declare const AssistantTestLayerWithTriggers: (options: TestLayerWithTriggersOptions) => Layer.Layer<AssistantTestServicesWithTriggers, never, TestContextService>;
+export {};
+//# sourceMappingURL=layer.d.ts.map

package/dist/types/src/testing/layer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"layer.d.ts","sourceRoot":"","sources":["../../../../src/testing/layer.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAGtC,OAAO,EACL,SAAS,EAET,cAAc,EACd,KAAK,SAAS,EACd,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EACzB,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,QAAQ,EAAO,IAAI,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvD,OAAO,EACL,kBAAkB,EAClB,yBAAyB,EACzB,YAAY,EACZ,KAAK,iBAAiB,EAEtB,KAAK,EACL,cAAc,EACf,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,OAAO,EACP,cAAc,EACd,eAAe,EAEf,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,yBAAyB,CAAC;AAIjC,OAAO,EAAE,SAAS,EAAU,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAa,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpF,OAAO,KAAK,aAAa,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,aAAa,MAAM,gCAAgC,CAAC;AAEhE,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAa/D,UAAU,gBAAgB;IACxB,eAAe,CAAC,EAAE,QAAQ,GAAG,YAAY,GAAG,aAAa,CAAC;IAC1D,KAAK,CAAC,EAAE,SAAS,CAAC;IAClB,iBAAiB,CAAC,EAAE,mBAAmB,CAAC,mBAAmB,GAAG,mBAAmB,CAAC,mBAAmB,EAAE,CAAC;IACxG,QAAQ,CAAC,EAAE,cAAc,CAAC,cAAc,EAAE,CAAC;IAC3C,KAAK,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,UAAU,CAAC,EAAE,SAAS,CAAC,SAAS,EAAE,CAAC;IACnC,WAAW,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAKlC,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,QAAQ,CAAC;IAExC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAEhC;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,qBAAqB,GAE7B,aAAa,CAAC,aAAa,GAC3B,IAAI,CAAC,OAAO,GACZ,kBAAkB,GAClB,YAAY,CAAC,YAAY,GACzB,SAAS,CAAC,SAAS,GACnB,QAAQ,CAAC,OAAO,GAChB,YAAY,GAEZ,SAAS,CAAC,eAAe,GACzB,iBAAiB,CAAC,OAAO,GACzB,cAAc,CAAC,sBAAsB,GAErC,cAAc,CAAC,qBAAqB,GACpC,OAAO,CAAC,qBAAqB,GAC7B,QAAQ,CAAC,YAAY,GAErB,mBAAmB,CAAC,wBAAwB,GAC5C,aAAa,CAAC,aAAa,GAC3B,eAAe,CAAC,eAAe,GAC/B,cAAc,GACd,KAAK,CAAC,YAAY,GAClB,KAAK,CAAC,SAAS,GAEf,oBAAoB,GACpB,mBAAmB,GACnB,yBAAyB,CAAC;AAE9B,eAAO,MAAM,kBAAkB,GAAI,yIAWhC,gBAAqB,KAAG,KAAK,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,EAAE,kBAAkB,CA6GtF,CAAC;AAEF,UAAU,4BAA6B,SAAQ,gBAAgB;CAAG;AAElE,MAAM,MAAM,iCAAiC,GAAG,qBAAqB,GAAG,iBAAiB,GAAG,iBAAiB,CAAC;AAE9G,eAAO,MAAM,8BAA8B,GACzC,SAAS,4BAA4B,KACpC,KAAK,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,EAAE,kBAAkB,CAOjE,CAAC"}