@opencode-ai/ai 0.0.0-next-16389 → 0.0.0-next-16396

package/dist/testing.js

@@ -0,0 +1,89 @@
+export * as TestLLM from "./testing";
+import { LLMClient } from "./route/client";
+import { LLMEvent, LLMResponse, } from "./schema";
+import { Context, Deferred, Effect, Latch, Layer, Queue, Scope, Stream } from "effect";
+export class Service extends Context.Service()("@opencode/ai/TestLLM") {
+}
+export const complete = (options, ...events) => [
+    LLMEvent.stepStart({ index: 0 }),
+    ...events,
+    LLMEvent.stepFinish({ index: 0, reason: options.reason, usage: options.usage }),
+    LLMEvent.finish({ reason: options.reason }),
+];
+export const stop = (...events) => complete({ reason: { normalized: "stop" } }, ...events);
+export const toolCalls = (...events) => complete({ reason: { normalized: "tool-calls" } }, ...events);
+const textEvents = (value, id) => [
+    LLMEvent.textStart({ id }),
+    LLMEvent.textDelta({ id, text: value }),
+    LLMEvent.textEnd({ id }),
+];
+export const text = (value, id) => stop(...textEvents(value, id));
+export const textWithUsage = (value, id, inputTokens) => complete({ reason: { normalized: "stop" }, usage: { inputTokens, nonCachedInputTokens: inputTokens } }, ...textEvents(value, id));
+export const tool = (id, name, input) => toolCalls(LLMEvent.toolCall({ id, name, input }));
+export const failAfter = (error, ...events) => Stream.fromIterable(events).pipe(Stream.concat(Stream.fail(error)));
+export const hangAfter = (...events) => Stream.concat(Stream.fromIterable(events), Stream.never);
+const toStream = (response) => (Stream.isStream(response) ? response : Stream.fromIterable(response));
+export const layer = (options = {}) => Layer.effect(Service, Effect.gen(function* () {
+    const requests = [];
+    const responses = [];
+    let started = Deferred.makeUnsafe();
+    let fallback = options.fallback;
+    let activeGate;
+    const wait = (count) => Effect.suspend(() => requests.length >= count ? Effect.void : Deferred.await(started).pipe(Effect.andThen(wait(count))));
+    const stream = ((request) => {
+        requests.push(options.transformRequest?.(request) ?? request);
+        const waiting = started;
+        started = Deferred.makeUnsafe();
+        Deferred.doneUnsafe(waiting, Effect.void);
+        const response = responses.shift() ?? fallback;
+        if (!response)
+            return Stream.die(new Error(`TestLLM has no response for request ${requests.length}`));
+        const streamed = toStream(response);
+        const gate = activeGate;
+        if (!gate)
+            return streamed;
+        return Stream.unwrap(Queue.offer(gate.started, undefined).pipe(Effect.andThen(gate.release.await), Effect.as(streamed)));
+    });
+    const client = LLMClient.Service.of({
+        prepare: () => Effect.die("TestLLM does not prepare provider-native requests"),
+        stream,
+        generate: (request) => stream(request).pipe(Stream.runFold(LLMResponse.empty, LLMResponse.reduce), Effect.flatMap((state) => {
+            const response = LLMResponse.complete(state);
+            if (response)
+                return Effect.succeed(response);
+            return Effect.die("TestLLM response ended without a terminal finish event");
+        })),
+    });
+    return Service.of({
+        requests,
+        push: (...input) => Effect.sync(() => {
+            responses.push(...input);
+        }),
+        always: (response) => Effect.sync(() => {
+            fallback = response;
+        }),
+        wait,
+        gate: Effect.gen(function* () {
+            const gate = {
+                started: yield* Effect.acquireRelease(Queue.unbounded(), Queue.shutdown),
+                release: yield* Latch.make(),
+            };
+            activeGate = gate;
+            const release = Effect.sync(() => {
+                if (activeGate === gate)
+                    activeGate = undefined;
+            }).pipe(Effect.andThen(gate.release.open), Effect.asVoid);
+            yield* Effect.addFinalizer(() => release);
+            return {
+                started: Queue.take(gate.started),
+                release,
+            };
+        }),
+        client,
+    });
+}));
+export const clientLayer = Layer.effect(LLMClient.Service, Effect.map(Service, (service) => service.client));
+export const push = (...responses) => Service.use((service) => service.push(...responses));
+export const always = (response) => Service.use((service) => service.always(response));
+export const wait = (count) => Service.use((service) => service.wait(count));
+export const gate = Service.use((service) => service.gate);

package/package.json

@@ -1,6 +1,6 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
-  "version": "0.0.0-next-16389",
+  "version": "0.0.0-next-16396",
   "name": "@opencode-ai/ai",
   "type": "module",
   "license": "MIT",
@@ -18,6 +18,10 @@
       "import": "./dist/index.js",
       "types": "./dist/index.d.ts"
     },
+    "./testing": {
+      "import": "./dist/testing.js",
+      "types": "./dist/testing.d.ts"
+    },
     "./*": {
       "import": "./dist/*.js",
       "types": "./dist/*.d.ts"
@@ -26,7 +30,7 @@
   "devDependencies": {
     "@clack/prompts": "1.0.0-alpha.1",
     "@effect/platform-node": "4.0.0-beta.101",
-    "@opencode-ai/http-recorder": "0.0.0-next-16389",
+    "@opencode-ai/http-recorder": "0.0.0-next-16396",
     "@tsconfig/bun": "1.0.9",
     "@types/bun": "1.3.13",
     "@typescript/native-preview": "7.0.0-dev.20251207.1",
@@ -35,7 +39,7 @@
   "dependencies": {
     "@smithy/eventstream-codec": "4.2.14",
     "@smithy/util-utf8": "4.2.2",
-    "@opencode-ai/schema": "0.0.0-next-16389",
+    "@opencode-ai/schema": "0.0.0-next-16396",
     "aws4fetch": "1.0.20",
     "effect": "4.0.0-beta.101",
     "google-auth-library": "10.5.0"