@arki-moe/agent-ts 1.0.1 → 1.0.3

package/README.md

@@ -35,29 +35,41 @@ console.log(msgs2);
 console.log(agent.context);
 ```
 
-## OpenRouter
+## Supported Adapters
 
-```ts
-import { Agent, Role } from "@arki-moe/agent-ts";
+| Adapter | Required | Optional |
+|---------|----------|----------|
+| `openai` | `apiKey` (or `OPENAI_API_KEY` env), `model` | `system`, `baseUrl` |
+| `openrouter` | `apiKey` (or `OPENROUTER_API_KEY` env), `model` | `system`, `baseUrl`, `httpReferer`, `title` |
 
-const agent = new Agent("openrouter", {
-  apiKey: "or-...",
-  model: "gpt-5-nano",
-  httpReferer: "https://your-site.example", // optional
-  title: "Your App", // optional
-});
-
-const msgs = await agent.step({ role: Role.User, content: "Hello" });
-console.log(msgs);
-```
+When `apiKey` is not provided in config, adapters read from the corresponding environment variable. An error is thrown only when both are missing.
 
 ## API
 
 - `Agent(adapterName, config)` - Create Agent, config contains `apiKey`, `model`, `system` (optional), etc.
 - `agent.context` - Public property, complete conversation history
 - `agent.registerTool(tool)` - Register tool
-- `agent.step(message)` - Call model once, returns new `Message[]`
-- `agent.run(message)` - Execute tool chain automatically, returns all new `Message[]`
+- `agent.step(message?)` - Call model once, returns new `Message[]`
+- `agent.run(message, endCondition?)` - Execute tool chain automatically, returns all new `Message[]`
+- `agent.fork()` - Create a new agent with a copied context
+
+`agent.step` and `agent.run` always append new messages to `agent.context`.
+`endCondition` receives `(context, last)` and stops the run when it returns `true`. Defaults to `last.role === Role.Ai`.
+
+`agent.fork()` shallow-copies the context array, but message objects are shared. This means:
+- Shallow copy: `forked.context !== agent.context`, so pushing new messages does not affect the other agent.
+- Shared messages: modifying a message object in one context will be visible in the other.
+
+Example:
+
+```ts
+const forked = agent.fork();
+forked.context.push({ role: Role.User, content: "hi" });
+// agent.context length is unchanged
+
+forked.context[0].content = "changed";
+// agent.context[0].content is also "changed" because messages are shared
+```
 
 ## Scripts
 

package/dist/adapter/openai.js

@@ -27,10 +27,10 @@ function toOpenAIMessages(context) {
 }
 async function openaiAdapter(config, context, tools) {
     const baseUrl = config.baseUrl ?? "https://api.openai.com";
-    const apiKey = config.apiKey ?? "";
+    const apiKey = config.apiKey || process.env.OPENAI_API_KEY || "";
     const model = config.model ?? "gpt-5-nano";
     if (!apiKey)
-        throw new Error("OpenAI adapter requires apiKey in config");
+        throw new Error("OpenAI adapter requires apiKey in config or OPENAI_API_KEY env");
     const contextMessages = toOpenAIMessages(context);
     const systemContent = config.system;
     const messages = systemContent

package/dist/adapter/openrouter.js

@@ -27,12 +27,12 @@ function toOpenRouterMessages(context) {
 }
 async function openrouterAdapter(config, context, tools) {
     const baseUrl = config.baseUrl ?? "https://openrouter.ai/api/v1";
-    const apiKey = config.apiKey ?? "";
+    const apiKey = config.apiKey || process.env.OPENROUTER_API_KEY || "";
     const model = config.model ?? "gpt-5-nano";
     const httpReferer = config.httpReferer;
     const title = config.title;
     if (!apiKey)
-        throw new Error("OpenRouter adapter requires apiKey in config");
+        throw new Error("OpenRouter adapter requires apiKey in config or OPENROUTER_API_KEY env");
     const contextMessages = toOpenRouterMessages(context);
     const systemContent = config.system;
     const messages = systemContent

package/dist/index.d.ts

@@ -6,10 +6,12 @@ export { Role } from "./types";
 export declare class Agent {
     context: Context;
     private adapter;
+    private adapterName;
     private config;
     private tools;
     constructor(adapterName: string, config: Record<string, unknown>);
     registerTool(tool: Tool): void;
-    step(message: Message): Promise<Message[]>;
-    run(message: Message): Promise<Message[]>;
+    step(message?: Message): Promise<Message[]>;
+    run(message: Message, endCondition?: (context: Message[], last: Message) => boolean): Promise<Message[]>;
+    fork(): Agent;
 }

package/dist/index.js

@@ -18,6 +18,7 @@ class Agent {
     constructor(adapterName, config) {
         this.context = [];
         this.tools = [];
+        this.adapterName = adapterName;
         this.adapter = adapters[adapterName] ?? (() => { throw new Error(`Adapter "${adapterName}" not found`); })();
         this.config = config;
     }
@@ -25,20 +26,20 @@ class Agent {
         this.tools.push(tool);
     }
     async step(message) {
-        this.context.push(message);
+        if (message) {
+            this.context.push(message);
+        }
         const msgs = await this.adapter(this.config, this.context, this.tools);
         this.context.push(...msgs);
         return msgs;
     }
-    async run(message) {
-        this.context.push(message);
+    async run(message, endCondition = (_context, last) => last.role === types_1.Role.Ai) {
         const all = [];
+        let msgs = await this.step(message);
+        all.push(...msgs);
         for (;;) {
-            const msgs = await this.adapter(this.config, this.context, this.tools);
-            this.context.push(...msgs);
-            all.push(...msgs);
             const last = msgs[msgs.length - 1];
-            if (last.role === types_1.Role.Ai)
+            if (endCondition(this.context, last))
                 return all;
             for (const m of msgs) {
                 if (m.role !== types_1.Role.ToolCall)
@@ -61,7 +62,16 @@ class Agent {
                 this.context.push(result);
                 all.push(result);
             }
+            msgs = await this.step();
+            all.push(...msgs);
         }
     }
+    fork() {
+        const agent = new Agent(this.adapterName, this.config);
+        agent.adapter = this.adapter;
+        agent.tools = [...this.tools];
+        agent.context = [...this.context];
+        return agent;
+    }
 }
 exports.Agent = Agent;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arki-moe/agent-ts",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "Minimal Agent library, zero dependencies",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",