replicate-stainless
TypeScript icon, indicating that this package has built-in type declarations

0.7.1 • Public • Published

Replicate TypeScript API Library

Banner

NPM version npm bundle size

This library provides convenient access to the Replicate REST API from server-side TypeScript or JavaScript.

The REST API documentation can be found on replicate.com. The full API of this library can be found in api.md.

It is generated with Stainless.

Installation

npm install replicate-stainless

Usage

The full API of this library can be found in api.md.

import Replicate from 'replicate-stainless';

const client = new Replicate({
  bearerToken: process.env['REPLICATE_API_TOKEN'], // This is the default and can be omitted
});

async function main() {
  const prediction = await client.predictions.get({ prediction_id: 'gm3qorzdhgbfurvjtvhg6dckhu' });

  console.log(prediction.id);
}

main();

The SDK exposes different layers of abstraction depending on what you are trying to achieve.

Run a model — replicate.run()

const output = await replicate.run('replicate/llama-3:latest', { prompt: 'Hello there!' });
console.log(output);

replicate.run raises ModelError if the prediction fails. You can access the exception's prediction property to get more information about the failure.

Resource API — replicate.predictions.*, replicate.models.*, …

The full API, including all definitions for all request params and responses, is exposed in the client. The full list of methods can be found in api.md.

You may import and use them like so:

import Replicate from 'replicate-stainless';

const client = new Replicate({
  bearerToken: process.env['REPLICATE_API_TOKEN'], // This is the default and can be omitted
});

async function main() {
  const params: Replicate.PredictionCreateParams = {
    input: { text: 'Alice' },
    version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa',
  };
  const prediction: Replicate.Prediction = await client.predictions.create(params);
}

  const collections = replicate.collections.list();

  const accounts = await replicate.accounts.list();
}
main();

Documentation for each method, request param, and response field are available in docstrings and will appear on hover in most modern editors: hover

Low‑level requests — replicate.get() / replicate.post() / …

The SDK also exposes methods to make generic requests to the API. This is helpful for handling experimental or undocumented endpoints, or other non-standard requests.

await replicate.post('/v1/experimental/feature', {
  body: { foo: 'bar' },
  query: { alpha: true },
});

File uploads

Request parameters that correspond to file uploads can be passed in many different forms:

  • File (or an object with the same structure)
  • a fetch Response (or an object with the same structure)
  • an fs.ReadStream
  • the return value of our toFile helper
import fs from 'fs';
import Replicate, { toFile } from 'replicate-stainless';

const client = new Replicate();

// If you have access to Node `fs` we recommend using `fs.createReadStream()`:
await client.files.create({ content: fs.createReadStream('/path/to/file') });

// Or if you have the web `File` API you can pass a `File` instance:
await client.files.create({ content: new File(['my bytes'], 'file') });

// You can also pass a `fetch` `Response`:
await client.files.create({ content: await fetch('https://somesite/file') });

// Finally, if none of the above are convenient, you can use our `toFile` helper:
await client.files.create({ content: await toFile(Buffer.from('my bytes'), 'file') });
await client.files.create({ content: await toFile(new Uint8Array([0, 1, 2]), 'file') });

Handling errors

When the library is unable to connect to the API, or if the API returns a non-success status code (i.e., 4xx or 5xx response), a subclass of APIError will be thrown:

async function main() {
  const prediction = await client.predictions
    .create({
      input: { text: 'Alice' },
      version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa',
    })
    .catch(async (err) => {
      if (err instanceof Replicate.APIError) {
        console.log(err.status); // 400
        console.log(err.name); // BadRequestError
        console.log(err.headers); // {server: 'nginx', ...}
      } else {
        throw err;
      }
    });
}

main();

Error codes are as follows:

Status Code Error Type
400 BadRequestError
401 AuthenticationError
403 PermissionDeniedError
404 NotFoundError
422 UnprocessableEntityError
429 RateLimitError
>=500 InternalServerError
N/A APIConnectionError

Retries

