Class OpenAI

integrations/openai/src/index.ts:26–323

export type OpenAIRunTask = InstanceType<typeof OpenAI>["runTask"];

export class OpenAI implements TriggerIntegration {
  // @internal
  private _options: OpenAIIntegrationOptions;
  // @internal
  private _client?: OpenAIApi;
  // @internal
  private _io?: IO;
  // @internal
  private _connectionKey?: string;

  /**
   * The native OpenAIApi client. This is exposed for use outside of Trigger.dev jobs
   *
   * @example
   * ```ts
   * import { OpenAI } from "@trigger.dev/openai";
   *
   * const openAI = new OpenAI({
   *   id: "my-openai",
   *   apiKey: process.env.OPENAI_API_KEY!,
   * });
   *
   * const response = await openAI.native.completions.create({}); // ...
   * ```
   */
  public readonly native: OpenAIApi;

  constructor(private options: OpenAIIntegrationOptions) {
    this._options = options;

    this.native = new OpenAIApi({
      apiKey: options.apiKey,
      organization: options.organization,
      baseURL: options.baseURL,
      defaultHeaders: options.defaultHeaders,
      defaultQuery: options.defaultQuery,
      maxRetries: 0,
    });
  }

  get authSource() {
    return "LOCAL" as const;
  }

  cloneForRun(io: IO, connectionKey: string, auth?: ConnectionAuth) {
    const apiKey = this._options.apiKey ?? auth?.accessToken;

    if (!apiKey) {
      throw new Error(
        `Can't initialize OpenAI integration (${this._options.id}) as apiKey was undefined`
      );
    }

    const openai = new OpenAI(this._options);
    openai._io = io;
    openai._connectionKey = connectionKey;
    openai._client = new OpenAIApi({
      apiKey,