Method createAndAwait

integrations/replicate/src/trainings.ts:51–84

Create a new training and await the result.

(
    key: IntegrationTaskKey,
    params: {
      model_owner: string;
      model_name: string;
      version_id: string;
    } & Omit<
      Parameters<ReplicateClient["trainings"]["create"]>[3],
      "webhook" | "webhook_events_filter"
    >,
    options: CallbackTimeout = { timeoutInSeconds: 3600 }
  )

  /** Create a new training and await the result. */
  createAndAwait(
    key: IntegrationTaskKey,
    params: {
      model_owner: string;
      model_name: string;
      version_id: string;
    } & Omit<
      Parameters<ReplicateClient["trainings"]["create"]>[3],
      "webhook" | "webhook_events_filter"
    >,
    options: CallbackTimeout = { timeoutInSeconds: 3600 }
  ): ReplicateReturnType<Training> {
    return this.runTask(
      key,
      (client, task) => {
        const { model_owner, model_name, version_id, ...options } = params;

        return client.trainings.create(model_owner, model_name, version_id, {
          ...options,
          webhook: task.callbackUrl ?? "",
          webhook_events_filter: ["completed"],
        });
      },
      {
        name: "Create And Await Training",
        params,
        properties: [...modelProperties(params), ...callbackProperties(options)],
        callback: {
          enabled: true,
          timeoutInSeconds: options.timeoutInSeconds,
        },
      }
    );
  }

  /** Fetch a training. */
  get(key: IntegrationTaskKey, params: { id: string }): ReplicateReturnType<Training> {