@dxos/agent-runtime 0.0.0

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

@@ -0,0 +1,28 @@
+import * as Context from 'effect/Context';
+import * as Deferred from 'effect/Deferred';
+import * as Effect from 'effect/Effect';
+import * as Layer from 'effect/Layer';
+import { Organization } from '@dxos/types';
+export declare const getTestData: () => {
+    organizations: import("@dxos/echo/Obj").OfShape<Organization.Organization & import("@dxos/echo/Entity").OfKind<import("@dxos/echo/internal").EntityKind.Object>>[];
+    research: Record<string, string>;
+};
+export interface Task {
+    id: string;
+    state: 'pending' | 'completed' | 'interrupted';
+    website: string;
+    deferred: Deferred.Deferred<string>;
+}
+export interface Service {
+    getTasks: () => readonly Task[];
+    research: (website: string) => Effect.Effect<string>;
+    waitForTaskToAppear: () => Effect.Effect<void>;
+    completeOneTask: () => Effect.Effect<void>;
+    completeAllTasks: () => Effect.Effect<void>;
+}
+declare const ResearchService_base: Context.TagClass<ResearchService, "@dxos/functions-runtime/testing/ResearchService", Service>;
+export declare class ResearchService extends ResearchService_base {
+}
+export declare const layer: Layer.Layer<ResearchService, never, never>;
+export {};
+//# sourceMappingURL=ResearchService.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ResearchService.d.ts","sourceRoot":"","sources":["../../../../src/testing/ResearchService.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,OAAO,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,iBAAiB,CAAC;AAC5C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAItC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAM3C,eAAO,MAAM,WAAW;IACtB,aAAa;IAcb,QAAQ,EAqBH,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;CAC3B,CAAC;AAEH,MAAM,WAAW,IAAI;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,SAAS,GAAG,WAAW,GAAG,aAAa,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,OAAO;IACtB,QAAQ,EAAE,MAAM,SAAS,IAAI,EAAE,CAAC;IAChC,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACrD,mBAAmB,EAAE,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC/C,eAAe,EAAE,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC3C,gBAAgB,EAAE,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;CAC7C;;AAED,qBAAa,eAAgB,SAAQ,oBAGlC;CAAG;AAEN,eAAO,MAAM,KAAK,4CAiEjB,CAAC"}

package/dist/types/src/testing/assistant-test-layer.d.ts