Certain errors will be automatically retried 2 times by default, with a short exponential backoff. Connection errors (for example, due to a network connectivity problem), 408 Request Timeout, 409 Conflict, 429 Rate Limit, and >=500 Internal errors will all be retried by default.

You can use the maxRetries option to configure or disable this:

// Configure the default for all requests:
const client = new Replicate({
  maxRetries: 0, // default is 2
});

// Or, configure per-request:
await client.predictions.create({ input: { text: 'Alice' }, version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa' }, {
  maxRetries: 5,
});

Timeouts

Requests time out after 1 minute by default. You can configure this with a timeout option:

// Configure the default for all requests:
const client = new Replicate({
  timeout: 20 * 1000, // 20 seconds (default is 1 minute)
});

// Override per-request:
await client.predictions.create({ input: { text: 'Alice' }, version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa' }, {
  timeout: 5 * 1000,
});

On timeout, an APIConnectionTimeoutError is thrown.

Note that requests which time out will be retried twice by default.

Auto-pagination

List methods in the Replicate API are paginated. You can use the for await … of syntax to iterate through items across all pages:

async function fetchAllModels(params) {
  const allModels = [];
  // Automatically fetches more pages as needed.
  for await (const modelListResponse of client.models.list()) {
    allModels.push(modelListResponse);
  }
  return allModels;
}

Alternatively, you can request a single page at a time:

let page = await client.models.list();
for (const modelListResponse of page.results) {
  console.log(modelListResponse);
}

// Convenience methods are provided for manually paginating:
while (page.hasNextPage()) {
  page = await page.getNextPage();
  // ...
}

Advanced Usage

Accessing raw Response data (e.g., headers)

The "raw" Response returned by fetch() can be accessed through the .asResponse() method on the APIPromise type that all methods return. This method returns as soon as the headers for a successful response are received and does not consume the response body, so you are free to write custom parsing or streaming logic.

You can also use the .withResponse() method to get the raw Response along with the parsed data. Unlike .asResponse() this method consumes the body, returning once it is parsed.

const client = new Replicate();

const response = await client.predictions
  .create({
    input: { text: 'Alice' },
    version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa',
  })
  .asResponse();
console.log(response.headers.get('X-My-Header'));
console.log(response.statusText); // access the underlying Response object

const { data: prediction, response: raw } = await client.predictions
  .create({
    input: { text: 'Alice' },
    version: 'replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa',
  })
  .withResponse();
console.log(raw.headers.get('X-My-Header'));
console.log(prediction.id);

Logging

[!IMPORTANT] All log messages are intended for debugging only. The format and content of log messages may change between releases.

Log levels

The log level can be configured in two ways:

  1. Via the REPLICATE_LOG environment variable
  2. Using the logLevel client option (overrides the environment variable if set)
import Replicate from 'replicate-stainless';

const client = new Replicate({
  logLevel: 'debug', // Show all log messages
});

Available log levels, from most to least verbose:

  • 'debug' - Show debug messages, info, warnings, and errors
  • 'info' - Show info messages, warnings, and errors
  • 'warn' - Show warnings and errors (default)
  • 'error' - Show only errors
  • 'off' - Disable all logging

At the 'debug' level, all HTTP requests and responses are logged, including headers and bodies. Some authentication-related headers are redacted, but sensitive data in request and response bodies may still be visible.

Custom logger

By default, this library logs to globalThis.console. You can also provide a custom logger. Most logging libraries are supported, including pino, winston, bunyan, consola, signale, and @std/log. If your logger doesn't work, please open an issue.

When providing a custom logger, the logLevel option still controls which messages are emitted, messages below the configured level will not be sent to your logger.

import Replicate from 'replicate-stainless';
import pino from 'pino';

const logger = pino();

const client = new Replicate({
  logger: logger.child({ name: 'Replicate' }),
  logLevel: 'debug', // Send all messages to pino, allowing it to filter
});

Making custom/undocumented requests

This library is typed for convenient access to the documented API. If you need to access undocumented endpoints, params, or response properties, the library can still be used.

