@mohanscodex/spectra-agent 0.4.0

package/README.md

@@ -0,0 +1,241 @@
+# @singularity-ai/spectra-agent
+
+**Agent runtime with multi-turn tool dispatch and streaming event delivery.**
+
+Orchestrates conversations with LLMs: streams responses, dispatches tool calls (parallel or sequential), injects intermediate messages mid-turn, and emits typed events for every phase of execution.
+
+## Features
+
+- **Multi-turn loop** — Automatically feeds tool results back to the model. Configurable `maxTurns`.
+- **Tool execution** — Parallel or sequential dispatch. Tools defined with Zod schemas for type-safe argument validation.
+- **Streaming events** — AsyncGenerator yields `AgentEvent` discriminated unions. Subscribe via `for await` or side-channel listeners.
+- **Hooks** — `beforeToolCall` (block/modify), `afterToolCall` (modify results), `transformContext` (rewrite messages), `getApiKey` (dynamic key resolution).
+- **Steering & follow-up queues** — Inject messages mid-turn (`steer()`) or queue them for the next run (`followUp()`).
+- **Automatic retry** — Exponential backoff (up to 3 retries) for transient API errors. Configurable max delay.
+- **Abort support** — `agent.abort()` cancels in-flight requests with `AbortController`.
+- **Subscriber pattern** — Push-based listeners for side-channel logging, metrics, or persistence.
+
+## Installation
+
+```bash
+bun add @singularity-ai/spectra-agent
+```
+
+Depends on `@singularity-ai/spectra-ai` (automatically resolved as a workspace dependency).
+
+## Quick Start
+
+```typescript
+import { Agent, defineTool } from "@singularity-ai/spectra-agent";
+import { z } from "zod";
+
+const weatherTool = defineTool({
+  name: "get_weather",
+  description: "Get current weather for a location",
+  parameters: z.object({
+    location: z.string().describe("City name"),
+  }),
+  execute: async ({ location }) => ({
+    content: [{ type: "text", text: `The weather in ${location} is sunny.` }],
+  }),
+});
+
+const agent = new Agent({
+  model: { id: "gpt-4o", name: "GPT-4o", provider: "openai-completions", api: "openai" },
+  systemPrompt: "You are a helpful assistant with weather data.",
+  tools: [weatherTool],
+});
+
+for await (const event of agent.run("What's the weather in Tokyo?")) {
+  switch (event.type) {
+    case "message_update":
+      const text = event.message.content
+        .filter((c) => c.type === "text")
+        .map((c) => c.text)
+        .join("");
+      process.stdout.write(text);
+      break;
+    case "tool_execution_start":
+      console.log(`\n[Tool: ${event.toolName}]`);
+      break;
+    case "tool_execution_end":
+      console.log(`\n[Result: ${event.isError ? "error" : "ok"}]`);
+      break;
+    case "agent_end":
+      console.log("\nDone. Transcript length:", event.messages.length);
+      break;
+  }
+}
+```
+
+## Architecture
+
+```
+agent.run("Hello")
+  │
+  ├─ emit agent_start
+  ├─ emit message_start/user
+  ├─ emit turn_start
+  │
+  ├─ LLM stream (via @spectra-ai)
+  │   ├─ emit message_start (assistant)
+  │   ├─ emit message_update (deltas)
+  │   └─ emit message_end (complete)
+  │
+  ├─ toolCalls detected?
+  │   ├─ NO  → emit turn_end → emit agent_end → done
+  │   └─ YES → prepare (beforeToolCall hook)
+  │              ├─ sequential: execute one-by-one
+  │              └─ parallel: prepare all, then execute concurrently
+  │             → attach tool results to transcript
+  │             → emit turn_end
+  │             → check steering queue / follow-up queue
+  │             → loop back to LLM stream
+  │
+  └─ agent_end (final transcript in event.messages)
+```
+
+## API
+
+### `new Agent(config)`
+
+```typescript
+interface AgentConfig {
+  model: Model;                         // LLM model to use
+  systemPrompt?: string;                // system prompt
+  tools?: AgentTool[];                  // registered tools
+  maxTurns?: number;                    // max loop iterations (default: unlimited)
+  toolExecution?: "parallel" | "sequential";  // default: "parallel"
+  beforeToolCall?: (ctx, signal?) => Promise<BeforeToolCallResult | undefined>;
+  afterToolCall?: (ctx, signal?) => Promise<AfterToolCallResult | undefined>;
+  transformContext?: (messages, signal?) => Promise<Message[]>;
+  getApiKey?: (provider) => string | undefined | Promise<string | undefined>;
+}
+```
+
+### `agent.run(input)`
+
+Returns `AsyncGenerator<AgentEvent>`. Input is a `string`, `Message`, or `Message[]`.
+
+### `agent.subscribe(listener)`
+
+Returns an unsubscribe function. Listeners fire for every event in parallel with the generator consumer.
+
+```typescript
+const unsub = agent.subscribe((event, signal) => {
+  if (event.type === "tool_execution_start") {
+    metrics.counter.inc({ tool: event.toolName });
+  }
+});
+```
+
+### `agent.steer(message)` / `agent.followUp(message)`
+
+- `steer()` — Injects a user message into the current run loop. Processed on the next turn.
+- `followUp()` — Queues a message for after the current `run()` completes.
+
+### State
+
+```typescript
+agent.messages           // Message[] — full conversation transcript
+agent.isStreaming        // boolean — run in progress
+agent.streamingMessage   // AssistantMessage | undefined
+agent.pendingToolCalls   // Set<string> — tool call IDs in flight
+agent.errorMessage       // string | undefined
+agent.signal             // AbortSignal | undefined
+```
+
+### `agent.abort()` / `agent.reset()` / `agent.restoreHistory(messages)`
+
+## Tool Definition
+
+```typescript
+import { defineTool, textResult, errorResult } from "@singularity-ai/spectra-agent";
+import { z } from "zod";
+
+const searchTool = defineTool({
+  name: "search_web",
+  description: "Search the web for information",
+  parameters: z.object({
+    query: z.string().describe("Search query"),
+    maxResults: z.number().optional().default(5),
+  }),
+  execute: async (args, { toolCallId, signal, onUpdate }) => {
+    // args is fully typed: { query: string; maxResults?: number }
+    if (!args.query) return errorResult("Query is required");
+    onUpdate?.({ content: [{ type: "text", text: "Searching..." }] });
+    const results = await fetchResults(args.query, args.maxResults, signal);
+    return textResult(results);
+  },
+});
+```
+
+## Events
+
+| Event | When | Payload |
+|---|---|---|
+| `agent_start` | Run begins | — |
+| `agent_end` | Run complete | `messages: Message[]` |
+| `turn_start` | LLM turn begins | — |
+| `turn_end` | Turn complete | `message`, `toolResults` |
+| `message_start` | Message added to transcript | `message` |
+| `message_update` | Assistant message delta | `message`, `assistantMessageEvent` |
+| `message_end` | Message fully formed | `message` |
+| `tool_execution_start` | Tool call begins | `toolCallId`, `toolName`, `args` |
+| `tool_execution_update` | Tool reports partial progress | `toolCallId`, `toolName`, `partialResult` |
+| `tool_execution_end` | Tool call completes | `toolCallId`, `toolName`, `result`, `isError` |
+
+## Hooks
+
+### `beforeToolCall`
+
+Block or modify a tool call before execution:
+
+```typescript
+beforeToolCall: async ({ toolCall, args }) => {
+  if (toolCall.name === "delete_file") {
+    return { block: true, reason: "Not allowed in current context" };
+  }
+}
+```
+
+### `afterToolCall`
+
+Transform tool results:
+
+```typescript
+afterToolCall: async ({ result, isError }) => {
+  if (isError) return { content: [{ type: "text", text: "Tool failed, retrying..." }], isError: false };
+}
+```
+
+### `transformContext`
+
+Rewrite messages before sending to the LLM:
+
+```typescript
+transformContext: async (messages) => {
+  return messages.filter(m => m.role !== "system");
+}
+```
+
+### `getApiKey`
+
+Resolve API keys dynamically:
+
+```typescript
+getApiKey: async (provider) => {
+  if (provider === "anthropic") return process.env.ANTHROPIC_KEY;
+  return process.env.OPENAI_API_KEY;
+}
+```
+
+## Retry Behavior
+
+The agent retries LLM calls up to 3 times with exponential backoff (1s, 2s, 4s) capped at `maxRetryDelayMs` (default 30s). Does not retry on:
+- 4xx client errors (400, 401, 403, 404)
+- Aborted requests
+
+## License
+
+MIT

