@agentfield/sdk 0.1.91 → 0.1.92-rc.1

package/dist/index.d.ts

@@ -45,6 +45,7 @@ declare class AIClient {
     getModel(options?: AIRequestOptions): _ai_sdk_provider.LanguageModelV3;
     private buildModel;
     private buildEmbeddingModel;
+    private openRouterHeaders;
     private getRateLimiter;
     private withRateLimitRetry;
 }
@@ -272,6 +273,14 @@ interface ExecutionStatusUpdate {
     progress?: number;
     statusReason?: string;
 }
+interface RestartExecutionOptions {
+    scope?: 'workflow' | 'execution';
+    reuse?: 'succeeded-before' | 'all-succeeded' | 'none';
+    fork?: boolean;
+    reason?: string;
+    input?: Record<string, unknown>;
+    context?: Record<string, unknown>;
+}
 declare class AgentFieldClient {
     private readonly http;
     private readonly config;
@@ -293,7 +302,11 @@ declare class AgentFieldClient {
         targetDid?: string;
         agentNodeDid?: string;
         agentNodeId?: string;
+        replaySourceRunId?: string;
+        replayBeforeExecutionId?: string;
+        replayMode?: string;
     }): Promise<T>;
+    restartExecution(executionId: string, options?: RestartExecutionOptions): Promise<any>;
     publishWorkflowEvent(event: {
         executionId: string;
         runId: string;
@@ -368,6 +381,9 @@ interface ExecutionMetadata {
     callerDid?: string;
     targetDid?: string;
     agentNodeDid?: string;
+    replaySourceRunId?: string;
+    replayBeforeExecutionId?: string;
+    replayMode?: string;
 }
 declare class ExecutionContext {
     readonly input: any;
@@ -869,6 +885,9 @@ interface AIConfig {
     embeddingModel?: string;
     apiKey?: string;
     baseUrl?: string;
+    openRouterSiteUrl?: string;
+    openRouterAppName?: string;
+    openRouterHeaders?: Record<string, string>;
     temperature?: number;
     maxTokens?: number;
     enableRateLimitRetry?: boolean;
@@ -1780,11 +1799,15 @@ declare class MediaRouter {
 interface OpenRouterMediaProviderOptions {
     apiKey?: string;
     baseUrl?: string;
+    openRouterSiteUrl?: string;
+    openRouterAppName?: string;
+    openRouterHeaders?: Record<string, string>;
 }
 declare class OpenRouterMediaProvider implements MediaProvider {
     readonly name = "openrouter";
     readonly supportedModalities: string[];
     private readonly baseUrl;
+    private readonly attributionHeaders;
     constructor(options?: OpenRouterMediaProviderOptions);
     /**
      * Seed the metadata cache for a model. Useful when running against test

package/dist/index.js

@@ -34,6 +34,107 @@ var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
+
+// src/ai/openrouterAttribution.ts
+function isOpenRouterRequest(options) {
+  const provider = clean(options.provider)?.toLowerCase();
+  if (provider === "openrouter") {
+    return true;
+  }
+  const model = clean(options.model)?.toLowerCase();
+  if (model?.startsWith("openrouter/")) {
+    return true;
+  }
+  const baseUrl = clean(options.baseUrl)?.toLowerCase();
+  return Boolean(baseUrl?.includes("openrouter.ai"));
+}
+function openRouterAttributionEnabled(env = process.env) {
+  const raw = clean(env.AGENTFIELD_OPENROUTER_ATTRIBUTION);
+  if (!raw) {
+    return true;
+  }
+  return !["0", "false", "no", "off"].includes(raw.toLowerCase());
+}
+function resolveOpenRouterAttribution(options = {}) {
+  const env = options.env ?? process.env;
+  if (!openRouterAttributionEnabled(env)) {
+    return void 0;
+  }
+  const siteUrl = clean(options.siteUrl) ?? clean(env.AGENTFIELD_OPENROUTER_SITE_URL) ?? clean(env.OR_SITE_URL) ?? DEFAULT_OPENROUTER_SITE_URL;
+  const appName = clean(options.appName) ?? clean(env.AGENTFIELD_OPENROUTER_APP_NAME) ?? clean(env.OR_APP_NAME) ?? DEFAULT_OPENROUTER_APP_NAME;
+  return { siteUrl, appName };
+}
+function openRouterAttributionHeaders(options = {}) {
+  const resolved = resolveOpenRouterAttribution(options);
+  if (!resolved) {
+    return {};
+  }
+  return {
+    "HTTP-Referer": resolved.siteUrl,
+    "X-OpenRouter-Title": resolved.appName,
+    "X-Title": resolved.appName
+  };
+}
+function mergeOpenRouterAttributionHeaders(existing = {}, options = {}) {
+  const merged = {};
+  const lowerKeys = /* @__PURE__ */ new Set();
+  for (const [key, value] of Object.entries(existing)) {
+    const cleaned = clean(value);
+    if (!cleaned) {
+      continue;
+    }
+    merged[key] = cleaned;
+    lowerKeys.add(key.toLowerCase());
+  }
+  for (const [key, value] of Object.entries(openRouterAttributionHeaders(options))) {
+    if (!lowerKeys.has(key.toLowerCase())) {
+      merged[key] = value;
+    }
+  }
+  return merged;
+}
+function openRouterAttributionEnv(env = process.env) {
+  const resolved = resolveOpenRouterAttribution({ env });
+  if (!resolved) {
+    return {};
+  }
+  return {
+    AGENTFIELD_OPENROUTER_SITE_URL: resolved.siteUrl,
+    AGENTFIELD_OPENROUTER_APP_NAME: resolved.appName,
+    OR_SITE_URL: resolved.siteUrl,
+    OR_APP_NAME: resolved.appName
+  };
+}
+function applyOpenRouterAttributionEnv(env) {
+  if (!openRouterAttributionEnabled(env)) {
+    for (const key of [
+      "AGENTFIELD_OPENROUTER_SITE_URL",
+      "AGENTFIELD_OPENROUTER_APP_NAME",
+      "OR_SITE_URL",
+      "OR_APP_NAME"
+    ]) {
+      delete env[key];
+    }
+    return;
+  }
+  const attribution = openRouterAttributionEnv(env);
+  for (const [key, value] of Object.entries(attribution)) {
+    if (clean(env[key]) == null) {
+      env[key] = value;
+    }
+  }
+}
+function clean(value) {
+  const trimmed = value?.trim();
+  return trimmed ? trimmed : void 0;
+}
+var DEFAULT_OPENROUTER_SITE_URL, DEFAULT_OPENROUTER_APP_NAME;
+var init_openrouterAttribution = __esm({
+  "src/ai/openrouterAttribution.ts"() {
+    DEFAULT_OPENROUTER_SITE_URL = "https://agentfield.ai";
+    DEFAULT_OPENROUTER_APP_NAME = "AgentField AI";
+  }
+});
 function getZodConverter() {
   if (zodConverter !== void 0) {
     return zodConverter;
@@ -340,8 +441,10 @@ var init_claude = __esm({
 function runCli(cmd, options) {
   return new Promise((resolve2, reject) => {
     const [bin, ...args] = cmd;
+    const env = { ...process.env, ...options?.env };
+    applyOpenRouterAttributionEnv(env);
     const proc = spawn(bin, args, {
-      env: { ...process.env, ...options?.env },
+      env,
       cwd: options?.cwd,
       stdio: ["pipe", "pipe", "pipe"]
     });
@@ -414,6 +517,7 @@ function extractFinalText(events) {
 }
 var init_cli = __esm({
   "src/harness/cli.ts"() {
+    init_openrouterAttribution();
   }
 });
 
@@ -566,6 +670,7 @@ var init_opencode = __esm({
   "src/harness/providers/opencode.ts"() {
     init_types();
     init_cli();
+    init_openrouterAttribution();
     OpenCodeProvider = class {
       bin;
       constructor(binPath = "opencode") {
@@ -593,6 +698,22 @@ USER REQUEST:
 ${prompt}`;
         }
         cmd.push(effectivePrompt);
+        const explicitModel = typeof options.model === "string" ? options.model : void 0;
+        if (explicitModel && isOpenRouterRequest({ model: explicitModel }) && !env.OPENCODE_CONFIG_CONTENT && !process.env.OPENCODE_CONFIG_CONTENT) {
+          const modelSlug = explicitModel.slice("openrouter/".length);
+          const headers = openRouterAttributionHeaders({ env: { ...process.env, ...env } });
+          if (modelSlug && Object.keys(headers).length > 0) {
+            env.OPENCODE_CONFIG_CONTENT = JSON.stringify({
+              provider: {
+                openrouter: {
+                  models: {
+                    [modelSlug]: { headers }
+                  }
+                }
+              }
+            });
+          }
+        }
         const startApi = Date.now();
         try {
           const { stdout, stderr, exitCode } = await runCli(cmd, { env });
@@ -1671,6 +1792,7 @@ function toError(error) {
 }
 
 // src/ai/AIClient.ts
+init_openrouterAttribution();
 function repairJsonText(text2) {
   let cleaned = text2.trim();
   const codeBlockMatch = cleaned.match(/```(?:json)?\s*([\s\S]*?)```/);
@@ -1772,6 +1894,7 @@ var AIClient = class {
   buildModel(options) {
     const provider = options.provider ?? this.config.provider ?? "openai";
     const modelName = options.model ?? this.config.model ?? "gpt-4o";
+    const openRouterHeaders = this.openRouterHeaders(provider, modelName);
     switch (provider) {
       case "anthropic": {
         const anthropic = createAnthropic({
@@ -1825,7 +1948,8 @@ var AIClient = class {
       case "openrouter": {
         const openrouter = createOpenAI({
           apiKey: this.config.apiKey,
-          baseURL: this.config.baseUrl ?? "https://openrouter.ai/api/v1"
+          baseURL: this.config.baseUrl ?? "https://openrouter.ai/api/v1",
+          headers: openRouterHeaders
         });
         return openrouter.chat(modelName);
       }
@@ -1841,7 +1965,8 @@ var AIClient = class {
       default: {
         const openai = createOpenAI({
           apiKey: this.config.apiKey,
-          baseURL: this.config.baseUrl
+          baseURL: this.config.baseUrl,
+          ...openRouterHeaders ? { headers: openRouterHeaders } : {}
         });
         return openai(modelName);
       }
@@ -1850,6 +1975,7 @@ var AIClient = class {
   buildEmbeddingModel(options) {
     const provider = options.provider ?? this.config.provider ?? "openai";
     const modelName = options.model ?? this.config.embeddingModel ?? "text-embedding-3-small";
+    const openRouterHeaders = this.openRouterHeaders(provider, modelName);
     const noEmbeddingProviders = ["anthropic", "xai", "deepseek", "groq"];
     if (noEmbeddingProviders.includes(provider)) {
       throw new Error(`Embedding generation is not supported for ${provider} provider`);
@@ -1882,12 +2008,22 @@ var AIClient = class {
       default: {
         const openai = createOpenAI({
           apiKey: this.config.apiKey ?? (provider === "ollama" ? "ollama" : void 0),
-          baseURL: this.config.baseUrl ?? (provider === "openrouter" ? "https://openrouter.ai/api/v1" : provider === "ollama" ? "http://localhost:11434/v1" : void 0)
+          baseURL: this.config.baseUrl ?? (provider === "openrouter" ? "https://openrouter.ai/api/v1" : provider === "ollama" ? "http://localhost:11434/v1" : void 0),
+          ...openRouterHeaders ? { headers: openRouterHeaders } : {}
         });
         return openai.embedding(modelName);
       }
     }
   }
+  openRouterHeaders(provider, model) {
+    if (!isOpenRouterRequest({ provider, model, baseUrl: this.config.baseUrl })) {
+      return void 0;
+    }
+    return mergeOpenRouterAttributionHeaders(this.config.openRouterHeaders, {
+      siteUrl: this.config.openRouterSiteUrl,
+      appName: this.config.openRouterAppName
+    });
+  }
   getRateLimiter() {
     if (!this.rateLimiter) {
       this.rateLimiter = new StatelessRateLimiter({
@@ -2187,6 +2323,9 @@ var AgentFieldClient = class {
     if (metadata?.targetDid) headers["X-Target-DID"] = metadata.targetDid;
     if (metadata?.agentNodeDid) headers["X-Agent-Node-DID"] = metadata.agentNodeDid;
     if (metadata?.agentNodeId) headers["X-Agent-Node-ID"] = metadata.agentNodeId;
+    if (metadata?.replaySourceRunId) headers["X-AgentField-Replay-Source-Run-ID"] = metadata.replaySourceRunId;
+    if (metadata?.replayBeforeExecutionId) headers["X-AgentField-Replay-Before-Execution-ID"] = metadata.replayBeforeExecutionId;
+    if (metadata?.replayMode) headers["X-AgentField-Replay-Mode"] = metadata.replayMode;
     const bodyStr = JSON.stringify({ input });
     const authHeaders = this.didAuthenticator.signRequest(Buffer.from(bodyStr));
     try {
@@ -2213,6 +2352,44 @@ var AgentFieldClient = class {
       throw err;
     }
   }
+  async restartExecution(executionId, options = {}) {
+    if (!executionId) {
+      throw new Error("executionId is required");
+    }
+    const body = {
+      scope: options.scope ?? "workflow",
+      reuse: options.reuse ?? "succeeded-before",
+      fork: options.fork,
+      reason: options.reason,
+      input: options.input,
+      context: options.context
+    };
+    const bodyStr = JSON.stringify(body);
+    const authHeaders = this.didAuthenticator.signRequest(Buffer.from(bodyStr));
+    try {
+      const res = await this.http.post(
+        `/api/v1/executions/${encodeURIComponent(executionId)}/restart`,
+        bodyStr,
+        { headers: this.mergeHeaders({ "Content-Type": "application/json", ...authHeaders }) }
+      );
+      return res.data;
+    } catch (err) {
+      const respData = err?.response?.data;
+      if (respData) {
+        const status = err.response.status;
+        const msg = respData.message || respData.error || JSON.stringify(respData);
+        const enriched = Object.assign(
+          new Error(`restart execution ${executionId} failed (${status}): ${msg}`),
+          {
+            status,
+            responseData: respData
+          }
+        );
+        throw enriched;
+      }
+      throw err;
+    }
+  }
   async publishWorkflowEvent(event) {
     const payload = {
       execution_id: event.executionId,
@@ -2409,6 +2586,9 @@ var AgentFieldClient = class {
     if (metadata.targetDid) headers["x-target-did"] = metadata.targetDid;
     if (metadata.agentNodeDid) headers["x-agent-node-did"] = metadata.agentNodeDid;
     if (metadata.agentNodeId) headers["x-agent-node-id"] = metadata.agentNodeId;
+    if (metadata.replaySourceRunId) headers["x-agentfield-replay-source-run-id"] = metadata.replaySourceRunId;
+    if (metadata.replayBeforeExecutionId) headers["x-agentfield-replay-before-execution-id"] = metadata.replayBeforeExecutionId;
+    if (metadata.replayMode) headers["x-agentfield-replay-mode"] = metadata.replayMode;
     return headers;
   }
   setDIDCredentials(did, privateKeyJwk) {
@@ -4116,7 +4296,10 @@ var Agent = class {
         callerDid: parentMetadata?.callerDid,
         targetDid: parentMetadata?.targetDid,
         agentNodeDid: parentMetadata?.agentNodeDid,
-        agentNodeId: this.config.nodeId
+        agentNodeId: this.config.nodeId,
+        replaySourceRunId: parentMetadata?.replaySourceRunId,
+        replayBeforeExecutionId: parentMetadata?.replayBeforeExecutionId,
+        replayMode: parentMetadata?.replayMode
       });
       this.executionLogger.system("agent.call.completed", "Remote agent call completed", {
         target,
@@ -4382,7 +4565,10 @@ var Agent = class {
       reasonerId: overrides?.reasonerId ?? normalized["x-reasoner-id"],
       callerDid: overrides?.callerDid ?? normalized["x-caller-did"],
       targetDid: overrides?.targetDid ?? normalized["x-target-did"],
-      agentNodeDid: overrides?.agentNodeDid ?? normalized["x-agent-node-did"] ?? normalized["x-agent-did"]
+      agentNodeDid: overrides?.agentNodeDid ?? normalized["x-agent-node-did"] ?? normalized["x-agent-did"],
+      replaySourceRunId: overrides?.replaySourceRunId ?? normalized["x-agentfield-replay-source-run-id"],
+      replayBeforeExecutionId: overrides?.replayBeforeExecutionId ?? normalized["x-agentfield-replay-before-execution-id"],
+      replayMode: overrides?.replayMode ?? normalized["x-agentfield-replay-mode"]
     };
   }
   handleHttpRequest(req, res) {
@@ -5644,6 +5830,7 @@ var MediaRouter = class {
 };
 
 // src/ai/OpenRouterMediaProvider.ts
+init_openrouterAttribution();
 var OPENROUTER_BASE = "https://openrouter.ai/api/v1";
 var DEFAULT_POLL_INTERVAL = 3e4;
 var DEFAULT_TIMEOUT = 6e5;
@@ -5735,9 +5922,14 @@ var OpenRouterMediaProvider = class {
   name = "openrouter";
   supportedModalities = ["image", "audio", "video"];
   baseUrl;
+  attributionHeaders;
   constructor(options = {}) {
     const key = options.apiKey ?? process.env.OPENROUTER_API_KEY ?? "";
     this.baseUrl = options.baseUrl ?? OPENROUTER_BASE;
+    this.attributionHeaders = mergeOpenRouterAttributionHeaders(options.openRouterHeaders, {
+      siteUrl: options.openRouterSiteUrl,
+      appName: options.openRouterAppName
+    });
     if (!key) {
       throw new MediaProviderError("OpenRouter API key required: pass apiKey or set OPENROUTER_API_KEY", {
         provider: "openrouter"
@@ -5897,7 +6089,10 @@ var OpenRouterMediaProvider = class {
       } catch {
       }
       const dlRes = await fetch(videoUrl, {
-        headers: downloadHeaders,
+        headers: mergeOpenRouterAttributionHeaders({
+          ...this.attributionHeaders,
+          ...downloadHeaders
+        }),
         signal: AbortSignal.timeout(DOWNLOAD_TIMEOUT),
         redirect: "error"
       });
@@ -6189,10 +6384,11 @@ var OpenRouterMediaProvider = class {
     const key = apiKeyStore.get(this);
     return fetch(url, {
       method: "POST",
-      headers: {
+      headers: mergeOpenRouterAttributionHeaders({
+        ...this.attributionHeaders,
         "Content-Type": "application/json",
         Authorization: `Bearer ${key}`
-      },
+      }),
       body: JSON.stringify(body),
       signal: AbortSignal.timeout(API_TIMEOUT)
     });
@@ -6201,9 +6397,10 @@ var OpenRouterMediaProvider = class {
     const key = apiKeyStore.get(this);
     return fetch(url, {
       method: "GET",
-      headers: {
+      headers: mergeOpenRouterAttributionHeaders({
+        ...this.attributionHeaders,
         Authorization: `Bearer ${key}`
-      },
+      }),
       signal: AbortSignal.timeout(API_TIMEOUT)
     });
   }