@arki-moe/agent-ts 5.3.0 → 5.4.0

package/README.md

@@ -38,6 +38,10 @@ const result = await agent.run("What time is it?");
 console.log(result.messages);
 console.log(result.usage, result.usageSum);
 
+// run also accepts multiple user messages or Message objects
+await agent.run(["Hello", "What's next?"]);
+await agent.run({ role: Role.System, content: "You are brief." });
+
 // context is a public property that can be read directly
 console.log(agent.context);
 ```
@@ -58,6 +62,7 @@ When `apiKey` is not provided in config, adapters read from the corresponding en
 - `agent.context` - Public property, complete conversation history
 - `agent.registerTool(tool)` - Register tool
 - `agent.run(message)` - Execute tool chain automatically, returns `{ messages, usage, usageSum }`
+  - `message` can be `string | string[] | Message | Message[]`
 
 ### Config
 

package/dist/index.d.ts

@@ -1,8 +1,8 @@
-import type { AgentConfig, AgentLike, Context, Message, Tool, Usage } from "./types";
+import type { AgentConfig, AgentLike, Context, Message, RunInput, Tool, Usage } from "./types";
 export { openaiAdapter } from "./adapter/openai";
 export { openrouterAdapter } from "./adapter/openrouter";
 export { selfhostChatCompletionsAdapter } from "./adapter/selfhost_chat_completions";
-export type { Adapter, AgentConfig, AgentLike, Context, Message, Tool, Usage, AdapterResult } from "./types";
+export type { Adapter, AgentConfig, AgentLike, Context, Message, RunInput, Tool, Usage, AdapterResult } from "./types";
 export { Role } from "./types";
 export declare class Agent implements AgentLike {
     context: Context;
@@ -15,9 +15,29 @@ export declare class Agent implements AgentLike {
     private onToolResult?;
     constructor(adapterName: string, config: AgentConfig);
     registerTool(tool: Tool): void;
+    run(message: RunInput): Promise<{
+        messages: Message[];
+        usage?: Usage;
+        usageSum?: Usage;
+    }>;
     run(message: string): Promise<{
         messages: Message[];
         usage?: Usage;
         usageSum?: Usage;
     }>;
+    run(message: string[]): Promise<{
+        messages: Message[];
+        usage?: Usage;
+        usageSum?: Usage;
+    }>;
+    run(message: Message): Promise<{
+        messages: Message[];
+        usage?: Usage;
+        usageSum?: Usage;
+    }>;
+    run(message: Message[]): Promise<{
+        messages: Message[];
+        usage?: Usage;
+        usageSum?: Usage;
+    }>;
 }

package/dist/index.js

@@ -19,6 +19,19 @@ const adapters = {
     openrouter: openrouter_1.openrouterAdapter,
     selfhost_chat_completions: selfhost_chat_completions_1.selfhostChatCompletionsAdapter,
 };
+const normalizeRunInput = (input) => {
+    if (typeof input === "string")
+        return [{ role: types_1.Role.User, content: input }];
+    if (Array.isArray(input)) {
+        if (input.length === 0)
+            return [];
+        if (typeof input[0] === "string") {
+            return input.map((content) => ({ role: types_1.Role.User, content }));
+        }
+        return input;
+    }
+    return [input];
+};
 class Agent {
     constructor(adapterName, config) {
         this.context = [];
@@ -47,9 +60,11 @@ class Agent {
         const pushToSession = (msgs) => {
             sessionContext.push(...msgs);
         };
-        const userMessage = { role: types_1.Role.User, content: message };
-        pushToSession([userMessage]);
-        persistToContext([userMessage]);
+        const inputMessages = normalizeRunInput(message);
+        if (inputMessages.length === 0)
+            return { messages: [] };
+        pushToSession(inputMessages);
+        persistToContext(inputMessages);
         const addUsage = (next) => {
             if (!next)
                 return;

package/dist/types.d.ts

@@ -27,6 +27,7 @@ export type Message = {
     isError?: boolean;
 };
 export type Context = Message[];
+export type RunInput = string | string[] | Message | Message[];
 export type Usage = {
     promptTokens: number;
     completionTokens: number;
@@ -35,7 +36,7 @@ export type Usage = {
 export interface AgentLike {
     context: Context;
     registerTool: (tool: Tool) => void;
-    run: (message: string) => Promise<{
+    run: (message: RunInput) => Promise<{
         messages: Message[];
         usage?: Usage;
         usageSum?: Usage;

package/package.json

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