package/dist/__tests__/agent-features.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=agent-features.test.d.ts.map

package/dist/__tests__/agent-features.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-features.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/agent-features.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/agent-features.test.js

@@ -0,0 +1,108 @@
+import { describe, it, expect } from "vitest";
+import { Agent } from "../agent.js";
+const testModel = {
+    id: "test-model",
+    name: "Test Model",
+    provider: "test",
+    api: "test",
+};
+describe("Agent Steering Queue", () => {
+    it("should queue steering messages", () => {
+        const agent = new Agent({ model: testModel });
+        agent.steer("Be more concise");
+        agent.steer("Use bullet points");
+        // Should not throw and should queue internally
+        expect(agent).toBeDefined();
+    });
+    it("should accept string steering messages", () => {
+        const agent = new Agent({ model: testModel });
+        expect(() => agent.steer("Hello")).not.toThrow();
+    });
+    it("should accept Message steering messages", () => {
+        const agent = new Agent({ model: testModel });
+        const message = {
+            role: "user",
+            content: "Custom steering",
+            timestamp: Date.now(),
+        };
+        expect(() => agent.steer(message)).not.toThrow();
+    });
+});
+describe("Agent Follow-Up Queue", () => {
+    it("should queue follow-up messages", () => {
+        const agent = new Agent({ model: testModel });
+        agent.followUp("What about tomorrow?");
+        agent.followUp("And the weekend?");
+        expect(agent).toBeDefined();
+    });
+    it("should accept string follow-up messages", () => {
+        const agent = new Agent({ model: testModel });
+        expect(() => agent.followUp("Hello")).not.toThrow();
+    });
+    it("should accept Message follow-up messages", () => {
+        const agent = new Agent({ model: testModel });
+        const message = {
+            role: "user",
+            content: "Custom follow-up",
+            timestamp: Date.now(),
+        };
+        expect(() => agent.followUp(message)).not.toThrow();
+    });
+});
+describe("Agent Configuration", () => {
+    it("should accept steeringMode option", () => {
+        const agent = new Agent({
+            model: testModel,
+            steeringMode: "all",
+        });
+        expect(agent).toBeDefined();
+    });
+    it("should accept followUpMode option", () => {
+        const agent = new Agent({
+            model: testModel,
+            followUpMode: "one-at-a-time",
+        });
+        expect(agent).toBeDefined();
+    });
+    it("should accept convertToLlm hook", () => {
+        const convertToLlm = (messages) => messages;
+        const agent = new Agent({
+            model: testModel,
+            convertToLlm,
+        });
+        expect(agent).toBeDefined();
+    });
+    it("should accept maxRetryDelayMs option", () => {
+        const agent = new Agent({
+            model: testModel,
+            maxRetryDelayMs: 5000,
+        });
+        expect(agent).toBeDefined();
+    });
+});
+describe("Agent Error Recovery", () => {
+    it("should configure retry parameters", () => {
+        const agent = new Agent({
+            model: testModel,
+            maxRetryDelayMs: 10000,
+        });
+        expect(agent).toBeDefined();
+    });
+});
+describe("Agent Queue Integration", () => {
+    it("should handle mixed steer and followUp calls", () => {
+        const agent = new Agent({ model: testModel });
+        agent.steer("First steering");
+        agent.followUp("First follow-up");
+        agent.steer("Second steering");
+        agent.followUp("Second follow-up");
+        expect(agent).toBeDefined();
+    });
+    it("should not throw when queuing during non-streaming state", () => {
+        const agent = new Agent({ model: testModel });
+        expect(agent.isStreaming).toBe(false);
+        expect(() => agent.steer("test")).not.toThrow();
+        expect(() => agent.followUp("test")).not.toThrow();
+    });
+});
+//# sourceMappingURL=agent-features.test.js.map