Undocumented endpoints

To make requests to undocumented endpoints, you can use client.get, client.post, and other HTTP verbs. Options on the client, such as retries, will be respected when making these requests.

await replicate.post('/some/path', {
  body: { some_prop: 'foo' },
  query: { some_query_arg: 'bar' },
});

Undocumented request params

To make requests using undocumented parameters, you may use // @ts-expect-error on the undocumented parameter. This library doesn't validate at runtime that the request matches the type, so any extra values you send will be sent as-is.

client.foo.create({
  foo: 'my_param',
  bar: 12,
  // @ts-expect-error baz is not yet public
  baz: 'undocumented option',
});

For requests with the GET verb, any extra params will be in the query, all other requests will send the extra param in the body.

If you want to explicitly send an extra argument, you can do so with the query, body, and headers request options.

Undocumented response properties

To access undocumented response properties, you may access the response object with // @ts-expect-error on the response object, or cast the response object to the requisite type. Like the request params, we do not validate or strip extra properties from the response from the API.

Customizing the fetch client

By default, this library expects a global fetch function is defined.

If you want to use a different fetch function, you can either polyfill the global:

import fetch from 'my-fetch';

globalThis.fetch = fetch;

Or pass it to the client:

import Replicate from 'replicate-stainless';
import fetch from 'my-fetch';

const client = new Replicate({ fetch });

Fetch options

If you want to set custom fetch options without overriding the fetch function, you can provide a fetchOptions object when instantiating the client or making a request. (Request-specific options override client options.)

import Replicate from 'replicate-stainless';

const client = new Replicate({
  fetchOptions: {
    // `RequestInit` options
  },
});

Configuring proxies

To modify proxy behavior, you can provide custom fetchOptions that add runtime-specific proxy options to requests:

Node [docs]

import Replicate from 'replicate-stainless';
import * as undici from 'undici';

const proxyAgent = new undici.ProxyAgent('http://localhost:8888');
const client = new Replicate({
  fetchOptions: {
    dispatcher: proxyAgent,
  },
});

Bun [docs]

import Replicate from 'replicate-stainless';

const client = new Replicate({
  fetchOptions: {
    proxy: 'http://localhost:8888',
  },
});

Deno [docs]

import Replicate from 'npm:replicate-stainless';

const httpClient = Deno.createHttpClient({ proxy: { url: 'http://localhost:8888' } });
const client = new Replicate({
  fetchOptions: {
    client: httpClient,
  },
});

Frequently Asked Questions

Semantic versioning

This package generally follows SemVer conventions, though certain backwards-incompatible changes may be released as minor versions:

  1. Changes that only affect static types, without breaking runtime behavior.
  2. Changes to library internals which are technically public but not intended or documented for external use. (Please open a GitHub issue to let us know if you are relying on such internals.)
  3. Changes that we do not expect to impact the vast majority of users in practice.

We take backwards-compatibility seriously and work hard to ensure you can rely on a smooth upgrade experience.

We are keen for your feedback; please open an issue with questions, bugs, or suggestions.

Requirements

TypeScript >= 4.9 is supported.

The following runtimes are supported:

  • Web browsers (Up-to-date Chrome, Firefox, Safari, Edge, and more)
  • Node.js 20 LTS or later (non-EOL) versions.
  • Deno v1.28.0 or higher.
  • Bun 1.0 or later.
  • Cloudflare Workers.
  • Vercel Edge Runtime.
  • Jest 28 or greater with the "node" environment ("jsdom" is not supported at this time).
  • Nitro v2.6 or greater.

Note that React Native is not supported at this time.

If you are interested in other runtime environments, please open or upvote an issue on GitHub.

Contributing

See the contributing documentation.

Readme

Keywords

none

Package Sidebar

Install

npm i replicate-stainless

Weekly Downloads

215

Version

0.7.1

License

Apache-2.0

Unpacked Size

1.18 MB

Total Files

591

Last publish

Collaborators

  • replicatebot