@orka-js/test 1.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Orka Team
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.cjs

@@ -0,0 +1,275 @@
+'use strict';
+
+var core = require('@orka-js/core');
+
+// src/mock-llm.ts
+var MockLLMAdapter = class {
+  name = "mock";
+  supportsStreaming = true;
+  responses;
+  calls = [];
+  defaultOutput;
+  constructor(responses = [], defaultOutput = "Mock response") {
+    this.responses = responses;
+    this.defaultOutput = defaultOutput;
+  }
+  findResponse(prompt) {
+    for (const response of this.responses) {
+      if (!response.when) return response;
+      if (typeof response.when === "string") {
+        if (prompt.includes(response.when)) return response;
+      } else if (response.when instanceof RegExp) {
+        if (response.when.test(prompt)) return response;
+      } else if (typeof response.when === "function") {
+        if (response.when(prompt)) return response;
+      }
+    }
+    return null;
+  }
+  getPromptText(prompt, options) {
+    const messagesText = options?.messages?.map((m) => typeof m.content === "string" ? m.content : JSON.stringify(m.content)).join(" ") ?? "";
+    return prompt + " " + messagesText;
+  }
+  async generate(prompt, options) {
+    const fullPrompt = this.getPromptText(prompt, options);
+    const response = this.findResponse(fullPrompt);
+    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });
+    if (response?.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, response.latencyMs));
+    }
+    if (response?.error) throw response.error;
+    const output = response?.output ?? this.defaultOutput;
+    return {
+      content: output,
+      usage: { promptTokens: 100, completionTokens: 50, totalTokens: 150 },
+      model: "mock-model",
+      finishReason: "stop",
+      cost: 0
+    };
+  }
+  async *stream(prompt, options) {
+    const fullPrompt = prompt + " " + (options?.messages?.map(
+      (m) => typeof m.content === "string" ? m.content : JSON.stringify(m.content)
+    ).join(" ") ?? "");
+    const response = this.findResponse(fullPrompt);
+    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });
+    if (response?.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, response.latencyMs));
+    }
+    if (response?.error) {
+      yield core.createStreamEvent("error", {
+        error: response.error,
+        message: response.error.message
+      });
+      return;
+    }
+    if (response?.toolCall) {
+      const callId = response.toolCall.id ?? `call_mock_${Date.now()}`;
+      yield core.createStreamEvent("tool_call", {
+        toolCallId: callId,
+        name: response.toolCall.name,
+        arguments: JSON.stringify(response.toolCall.args)
+      });
+      yield core.createStreamEvent("done", {
+        content: "",
+        finishReason: "tool_calls"
+      });
+      return;
+    }
+    const output = response?.output ?? this.defaultOutput;
+    const tokens = output.split(" ");
+    let index = 0;
+    for (const token of tokens) {
+      const tokenStr = index === 0 ? token : " " + token;
+      yield core.createStreamEvent("token", { token: tokenStr, index });
+      index++;
+    }
+    yield core.createStreamEvent("usage", {
+      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length }
+    });
+    yield core.createStreamEvent("done", {
+      content: output,
+      finishReason: "stop",
+      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },
+      cost: 0
+    });
+  }
+  async streamGenerate(prompt, options) {
+    let content = "";
+    const startTime = Date.now();
+    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+    for await (const event of this.stream(prompt, options)) {
+      if (event.type === "token") content += event.token;
+      if (event.type === "done") {
+        content = event.content || content;
+        if (event.usage) usage = event.usage;
+      }
+      if (event.type === "error") throw event.error;
+    }
+    return {
+      content,
+      usage,
+      model: "mock-model",
+      finishReason: "stop",
+      durationMs: Date.now() - startTime
+    };
+  }
+  async embed(text) {
+    const inputs = Array.isArray(text) ? text : [text];
+    return inputs.map(() => Array.from({ length: 1536 }, (_, i) => Math.sin(i * 0.1)));
+  }
+  async generateObject(schema, prompt, options) {
+    const result = await this.generate(prompt, options);
+    try {
+      return schema.parse(JSON.parse(result.content));
+    } catch {
+      return schema.parse({});
+    }
+  }
+  // ---- Assertion helpers ----
+  getCalls() {
+    return [...this.calls];
+  }
+  getCallCount() {
+    return this.calls.length;
+  }
+  wasCalledWith(pattern) {
+    return this.calls.some(
+      (c) => typeof pattern === "string" ? c.prompt.includes(pattern) : pattern.test(c.prompt)
+    );
+  }
+  getLastCall() {
+    return this.calls[this.calls.length - 1];
+  }
+  reset() {
+    this.calls = [];
+  }
+};
+function mockLLM(responses = [], defaultOutput) {
+  return new MockLLMAdapter(responses, defaultOutput);
+}
+
+// src/test-bed.ts
+var AgentTestBed = class {
+  agent;
+  llm;
+  constructor(config) {
+    this.agent = config.agent;
+    this.llm = config.llm ?? new MockLLMAdapter();
+  }
+  async run(input) {
+    this.llm.reset();
+    const events = [];
+    const toolCalls = [];
+    const streamAgent = this.agent;
+    let output = "";
+    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+    let steps = 0;
+    if (typeof streamAgent.runStream === "function") {
+      for await (const event of streamAgent.runStream(input)) {
+        events.push(event);
+        if (event.type === "tool_call") {
+          let args = {};
+          try {
+            args = JSON.parse(event.arguments);
+          } catch {
+            args = {};
+          }
+          toolCalls.push({ name: event.name, args, callId: event.toolCallId });
+          steps++;
+        }
+        if (event.type === "done") {
+          output = event.content;
+          if (event.usage) usage = event.usage;
+        }
+      }
+    } else {
+      const result2 = await this.agent.run(input);
+      output = result2.output;
+      steps = result2.steps.length;
+      usage = { promptTokens: 0, completionTokens: 0, totalTokens: result2.totalTokens };
+      for (const step of result2.steps) {
+        toolCalls.push({ name: step.action, args: step.actionInput, callId: "" });
+      }
+    }
+    const result = { output, toolCalls, steps, usage, events };
+    return Object.assign(result, makeAssertions(result));
+  }
+  getLLMMock() {
+    return this.llm;
+  }
+  reset() {
+    this.llm.reset();
+  }
+};
+function makeAssertions(result) {
+  return {
+    toHaveCalledTool(name) {
+      const called = result.toolCalls.some((tc) => tc.name === name);
+      if (!called) {
+        const calledTools = result.toolCalls.map((tc) => tc.name).join(", ") || "none";
+        throw new Error(
+          "Expected agent to have called tool " + JSON.stringify(name) + ", but it called: " + calledTools
+        );
+      }
+      return result;
+    },
+    toHaveOutput(pattern) {
+      const matches = typeof pattern === "string" ? result.output.includes(pattern) : pattern.test(result.output);
+      if (!matches) {
+        throw new Error(
+          "Expected output to match " + String(pattern) + ", got: " + JSON.stringify(result.output.slice(0, 200))
+        );
+      }
+      return result;
+    },
+    not: {
+      toHaveCalledTool(name) {
+        const called = result.toolCalls.some((tc) => tc.name === name);
+        if (called) throw new Error("Expected agent NOT to have called tool " + JSON.stringify(name) + ", but it did");
+        return result;
+      }
+    }
+  };
+}
+
+// src/matchers.ts
+function extendExpect(expect) {
+  expect.extend({
+    toHaveCalledTool(received, toolName) {
+      const called = received.toolCalls?.some((tc) => tc.name === toolName);
+      return {
+        pass: called,
+        message: () => called ? "Expected agent NOT to have called tool " + JSON.stringify(toolName) : "Expected agent to have called tool " + JSON.stringify(toolName) + ". Called tools: [" + (received.toolCalls?.map((t) => t.name).join(", ") || "none") + "]"
+      };
+    },
+    toHaveOutput(received, pattern) {
+      const matches = typeof pattern === "string" ? received.output?.includes(pattern) : pattern.test(received.output ?? "");
+      return {
+        pass: matches,
+        message: () => matches ? "Expected output NOT to match " + String(pattern) : "Expected output to match " + String(pattern) + ". Got: " + JSON.stringify(received.output?.slice(0, 200))
+      };
+    },
+    toHaveBeenCalledWithPrompt(received, pattern) {
+      const called = received.wasCalledWith(pattern);
+      return {
+        pass: called,
+        message: () => called ? "Expected MockLLM NOT to have been called with " + String(pattern) : "Expected MockLLM to have been called with " + String(pattern)
+      };
+    },
+    toHaveCallCount(received, count) {
+      const actual = received.getCallCount();
+      return {
+        pass: actual === count,
+        message: () => "Expected MockLLM call count to be " + count + ", got " + actual
+      };
+    }
+  });
+}
+
+exports.AgentTestBed = AgentTestBed;
+exports.MockLLMAdapter = MockLLMAdapter;
+exports.extendExpect = extendExpect;
+exports.mockLLM = mockLLM;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/mock-llm.ts","../src/test-bed.ts","../src/matchers.ts"],"names":["createStreamEvent","result"],"mappings":";;;;;AAYO,IAAM,iBAAN,MAAgE;AAAA,EAC5D,IAAA,GAAO,MAAA;AAAA,EACP,iBAAA,GAAoB,IAAA;AAAA,EACrB,SAAA;AAAA,EACA,QAAoB,EAAC;AAAA,EACrB,aAAA;AAAA,EAER,WAAA,CAAY,SAAA,GAA4B,EAAC,EAAG,gBAAgB,eAAA,EAAiB;AAC3E,IAAA,IAAA,CAAK,SAAA,GAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,GAAgB,aAAA;AAAA,EACvB;AAAA,EAEQ,aAAa,MAAA,EAAqC;AACxD,IAAA,KAAA,MAAW,QAAA,IAAY,KAAK,SAAA,EAAW;AACrC,MAAA,IAAI,CAAC,QAAA,CAAS,IAAA,EAAM,OAAO,QAAA;AAC3B,MAAA,IAAI,OAAO,QAAA,CAAS,IAAA,KAAS,QAAA,EAAU;AACrC,QAAA,IAAI,MAAA,CAAO,QAAA,CAAS,QAAA,CAAS,IAAI,GAAG,OAAO,QAAA;AAAA,MAC7C,CAAA,MAAA,IAAW,QAAA,CAAS,IAAA,YAAgB,MAAA,EAAQ;AAC1C,QAAA,IAAI,QAAA,CAAS,IAAA,CAAK,IAAA,CAAK,MAAM,GAAG,OAAO,QAAA;AAAA,MACzC,CAAA,MAAA,IAAW,OAAO,QAAA,CAAS,IAAA,KAAS,UAAA,EAAY;AAC9C,QAAA,IAAI,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA,EAAG,OAAO,QAAA;AAAA,MACpC;AAAA,IACF;AACA,IAAA,OAAO,IAAA;AAAA,EACT;AAAA,EAEQ,aAAA,CAAc,QAAgB,OAAA,EAAsC;AAE1E,IAAA,MAAM,eAAe,OAAA,EAAS,QAAA,EAC1B,IAAI,CAAA,CAAA,KAAM,OAAO,EAAE,OAAA,KAAY,QAAA,GAAW,EAAE,OAAA,GAAU,IAAA,CAAK,UAAU,CAAA,CAAE,OAAO,CAAE,CAAA,CACjF,IAAA,CAAK,GAAG,CAAA,IAAK,EAAA;AAChB,IAAA,OAAO,SAAS,GAAA,GAAM,YAAA;AAAA,EACxB;AAAA,EAEA,MAAM,QAAA,CAAS,MAAA,EAAgB,OAAA,EAAkD;AAC/E,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,MAAA,EAAQ,OAAO,CAAA;AACrD,IAAA,MAAM,QAAA,GAAW,IAAA,CAAK,YAAA,CAAa,UAAU,CAAA;AAE7C,IAAA,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,EAAG,QAAA,EAAU,CAAA;AAEhF,IAAA,IAAI,UAAU,SAAA,EAAW;AACvB,MAAA,MAAM,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,WAAW,OAAA,EAAS,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,QAAA,EAAU,KAAA,EAAO,MAAM,QAAA,CAAS,KAAA;AAEpC,IAAA,MAAM,MAAA,GAAS,QAAA,EAAU,MAAA,IAAU,IAAA,CAAK,aAAA;AAExC,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,MAAA;AAAA,MACT,OAAO,EAAE,YAAA,EAAc,KAAK,gBAAA,EAAkB,EAAA,EAAI,aAAa,GAAA,EAAI;AAAA,MACnE,KAAA,EAAO,YAAA;AAAA,MACP,YAAA,EAAc,MAAA;AAAA,MACd,IAAA,EAAM;AAAA,KACR;AAAA,EACF;AAAA,EAEA,OAAO,MAAA,CAAO,MAAA,EAAgB,OAAA,EAAgE;AAC5F,IAAA,MAAM,UAAA,GAAa,MAAA,GAAS,GAAA,IAAO,OAAA,EAAS,QAAA,EAAU,GAAA;AAAA,MAAI,CAAA,CAAA,KACxD,OAAO,CAAA,CAAE,OAAA,KAAY,QAAA,GAAW,EAAE,OAAA,GAAU,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,OAAO;AAAA,KACtE,CAAE,IAAA,CAAK,GAAG,CAAA,IAAK,EAAA,CAAA;AAEf,IAAA,MAAM,QAAA,GAAW,IAAA,CAAK,YAAA,CAAa,UAAU,CAAA;AAC7C,IAAA,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,EAAG,QAAA,EAAU,CAAA;AAEhF,IAAA,IAAI,UAAU,SAAA,EAAW;AACvB,MAAA,MAAM,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,WAAW,OAAA,EAAS,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,UAAU,KAAA,EAAO;AACnB,MAAA,MAAMA,uBAAkB,OAAA,EAAS;AAAA,QAC/B,OAAO,QAAA,CAAS,KAAA;AAAA,QAChB,OAAA,EAAS,SAAS,KAAA,CAAM;AAAA,OAChB,CAAA;AACV,MAAA;AAAA,IACF;AAGA,IAAA,IAAI,UAAU,QAAA,EAAU;AACtB,MAAA,MAAM,SAAS,QAAA,CAAS,QAAA,CAAS,MAAM,CAAA,UAAA,EAAa,IAAA,CAAK,KAAK,CAAA,CAAA;AAC9D,MAAA,MAAMA,uBAAkB,WAAA,EAAa;AAAA,QACnC,UAAA,EAAY,MAAA;AAAA,QACZ,IAAA,EAAM,SAAS,QAAA,CAAS,IAAA;AAAA,QACxB,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,SAAS,IAAI;AAAA,OACxC,CAAA;AACV,MAAA,MAAMA,uBAAkB,MAAA,EAAQ;AAAA,QAC9B,OAAA,EAAS,EAAA;AAAA,QACT,YAAA,EAAc;AAAA,OACN,CAAA;AACV,MAAA;AAAA,IACF;AAGA,IAAA,MAAM,MAAA,GAAS,QAAA,EAAU,MAAA,IAAU,IAAA,CAAK,aAAA;AACxC,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,KAAA,CAAM,GAAG,CAAA;AAC/B,IAAA,IAAI,KAAA,GAAQ,CAAA;AAEZ,IAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,MAAA,MAAM,QAAA,GAAW,KAAA,KAAU,CAAA,GAAI,KAAA,GAAQ,GAAA,GAAM,KAAA;AAC7C,MAAA,MAAMA,uBAAkB,OAAA,EAAS,EAAE,KAAA,EAAO,QAAA,EAAU,OAAgB,CAAA;AACpE,MAAA,KAAA,EAAA;AAAA,IACF;AAEA,IAAA,MAAMA,uBAAkB,OAAA,EAAS;AAAA,MAC/B,KAAA,EAAO,EAAE,YAAA,EAAc,GAAA,EAAK,gBAAA,EAAkB,OAAO,MAAA,EAAQ,WAAA,EAAa,GAAA,GAAM,MAAA,CAAO,MAAA;AAAO,KACtF,CAAA;AAEV,IAAA,MAAMA,uBAAkB,MAAA,EAAQ;AAAA,MAC9B,OAAA,EAAS,MAAA;AAAA,MACT,YAAA,EAAc,MAAA;AAAA,MACd,KAAA,EAAO,EAAE,YAAA,EAAc,GAAA,EAAK,gBAAA,EAAkB,OAAO,MAAA,EAAQ,WAAA,EAAa,GAAA,GAAM,MAAA,CAAO,MAAA,EAAO;AAAA,MAC9F,IAAA,EAAM;AAAA,KACE,CAAA;AAAA,EACZ;AAAA,EAEA,MAAM,cAAA,CAAe,MAAA,EAAgB,OAAA,EAAwD;AAC3F,IAAA,IAAI,OAAA,GAAU,EAAA;AACd,IAAA,MAAM,SAAA,GAAY,KAAK,GAAA,EAAI;AAC3B,IAAA,IAAI,QAAQ,EAAE,YAAA,EAAc,GAAG,gBAAA,EAAkB,CAAA,EAAG,aAAa,CAAA,EAAE;AAEnE,IAAA,WAAA,MAAiB,KAAA,IAAS,IAAA,CAAK,MAAA,CAAO,MAAA,EAAQ,OAAO,CAAA,EAAG;AACtD,MAAA,IAAI,KAAA,CAAM,IAAA,KAAS,OAAA,EAAS,OAAA,IAAW,KAAA,CAAM,KAAA;AAC7C,MAAA,IAAI,KAAA,CAAM,SAAS,MAAA,EAAQ;AACzB,QAAA,OAAA,GAAU,MAAM,OAAA,IAAW,OAAA;AAC3B,QAAA,IAAI,KAAA,CAAM,KAAA,EAAO,KAAA,GAAQ,KAAA,CAAM,KAAA;AAAA,MACjC;AACA,MAAA,IAAI,KAAA,CAAM,IAAA,KAAS,OAAA,EAAS,MAAM,KAAA,CAAM,KAAA;AAAA,IAC1C;AAEA,IAAA,OAAO;AAAA,MACL,OAAA;AAAA,MACA,KAAA;AAAA,MACA,KAAA,EAAO,YAAA;AAAA,MACP,YAAA,EAAc,MAAA;AAAA,MACd,UAAA,EAAY,IAAA,CAAK,GAAA,EAAI,GAAI;AAAA,KAC3B;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,IAAA,EAA8C;AACxD,IAAA,MAAM,SAAS,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,GAAI,IAAA,GAAO,CAAC,IAAI,CAAA;AAEjD,IAAA,OAAO,OAAO,GAAA,CAAI,MAAM,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAG,CAAC,GAAG,CAAA,KAAM,IAAA,CAAK,IAAI,CAAA,GAAI,GAAG,CAAC,CAAC,CAAA;AAAA,EACnF;AAAA,EAEA,MAAM,cAAA,CAAkB,MAAA,EAAkC,MAAA,EAAgB,OAAA,EAA0C;AAClH,IAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,QAAA,CAAS,QAAQ,OAAO,CAAA;AAClD,IAAA,IAAI;AACF,MAAA,OAAO,OAAO,KAAA,CAAM,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,OAAO,CAAC,CAAA;AAAA,IAChD,CAAA,CAAA,MAAQ;AACN,MAAA,OAAO,MAAA,CAAO,KAAA,CAAM,EAAE,CAAA;AAAA,IACxB;AAAA,EACF;AAAA;AAAA,EAIA,QAAA,GAAuB;AACrB,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,KAAK,CAAA;AAAA,EACvB;AAAA,EAEA,YAAA,GAAuB;AACrB,IAAA,OAAO,KAAK,KAAA,CAAM,MAAA;AAAA,EACpB;AAAA,EAEA,cAAc,OAAA,EAAmC;AAC/C,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA;AAAA,MAAK,CAAA,CAAA,KACrB,OAAO,OAAA,KAAY,QAAA,GAAW,CAAA,CAAE,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAAI,OAAA,CAAQ,IAAA,CAAK,CAAA,CAAE,MAAM;AAAA,KAClF;AAAA,EACF;AAAA,EAEA,WAAA,GAAoC;AAClC,IAAA,OAAO,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,KAAA,CAAM,SAAS,CAAC,CAAA;AAAA,EACzC;AAAA,EAEA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,QAAQ,EAAC;AAAA,EAChB;AACF;AAEO,SAAS,OAAA,CAAQ,SAAA,GAA4B,EAAC,EAAG,aAAA,EAAwC;AAC9F,EAAA,OAAO,IAAI,cAAA,CAAe,SAAA,EAAW,aAAa,CAAA;AACpD;;;ACtLO,IAAM,eAAN,MAAmB;AAAA,EAChB,KAAA;AAAA,EACA,GAAA;AAAA,EAER,YAAY,MAAA,EAA4B;AACtC,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,GAAA,GAAM,MAAA,CAAO,GAAA,IAAO,IAAI,cAAA,EAAe;AAAA,EAC9C;AAAA,EAEA,MAAM,IAAI,KAAA,EAA2D;AACnE,IAAA,IAAA,CAAK,IAAI,KAAA,EAAM;AACf,IAAA,MAAM,SAA2B,EAAC;AAClC,IAAA,MAAM,YAA0C,EAAC;AAGjD,IAAA,MAAM,cAAc,IAAA,CAAK,KAAA;AAEzB,IAAA,IAAI,MAAA,GAAS,EAAA;AACb,IAAA,IAAI,QAAQ,EAAE,YAAA,EAAc,GAAG,gBAAA,EAAkB,CAAA,EAAG,aAAa,CAAA,EAAE;AACnE,IAAA,IAAI,KAAA,GAAQ,CAAA;AAEZ,IAAA,IAAI,OAAO,WAAA,CAAY,SAAA,KAAc,UAAA,EAAY;AAC/C,MAAA,WAAA,MAAiB,KAAA,IAAS,WAAA,CAAY,SAAA,CAAU,KAAK,CAAA,EAAG;AACtD,QAAA,MAAA,CAAO,KAAK,KAAK,CAAA;AACjB,QAAA,IAAI,KAAA,CAAM,SAAS,WAAA,EAAa;AAC9B,UAAA,IAAI,OAAgC,EAAC;AACrC,UAAA,IAAI;AAAE,YAAA,IAAA,GAAO,IAAA,CAAK,KAAA,CAAM,KAAA,CAAM,SAAS,CAAA;AAAA,UAAG,CAAA,CAAA,MAAQ;AAAE,YAAA,IAAA,GAAO,EAAC;AAAA,UAAG;AAC/D,UAAA,SAAA,CAAU,IAAA,CAAK,EAAE,IAAA,EAAM,KAAA,CAAM,MAAM,IAAA,EAAM,MAAA,EAAQ,KAAA,CAAM,UAAA,EAAY,CAAA;AACnE,UAAA,KAAA,EAAA;AAAA,QACF;AACA,QAAA,IAAI,KAAA,CAAM,SAAS,MAAA,EAAQ;AACzB,UAAA,MAAA,GAAS,KAAA,CAAM,OAAA;AACf,UAAA,IAAI,KAAA,CAAM,KAAA,EAAO,KAAA,GAAQ,KAAA,CAAM,KAAA;AAAA,QACjC;AAAA,MACF;AAAA,IACF,CAAA,MAAO;AACL,MAAA,MAAMC,OAAAA,GAAsB,MAAM,IAAA,CAAK,KAAA,CAAM,IAAI,KAAK,CAAA;AACtD,MAAA,MAAA,GAASA,OAAAA,CAAO,MAAA;AAChB,MAAA,KAAA,GAAQA,QAAO,KAAA,CAAM,MAAA;AACrB,MAAA,KAAA,GAAQ,EAAE,YAAA,EAAc,CAAA,EAAG,kBAAkB,CAAA,EAAG,WAAA,EAAaA,QAAO,WAAA,EAAY;AAChF,MAAA,KAAA,MAAW,IAAA,IAAQA,QAAO,KAAA,EAAO;AAC/B,QAAA,SAAA,CAAU,IAAA,CAAK,EAAE,IAAA,EAAM,IAAA,CAAK,MAAA,EAAQ,MAAM,IAAA,CAAK,WAAA,EAAa,MAAA,EAAQ,EAAA,EAAI,CAAA;AAAA,MAC1E;AAAA,IACF;AAEA,IAAA,MAAM,SAA0B,EAAE,MAAA,EAAQ,SAAA,EAAW,KAAA,EAAO,OAAO,MAAA,EAAO;AAC1E,IAAA,OAAO,MAAA,CAAO,MAAA,CAAO,MAAA,EAAQ,cAAA,CAAe,MAAM,CAAC,CAAA;AAAA,EACrD;AAAA,EAEA,UAAA,GAA6B;AAC3B,IAAA,OAAO,IAAA,CAAK,GAAA;AAAA,EACd;AAAA,EAEA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,IAAI,KAAA,EAAM;AAAA,EACjB;AACF;AAEA,SAAS,eAAe,MAAA,EAA0C;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,IAAA,EAAc;AAC7B,MAAA,MAAM,SAAS,MAAA,CAAO,SAAA,CAAU,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,IAAI,CAAA;AAC3D,MAAA,IAAI,CAAC,MAAA,EAAQ;AACX,QAAA,MAAM,WAAA,GAAc,MAAA,CAAO,SAAA,CAAU,GAAA,CAAI,CAAA,EAAA,KAAM,GAAG,IAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA,IAAK,MAAA;AACtE,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,qCAAA,GAAwC,IAAA,CAAK,SAAA,CAAU,IAAI,IAAI,mBAAA,GAAsB;AAAA,SACvF;AAAA,MACF;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA,aAAa,OAAA,EAA0B;AACrC,MAAA,MAAM,OAAA,GAAU,OAAO,OAAA,KAAY,QAAA,GAC/B,MAAA,CAAO,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAC9B,OAAA,CAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,CAAA;AAC9B,MAAA,IAAI,CAAC,OAAA,EAAS;AACZ,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,2BAAA,GAA8B,MAAA,CAAO,OAAO,CAAA,GAAI,SAAA,GAAY,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC;AAAA,SACxG;AAAA,MACF;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA,GAAA,EAAK;AAAA,MACH,iBAAiB,IAAA,EAAc;AAC7B,QAAA,MAAM,SAAS,MAAA,CAAO,SAAA,CAAU,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,IAAI,CAAA;AAC3D,QAAA,IAAI,MAAA,QAAc,IAAI,KAAA,CAAM,4CAA4C,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,GAAI,cAAc,CAAA;AAC7G,QAAA,OAAO,MAAA;AAAA,MACT;AAAA;AACF,GACF;AACF;;;AClFO,SAAS,aAAa,MAAA,EAEpB;AACP,EAAA,MAAA,CAAO,MAAA,CAAO;AAAA,IACZ,gBAAA,CAAiB,UAA2B,QAAA,EAAkB;AAC5D,MAAA,MAAM,SAAS,QAAA,CAAS,SAAA,EAAW,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,QAAQ,CAAA;AAClE,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,OAAA,EAAS,MAAM,MAAA,GACX,yCAAA,GAA4C,IAAA,CAAK,UAAU,QAAQ,CAAA,GACnE,qCAAA,GAAwC,IAAA,CAAK,SAAA,CAAU,QAAQ,IAAI,mBAAA,IAAuB,QAAA,CAAS,SAAA,EAAW,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA,IAAK,MAAA,CAAA,GAAU;AAAA,OAC7J;AAAA,IACF,CAAA;AAAA,IAEA,YAAA,CAAa,UAA2B,OAAA,EAA0B;AAChE,MAAA,MAAM,OAAA,GAAU,OAAO,OAAA,KAAY,QAAA,GAC/B,QAAA,CAAS,MAAA,EAAQ,QAAA,CAAS,OAAO,CAAA,GACjC,OAAA,CAAQ,IAAA,CAAK,QAAA,CAAS,UAAU,EAAE,CAAA;AACtC,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,OAAA;AAAA,QACN,SAAS,MAAM,OAAA,GACX,kCAAkC,MAAA,CAAO,OAAO,IAChD,2BAAA,GAA8B,MAAA,CAAO,OAAO,CAAA,GAAI,SAAA,GAAY,KAAK,SAAA,CAAU,QAAA,CAAS,QAAQ,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC;AAAA,OAC/G;AAAA,IACF,CAAA;AAAA,IAEA,0BAAA,CAA2B,UAA0B,OAAA,EAA0B;AAC7E,MAAA,MAAM,MAAA,GAAS,QAAA,CAAS,aAAA,CAAc,OAAO,CAAA;AAC7C,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,OAAA,EAAS,MAAM,MAAA,GACX,gDAAA,GAAmD,OAAO,OAAO,CAAA,GACjE,4CAAA,GAA+C,MAAA,CAAO,OAAO;AAAA,OACnE;AAAA,IACF,CAAA;AAAA,IAEA,eAAA,CAAgB,UAA0B,KAAA,EAAe;AACvD,MAAA,MAAM,MAAA,GAAS,SAAS,YAAA,EAAa;AACrC,MAAA,OAAO;AAAA,QACL,MAAM,MAAA,KAAW,KAAA;AAAA,QACjB,OAAA,EAAS,MAAM,oCAAA,GAAuC,KAAA,GAAQ,QAAA,GAAW;AAAA,OAC3E;AAAA,IACF;AAAA,GACD,CAAA;AACH","file":"index.cjs","sourcesContent":["import type {\n  LLMAdapter,\n  LLMGenerateOptions,\n  LLMResult,\n  StreamingLLMAdapter,\n  StreamGenerateOptions,\n  StreamResult,\n  LLMStreamEvent,\n} from '@orka-js/core';\nimport { createStreamEvent } from '@orka-js/core';\nimport type { MockResponse, MockCall } from './types.js';\n\nexport class MockLLMAdapter implements LLMAdapter, StreamingLLMAdapter {\n  readonly name = 'mock';\n  readonly supportsStreaming = true;\n  private responses: MockResponse[];\n  private calls: MockCall[] = [];\n  private defaultOutput: string;\n\n  constructor(responses: MockResponse[] = [], defaultOutput = 'Mock response') {\n    this.responses = responses;\n    this.defaultOutput = defaultOutput;\n  }\n\n  private findResponse(prompt: string): MockResponse | null {\n    for (const response of this.responses) {\n      if (!response.when) return response;\n      if (typeof response.when === 'string') {\n        if (prompt.includes(response.when)) return response;\n      } else if (response.when instanceof RegExp) {\n        if (response.when.test(prompt)) return response;\n      } else if (typeof response.when === 'function') {\n        if (response.when(prompt)) return response;\n      }\n    }\n    return null;\n  }\n\n  private getPromptText(prompt: string, options?: LLMGenerateOptions): string {\n    // Also search in messages\n    const messagesText = options?.messages\n      ?.map(m => (typeof m.content === 'string' ? m.content : JSON.stringify(m.content)))\n      .join(' ') ?? '';\n    return prompt + ' ' + messagesText;\n  }\n\n  async generate(prompt: string, options?: LLMGenerateOptions): Promise<LLMResult> {\n    const fullPrompt = this.getPromptText(prompt, options);\n    const response = this.findResponse(fullPrompt);\n\n    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });\n\n    if (response?.latencyMs) {\n      await new Promise(resolve => setTimeout(resolve, response.latencyMs));\n    }\n\n    if (response?.error) throw response.error;\n\n    const output = response?.output ?? this.defaultOutput;\n\n    return {\n      content: output,\n      usage: { promptTokens: 100, completionTokens: 50, totalTokens: 150 },\n      model: 'mock-model',\n      finishReason: 'stop',\n      cost: 0,\n    };\n  }\n\n  async *stream(prompt: string, options?: StreamGenerateOptions): AsyncIterable<LLMStreamEvent> {\n    const fullPrompt = prompt + ' ' + (options?.messages?.map(m =>\n      typeof m.content === 'string' ? m.content : JSON.stringify(m.content)\n    ).join(' ') ?? '');\n\n    const response = this.findResponse(fullPrompt);\n    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });\n\n    if (response?.latencyMs) {\n      await new Promise(resolve => setTimeout(resolve, response.latencyMs));\n    }\n\n    if (response?.error) {\n      yield createStreamEvent('error', {\n        error: response.error,\n        message: response.error.message,\n      } as never);\n      return;\n    }\n\n    // Simulate tool call if configured\n    if (response?.toolCall) {\n      const callId = response.toolCall.id ?? `call_mock_${Date.now()}`;\n      yield createStreamEvent('tool_call', {\n        toolCallId: callId,\n        name: response.toolCall.name,\n        arguments: JSON.stringify(response.toolCall.args),\n      } as never);\n      yield createStreamEvent('done', {\n        content: '',\n        finishReason: 'tool_calls',\n      } as never);\n      return;\n    }\n\n    // Stream tokens one by one\n    const output = response?.output ?? this.defaultOutput;\n    const tokens = output.split(' ');\n    let index = 0;\n\n    for (const token of tokens) {\n      const tokenStr = index === 0 ? token : ' ' + token;\n      yield createStreamEvent('token', { token: tokenStr, index } as never);\n      index++;\n    }\n\n    yield createStreamEvent('usage', {\n      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },\n    } as never);\n\n    yield createStreamEvent('done', {\n      content: output,\n      finishReason: 'stop',\n      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },\n      cost: 0,\n    } as never);\n  }\n\n  async streamGenerate(prompt: string, options?: StreamGenerateOptions): Promise<StreamResult> {\n    let content = '';\n    const startTime = Date.now();\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };\n\n    for await (const event of this.stream(prompt, options)) {\n      if (event.type === 'token') content += event.token;\n      if (event.type === 'done') {\n        content = event.content || content;\n        if (event.usage) usage = event.usage;\n      }\n      if (event.type === 'error') throw event.error;\n    }\n\n    return {\n      content,\n      usage,\n      model: 'mock-model',\n      finishReason: 'stop',\n      durationMs: Date.now() - startTime,\n    };\n  }\n\n  async embed(text: string | string[]): Promise<number[][]> {\n    const inputs = Array.isArray(text) ? text : [text];\n    // Return deterministic mock embeddings\n    return inputs.map(() => Array.from({ length: 1536 }, (_, i) => Math.sin(i * 0.1)));\n  }\n\n  async generateObject<T>(schema: { parse(d: unknown): T }, prompt: string, options?: LLMGenerateOptions): Promise<T> {\n    const result = await this.generate(prompt, options);\n    try {\n      return schema.parse(JSON.parse(result.content));\n    } catch {\n      return schema.parse({});\n    }\n  }\n\n  // ---- Assertion helpers ----\n\n  getCalls(): MockCall[] {\n    return [...this.calls];\n  }\n\n  getCallCount(): number {\n    return this.calls.length;\n  }\n\n  wasCalledWith(pattern: string | RegExp): boolean {\n    return this.calls.some(c =>\n      typeof pattern === 'string' ? c.prompt.includes(pattern) : pattern.test(c.prompt)\n    );\n  }\n\n  getLastCall(): MockCall | undefined {\n    return this.calls[this.calls.length - 1];\n  }\n\n  reset(): void {\n    this.calls = [];\n  }\n}\n\nexport function mockLLM(responses: MockResponse[] = [], defaultOutput?: string): MockLLMAdapter {\n  return new MockLLMAdapter(responses, defaultOutput);\n}\n","import type { LLMStreamEvent } from '@orka-js/core';\nimport type { BaseAgent, AgentResult } from '@orka-js/agent';\nimport type { AgentTestResult } from './types.js';\nimport { MockLLMAdapter } from './mock-llm.js';\n\nexport interface AgentTestBedConfig {\n  agent: BaseAgent;\n  llm?: MockLLMAdapter;\n}\n\nexport class AgentTestBed {\n  private agent: BaseAgent;\n  private llm: MockLLMAdapter;\n\n  constructor(config: AgentTestBedConfig) {\n    this.agent = config.agent;\n    this.llm = config.llm ?? new MockLLMAdapter();\n  }\n\n  async run(input: string): Promise<AgentTestResult & AgentAssertions> {\n    this.llm.reset();\n    const events: LLMStreamEvent[] = [];\n    const toolCalls: AgentTestResult['toolCalls'] = [];\n\n    // Check if agent has runStream (StreamingToolAgent)\n    const streamAgent = this.agent as BaseAgent & { runStream?: (input: string) => AsyncIterable<LLMStreamEvent> };\n\n    let output = '';\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };\n    let steps = 0;\n\n    if (typeof streamAgent.runStream === 'function') {\n      for await (const event of streamAgent.runStream(input)) {\n        events.push(event);\n        if (event.type === 'tool_call') {\n          let args: Record<string, unknown> = {};\n          try { args = JSON.parse(event.arguments); } catch { args = {}; }\n          toolCalls.push({ name: event.name, args, callId: event.toolCallId });\n          steps++;\n        }\n        if (event.type === 'done') {\n          output = event.content;\n          if (event.usage) usage = event.usage;\n        }\n      }\n    } else {\n      const result: AgentResult = await this.agent.run(input);\n      output = result.output;\n      steps = result.steps.length;\n      usage = { promptTokens: 0, completionTokens: 0, totalTokens: result.totalTokens };\n      for (const step of result.steps) {\n        toolCalls.push({ name: step.action, args: step.actionInput, callId: '' });\n      }\n    }\n\n    const result: AgentTestResult = { output, toolCalls, steps, usage, events };\n    return Object.assign(result, makeAssertions(result));\n  }\n\n  getLLMMock(): MockLLMAdapter {\n    return this.llm;\n  }\n\n  reset(): void {\n    this.llm.reset();\n  }\n}\n\nfunction makeAssertions(result: AgentTestResult): AgentAssertions {\n  return {\n    toHaveCalledTool(name: string) {\n      const called = result.toolCalls.some(tc => tc.name === name);\n      if (!called) {\n        const calledTools = result.toolCalls.map(tc => tc.name).join(', ') || 'none';\n        throw new Error(\n          'Expected agent to have called tool ' + JSON.stringify(name) + ', but it called: ' + calledTools\n        );\n      }\n      return result as AgentTestResult & AgentAssertions;\n    },\n    toHaveOutput(pattern: string | RegExp) {\n      const matches = typeof pattern === 'string'\n        ? result.output.includes(pattern)\n        : pattern.test(result.output);\n      if (!matches) {\n        throw new Error(\n          'Expected output to match ' + String(pattern) + ', got: ' + JSON.stringify(result.output.slice(0, 200))\n        );\n      }\n      return result as AgentTestResult & AgentAssertions;\n    },\n    not: {\n      toHaveCalledTool(name: string) {\n        const called = result.toolCalls.some(tc => tc.name === name);\n        if (called) throw new Error('Expected agent NOT to have called tool ' + JSON.stringify(name) + ', but it did');\n        return result as AgentTestResult & AgentAssertions;\n      },\n    },\n  };\n}\n\nexport interface AgentAssertions {\n  toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;\n  toHaveOutput(pattern: string | RegExp): AgentTestResult & AgentAssertions;\n  not: {\n    toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;\n  };\n}","import type { AgentTestResult } from './types.js';\nimport type { MockLLMAdapter } from './mock-llm.js';\n\n/**\n * Extend Vitest/Jest expect with OrkaJS agent matchers.\n * \n * @example\n * ```typescript\n * import { expect } from 'vitest'\n * import { extendExpect } from '@orka-js/test'\n * extendExpect(expect)\n * \n * const result = await bed.run('hello')\n * expect(result).toHaveCalledTool('search')\n * expect(result).toHaveOutput(/hello/)\n * ```\n */\nexport function extendExpect(expect: {\n  extend(matchers: Record<string, unknown>): void;\n}): void {\n  expect.extend({\n    toHaveCalledTool(received: AgentTestResult, toolName: string) {\n      const called = received.toolCalls?.some(tc => tc.name === toolName);\n      return {\n        pass: called,\n        message: () => called\n          ? 'Expected agent NOT to have called tool ' + JSON.stringify(toolName)\n          : 'Expected agent to have called tool ' + JSON.stringify(toolName) + '. Called tools: [' + (received.toolCalls?.map(t => t.name).join(', ') || 'none') + ']',\n      };\n    },\n\n    toHaveOutput(received: AgentTestResult, pattern: string | RegExp) {\n      const matches = typeof pattern === 'string'\n        ? received.output?.includes(pattern)\n        : pattern.test(received.output ?? '');\n      return {\n        pass: matches,\n        message: () => matches\n          ? 'Expected output NOT to match ' + String(pattern)\n          : 'Expected output to match ' + String(pattern) + '. Got: ' + JSON.stringify(received.output?.slice(0, 200)),\n      };\n    },\n\n    toHaveBeenCalledWithPrompt(received: MockLLMAdapter, pattern: string | RegExp) {\n      const called = received.wasCalledWith(pattern);\n      return {\n        pass: called,\n        message: () => called\n          ? 'Expected MockLLM NOT to have been called with ' + String(pattern)\n          : 'Expected MockLLM to have been called with ' + String(pattern),\n      };\n    },\n\n    toHaveCallCount(received: MockLLMAdapter, count: number) {\n      const actual = received.getCallCount();\n      return {\n        pass: actual === count,\n        message: () => 'Expected MockLLM call count to be ' + count + ', got ' + actual,\n      };\n    },\n  });\n}"]}

package/dist/index.d.cts

@@ -0,0 +1,108 @@
+import { LLMStreamEvent, LLMAdapter, StreamingLLMAdapter, LLMGenerateOptions, LLMResult, StreamGenerateOptions, StreamResult } from '@orka-js/core';
+import { BaseAgent } from '@orka-js/agent';
+
+interface MockResponse {
+    /** Match condition — string (exact/contains), RegExp, or function */
+    when?: string | RegExp | ((prompt: string) => boolean);
+    /** Text output to return */
+    output?: string;
+    /** Tool call to simulate */
+    toolCall?: {
+        name: string;
+        args: Record<string, unknown>;
+        id?: string;
+    };
+    /** Error to throw */
+    error?: Error;
+    /** Simulate latency in ms */
+    latencyMs?: number;
+}
+interface MockCall {
+    prompt: string;
+    options?: unknown;
+    timestamp: number;
+    response: MockResponse | null;
+}
+interface AgentTestResult {
+    output: string;
+    toolCalls: Array<{
+        name: string;
+        args: Record<string, unknown>;
+        callId: string;
+    }>;
+    steps: number;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    events: LLMStreamEvent[];
+}
+interface AgentSnapshot {
+    output: string;
+    toolCalls: string[];
+}
+
+declare class MockLLMAdapter implements LLMAdapter, StreamingLLMAdapter {
+    readonly name = "mock";
+    readonly supportsStreaming = true;
+    private responses;
+    private calls;
+    private defaultOutput;
+    constructor(responses?: MockResponse[], defaultOutput?: string);
+    private findResponse;
+    private getPromptText;
+    generate(prompt: string, options?: LLMGenerateOptions): Promise<LLMResult>;
+    stream(prompt: string, options?: StreamGenerateOptions): AsyncIterable<LLMStreamEvent>;
+    streamGenerate(prompt: string, options?: StreamGenerateOptions): Promise<StreamResult>;
+    embed(text: string | string[]): Promise<number[][]>;
+    generateObject<T>(schema: {
+        parse(d: unknown): T;
+    }, prompt: string, options?: LLMGenerateOptions): Promise<T>;
+    getCalls(): MockCall[];
+    getCallCount(): number;
+    wasCalledWith(pattern: string | RegExp): boolean;
+    getLastCall(): MockCall | undefined;
+    reset(): void;
+}
+declare function mockLLM(responses?: MockResponse[], defaultOutput?: string): MockLLMAdapter;
+
+interface AgentTestBedConfig {
+    agent: BaseAgent;
+    llm?: MockLLMAdapter;
+}
+declare class AgentTestBed {
+    private agent;
+    private llm;
+    constructor(config: AgentTestBedConfig);
+    run(input: string): Promise<AgentTestResult & AgentAssertions>;
+    getLLMMock(): MockLLMAdapter;
+    reset(): void;
+}
+interface AgentAssertions {
+    toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;
+    toHaveOutput(pattern: string | RegExp): AgentTestResult & AgentAssertions;
+    not: {
+        toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;
+    };
+}
+
+/**
+ * Extend Vitest/Jest expect with OrkaJS agent matchers.
+ *
+ * @example
+ * ```typescript
+ * import { expect } from 'vitest'
+ * import { extendExpect } from '@orka-js/test'
+ * extendExpect(expect)
+ *
+ * const result = await bed.run('hello')
+ * expect(result).toHaveCalledTool('search')
+ * expect(result).toHaveOutput(/hello/)
+ * ```
+ */
+declare function extendExpect(expect: {
+    extend(matchers: Record<string, unknown>): void;
+}): void;
+
+export { type AgentAssertions, type AgentSnapshot, AgentTestBed, type AgentTestResult, type MockCall, MockLLMAdapter, type MockResponse, extendExpect, mockLLM };

package/dist/index.d.ts

@@ -0,0 +1,108 @@
+import { LLMStreamEvent, LLMAdapter, StreamingLLMAdapter, LLMGenerateOptions, LLMResult, StreamGenerateOptions, StreamResult } from '@orka-js/core';
+import { BaseAgent } from '@orka-js/agent';
+
+interface MockResponse {
+    /** Match condition — string (exact/contains), RegExp, or function */
+    when?: string | RegExp | ((prompt: string) => boolean);
+    /** Text output to return */
+    output?: string;
+    /** Tool call to simulate */
+    toolCall?: {
+        name: string;
+        args: Record<string, unknown>;
+        id?: string;
+    };
+    /** Error to throw */
+    error?: Error;
+    /** Simulate latency in ms */
+    latencyMs?: number;
+}
+interface MockCall {
+    prompt: string;
+    options?: unknown;
+    timestamp: number;
+    response: MockResponse | null;
+}
+interface AgentTestResult {
+    output: string;
+    toolCalls: Array<{
+        name: string;
+        args: Record<string, unknown>;
+        callId: string;
+    }>;
+    steps: number;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    events: LLMStreamEvent[];
+}
+interface AgentSnapshot {
+    output: string;
+    toolCalls: string[];
+}
+
+declare class MockLLMAdapter implements LLMAdapter, StreamingLLMAdapter {
+    readonly name = "mock";
+    readonly supportsStreaming = true;
+    private responses;
+    private calls;
+    private defaultOutput;
+    constructor(responses?: MockResponse[], defaultOutput?: string);
+    private findResponse;
+    private getPromptText;
+    generate(prompt: string, options?: LLMGenerateOptions): Promise<LLMResult>;
+    stream(prompt: string, options?: StreamGenerateOptions): AsyncIterable<LLMStreamEvent>;
+    streamGenerate(prompt: string, options?: StreamGenerateOptions): Promise<StreamResult>;
+    embed(text: string | string[]): Promise<number[][]>;
+    generateObject<T>(schema: {
+        parse(d: unknown): T;
+    }, prompt: string, options?: LLMGenerateOptions): Promise<T>;
+    getCalls(): MockCall[];
+    getCallCount(): number;
+    wasCalledWith(pattern: string | RegExp): boolean;
+    getLastCall(): MockCall | undefined;
+    reset(): void;
+}
+declare function mockLLM(responses?: MockResponse[], defaultOutput?: string): MockLLMAdapter;
+
+interface AgentTestBedConfig {
+    agent: BaseAgent;
+    llm?: MockLLMAdapter;
+}
+declare class AgentTestBed {
+    private agent;
+    private llm;
+    constructor(config: AgentTestBedConfig);
+    run(input: string): Promise<AgentTestResult & AgentAssertions>;
+    getLLMMock(): MockLLMAdapter;
+    reset(): void;
+}
+interface AgentAssertions {
+    toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;
+    toHaveOutput(pattern: string | RegExp): AgentTestResult & AgentAssertions;
+    not: {
+        toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;
+    };
+}
+
+/**
+ * Extend Vitest/Jest expect with OrkaJS agent matchers.
+ *
+ * @example
+ * ```typescript
+ * import { expect } from 'vitest'
+ * import { extendExpect } from '@orka-js/test'
+ * extendExpect(expect)
+ *
+ * const result = await bed.run('hello')
+ * expect(result).toHaveCalledTool('search')
+ * expect(result).toHaveOutput(/hello/)
+ * ```
+ */
+declare function extendExpect(expect: {
+    extend(matchers: Record<string, unknown>): void;
+}): void;
+
+export { type AgentAssertions, type AgentSnapshot, AgentTestBed, type AgentTestResult, type MockCall, MockLLMAdapter, type MockResponse, extendExpect, mockLLM };

package/dist/index.js

@@ -0,0 +1,270 @@
+import { createStreamEvent } from '@orka-js/core';
+
+// src/mock-llm.ts
+var MockLLMAdapter = class {
+  name = "mock";
+  supportsStreaming = true;
+  responses;
+  calls = [];
+  defaultOutput;
+  constructor(responses = [], defaultOutput = "Mock response") {
+    this.responses = responses;
+    this.defaultOutput = defaultOutput;
+  }
+  findResponse(prompt) {
+    for (const response of this.responses) {
+      if (!response.when) return response;
+      if (typeof response.when === "string") {
+        if (prompt.includes(response.when)) return response;
+      } else if (response.when instanceof RegExp) {
+        if (response.when.test(prompt)) return response;
+      } else if (typeof response.when === "function") {
+        if (response.when(prompt)) return response;
+      }
+    }
+    return null;
+  }
+  getPromptText(prompt, options) {
+    const messagesText = options?.messages?.map((m) => typeof m.content === "string" ? m.content : JSON.stringify(m.content)).join(" ") ?? "";
+    return prompt + " " + messagesText;
+  }
+  async generate(prompt, options) {
+    const fullPrompt = this.getPromptText(prompt, options);
+    const response = this.findResponse(fullPrompt);
+    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });
+    if (response?.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, response.latencyMs));
+    }
+    if (response?.error) throw response.error;
+    const output = response?.output ?? this.defaultOutput;
+    return {
+      content: output,
+      usage: { promptTokens: 100, completionTokens: 50, totalTokens: 150 },
+      model: "mock-model",
+      finishReason: "stop",
+      cost: 0
+    };
+  }
+  async *stream(prompt, options) {
+    const fullPrompt = prompt + " " + (options?.messages?.map(
+      (m) => typeof m.content === "string" ? m.content : JSON.stringify(m.content)
+    ).join(" ") ?? "");
+    const response = this.findResponse(fullPrompt);
+    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });
+    if (response?.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, response.latencyMs));
+    }
+    if (response?.error) {
+      yield createStreamEvent("error", {
+        error: response.error,
+        message: response.error.message
+      });
+      return;
+    }
+    if (response?.toolCall) {
+      const callId = response.toolCall.id ?? `call_mock_${Date.now()}`;
+      yield createStreamEvent("tool_call", {
+        toolCallId: callId,
+        name: response.toolCall.name,
+        arguments: JSON.stringify(response.toolCall.args)
+      });
+      yield createStreamEvent("done", {
+        content: "",
+        finishReason: "tool_calls"
+      });
+      return;
+    }
+    const output = response?.output ?? this.defaultOutput;
+    const tokens = output.split(" ");
+    let index = 0;
+    for (const token of tokens) {
+      const tokenStr = index === 0 ? token : " " + token;
+      yield createStreamEvent("token", { token: tokenStr, index });
+      index++;
+    }
+    yield createStreamEvent("usage", {
+      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length }
+    });
+    yield createStreamEvent("done", {
+      content: output,
+      finishReason: "stop",
+      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },
+      cost: 0
+    });
+  }
+  async streamGenerate(prompt, options) {
+    let content = "";
+    const startTime = Date.now();
+    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+    for await (const event of this.stream(prompt, options)) {
+      if (event.type === "token") content += event.token;
+      if (event.type === "done") {
+        content = event.content || content;
+        if (event.usage) usage = event.usage;
+      }
+      if (event.type === "error") throw event.error;
+    }
+    return {
+      content,
+      usage,
+      model: "mock-model",
+      finishReason: "stop",
+      durationMs: Date.now() - startTime
+    };
+  }
+  async embed(text) {
+    const inputs = Array.isArray(text) ? text : [text];
+    return inputs.map(() => Array.from({ length: 1536 }, (_, i) => Math.sin(i * 0.1)));
+  }
+  async generateObject(schema, prompt, options) {
+    const result = await this.generate(prompt, options);
+    try {
+      return schema.parse(JSON.parse(result.content));
+    } catch {
+      return schema.parse({});
+    }
+  }
+  // ---- Assertion helpers ----
+  getCalls() {
+    return [...this.calls];
+  }
+  getCallCount() {
+    return this.calls.length;
+  }
+  wasCalledWith(pattern) {
+    return this.calls.some(
+      (c) => typeof pattern === "string" ? c.prompt.includes(pattern) : pattern.test(c.prompt)
+    );
+  }
+  getLastCall() {
+    return this.calls[this.calls.length - 1];
+  }
+  reset() {
+    this.calls = [];
+  }
+};
+function mockLLM(responses = [], defaultOutput) {
+  return new MockLLMAdapter(responses, defaultOutput);
+}
+
+// src/test-bed.ts
+var AgentTestBed = class {
+  agent;
+  llm;
+  constructor(config) {
+    this.agent = config.agent;
+    this.llm = config.llm ?? new MockLLMAdapter();
+  }
+  async run(input) {
+    this.llm.reset();
+    const events = [];
+    const toolCalls = [];
+    const streamAgent = this.agent;
+    let output = "";
+    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+    let steps = 0;
+    if (typeof streamAgent.runStream === "function") {
+      for await (const event of streamAgent.runStream(input)) {
+        events.push(event);
+        if (event.type === "tool_call") {
+          let args = {};
+          try {
+            args = JSON.parse(event.arguments);
+          } catch {
+            args = {};
+          }
+          toolCalls.push({ name: event.name, args, callId: event.toolCallId });
+          steps++;
+        }
+        if (event.type === "done") {
+          output = event.content;
+          if (event.usage) usage = event.usage;
+        }
+      }
+    } else {
+      const result2 = await this.agent.run(input);
+      output = result2.output;
+      steps = result2.steps.length;
+      usage = { promptTokens: 0, completionTokens: 0, totalTokens: result2.totalTokens };
+      for (const step of result2.steps) {
+        toolCalls.push({ name: step.action, args: step.actionInput, callId: "" });
+      }
+    }
+    const result = { output, toolCalls, steps, usage, events };
+    return Object.assign(result, makeAssertions(result));
+  }
+  getLLMMock() {
+    return this.llm;
+  }
+  reset() {
+    this.llm.reset();
+  }
+};
+function makeAssertions(result) {
+  return {
+    toHaveCalledTool(name) {
+      const called = result.toolCalls.some((tc) => tc.name === name);
+      if (!called) {
+        const calledTools = result.toolCalls.map((tc) => tc.name).join(", ") || "none";
+        throw new Error(
+          "Expected agent to have called tool " + JSON.stringify(name) + ", but it called: " + calledTools
+        );
+      }
+      return result;
+    },
+    toHaveOutput(pattern) {
+      const matches = typeof pattern === "string" ? result.output.includes(pattern) : pattern.test(result.output);
+      if (!matches) {
+        throw new Error(
+          "Expected output to match " + String(pattern) + ", got: " + JSON.stringify(result.output.slice(0, 200))
+        );
+      }
+      return result;
+    },
+    not: {
+      toHaveCalledTool(name) {
+        const called = result.toolCalls.some((tc) => tc.name === name);
+        if (called) throw new Error("Expected agent NOT to have called tool " + JSON.stringify(name) + ", but it did");
+        return result;
+      }
+    }
+  };
+}
+
+// src/matchers.ts
+function extendExpect(expect) {
+  expect.extend({
+    toHaveCalledTool(received, toolName) {
+      const called = received.toolCalls?.some((tc) => tc.name === toolName);
+      return {
+        pass: called,
+        message: () => called ? "Expected agent NOT to have called tool " + JSON.stringify(toolName) : "Expected agent to have called tool " + JSON.stringify(toolName) + ". Called tools: [" + (received.toolCalls?.map((t) => t.name).join(", ") || "none") + "]"
+      };
+    },
+    toHaveOutput(received, pattern) {
+      const matches = typeof pattern === "string" ? received.output?.includes(pattern) : pattern.test(received.output ?? "");
+      return {
+        pass: matches,
+        message: () => matches ? "Expected output NOT to match " + String(pattern) : "Expected output to match " + String(pattern) + ". Got: " + JSON.stringify(received.output?.slice(0, 200))
+      };
+    },
+    toHaveBeenCalledWithPrompt(received, pattern) {
+      const called = received.wasCalledWith(pattern);
+      return {
+        pass: called,
+        message: () => called ? "Expected MockLLM NOT to have been called with " + String(pattern) : "Expected MockLLM to have been called with " + String(pattern)
+      };
+    },
+    toHaveCallCount(received, count) {
+      const actual = received.getCallCount();
+      return {
+        pass: actual === count,
+        message: () => "Expected MockLLM call count to be " + count + ", got " + actual
+      };
+    }
+  });
+}
+
+export { AgentTestBed, MockLLMAdapter, extendExpect, mockLLM };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/mock-llm.ts","../src/test-bed.ts","../src/matchers.ts"],"names":["result"],"mappings":";;;AAYO,IAAM,iBAAN,MAAgE;AAAA,EAC5D,IAAA,GAAO,MAAA;AAAA,EACP,iBAAA,GAAoB,IAAA;AAAA,EACrB,SAAA;AAAA,EACA,QAAoB,EAAC;AAAA,EACrB,aAAA;AAAA,EAER,WAAA,CAAY,SAAA,GAA4B,EAAC,EAAG,gBAAgB,eAAA,EAAiB;AAC3E,IAAA,IAAA,CAAK,SAAA,GAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,GAAgB,aAAA;AAAA,EACvB;AAAA,EAEQ,aAAa,MAAA,EAAqC;AACxD,IAAA,KAAA,MAAW,QAAA,IAAY,KAAK,SAAA,EAAW;AACrC,MAAA,IAAI,CAAC,QAAA,CAAS,IAAA,EAAM,OAAO,QAAA;AAC3B,MAAA,IAAI,OAAO,QAAA,CAAS,IAAA,KAAS,QAAA,EAAU;AACrC,QAAA,IAAI,MAAA,CAAO,QAAA,CAAS,QAAA,CAAS,IAAI,GAAG,OAAO,QAAA;AAAA,MAC7C,CAAA,MAAA,IAAW,QAAA,CAAS,IAAA,YAAgB,MAAA,EAAQ;AAC1C,QAAA,IAAI,QAAA,CAAS,IAAA,CAAK,IAAA,CAAK,MAAM,GAAG,OAAO,QAAA;AAAA,MACzC,CAAA,MAAA,IAAW,OAAO,QAAA,CAAS,IAAA,KAAS,UAAA,EAAY;AAC9C,QAAA,IAAI,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA,EAAG,OAAO,QAAA;AAAA,MACpC;AAAA,IACF;AACA,IAAA,OAAO,IAAA;AAAA,EACT;AAAA,EAEQ,aAAA,CAAc,QAAgB,OAAA,EAAsC;AAE1E,IAAA,MAAM,eAAe,OAAA,EAAS,QAAA,EAC1B,IAAI,CAAA,CAAA,KAAM,OAAO,EAAE,OAAA,KAAY,QAAA,GAAW,EAAE,OAAA,GAAU,IAAA,CAAK,UAAU,CAAA,CAAE,OAAO,CAAE,CAAA,CACjF,IAAA,CAAK,GAAG,CAAA,IAAK,EAAA;AAChB,IAAA,OAAO,SAAS,GAAA,GAAM,YAAA;AAAA,EACxB;AAAA,EAEA,MAAM,QAAA,CAAS,MAAA,EAAgB,OAAA,EAAkD;AAC/E,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,MAAA,EAAQ,OAAO,CAAA;AACrD,IAAA,MAAM,QAAA,GAAW,IAAA,CAAK,YAAA,CAAa,UAAU,CAAA;AAE7C,IAAA,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,EAAG,QAAA,EAAU,CAAA;AAEhF,IAAA,IAAI,UAAU,SAAA,EAAW;AACvB,MAAA,MAAM,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,WAAW,OAAA,EAAS,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,QAAA,EAAU,KAAA,EAAO,MAAM,QAAA,CAAS,KAAA;AAEpC,IAAA,MAAM,MAAA,GAAS,QAAA,EAAU,MAAA,IAAU,IAAA,CAAK,aAAA;AAExC,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,MAAA;AAAA,MACT,OAAO,EAAE,YAAA,EAAc,KAAK,gBAAA,EAAkB,EAAA,EAAI,aAAa,GAAA,EAAI;AAAA,MACnE,KAAA,EAAO,YAAA;AAAA,MACP,YAAA,EAAc,MAAA;AAAA,MACd,IAAA,EAAM;AAAA,KACR;AAAA,EACF;AAAA,EAEA,OAAO,MAAA,CAAO,MAAA,EAAgB,OAAA,EAAgE;AAC5F,IAAA,MAAM,UAAA,GAAa,MAAA,GAAS,GAAA,IAAO,OAAA,EAAS,QAAA,EAAU,GAAA;AAAA,MAAI,CAAA,CAAA,KACxD,OAAO,CAAA,CAAE,OAAA,KAAY,QAAA,GAAW,EAAE,OAAA,GAAU,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,OAAO;AAAA,KACtE,CAAE,IAAA,CAAK,GAAG,CAAA,IAAK,EAAA,CAAA;AAEf,IAAA,MAAM,QAAA,GAAW,IAAA,CAAK,YAAA,CAAa,UAAU,CAAA;AAC7C,IAAA,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,UAAA,EAAY,OAAA,EAAS,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,EAAG,QAAA,EAAU,CAAA;AAEhF,IAAA,IAAI,UAAU,SAAA,EAAW;AACvB,MAAA,MAAM,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,WAAW,OAAA,EAAS,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,UAAU,KAAA,EAAO;AACnB,MAAA,MAAM,kBAAkB,OAAA,EAAS;AAAA,QAC/B,OAAO,QAAA,CAAS,KAAA;AAAA,QAChB,OAAA,EAAS,SAAS,KAAA,CAAM;AAAA,OAChB,CAAA;AACV,MAAA;AAAA,IACF;AAGA,IAAA,IAAI,UAAU,QAAA,EAAU;AACtB,MAAA,MAAM,SAAS,QAAA,CAAS,QAAA,CAAS,MAAM,CAAA,UAAA,EAAa,IAAA,CAAK,KAAK,CAAA,CAAA;AAC9D,MAAA,MAAM,kBAAkB,WAAA,EAAa;AAAA,QACnC,UAAA,EAAY,MAAA;AAAA,QACZ,IAAA,EAAM,SAAS,QAAA,CAAS,IAAA;AAAA,QACxB,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,SAAS,IAAI;AAAA,OACxC,CAAA;AACV,MAAA,MAAM,kBAAkB,MAAA,EAAQ;AAAA,QAC9B,OAAA,EAAS,EAAA;AAAA,QACT,YAAA,EAAc;AAAA,OACN,CAAA;AACV,MAAA;AAAA,IACF;AAGA,IAAA,MAAM,MAAA,GAAS,QAAA,EAAU,MAAA,IAAU,IAAA,CAAK,aAAA;AACxC,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,KAAA,CAAM,GAAG,CAAA;AAC/B,IAAA,IAAI,KAAA,GAAQ,CAAA;AAEZ,IAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,MAAA,MAAM,QAAA,GAAW,KAAA,KAAU,CAAA,GAAI,KAAA,GAAQ,GAAA,GAAM,KAAA;AAC7C,MAAA,MAAM,kBAAkB,OAAA,EAAS,EAAE,KAAA,EAAO,QAAA,EAAU,OAAgB,CAAA;AACpE,MAAA,KAAA,EAAA;AAAA,IACF;AAEA,IAAA,MAAM,kBAAkB,OAAA,EAAS;AAAA,MAC/B,KAAA,EAAO,EAAE,YAAA,EAAc,GAAA,EAAK,gBAAA,EAAkB,OAAO,MAAA,EAAQ,WAAA,EAAa,GAAA,GAAM,MAAA,CAAO,MAAA;AAAO,KACtF,CAAA;AAEV,IAAA,MAAM,kBAAkB,MAAA,EAAQ;AAAA,MAC9B,OAAA,EAAS,MAAA;AAAA,MACT,YAAA,EAAc,MAAA;AAAA,MACd,KAAA,EAAO,EAAE,YAAA,EAAc,GAAA,EAAK,gBAAA,EAAkB,OAAO,MAAA,EAAQ,WAAA,EAAa,GAAA,GAAM,MAAA,CAAO,MAAA,EAAO;AAAA,MAC9F,IAAA,EAAM;AAAA,KACE,CAAA;AAAA,EACZ;AAAA,EAEA,MAAM,cAAA,CAAe,MAAA,EAAgB,OAAA,EAAwD;AAC3F,IAAA,IAAI,OAAA,GAAU,EAAA;AACd,IAAA,MAAM,SAAA,GAAY,KAAK,GAAA,EAAI;AAC3B,IAAA,IAAI,QAAQ,EAAE,YAAA,EAAc,GAAG,gBAAA,EAAkB,CAAA,EAAG,aAAa,CAAA,EAAE;AAEnE,IAAA,WAAA,MAAiB,KAAA,IAAS,IAAA,CAAK,MAAA,CAAO,MAAA,EAAQ,OAAO,CAAA,EAAG;AACtD,MAAA,IAAI,KAAA,CAAM,IAAA,KAAS,OAAA,EAAS,OAAA,IAAW,KAAA,CAAM,KAAA;AAC7C,MAAA,IAAI,KAAA,CAAM,SAAS,MAAA,EAAQ;AACzB,QAAA,OAAA,GAAU,MAAM,OAAA,IAAW,OAAA;AAC3B,QAAA,IAAI,KAAA,CAAM,KAAA,EAAO,KAAA,GAAQ,KAAA,CAAM,KAAA;AAAA,MACjC;AACA,MAAA,IAAI,KAAA,CAAM,IAAA,KAAS,OAAA,EAAS,MAAM,KAAA,CAAM,KAAA;AAAA,IAC1C;AAEA,IAAA,OAAO;AAAA,MACL,OAAA;AAAA,MACA,KAAA;AAAA,MACA,KAAA,EAAO,YAAA;AAAA,MACP,YAAA,EAAc,MAAA;AAAA,MACd,UAAA,EAAY,IAAA,CAAK,GAAA,EAAI,GAAI;AAAA,KAC3B;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,IAAA,EAA8C;AACxD,IAAA,MAAM,SAAS,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,GAAI,IAAA,GAAO,CAAC,IAAI,CAAA;AAEjD,IAAA,OAAO,OAAO,GAAA,CAAI,MAAM,KAAA,CAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAG,CAAC,GAAG,CAAA,KAAM,IAAA,CAAK,IAAI,CAAA,GAAI,GAAG,CAAC,CAAC,CAAA;AAAA,EACnF;AAAA,EAEA,MAAM,cAAA,CAAkB,MAAA,EAAkC,MAAA,EAAgB,OAAA,EAA0C;AAClH,IAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,QAAA,CAAS,QAAQ,OAAO,CAAA;AAClD,IAAA,IAAI;AACF,MAAA,OAAO,OAAO,KAAA,CAAM,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,OAAO,CAAC,CAAA;AAAA,IAChD,CAAA,CAAA,MAAQ;AACN,MAAA,OAAO,MAAA,CAAO,KAAA,CAAM,EAAE,CAAA;AAAA,IACxB;AAAA,EACF;AAAA;AAAA,EAIA,QAAA,GAAuB;AACrB,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,KAAK,CAAA;AAAA,EACvB;AAAA,EAEA,YAAA,GAAuB;AACrB,IAAA,OAAO,KAAK,KAAA,CAAM,MAAA;AAAA,EACpB;AAAA,EAEA,cAAc,OAAA,EAAmC;AAC/C,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA;AAAA,MAAK,CAAA,CAAA,KACrB,OAAO,OAAA,KAAY,QAAA,GAAW,CAAA,CAAE,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAAI,OAAA,CAAQ,IAAA,CAAK,CAAA,CAAE,MAAM;AAAA,KAClF;AAAA,EACF;AAAA,EAEA,WAAA,GAAoC;AAClC,IAAA,OAAO,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,KAAA,CAAM,SAAS,CAAC,CAAA;AAAA,EACzC;AAAA,EAEA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,QAAQ,EAAC;AAAA,EAChB;AACF;AAEO,SAAS,OAAA,CAAQ,SAAA,GAA4B,EAAC,EAAG,aAAA,EAAwC;AAC9F,EAAA,OAAO,IAAI,cAAA,CAAe,SAAA,EAAW,aAAa,CAAA;AACpD;;;ACtLO,IAAM,eAAN,MAAmB;AAAA,EAChB,KAAA;AAAA,EACA,GAAA;AAAA,EAER,YAAY,MAAA,EAA4B;AACtC,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,GAAA,GAAM,MAAA,CAAO,GAAA,IAAO,IAAI,cAAA,EAAe;AAAA,EAC9C;AAAA,EAEA,MAAM,IAAI,KAAA,EAA2D;AACnE,IAAA,IAAA,CAAK,IAAI,KAAA,EAAM;AACf,IAAA,MAAM,SAA2B,EAAC;AAClC,IAAA,MAAM,YAA0C,EAAC;AAGjD,IAAA,MAAM,cAAc,IAAA,CAAK,KAAA;AAEzB,IAAA,IAAI,MAAA,GAAS,EAAA;AACb,IAAA,IAAI,QAAQ,EAAE,YAAA,EAAc,GAAG,gBAAA,EAAkB,CAAA,EAAG,aAAa,CAAA,EAAE;AACnE,IAAA,IAAI,KAAA,GAAQ,CAAA;AAEZ,IAAA,IAAI,OAAO,WAAA,CAAY,SAAA,KAAc,UAAA,EAAY;AAC/C,MAAA,WAAA,MAAiB,KAAA,IAAS,WAAA,CAAY,SAAA,CAAU,KAAK,CAAA,EAAG;AACtD,QAAA,MAAA,CAAO,KAAK,KAAK,CAAA;AACjB,QAAA,IAAI,KAAA,CAAM,SAAS,WAAA,EAAa;AAC9B,UAAA,IAAI,OAAgC,EAAC;AACrC,UAAA,IAAI;AAAE,YAAA,IAAA,GAAO,IAAA,CAAK,KAAA,CAAM,KAAA,CAAM,SAAS,CAAA;AAAA,UAAG,CAAA,CAAA,MAAQ;AAAE,YAAA,IAAA,GAAO,EAAC;AAAA,UAAG;AAC/D,UAAA,SAAA,CAAU,IAAA,CAAK,EAAE,IAAA,EAAM,KAAA,CAAM,MAAM,IAAA,EAAM,MAAA,EAAQ,KAAA,CAAM,UAAA,EAAY,CAAA;AACnE,UAAA,KAAA,EAAA;AAAA,QACF;AACA,QAAA,IAAI,KAAA,CAAM,SAAS,MAAA,EAAQ;AACzB,UAAA,MAAA,GAAS,KAAA,CAAM,OAAA;AACf,UAAA,IAAI,KAAA,CAAM,KAAA,EAAO,KAAA,GAAQ,KAAA,CAAM,KAAA;AAAA,QACjC;AAAA,MACF;AAAA,IACF,CAAA,MAAO;AACL,MAAA,MAAMA,OAAAA,GAAsB,MAAM,IAAA,CAAK,KAAA,CAAM,IAAI,KAAK,CAAA;AACtD,MAAA,MAAA,GAASA,OAAAA,CAAO,MAAA;AAChB,MAAA,KAAA,GAAQA,QAAO,KAAA,CAAM,MAAA;AACrB,MAAA,KAAA,GAAQ,EAAE,YAAA,EAAc,CAAA,EAAG,kBAAkB,CAAA,EAAG,WAAA,EAAaA,QAAO,WAAA,EAAY;AAChF,MAAA,KAAA,MAAW,IAAA,IAAQA,QAAO,KAAA,EAAO;AAC/B,QAAA,SAAA,CAAU,IAAA,CAAK,EAAE,IAAA,EAAM,IAAA,CAAK,MAAA,EAAQ,MAAM,IAAA,CAAK,WAAA,EAAa,MAAA,EAAQ,EAAA,EAAI,CAAA;AAAA,MAC1E;AAAA,IACF;AAEA,IAAA,MAAM,SAA0B,EAAE,MAAA,EAAQ,SAAA,EAAW,KAAA,EAAO,OAAO,MAAA,EAAO;AAC1E,IAAA,OAAO,MAAA,CAAO,MAAA,CAAO,MAAA,EAAQ,cAAA,CAAe,MAAM,CAAC,CAAA;AAAA,EACrD;AAAA,EAEA,UAAA,GAA6B;AAC3B,IAAA,OAAO,IAAA,CAAK,GAAA;AAAA,EACd;AAAA,EAEA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,IAAI,KAAA,EAAM;AAAA,EACjB;AACF;AAEA,SAAS,eAAe,MAAA,EAA0C;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,IAAA,EAAc;AAC7B,MAAA,MAAM,SAAS,MAAA,CAAO,SAAA,CAAU,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,IAAI,CAAA;AAC3D,MAAA,IAAI,CAAC,MAAA,EAAQ;AACX,QAAA,MAAM,WAAA,GAAc,MAAA,CAAO,SAAA,CAAU,GAAA,CAAI,CAAA,EAAA,KAAM,GAAG,IAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA,IAAK,MAAA;AACtE,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,qCAAA,GAAwC,IAAA,CAAK,SAAA,CAAU,IAAI,IAAI,mBAAA,GAAsB;AAAA,SACvF;AAAA,MACF;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA,aAAa,OAAA,EAA0B;AACrC,MAAA,MAAM,OAAA,GAAU,OAAO,OAAA,KAAY,QAAA,GAC/B,MAAA,CAAO,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAC9B,OAAA,CAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,CAAA;AAC9B,MAAA,IAAI,CAAC,OAAA,EAAS;AACZ,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,2BAAA,GAA8B,MAAA,CAAO,OAAO,CAAA,GAAI,SAAA,GAAY,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC;AAAA,SACxG;AAAA,MACF;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA,GAAA,EAAK;AAAA,MACH,iBAAiB,IAAA,EAAc;AAC7B,QAAA,MAAM,SAAS,MAAA,CAAO,SAAA,CAAU,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,IAAI,CAAA;AAC3D,QAAA,IAAI,MAAA,QAAc,IAAI,KAAA,CAAM,4CAA4C,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,GAAI,cAAc,CAAA;AAC7G,QAAA,OAAO,MAAA;AAAA,MACT;AAAA;AACF,GACF;AACF;;;AClFO,SAAS,aAAa,MAAA,EAEpB;AACP,EAAA,MAAA,CAAO,MAAA,CAAO;AAAA,IACZ,gBAAA,CAAiB,UAA2B,QAAA,EAAkB;AAC5D,MAAA,MAAM,SAAS,QAAA,CAAS,SAAA,EAAW,KAAK,CAAA,EAAA,KAAM,EAAA,CAAG,SAAS,QAAQ,CAAA;AAClE,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,OAAA,EAAS,MAAM,MAAA,GACX,yCAAA,GAA4C,IAAA,CAAK,UAAU,QAAQ,CAAA,GACnE,qCAAA,GAAwC,IAAA,CAAK,SAAA,CAAU,QAAQ,IAAI,mBAAA,IAAuB,QAAA,CAAS,SAAA,EAAW,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA,IAAK,MAAA,CAAA,GAAU;AAAA,OAC7J;AAAA,IACF,CAAA;AAAA,IAEA,YAAA,CAAa,UAA2B,OAAA,EAA0B;AAChE,MAAA,MAAM,OAAA,GAAU,OAAO,OAAA,KAAY,QAAA,GAC/B,QAAA,CAAS,MAAA,EAAQ,QAAA,CAAS,OAAO,CAAA,GACjC,OAAA,CAAQ,IAAA,CAAK,QAAA,CAAS,UAAU,EAAE,CAAA;AACtC,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,OAAA;AAAA,QACN,SAAS,MAAM,OAAA,GACX,kCAAkC,MAAA,CAAO,OAAO,IAChD,2BAAA,GAA8B,MAAA,CAAO,OAAO,CAAA,GAAI,SAAA,GAAY,KAAK,SAAA,CAAU,QAAA,CAAS,QAAQ,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC;AAAA,OAC/G;AAAA,IACF,CAAA;AAAA,IAEA,0BAAA,CAA2B,UAA0B,OAAA,EAA0B;AAC7E,MAAA,MAAM,MAAA,GAAS,QAAA,CAAS,aAAA,CAAc,OAAO,CAAA;AAC7C,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,OAAA,EAAS,MAAM,MAAA,GACX,gDAAA,GAAmD,OAAO,OAAO,CAAA,GACjE,4CAAA,GAA+C,MAAA,CAAO,OAAO;AAAA,OACnE;AAAA,IACF,CAAA;AAAA,IAEA,eAAA,CAAgB,UAA0B,KAAA,EAAe;AACvD,MAAA,MAAM,MAAA,GAAS,SAAS,YAAA,EAAa;AACrC,MAAA,OAAO;AAAA,QACL,MAAM,MAAA,KAAW,KAAA;AAAA,QACjB,OAAA,EAAS,MAAM,oCAAA,GAAuC,KAAA,GAAQ,QAAA,GAAW;AAAA,OAC3E;AAAA,IACF;AAAA,GACD,CAAA;AACH","file":"index.js","sourcesContent":["import type {\n  LLMAdapter,\n  LLMGenerateOptions,\n  LLMResult,\n  StreamingLLMAdapter,\n  StreamGenerateOptions,\n  StreamResult,\n  LLMStreamEvent,\n} from '@orka-js/core';\nimport { createStreamEvent } from '@orka-js/core';\nimport type { MockResponse, MockCall } from './types.js';\n\nexport class MockLLMAdapter implements LLMAdapter, StreamingLLMAdapter {\n  readonly name = 'mock';\n  readonly supportsStreaming = true;\n  private responses: MockResponse[];\n  private calls: MockCall[] = [];\n  private defaultOutput: string;\n\n  constructor(responses: MockResponse[] = [], defaultOutput = 'Mock response') {\n    this.responses = responses;\n    this.defaultOutput = defaultOutput;\n  }\n\n  private findResponse(prompt: string): MockResponse | null {\n    for (const response of this.responses) {\n      if (!response.when) return response;\n      if (typeof response.when === 'string') {\n        if (prompt.includes(response.when)) return response;\n      } else if (response.when instanceof RegExp) {\n        if (response.when.test(prompt)) return response;\n      } else if (typeof response.when === 'function') {\n        if (response.when(prompt)) return response;\n      }\n    }\n    return null;\n  }\n\n  private getPromptText(prompt: string, options?: LLMGenerateOptions): string {\n    // Also search in messages\n    const messagesText = options?.messages\n      ?.map(m => (typeof m.content === 'string' ? m.content : JSON.stringify(m.content)))\n      .join(' ') ?? '';\n    return prompt + ' ' + messagesText;\n  }\n\n  async generate(prompt: string, options?: LLMGenerateOptions): Promise<LLMResult> {\n    const fullPrompt = this.getPromptText(prompt, options);\n    const response = this.findResponse(fullPrompt);\n\n    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });\n\n    if (response?.latencyMs) {\n      await new Promise(resolve => setTimeout(resolve, response.latencyMs));\n    }\n\n    if (response?.error) throw response.error;\n\n    const output = response?.output ?? this.defaultOutput;\n\n    return {\n      content: output,\n      usage: { promptTokens: 100, completionTokens: 50, totalTokens: 150 },\n      model: 'mock-model',\n      finishReason: 'stop',\n      cost: 0,\n    };\n  }\n\n  async *stream(prompt: string, options?: StreamGenerateOptions): AsyncIterable<LLMStreamEvent> {\n    const fullPrompt = prompt + ' ' + (options?.messages?.map(m =>\n      typeof m.content === 'string' ? m.content : JSON.stringify(m.content)\n    ).join(' ') ?? '');\n\n    const response = this.findResponse(fullPrompt);\n    this.calls.push({ prompt: fullPrompt, options, timestamp: Date.now(), response });\n\n    if (response?.latencyMs) {\n      await new Promise(resolve => setTimeout(resolve, response.latencyMs));\n    }\n\n    if (response?.error) {\n      yield createStreamEvent('error', {\n        error: response.error,\n        message: response.error.message,\n      } as never);\n      return;\n    }\n\n    // Simulate tool call if configured\n    if (response?.toolCall) {\n      const callId = response.toolCall.id ?? `call_mock_${Date.now()}`;\n      yield createStreamEvent('tool_call', {\n        toolCallId: callId,\n        name: response.toolCall.name,\n        arguments: JSON.stringify(response.toolCall.args),\n      } as never);\n      yield createStreamEvent('done', {\n        content: '',\n        finishReason: 'tool_calls',\n      } as never);\n      return;\n    }\n\n    // Stream tokens one by one\n    const output = response?.output ?? this.defaultOutput;\n    const tokens = output.split(' ');\n    let index = 0;\n\n    for (const token of tokens) {\n      const tokenStr = index === 0 ? token : ' ' + token;\n      yield createStreamEvent('token', { token: tokenStr, index } as never);\n      index++;\n    }\n\n    yield createStreamEvent('usage', {\n      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },\n    } as never);\n\n    yield createStreamEvent('done', {\n      content: output,\n      finishReason: 'stop',\n      usage: { promptTokens: 100, completionTokens: tokens.length, totalTokens: 100 + tokens.length },\n      cost: 0,\n    } as never);\n  }\n\n  async streamGenerate(prompt: string, options?: StreamGenerateOptions): Promise<StreamResult> {\n    let content = '';\n    const startTime = Date.now();\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };\n\n    for await (const event of this.stream(prompt, options)) {\n      if (event.type === 'token') content += event.token;\n      if (event.type === 'done') {\n        content = event.content || content;\n        if (event.usage) usage = event.usage;\n      }\n      if (event.type === 'error') throw event.error;\n    }\n\n    return {\n      content,\n      usage,\n      model: 'mock-model',\n      finishReason: 'stop',\n      durationMs: Date.now() - startTime,\n    };\n  }\n\n  async embed(text: string | string[]): Promise<number[][]> {\n    const inputs = Array.isArray(text) ? text : [text];\n    // Return deterministic mock embeddings\n    return inputs.map(() => Array.from({ length: 1536 }, (_, i) => Math.sin(i * 0.1)));\n  }\n\n  async generateObject<T>(schema: { parse(d: unknown): T }, prompt: string, options?: LLMGenerateOptions): Promise<T> {\n    const result = await this.generate(prompt, options);\n    try {\n      return schema.parse(JSON.parse(result.content));\n    } catch {\n      return schema.parse({});\n    }\n  }\n\n  // ---- Assertion helpers ----\n\n  getCalls(): MockCall[] {\n    return [...this.calls];\n  }\n\n  getCallCount(): number {\n    return this.calls.length;\n  }\n\n  wasCalledWith(pattern: string | RegExp): boolean {\n    return this.calls.some(c =>\n      typeof pattern === 'string' ? c.prompt.includes(pattern) : pattern.test(c.prompt)\n    );\n  }\n\n  getLastCall(): MockCall | undefined {\n    return this.calls[this.calls.length - 1];\n  }\n\n  reset(): void {\n    this.calls = [];\n  }\n}\n\nexport function mockLLM(responses: MockResponse[] = [], defaultOutput?: string): MockLLMAdapter {\n  return new MockLLMAdapter(responses, defaultOutput);\n}\n","import type { LLMStreamEvent } from '@orka-js/core';\nimport type { BaseAgent, AgentResult } from '@orka-js/agent';\nimport type { AgentTestResult } from './types.js';\nimport { MockLLMAdapter } from './mock-llm.js';\n\nexport interface AgentTestBedConfig {\n  agent: BaseAgent;\n  llm?: MockLLMAdapter;\n}\n\nexport class AgentTestBed {\n  private agent: BaseAgent;\n  private llm: MockLLMAdapter;\n\n  constructor(config: AgentTestBedConfig) {\n    this.agent = config.agent;\n    this.llm = config.llm ?? new MockLLMAdapter();\n  }\n\n  async run(input: string): Promise<AgentTestResult & AgentAssertions> {\n    this.llm.reset();\n    const events: LLMStreamEvent[] = [];\n    const toolCalls: AgentTestResult['toolCalls'] = [];\n\n    // Check if agent has runStream (StreamingToolAgent)\n    const streamAgent = this.agent as BaseAgent & { runStream?: (input: string) => AsyncIterable<LLMStreamEvent> };\n\n    let output = '';\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };\n    let steps = 0;\n\n    if (typeof streamAgent.runStream === 'function') {\n      for await (const event of streamAgent.runStream(input)) {\n        events.push(event);\n        if (event.type === 'tool_call') {\n          let args: Record<string, unknown> = {};\n          try { args = JSON.parse(event.arguments); } catch { args = {}; }\n          toolCalls.push({ name: event.name, args, callId: event.toolCallId });\n          steps++;\n        }\n        if (event.type === 'done') {\n          output = event.content;\n          if (event.usage) usage = event.usage;\n        }\n      }\n    } else {\n      const result: AgentResult = await this.agent.run(input);\n      output = result.output;\n      steps = result.steps.length;\n      usage = { promptTokens: 0, completionTokens: 0, totalTokens: result.totalTokens };\n      for (const step of result.steps) {\n        toolCalls.push({ name: step.action, args: step.actionInput, callId: '' });\n      }\n    }\n\n    const result: AgentTestResult = { output, toolCalls, steps, usage, events };\n    return Object.assign(result, makeAssertions(result));\n  }\n\n  getLLMMock(): MockLLMAdapter {\n    return this.llm;\n  }\n\n  reset(): void {\n    this.llm.reset();\n  }\n}\n\nfunction makeAssertions(result: AgentTestResult): AgentAssertions {\n  return {\n    toHaveCalledTool(name: string) {\n      const called = result.toolCalls.some(tc => tc.name === name);\n      if (!called) {\n        const calledTools = result.toolCalls.map(tc => tc.name).join(', ') || 'none';\n        throw new Error(\n          'Expected agent to have called tool ' + JSON.stringify(name) + ', but it called: ' + calledTools\n        );\n      }\n      return result as AgentTestResult & AgentAssertions;\n    },\n    toHaveOutput(pattern: string | RegExp) {\n      const matches = typeof pattern === 'string'\n        ? result.output.includes(pattern)\n        : pattern.test(result.output);\n      if (!matches) {\n        throw new Error(\n          'Expected output to match ' + String(pattern) + ', got: ' + JSON.stringify(result.output.slice(0, 200))\n        );\n      }\n      return result as AgentTestResult & AgentAssertions;\n    },\n    not: {\n      toHaveCalledTool(name: string) {\n        const called = result.toolCalls.some(tc => tc.name === name);\n        if (called) throw new Error('Expected agent NOT to have called tool ' + JSON.stringify(name) + ', but it did');\n        return result as AgentTestResult & AgentAssertions;\n      },\n    },\n  };\n}\n\nexport interface AgentAssertions {\n  toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;\n  toHaveOutput(pattern: string | RegExp): AgentTestResult & AgentAssertions;\n  not: {\n    toHaveCalledTool(name: string): AgentTestResult & AgentAssertions;\n  };\n}","import type { AgentTestResult } from './types.js';\nimport type { MockLLMAdapter } from './mock-llm.js';\n\n/**\n * Extend Vitest/Jest expect with OrkaJS agent matchers.\n * \n * @example\n * ```typescript\n * import { expect } from 'vitest'\n * import { extendExpect } from '@orka-js/test'\n * extendExpect(expect)\n * \n * const result = await bed.run('hello')\n * expect(result).toHaveCalledTool('search')\n * expect(result).toHaveOutput(/hello/)\n * ```\n */\nexport function extendExpect(expect: {\n  extend(matchers: Record<string, unknown>): void;\n}): void {\n  expect.extend({\n    toHaveCalledTool(received: AgentTestResult, toolName: string) {\n      const called = received.toolCalls?.some(tc => tc.name === toolName);\n      return {\n        pass: called,\n        message: () => called\n          ? 'Expected agent NOT to have called tool ' + JSON.stringify(toolName)\n          : 'Expected agent to have called tool ' + JSON.stringify(toolName) + '. Called tools: [' + (received.toolCalls?.map(t => t.name).join(', ') || 'none') + ']',\n      };\n    },\n\n    toHaveOutput(received: AgentTestResult, pattern: string | RegExp) {\n      const matches = typeof pattern === 'string'\n        ? received.output?.includes(pattern)\n        : pattern.test(received.output ?? '');\n      return {\n        pass: matches,\n        message: () => matches\n          ? 'Expected output NOT to match ' + String(pattern)\n          : 'Expected output to match ' + String(pattern) + '. Got: ' + JSON.stringify(received.output?.slice(0, 200)),\n      };\n    },\n\n    toHaveBeenCalledWithPrompt(received: MockLLMAdapter, pattern: string | RegExp) {\n      const called = received.wasCalledWith(pattern);\n      return {\n        pass: called,\n        message: () => called\n          ? 'Expected MockLLM NOT to have been called with ' + String(pattern)\n          : 'Expected MockLLM to have been called with ' + String(pattern),\n      };\n    },\n\n    toHaveCallCount(received: MockLLMAdapter, count: number) {\n      const actual = received.getCallCount();\n      return {\n        pass: actual === count,\n        message: () => 'Expected MockLLM call count to be ' + count + ', got ' + actual,\n      };\n    },\n  });\n}"]}

package/package.json

@@ -0,0 +1,70 @@
+{
+  "name": "@orka-js/test",
+  "version": "1.1.0",
+  "description": "Testing utilities for OrkaJS agents — mock LLM, AgentTestBed, Vitest matchers",
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "sideEffects": false,
+  "keywords": [
+    "orkajs",
+    "orka",
+    "testing",
+    "mock",
+    "llm",
+    "ai",
+    "test",
+    "vitest"
+  ],
+  "author": "OrkaJS Team",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/orka-js/orkajs.git",
+    "directory": "packages/test"
+  },
+  "bugs": {
+    "url": "https://github.com/orka-js/orkajs/issues"
+  },
+  "homepage": "https://orkajs.com/en/test",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@orka-js/core": "^1.5.0",
+    "@orka-js/agent": "^1.5.2"
+  },
+  "devDependencies": {
+    "tsup": "^8.0.1",
+    "typescript": "^5.3.3",
+    "vitest": "^1.6.0"
+  },
+  "peerDependencies": {
+    "vitest": ">=1.0.0"
+  },
+  "peerDependenciesMeta": {
+    "vitest": {
+      "optional": true
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run --passWithNoTests"
+  }
+}