@@ -0,0 +1,70 @@
+import { Registry as AtomRegistry } from '@effect-atom/atom';
+import * as LanguageModel from '@effect/ai/LanguageModel';
+import * as KeyValueStore from '@effect/platform/KeyValueStore';
+import * as Context from 'effect/Context';
+import * as Layer from 'effect/Layer';
+import { AiService, OpaqueToolkit } from '@dxos/ai';
+import { AgentService, Credential, Operation, OperationHandlerSet, Process, ServiceResolver, Skill, Trace } from '@dxos/compute';
+import { FeedTraceSink, ProcessManager, TriggerDispatcher, TriggerStateStore } from '@dxos/compute-runtime';
+import { Database, Registry, Type } from '@dxos/echo';
+import { type TestContextService } from '@dxos/effect/testing';
+import { DXN } from '@dxos/keys';
+import { AgentService as AgentServiceRuntime } from '../agent-service';
+interface TestLayerOptions {
+    aiServicePreset?: 'direct' | 'edge-local' | 'edge-remote' | 'ollama';
+    /**
+     * Overrides the AI service entirely (e.g. a scripted model for deterministic e2e tests).
+     * When set, `aiServicePreset` and `disableLlmMemoization` are ignored.
+     */
+    aiService?: Layer.Layer<AiService.AiService>;
+    model?: DXN.DXN;
+    /** Provider the model resolves through; defaults to `ollama` for the ollama preset, else `edge`. */
+    provider?: DXN.DXN;
+    operationHandlers?: OperationHandlerSet.OperationHandlerSet | OperationHandlerSet.OperationHandlerSet[];
+    toolkits?: OpaqueToolkit.OpaqueToolkit[];
+    types?: Type.AnyEntity[];
+    skills?: Skill.Skill[];
+    credentials?: Credential.ServiceCredential[];
+    /**
+     * Tracing configuration.
+     * - `'feed'` persists trace events to a FeedTraceSink (queryable from the database).
+     * @default 'noop'
+     */
+    tracing?: 'noop' | 'console' | 'pretty' | 'feed';
+    disableLlmMemoization?: boolean;
+    /**
+     * Patterns matching run-specific values (e.g. timestamps derived from TestClock) that should be
+     * canonicalized for memoized-conversation matching. Forwarded to `TestAiService`. Opt-in.
+     */
+    dynamicValuePatterns?: readonly RegExp[];
+    /**
+     * Options for the agent process (system prompt, tool backgrounding, delegation strategy, etc.).
+     * The model defaults to the resolved test-layer model when not set here.
+     */
+    agent?: AgentServiceRuntime.AgentServiceOptions;
+    /**
+     * Extra services to make available in the service resolver.
+     * Operations can depend on those services.
+     */
+    extraServices?: Layer.Layer<never, never, never>;
+}
+export type AssistantTestServices = LanguageModel.LanguageModel | Credential.CredentialsService | AgentService.AgentService | AiService.AiService | Database.Service | Registry.Service | OpaqueToolkit.OpaqueToolkitProvider | Operation.Service | ProcessManager.Service | Process.ProcessMonitorService | AtomRegistry.AtomRegistry | OperationHandlerSet.OperationHandlerProvider | KeyValueStore.KeyValueStore | ServiceResolver.ServiceResolver | Trace.TraceService | Trace.TraceSink | FeedTraceSink.FeedTraceSink;
+export declare const AssistantTestLayer: (options?: TestLayerOptions) => Layer.Layer<AssistantTestServices, never, TestContextService>;
+/** Late-bound reference to the {@link ProcessManager.Service}, filled once the manager is built. */
+interface ProcessManagerHolder {
+    current?: Context.Tag.Service<ProcessManager.Service>;
+}
+/**
+ * Service resolver for testing.
+ */
+export declare const AssistantTestServiceResolverLayer: ({ extraServices }: Pick<TestLayerOptions, 'extraServices'>, processManagerHolder: ProcessManagerHolder) => Layer.Layer<ServiceResolver.ServiceResolver, never, AiService.AiService | Credential.CredentialsService | OpaqueToolkit.OpaqueToolkitProvider | Registry.Service | Database.Service>;
+/**
+ * Only storage + registry.
+ */
+export declare const AssistantTestBaseLayer: ({ operationHandlers, toolkits, types, credentials, skills, }: Pick<TestLayerOptions, 'operationHandlers' | 'toolkits' | 'types' | 'skills' | 'tracing' | 'credentials'>) => Layer.Layer<AtomRegistry.AtomRegistry | Credential.CredentialsService | KeyValueStore.KeyValueStore | OpaqueToolkit.OpaqueToolkitProvider | OperationHandlerSet.OperationHandlerProvider | Registry.Service | Database.Service, never, never>;
+interface TestLayerWithTriggersOptions extends TestLayerOptions {
+}
+export type AssistantTestServicesWithTriggers = AssistantTestServices | TriggerDispatcher | TriggerStateStore;
+export declare const AssistantTestLayerWithTriggers: (options: TestLayerWithTriggersOptions) => Layer.Layer<AssistantTestServicesWithTriggers, never, TestContextService>;
+export {};
+//# sourceMappingURL=assistant-test-layer.d.ts.map