package/dist/__tests__/agent-features.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-features.test.js","sourceRoot":"","sources":["../../src/__tests__/agent-features.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAM,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAGpC,MAAM,SAAS,GAAU;IACvB,EAAE,EAAE,YAAY;IAChB,IAAI,EAAE,YAAY;IAClB,QAAQ,EAAE,MAAM;IAChB,GAAG,EAAE,MAAM;CACZ,CAAC;AAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC/B,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAEjC,+CAA+C;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,iBAAiB;YAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,KAAK,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QACvC,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,kBAAkB;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,eAAe;SAC9B,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,YAAY,GAAG,CAAC,QAAmB,EAAE,EAAE,CAAC,QAAQ,CAAC;QAEvD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,eAAe,EAAE,KAAK;SACvB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,KAAK,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;QAClC,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC/B,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAChD,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/agent.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=agent.test.d.ts.map

package/dist/__tests__/agent.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/agent.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/agent.test.js

@@ -0,0 +1,109 @@
+import { describe, it, expect, vi } from "vitest";
+import { Agent } from "../agent.js";
+import { defineTool } from "../define-tool.js";
+import { z } from "zod";
+const testModel = { id: "claude-sonnet-4-20250514", name: "Claude Sonnet 4", provider: "anthropic", api: "anthropic-messages" };
+describe("Agent", () => {
+    it("should create agent instance", () => {
+        const agent = new Agent({
+            model: testModel,
+            systemPrompt: "You are a helpful assistant.",
+        });
+        expect(agent).toBeDefined();
+    });
+    it("should store messages", () => {
+        const agent = new Agent({ model: testModel });
+        expect(agent.messages).toEqual([]);
+    });
+    it("should register tools", () => {
+        const agent = new Agent({ model: testModel });
+        const tool = defineTool({
+            name: "get_weather",
+            description: "Get the current weather",
+            parameters: z.object({
+                location: z.string().describe("The location to get weather for"),
+            }),
+            execute: async (args, context) => {
+                return {
+                    content: [{ type: "text", text: `The weather in ${args.location} is sunny.` }],
+                };
+            },
+        });
+        agent.registerTool(tool);
+        expect(agent.messages).toEqual([]);
+    });
+    it("should subscribe and unsubscribe listeners", () => {
+        const agent = new Agent({ model: testModel });
+        const listener = vi.fn();
+        const unsubscribe = agent.subscribe(listener);
+        expect(unsubscribe).toBeDefined();
+        unsubscribe();
+    });
+    it("should abort request", () => {
+        const agent = new Agent({ model: testModel });
+        expect(() => agent.abort()).not.toThrow();
+    });
+    it("should reset state", () => {
+        const agent = new Agent({ model: testModel });
+        agent.reset();
+        expect(agent.messages).toEqual([]);
+        expect(agent.isStreaming).toBe(false);
+    });
+    it("should return signal", () => {
+        const agent = new Agent({ model: testModel });
+        const signal = agent.signal;
+        expect(signal).toBeUndefined();
+    });
+});
+describe("Agent Queues and Hooks", () => {
+    it("should add steering messages to queue", () => {
+        const agent = new Agent({ model: testModel });
+        agent.steer("Please be more concise");
+        // Steering message should be queued (internal state)
+        // We can't directly check the queue, but we can verify it doesn't throw
+        expect(() => agent.steer("Another message")).not.toThrow();
+    });
+    it("should add follow-up messages to queue", () => {
+        const agent = new Agent({ model: testModel });
+        agent.followUp("Can you also check the forecast?");
+        // Follow-up message should be queued
+        expect(() => agent.followUp("And the temperature?")).not.toThrow();
+    });
+    it("should accept Message objects in steer and followUp", () => {
+        const agent = new Agent({ model: testModel });
+        const message = {
+            role: "user",
+            content: "Custom message",
+            timestamp: Date.now(),
+        };
+        expect(() => agent.steer(message)).not.toThrow();
+        expect(() => agent.followUp(message)).not.toThrow();
+    });
+});
+describe("Agent Retry Logic", () => {
+    it("should configure max retry delay", () => {
+        const agent = new Agent({
+            model: testModel,
+            maxRetryDelayMs: 5000,
+        });
+        expect(agent).toBeDefined();
+    });
+});
+describe("defineTool", () => {
+    it("should create tool with Zod schema", () => {
+        const tool = defineTool({
+            name: "get_weather",
+            description: "Get the current weather",
+            parameters: z.object({
+                location: z.string().describe("The location"),
+            }),
+            execute: async (args, context) => ({
+                content: [{ type: "text", text: "sunny" }],
+            }),
+        });
+        expect(tool.name).toBe("get_weather");
+        expect(tool.description).toBe("Get the current weather");
+        expect(tool.parameters).toBeDefined();
+    });
+});
+//# sourceMappingURL=agent.test.js.map

package/dist/__tests__/agent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.test.js","sourceRoot":"","sources":["../../src/__tests__/agent.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAc,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,SAAS,GAAU,EAAE,EAAE,EAAE,0BAA0B,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,oBAAoB,EAAE,CAAC;AAEvI,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;IACrB,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,8BAA8B;SAC7C,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,IAAI,GAAG,UAAU,CAAC;YACtB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,yBAAyB;YACtC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;aACjE,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE;gBAC/B,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,IAAI,CAAC,QAAQ,YAAY,EAAE,CAAC;iBAC/E,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEzB,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;QAElC,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,KAAK,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAEtC,qDAAqD;QACrD,wEAAwE;QACxE,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,KAAK,CAAC,QAAQ,CAAC,kCAAkC,CAAC,CAAC;QAEnD,qCAAqC;QACrC,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,gBAAgB;YACzB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACjD,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,SAAS;YAChB,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,IAAI,GAAG,UAAU,CAAC;YACtB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,yBAAyB;YACtC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;gBACnB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;aAC9C,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACzD,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/e2e.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=e2e.test.d.ts.map

package/dist/__tests__/e2e.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"e2e.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/e2e.test.ts"],"names":[],"mappings":""}