@agentfield/sdk 0.1.91 → 0.1.92-rc.2

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;
 }
@@ -869,6 +870,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 +1784,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({
@@ -5644,6 +5780,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 +5872,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 +6039,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 +6334,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 +6347,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)
     });
   }