@lupaflow/agent 0.1.0 → 0.1.2

package/dist/index.js

@@ -0,0 +1,620 @@
+// src/runner.ts
+import { providerRegistry } from "@lupaflow/providers";
+
+// src/context.ts
+import { estimateTokens } from "@lupaflow/core";
+var ContextManager = class {
+  messages = [];
+  config;
+  constructor(config = {}) {
+    this.config = {
+      maxMessages: config.maxMessages || 50,
+      maxTokens: config.maxTokens || 4e3,
+      preserveSystem: config.preserveSystem ?? true
+    };
+  }
+  add(message) {
+    this.messages.push(message);
+    this.trim();
+  }
+  getMessages() {
+    return [...this.messages];
+  }
+  getTotalTokens() {
+    return this.messages.reduce((sum, m) => sum + estimateTokens(m.content || ""), 0);
+  }
+  trim() {
+    if (this.messages.length <= this.config.maxMessages && this.getTotalTokens() <= this.config.maxTokens) {
+      return;
+    }
+    while (this.messages.length > 2 && (this.messages.length > this.config.maxMessages || this.getTotalTokens() > this.config.maxTokens)) {
+      const startIdx = this.config.preserveSystem && this.messages[0]?.role === "system" ? 1 : 0;
+      if (startIdx < this.messages.length) {
+        this.messages.splice(startIdx, 1);
+      } else {
+        break;
+      }
+    }
+  }
+  clear() {
+    const systemMsg = this.config.preserveSystem ? this.messages.find((m) => m.role === "system") : void 0;
+    this.messages = systemMsg ? [systemMsg] : [];
+  }
+  toJSON() {
+    return this.getMessages();
+  }
+};
+
+// src/retry.ts
+import { delay } from "@lupaflow/core";
+import { RetryError } from "@lupaflow/core";
+import { globalEventBus } from "@lupaflow/core";
+async function withRetry(fn, config = {}, context) {
+  const cfg = {
+    maxRetries: config.maxRetries ?? 3,
+    baseDelay: config.baseDelay ?? 1e3,
+    maxDelay: config.maxDelay ?? 3e4,
+    backoffFactor: config.backoffFactor ?? 2
+  };
+  let lastError = null;
+  for (let attempt = 0; attempt <= cfg.maxRetries; attempt++) {
+    try {
+      return await fn();
+    } catch (err) {
+      lastError = err;
+      if (attempt < cfg.maxRetries) {
+        const waitTime = Math.min(cfg.baseDelay * Math.pow(cfg.backoffFactor, attempt), cfg.maxDelay);
+        await globalEventBus.emit("agent:error", {
+          error: err.message,
+          retryAttempt: attempt + 1
+        });
+        await delay(waitTime);
+      }
+    }
+  }
+  throw new RetryError(
+    `${context ? `[${context}] ` : ""}Failed after ${cfg.maxRetries + 1} attempts`,
+    cfg.maxRetries + 1,
+    lastError
+  );
+}
+
+// src/config.ts
+import { ConfigError } from "@lupaflow/core";
+function validateAgentConfig(config) {
+  if (!config.name) throw new ConfigError("Agent name is required");
+  if (!config.provider) throw new ConfigError("Provider is required for agent");
+  if (config.temperature !== void 0 && (config.temperature < 0 || config.temperature > 2)) {
+    throw new ConfigError("Temperature must be between 0 and 2");
+  }
+  if (config.maxRetries !== void 0 && config.maxRetries < 0) {
+    throw new ConfigError("maxRetries must be >= 0");
+  }
+}
+function buildSystemPrompt(config) {
+  const parts = [];
+  parts.push(config.systemPrompt || `You are ${config.name}, an AI assistant.`);
+  if (config.personality) {
+    const p = config.personality;
+    parts.push("");
+    parts.push("--- Personality ---");
+    if (p.name) parts.push(`Name: ${p.name}`);
+    if (p.tone) parts.push(`Tone: ${p.tone}`);
+    if (p.style) parts.push(`Style: ${p.style}`);
+    if (p.traits?.length) parts.push(`Traits: ${p.traits.join(", ")}`);
+  }
+  if (config.goals?.length) {
+    parts.push("");
+    parts.push("--- Goals ---");
+    config.goals.forEach((g, i) => parts.push(`${i + 1}. ${g.description}`));
+  }
+  if (config.constraints?.length) {
+    parts.push("");
+    parts.push("--- Constraints ---");
+    config.constraints.forEach((c) => parts.push(`- ${c.description}`));
+  }
+  if (config.tools?.length) {
+    parts.push("");
+    parts.push(`You have access to these tools: ${config.tools.map((t) => t.definition.name).join(", ")}`);
+    parts.push("Use them when appropriate to fulfill the user's request.");
+  }
+  return parts.join("\n");
+}
+
+// src/runner.ts
+import { generateId } from "@lupaflow/core";
+import { globalEventBus as globalEventBus2 } from "@lupaflow/core";
+import { ToolError } from "@lupaflow/core";
+import {
+  runBeforeAgentRunHooks,
+  runAfterAgentRunHooks,
+  runBeforeToolCallHooks,
+  runAfterToolCallHooks,
+  runBeforeProviderCallHooks,
+  runAfterProviderCallHooks
+} from "@lupaflow/plugins";
+var AgentRunner = class {
+  config;
+  context;
+  memory;
+  toolCallCount = 0;
+  totalPromptTokens = 0;
+  totalCompletionTokens = 0;
+  constructor(config, memory) {
+    this.config = config;
+    this.context = new ContextManager({
+      maxMessages: config.maxContextMessages || 50
+    });
+    this.memory = memory;
+  }
+  async run(options) {
+    const runId = generateId();
+    const startTime = Date.now();
+    this.toolCallCount = 0;
+    this.totalPromptTokens = 0;
+    this.totalCompletionTokens = 0;
+    const provider = providerRegistry.get(this.config.provider, {
+      model: this.config.model,
+      temperature: this.config.temperature,
+      maxTokens: this.config.maxTokens,
+      topP: this.config.topP
+    });
+    const systemPrompt = buildSystemPrompt(this.config);
+    runBeforeAgentRunHooks({
+      input: options.input,
+      ...this.config
+    });
+    this.context.add({ role: "system", content: systemPrompt });
+    this.context.add({ role: "user", content: options.input });
+    await globalEventBus2.emit("agent:start", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      config: this.config
+    });
+    await globalEventBus2.emit("agent:thinking", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      message: options.input
+    });
+    let finalContent = "";
+    let finishReason = "stop";
+    let iterationCount = 0;
+    const maxIterations = 20;
+    while (iterationCount < maxIterations) {
+      iterationCount++;
+      const request = runBeforeProviderCallHooks(this.config.provider, {
+        model: this.config.model,
+        messages: this.context.getMessages(),
+        tools: this.config.tools,
+        systemPrompt: void 0,
+        temperature: this.config.temperature,
+        maxTokens: this.config.maxTokens,
+        topP: this.config.topP
+      });
+      const response = await withRetry(
+        () => provider.complete({
+          messages: request.messages || this.context.getMessages(),
+          tools: this.config.tools,
+          systemPrompt: void 0,
+          temperature: this.config.temperature,
+          maxTokens: this.config.maxTokens,
+          topP: this.config.topP,
+          model: this.config.model
+        }),
+        {
+          maxRetries: this.config.maxRetries ?? 3,
+          baseDelay: this.config.retryDelay ?? 1e3
+        },
+        this.config.name
+      );
+      runAfterProviderCallHooks(response);
+      this.totalPromptTokens += response.usage.promptTokens;
+      this.totalCompletionTokens += response.usage.completionTokens;
+      if (response.toolCalls.length > 0) {
+        this.context.add({
+          role: "assistant",
+          content: response.content || null,
+          toolCalls: response.toolCalls.map((tc) => ({
+            id: tc.id,
+            type: "function",
+            function: { name: tc.name, arguments: JSON.stringify(tc.args) }
+          }))
+        });
+      } else if (response.content) {
+        this.context.add({ role: "assistant", content: response.content });
+      }
+      if (response.toolCalls.length === 0) {
+        finalContent = response.content || "";
+        finishReason = response.finishReason;
+        break;
+      }
+      for (const tc of response.toolCalls) {
+        this.toolCallCount++;
+        await globalEventBus2.emit("tool:start", {
+          agentId: this.config.id || this.config.name,
+          runId,
+          tool: tc.name,
+          args: tc.args
+        });
+        options.onToolCall?.(tc.name, tc.args);
+        try {
+          const tool = this.config.tools?.find((t) => t.definition.name === tc.name);
+          if (!tool) throw new ToolError(tc.name, `Tool "${tc.name}" not found on agent`);
+          const hookArgs = runBeforeToolCallHooks(tc.name, tc.args);
+          const result = await tool.execute(hookArgs);
+          const hookResult = runAfterToolCallHooks(result);
+          const resultStr = typeof hookResult === "string" ? hookResult : JSON.stringify(hookResult, null, 2);
+          this.context.add({
+            role: "tool",
+            toolCallId: tc.id,
+            name: tc.name,
+            content: resultStr
+          });
+          await globalEventBus2.emit("tool:finish", {
+            agentId: this.config.id || this.config.name,
+            runId,
+            tool: tc.name,
+            result: hookResult,
+            durationMs: Date.now() - startTime
+          });
+          options.onToolResult?.(tc.name, hookResult);
+        } catch (err) {
+          this.context.add({
+            role: "tool",
+            toolCallId: tc.id,
+            name: tc.name,
+            content: `Error: ${err.message}`
+          });
+          await globalEventBus2.emit("tool:error", {
+            agentId: this.config.id || this.config.name,
+            runId,
+            tool: tc.name,
+            error: err.message
+          });
+        }
+      }
+    }
+    const totalLatency = Date.now() - startTime;
+    if (this.memory) {
+      await this.memory.store({
+        id: generateId(),
+        type: "short-term",
+        key: `agent:${this.config.name}:input:${generateId()}`,
+        content: options.input,
+        metadata: { runId, agentName: this.config.name },
+        timestamp: Date.now()
+      });
+      if (finalContent) {
+        await this.memory.store({
+          id: generateId(),
+          type: "short-term",
+          key: `agent:${this.config.name}:output:${generateId()}`,
+          content: finalContent,
+          metadata: { runId, agentName: this.config.name },
+          timestamp: Date.now()
+        });
+      }
+    }
+    await globalEventBus2.emit("agent:response", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      content: finalContent,
+      tokens: this.totalCompletionTokens,
+      latencyMs: totalLatency
+    });
+    runAfterAgentRunHooks({
+      id: runId,
+      output: finalContent,
+      messages: this.context.getMessages(),
+      usage: {
+        totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+        promptTokens: this.totalPromptTokens,
+        completionTokens: this.totalCompletionTokens
+      },
+      latencyMs: totalLatency,
+      toolCalls: this.toolCallCount,
+      finishReason
+    });
+    await globalEventBus2.emit("agent:complete", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+      totalLatencyMs: totalLatency,
+      toolCalls: this.toolCallCount
+    });
+    return {
+      id: runId,
+      output: finalContent,
+      messages: this.context.getMessages(),
+      usage: {
+        totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+        promptTokens: this.totalPromptTokens,
+        completionTokens: this.totalCompletionTokens
+      },
+      latencyMs: totalLatency,
+      toolCalls: this.toolCallCount,
+      finishReason
+    };
+  }
+  async *streamRun(options) {
+    const runId = generateId();
+    const startTime = Date.now();
+    this.toolCallCount = 0;
+    this.totalPromptTokens = 0;
+    this.totalCompletionTokens = 0;
+    const provider = providerRegistry.get(this.config.provider, {
+      model: this.config.model,
+      temperature: this.config.temperature,
+      maxTokens: this.config.maxTokens,
+      topP: this.config.topP
+    });
+    const systemPrompt = buildSystemPrompt(this.config);
+    runBeforeAgentRunHooks({
+      input: options.input,
+      ...this.config
+    });
+    this.context.add({ role: "system", content: systemPrompt });
+    this.context.add({ role: "user", content: options.input });
+    await globalEventBus2.emit("agent:start", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      config: this.config
+    });
+    await globalEventBus2.emit("agent:thinking", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      message: options.input
+    });
+    let finalContent = "";
+    let finishReason = "stop";
+    let iterationCount = 0;
+    const maxIterations = 20;
+    while (iterationCount < maxIterations) {
+      iterationCount++;
+      const request = runBeforeProviderCallHooks(this.config.provider, {
+        model: this.config.model,
+        messages: this.context.getMessages(),
+        tools: this.config.tools,
+        systemPrompt: void 0,
+        temperature: this.config.temperature,
+        maxTokens: this.config.maxTokens,
+        topP: this.config.topP
+      });
+      const messages = request.messages || this.context.getMessages();
+      const stream = provider.streamComplete({
+        messages,
+        tools: this.config.tools,
+        systemPrompt: void 0,
+        temperature: this.config.temperature,
+        maxTokens: this.config.maxTokens,
+        topP: this.config.topP,
+        model: this.config.model
+      });
+      let roundContent = "";
+      let roundToolCalls = [];
+      for await (const chunk of stream) {
+        if (chunk.type === "text") {
+          roundContent += chunk.text;
+          yield { type: "text", text: chunk.text };
+        } else if (chunk.type === "tool_call") {
+          roundToolCalls.push(chunk.toolCall);
+        } else if (chunk.type === "finish") {
+          roundContent = chunk.content;
+          roundToolCalls = chunk.toolCalls;
+          this.totalPromptTokens += chunk.usage.promptTokens;
+          this.totalCompletionTokens += chunk.usage.completionTokens;
+          finishReason = chunk.finishReason;
+        }
+      }
+      runAfterProviderCallHooks({
+        content: roundContent,
+        toolCalls: roundToolCalls
+      });
+      if (roundToolCalls.length > 0) {
+        this.context.add({
+          role: "assistant",
+          content: roundContent || null,
+          toolCalls: roundToolCalls.map((tc) => ({
+            id: tc.id,
+            type: "function",
+            function: { name: tc.name, arguments: JSON.stringify(tc.args) }
+          }))
+        });
+      } else if (roundContent) {
+        this.context.add({ role: "assistant", content: roundContent });
+      }
+      if (roundToolCalls.length === 0) {
+        finalContent = roundContent;
+        finishReason = finishReason || "stop";
+        break;
+      }
+      for (const tc of roundToolCalls) {
+        this.toolCallCount++;
+        await globalEventBus2.emit("tool:start", {
+          agentId: this.config.id || this.config.name,
+          runId,
+          tool: tc.name,
+          args: tc.args
+        });
+        options.onToolCall?.(tc.name, tc.args);
+        yield { type: "tool_call", toolCall: tc };
+        try {
+          const tool = this.config.tools?.find((t) => t.definition.name === tc.name);
+          if (!tool) throw new ToolError(tc.name, `Tool "${tc.name}" not found on agent`);
+          const hookArgs = runBeforeToolCallHooks(tc.name, tc.args);
+          const result = await tool.execute(hookArgs);
+          const hookResult = runAfterToolCallHooks(result);
+          const resultStr = typeof hookResult === "string" ? hookResult : JSON.stringify(hookResult, null, 2);
+          this.context.add({
+            role: "tool",
+            toolCallId: tc.id,
+            name: tc.name,
+            content: resultStr
+          });
+          await globalEventBus2.emit("tool:finish", {
+            agentId: this.config.id || this.config.name,
+            runId,
+            tool: tc.name,
+            result: hookResult,
+            durationMs: Date.now() - startTime
+          });
+          options.onToolResult?.(tc.name, hookResult);
+          yield { type: "tool_result", toolCall: tc, result: hookResult };
+        } catch (err) {
+          this.context.add({
+            role: "tool",
+            toolCallId: tc.id,
+            name: tc.name,
+            content: `Error: ${err.message}`
+          });
+          await globalEventBus2.emit("tool:error", {
+            agentId: this.config.id || this.config.name,
+            runId,
+            tool: tc.name,
+            error: err.message
+          });
+          yield { type: "error", error: err.message };
+        }
+      }
+    }
+    const totalLatency = Date.now() - startTime;
+    if (this.memory) {
+      await this.memory.store({
+        id: generateId(),
+        type: "short-term",
+        key: `agent:${this.config.name}:input:${generateId()}`,
+        content: options.input,
+        metadata: { runId, agentName: this.config.name },
+        timestamp: Date.now()
+      });
+      if (finalContent) {
+        await this.memory.store({
+          id: generateId(),
+          type: "short-term",
+          key: `agent:${this.config.name}:output:${generateId()}`,
+          content: finalContent,
+          metadata: { runId, agentName: this.config.name },
+          timestamp: Date.now()
+        });
+      }
+    }
+    await globalEventBus2.emit("agent:response", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      content: finalContent,
+      tokens: this.totalCompletionTokens,
+      latencyMs: totalLatency
+    });
+    runAfterAgentRunHooks({
+      id: runId,
+      output: finalContent,
+      messages: this.context.getMessages(),
+      usage: {
+        totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+        promptTokens: this.totalPromptTokens,
+        completionTokens: this.totalCompletionTokens
+      },
+      latencyMs: totalLatency,
+      toolCalls: this.toolCallCount,
+      finishReason
+    });
+    await globalEventBus2.emit("agent:complete", {
+      agentId: this.config.id || this.config.name,
+      runId,
+      totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+      totalLatencyMs: totalLatency,
+      toolCalls: this.toolCallCount
+    });
+    yield {
+      type: "finish",
+      output: finalContent,
+      finishReason,
+      usage: {
+        totalTokens: this.totalPromptTokens + this.totalCompletionTokens,
+        promptTokens: this.totalPromptTokens,
+        completionTokens: this.totalCompletionTokens
+      },
+      toolCalls: this.toolCallCount,
+      latencyMs: totalLatency
+    };
+  }
+};
+
+// src/agent.ts
+import { generateId as generateId2 } from "@lupaflow/core";
+import { toolRegistry } from "@lupaflow/tools";
+var Agent = class {
+  config;
+  memoryStore;
+  constructor(config) {
+    validateAgentConfig(config);
+    this.config = {
+      ...config,
+      id: config.id || generateId2()
+    };
+  }
+  tool(t) {
+    this.config.tools = this.config.tools || [];
+    this.config.tools.push(t);
+    return this;
+  }
+  useTool(name) {
+    try {
+      const tool = toolRegistry.get(name);
+      this.config.tools = this.config.tools || [];
+      this.config.tools.push(tool);
+    } catch {
+      throw new Error(`Tool "${name}" not found in registry`);
+    }
+    return this;
+  }
+  memory(memory) {
+    this.memoryStore = memory;
+    return this;
+  }
+  personality(p) {
+    this.config.personality = p;
+    return this;
+  }
+  goal(g) {
+    this.config.goals = this.config.goals || [];
+    this.config.goals.push(g);
+    return this;
+  }
+  constraint(c) {
+    this.config.constraints = this.config.constraints || [];
+    this.config.constraints.push(c);
+    return this;
+  }
+  temperature(t) {
+    this.config.temperature = t;
+    return this;
+  }
+  model(m) {
+    this.config.model = m;
+    return this;
+  }
+  async run(input, options) {
+    const runner = new AgentRunner(this.config, this.memoryStore);
+    return runner.run({
+      input,
+      ...options
+    });
+  }
+  async *streamRun(input, options) {
+    const runner = new AgentRunner(this.config, this.memoryStore);
+    yield* runner.streamRun({
+      input,
+      ...options
+    });
+  }
+};
+export {
+  Agent,
+  AgentRunner,
+  ContextManager,
+  buildSystemPrompt,
+  validateAgentConfig,
+  withRetry
+};
+//# sourceMappingURL=index.js.map