package/dist/types/src/testing/assistant-test-layer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"assistant-test-layer.d.ts","sourceRoot":"","sources":["../../../../src/testing/assistant-test-layer.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,KAAK,aAAa,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,aAAa,MAAM,gCAAgC,CAAC;AAEhE,OAAO,KAAK,OAAO,MAAM,gBAAgB,CAAC;AAE1C,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAGtC,OAAO,EAAE,SAAS,EAAE,aAAa,EAAY,MAAM,UAAU,CAAC;AAG9D,OAAO,EACL,YAAY,EACZ,UAAU,EAEV,SAAS,EACT,mBAAmB,EACnB,OAAO,EAEP,eAAe,EACf,KAAK,EACL,KAAK,EAEN,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,aAAa,EACb,cAAc,EAGd,iBAAiB,EACjB,iBAAiB,EAElB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,QAAQ,EAAQ,QAAQ,EAAO,IAAI,EAAE,MAAM,YAAY,CAAC;AAEjE,OAAO,EAAE,KAAK,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AAEjC,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAGvE,UAAU,gBAAgB;IACxB,eAAe,CAAC,EAAE,QAAQ,GAAG,YAAY,GAAG,aAAa,GAAG,QAAQ,CAAC;IAErE;;;OAGG;IACH,SAAS,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC7C,KAAK,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC;IAChB,oGAAoG;IACpG,QAAQ,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC;IACnB,iBAAiB,CAAC,EAAE,mBAAmB,CAAC,mBAAmB,GAAG,mBAAmB,CAAC,mBAAmB,EAAE,CAAC;IACxG,QAAQ,CAAC,EAAE,aAAa,CAAC,aAAa,EAAE,CAAC;IACzC,KAAK,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;IACvB,WAAW,CAAC,EAAE,UAAU,CAAC,iBAAiB,EAAE,CAAC;IAE7C;;;;OAIG;IACH,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,QAAQ,GAAG,MAAM,CAAC;IAEjD,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAEhC;;;OAGG;IACH,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAEzC;;;OAGG;IACH,KAAK,CAAC,EAAE,mBAAmB,CAAC,mBAAmB,CAAC;IAEhD;;;OAGG;IACH,aAAa,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;CAClD;AAED,MAAM,MAAM,qBAAqB,GAC7B,aAAa,CAAC,aAAa,GAC3B,UAAU,CAAC,kBAAkB,GAC7B,YAAY,CAAC,YAAY,GACzB,SAAS,CAAC,SAAS,GACnB,QAAQ,CAAC,OAAO,GAChB,QAAQ,CAAC,OAAO,GAChB,aAAa,CAAC,qBAAqB,GACnC,SAAS,CAAC,OAAO,GACjB,cAAc,CAAC,OAAO,GACtB,OAAO,CAAC,qBAAqB,GAC7B,YAAY,CAAC,YAAY,GACzB,mBAAmB,CAAC,wBAAwB,GAC5C,aAAa,CAAC,aAAa,GAC3B,eAAe,CAAC,eAAe,GAC/B,KAAK,CAAC,YAAY,GAClB,KAAK,CAAC,SAAS,GACf,aAAa,CAAC,aAAa,CAAC;AAEhC,eAAO,MAAM,kBAAkB,aACpB,gBAAgB,KACxB,KAAK,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,EAAE,kBAAkB,CAgD9D,CAAC;AAEF,oGAAoG;AACpG,UAAU,oBAAoB;IAC5B,OAAO,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;CACvD;AAUD;;GAEG;AACH,eAAO,MAAM,iCAAiC,sBACX,IAAI,CAAC,gBAAgB,EAAE,eAAe,CAAC,wBAClD,oBAAoB,yLAsCzC,CAAC;AAEJ;;GAEG;AACH,eAAO,MAAM,sBAAsB,iEAMhC,IAAI,CAAC,gBAAgB,EAAE,mBAAmB,GAAG,UAAU,GAAG,OAAO,GAAG,QAAQ,GAAG,SAAS,GAAG,aAAa,CAAC,kPA0C3G,CAAC;AAaF,UAAU,4BAA6B,SAAQ,gBAAgB;CAAG;AAElE,MAAM,MAAM,iCAAiC,GAAG,qBAAqB,GAAG,iBAAiB,GAAG,iBAAiB,CAAC;AAE9G,eAAO,MAAM,8BAA8B,YAChC,4BAA4B,KACpC,KAAK,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,EAAE,kBAAkB,CAOjE,CAAC"}

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

@@ -0,0 +1,4 @@
+export * from './assistant-test-layer';
+export * from './trace-pretty-print';
+export * as ResearchService from './ResearchService';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/testing/index.ts"],"names":[],"mappings":"AAIA,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,OAAO,KAAK,eAAe,MAAM,mBAAmB,CAAC"}

package/dist/types/src/testing/trace-pretty-print.d.ts

@@ -0,0 +1,16 @@
+import * as Effect from 'effect/Effect';
+import * as Layer from 'effect/Layer';
+import { Trace } from '@dxos/compute';
+import { Database } from '@dxos/echo';
+/**
+ * Pretty-prints assistant trace messages to stdout (CompleteBlock, process spawn/exit).
+ */
+export declare const prettyPrintTraceMessage: (message: Trace.Message) => void;
+/** {@link Trace.TraceSink} layer that mirrors {@link AssistantTestLayer}'s `tracing: 'pretty'` mode. */
+export declare const traceSinkPrettyLayer: () => Layer.Layer<Trace.TraceSink>;
+/**
+ * Subscribes to the space trace feed and pretty-prints newly appended messages.
+ * Returns an unsubscribe function. Requires ambient {@link Database.Service}.
+ */
+export declare const traceFeedPrettyPrintSubscription: Effect.Effect<import("@dxos/async").CleanupFn, never, Database.Service>;
+//# sourceMappingURL=trace-pretty-print.d.ts.map

package/dist/types/src/testing/trace-pretty-print.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-pretty-print.d.ts","sourceRoot":"","sources":["../../../../src/testing/trace-pretty-print.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAItC,OAAO,EAAW,KAAK,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,EAAE,QAAQ,EAAiB,MAAM,YAAY,CAAC;AAErD;;GAEG;AACH,eAAO,MAAM,uBAAuB,YAAa,KAAK,CAAC,OAAO,KAAG,IAYhE,CAAC;AAEF,wGAAwG;AACxG,eAAO,MAAM,oBAAoB,QAAO,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAG/D,CAAC;AAEL;;;GAGG;AACH,eAAO,MAAM,gCAAgC,yEAc3C,CAAC"}