getaiapi 0.3.4 → 0.4.0

package/README.md

@@ -6,7 +6,7 @@
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
 [![TypeScript](https://img.shields.io/badge/TypeScript-strict-blue.svg)](https://www.typescriptlang.org/)
 
-A unified TypeScript library that wraps 1,930 AI models across 3 providers into a single `generate()` function. One input shape. One output shape. Any model.
+A unified TypeScript library that wraps 1,940+ AI models across 4 providers into a single `generate()` function. One input shape. One output shape. Any model.
 
 ## Install
 
@@ -29,6 +29,31 @@ console.log(result.outputs[0].url)
 
 ## More Examples
 
+**Text generation (LLMs)**
+
+```typescript
+const answer = await generate({
+  model: 'claude-sonnet-4-6',
+  prompt: 'Explain quantum computing in one paragraph'
+})
+
+console.log(answer.outputs[0].content)
+```
+
+With system prompt and parameters:
+
+```typescript
+const reply = await generate({
+  model: 'gpt-4o',
+  prompt: 'Write a haiku about TypeScript',
+  options: {
+    system: 'You are a creative poet.',
+    temperature: 0.9,
+    max_tokens: 100,
+  }
+})
+```
+
 **Text-to-video**
 
 ```typescript
@@ -105,6 +130,9 @@ export REPLICATE_API_TOKEN="your-replicate-token"
 
 # WaveSpeed (66 models)
 export WAVESPEED_API_KEY="your-wavespeed-key"
+
+# OpenRouter (10+ LLM models — Claude, GPT, Gemini, Llama, etc.)
+export OPENROUTER_API_KEY="your-openrouter-key"
 ```
 
 ### Option 2: Programmatic Configuration
@@ -119,6 +147,7 @@ configure({
     'fal-ai': process.env.MY_FAL_TOKEN,
     'replicate': process.env.MY_REPLICATE_TOKEN,
     'wavespeed': process.env.MY_WAVESPEED_TOKEN,
+    'openrouter': process.env.MY_OPENROUTER_TOKEN,
   },
 })
 ```
@@ -197,6 +226,7 @@ const model = getModel('flux-schnell')
 | `video-to-audio` | video | audio | 17 |
 | `upscale-video` | video | video | 15 |
 | `video-to-video` | video | video | 15 |
+| `text-generation` | text | text | 10 |
 | `moderation` | text/image/video | text | 8 |
 | `audio-to-video` | audio | video | 4 |
 | `audio-edit` | audio | audio | 2 |
@@ -209,9 +239,12 @@ const model = getModel('flux-schnell')
 | fal-ai | 1,201 | `FAL_KEY` | Native fetch |
 | Replicate | 687 | `REPLICATE_API_TOKEN` | Native fetch |
 | WaveSpeed | 66 | `WAVESPEED_API_KEY` | Native fetch |
+| OpenRouter | 10 | `OPENROUTER_API_KEY` | Native fetch |
 
 Zero external dependencies -- all provider communication uses native `fetch`.
 
+See the full [Model Directory](docs/MODELS.md) for all 1,940 models with provider availability.
+
 ## API Reference
 
 ### `generate(request: GenerateRequest): Promise<GenerateResponse>`
@@ -256,12 +289,16 @@ interface GenerateResponse {
     inference_time_ms?: number
     cost?: number
     safety_flagged?: boolean
+    tokens?: number           // total tokens (LLM only)
+    prompt_tokens?: number    // input tokens (LLM only)
+    completion_tokens?: number // output tokens (LLM only)
   }
 }
 
 interface OutputItem {
   type: 'image' | 'video' | 'audio' | 'text' | '3d' | 'segmentation'
-  url: string
+  url?: string      // URL for media outputs
+  content?: string  // text content for LLM outputs
   content_type: string
   size_bytes?: number
 }

package/dist/{chunk-RC2KZLI2.js → chunk-7UTLP4YN.js}

@@ -94,7 +94,8 @@ var StorageError = class extends GetAIApiError {
 var ENV_MAP = {
   "fal-ai": "FAL_KEY",
   replicate: "REPLICATE_API_TOKEN",
-  wavespeed: "WAVESPEED_API_KEY"
+  wavespeed: "WAVESPEED_API_KEY",
+  openrouter: "OPENROUTER_API_KEY"
 };
 var keyOverrides = /* @__PURE__ */ new Map();
 function configureAuth(keys) {
@@ -381,6 +382,9 @@ function genericExtract(data, path, type, contentType) {
     current = current[seg];
   }
   if (typeof current === "string") {
+    if (type === "text") {
+      return [{ type, content: current, content_type: contentType }];
+    }
     return [{ type, url: current, content_type: contentType }];
   }
   return [];
@@ -504,6 +508,14 @@ var imageEditTemplate = {
       },
       required: true
     },
+    {
+      universal: "images",
+      providers: {
+        "fal-ai": "image_urls",
+        "replicate": "image_urls",
+        "wavespeed": "image_urls"
+      }
+    },
     {
       universal: "prompt",
       providers: {
@@ -897,6 +909,25 @@ var removeBackgroundTemplate = {
   default_timeout_ms: 6e4
 };
 
+// src/categories/text-generation.ts
+var textGenerationTemplate = {
+  category: "text-generation",
+  input_mappings: [
+    {
+      universal: "prompt",
+      providers: {
+        openrouter: "prompt"
+      },
+      required: true
+    }
+  ],
+  output_type: "text",
+  output_extract: {
+    openrouter: "choices[0].message.content"
+  },
+  default_timeout_ms: 12e4
+};
+
 // src/categories/index.ts
 var templates = {
   "text-to-image": textToImageTemplate,
@@ -906,7 +937,8 @@ var templates = {
   "upscale-image": upscaleImageTemplate,
   "text-to-audio": textToAudioTemplate,
   "audio-to-text": audioToTextTemplate,
-  "remove-background": removeBackgroundTemplate
+  "remove-background": removeBackgroundTemplate,
+  "text-generation": textGenerationTemplate
 };
 function getCategoryTemplate(category) {
   return templates[category];
@@ -914,6 +946,11 @@ function getCategoryTemplate(category) {
 
 // src/adapters/fal-ai.ts
 var BASE_URL = "https://queue.fal.run";
+function getBaseEndpoint(endpoint) {
+  const parts = endpoint.split("/");
+  if (parts.length <= 2) return endpoint;
+  return `${parts[0]}/${parts[1]}`;
+}
 async function handleHttpErrors(response, endpoint) {
   if (response.ok) return;
   const status = response.status;
@@ -959,7 +996,8 @@ var falAiAdapter = {
     if (!endpoint) {
       throw new ProviderError("fal-ai", "unknown", 400, "endpoint is required for polling");
     }
-    const statusUrl = `${BASE_URL}/${endpoint}/requests/${taskId}/status`;
+    const baseEndpoint = getBaseEndpoint(endpoint);
+    const statusUrl = `${BASE_URL}/${baseEndpoint}/requests/${taskId}/status`;
     const statusResponse = await fetch(statusUrl, {
       headers: { Authorization: `Key ${auth}` }
     });
@@ -978,7 +1016,7 @@ var falAiAdapter = {
         status: "processing"
       };
     }
-    const resultUrl = `${BASE_URL}/${endpoint}/requests/${taskId}`;
+    const resultUrl = `${BASE_URL}/${baseEndpoint}/requests/${taskId}`;
     const resultResponse = await fetch(resultUrl, {
       headers: { Authorization: `Key ${auth}` }
     });
@@ -1002,6 +1040,17 @@ var falAiAdapter = {
         content_type: img.content_type ?? outputMapping.content_type ?? "image/jpeg"
       }));
     }
+    if (path === "image.url") {
+      const image = data.image;
+      if (!image?.url) return [];
+      return [
+        {
+          type: outputMapping.type,
+          url: image.url,
+          content_type: image.content_type ?? outputMapping.content_type ?? "image/png"
+        }
+      ];
+    }
     if (path === "video.url") {
       const video = data.video;
       if (!video?.url) return [];
@@ -1024,6 +1073,41 @@ var falAiAdapter = {
         }
       ];
     }
+    if (path === "audio_file.url") {
+      const audioFile = data.audio_file;
+      if (!audioFile?.url) return [];
+      return [
+        {
+          type: outputMapping.type,
+          url: audioFile.url,
+          content_type: audioFile.content_type ?? outputMapping.content_type ?? "audio/mpeg"
+        }
+      ];
+    }
+    if (path === "audio_url") {
+      const audioUrl = data.audio_url;
+      if (!audioUrl) return [];
+      const url = typeof audioUrl === "string" ? audioUrl : audioUrl.url;
+      if (!url) return [];
+      return [
+        {
+          type: outputMapping.type,
+          url,
+          content_type: (typeof audioUrl === "object" ? audioUrl.content_type : void 0) ?? outputMapping.content_type ?? "audio/mpeg"
+        }
+      ];
+    }
+    if (path === "video_url") {
+      const videoUrl = data.video_url;
+      if (!videoUrl) return [];
+      return [
+        {
+          type: outputMapping.type,
+          url: videoUrl,
+          content_type: outputMapping.content_type ?? "video/mp4"
+        }
+      ];
+    }
     return [];
   }
 };
@@ -1221,6 +1305,87 @@ var wavespeedAdapter = {
   }
 };
 
+// src/adapters/openrouter.ts
+var BASE_URL4 = "https://openrouter.ai/api/v1";
+async function handleHttpErrors4(response, endpoint) {
+  if (response.ok) return;
+  const status = response.status;
+  if (status === 401) {
+    throw new AuthError("openrouter", "OPENROUTER_API_KEY");
+  }
+  if (status === 429) {
+    const retryAfter = response.headers.get("retry-after");
+    const retryMs = retryAfter ? parseInt(retryAfter, 10) * 1e3 : 6e4;
+    throw new RateLimitError("openrouter", retryMs);
+  }
+  let raw;
+  try {
+    raw = await response.json();
+  } catch {
+    raw = await response.text().catch(() => null);
+  }
+  throw new ProviderError("openrouter", endpoint, status, raw);
+}
+function authHeaders4(auth) {
+  return {
+    Authorization: `Bearer ${auth}`,
+    "Content-Type": "application/json",
+    "X-Title": "getaiapi"
+  };
+}
+var openRouterAdapter = {
+  name: "openrouter",
+  async submit(endpoint, params, auth) {
+    const { prompt, system, temperature, max_tokens, top_p, ...rest } = params;
+    const messages = [];
+    if (system && typeof system === "string") {
+      messages.push({ role: "system", content: system });
+    }
+    messages.push({ role: "user", content: prompt ?? "" });
+    const body = {
+      model: endpoint,
+      messages,
+      ...rest
+    };
+    if (temperature !== void 0) body.temperature = temperature;
+    if (max_tokens !== void 0) body.max_tokens = max_tokens;
+    if (top_p !== void 0) body.top_p = top_p;
+    const url = `${BASE_URL4}/chat/completions`;
+    const response = await fetch(url, {
+      method: "POST",
+      headers: authHeaders4(auth),
+      body: JSON.stringify(body)
+    });
+    await handleHttpErrors4(response, endpoint);
+    const data = await response.json();
+    return {
+      id: data.id,
+      status: "completed",
+      output: data
+    };
+  },
+  async poll(taskId) {
+    return {
+      id: taskId,
+      status: "completed"
+    };
+  },
+  parseOutput(raw, outputMapping) {
+    const data = raw;
+    const choices = data.choices;
+    if (!Array.isArray(choices) || choices.length === 0) return [];
+    const content = choices[0].message?.content;
+    if (typeof content !== "string") return [];
+    return [
+      {
+        type: outputMapping.type,
+        content,
+        content_type: outputMapping.content_type ?? "text/plain"
+      }
+    ];
+  }
+};
+
 // src/retry.ts
 var DEFAULT_OPTIONS = {
   maxRetries: 3,
@@ -1614,7 +1779,8 @@ async function processParamsForUpload(params, options) {
 var adapters = {
   "fal-ai": falAiAdapter,
   "replicate": replicateAdapter,
-  "wavespeed": wavespeedAdapter
+  "wavespeed": wavespeedAdapter,
+  "openrouter": openRouterAdapter
 };
 async function generate(request) {
   const startTime = Date.now();
@@ -1660,17 +1826,25 @@ async function generate(request) {
     );
   }
   const outputs = mapOutput(result.output, binding.output_map);
+  const rawOutput = typeof result.output === "object" && result.output !== null ? result.output : void 0;
+  const metadata = {
+    inference_time_ms: Date.now() - startTime,
+    seed: rawOutput?.seed,
+    safety_flagged: rawOutput ? Array.isArray(rawOutput.has_nsfw_concepts) ? rawOutput.has_nsfw_concepts.some((v) => v) : void 0 : void 0
+  };
+  if (rawOutput?.usage) {
+    const usage = rawOutput.usage;
+    metadata.tokens = usage.total_tokens;
+    metadata.prompt_tokens = usage.prompt_tokens;
+    metadata.completion_tokens = usage.completion_tokens;
+  }
   return {
     id: randomUUID2(),
     model: model.canonical_name,
     provider: binding.provider,
     status: "completed",
     outputs,
-    metadata: {
-      inference_time_ms: Date.now() - startTime,
-      seed: typeof result.output === "object" && result.output !== null ? result.output.seed : void 0,
-      safety_flagged: typeof result.output === "object" && result.output !== null ? Array.isArray(result.output.has_nsfw_concepts) ? result.output.has_nsfw_concepts.some((v) => v) : void 0 : void 0
-    }
+    metadata
   };
 }
 
@@ -1730,4 +1904,4 @@ export {
   listModels,
   getModel
 };
-//# sourceMappingURL=chunk-RC2KZLI2.js.map
+//# sourceMappingURL=chunk-7UTLP4YN.js.map