Method run

integrations/replicate/src/index.ts:180–207

Run a model and await the result.

(
    key: IntegrationTaskKey,
    params: {
      identifier: Parameters<ReplicateClient["run"]>[0];
    } & Omit<
      Parameters<ReplicateClient["run"]>[1],
      "webhook" | "webhook_events_filter" | "wait" | "signal"
    >
  )

  /** Run a model and await the result. */
  run(
    key: IntegrationTaskKey,
    params: {
      identifier: Parameters<ReplicateClient["run"]>[0];
    } & Omit<
      Parameters<ReplicateClient["run"]>[1],
      "webhook" | "webhook_events_filter" | "wait" | "signal"
    >
  ): ReplicateReturnType<Prediction> {
    const { identifier, ...paramsWithoutIdentifier } = params;

    // see: https://github.com/replicate/replicate-javascript/blob/4b0d9cb0e226fab3d3d31de5b32261485acf5626/index.js#L102

    const namePattern = /[a-zA-Z0-9]+(?:(?:[._]|__|[-]*)[a-zA-Z0-9]+)*/;
    const pattern = new RegExp(
      `^(?<owner>${namePattern.source})/(?<name>${namePattern.source}):(?<version>[0-9a-fA-F]+)$`
    );

    const match = identifier.match(pattern);

    if (!match || !match.groups) {
      throw new Error('Invalid version. It must be in the format "owner/name:version"');
    }

    const { version } = match.groups;

    return this.predictions.createAndAwait(key, { ...paramsWithoutIdentifier, version });
  }

  // TODO: wait(prediction) - needs polling
}