Class Chat

integrations/openai/src/chat.ts:14–112

import { FetchRetryOptions, FetchTimeoutOptions } from "@trigger.dev/integration-kit";

export class Chat {
  constructor(
    private runTask: OpenAIRunTask,
    private options: OpenAIIntegrationOptions
  ) {}

  completions = {
    create: (
      key: IntegrationTaskKey,
      params: Prettify<OpenAI.Chat.ChatCompletionCreateParamsNonStreaming>,
      options: OpenAIRequestOptions = {}
    ): Promise<OpenAI.Chat.ChatCompletion> => {
      return this.runTask(
        key,
        async (client, task) => {
          const { data, response } = await client.chat.completions
            .create(params, {
              idempotencyKey: task.idempotencyKey,
              ...options,
            })
            .withResponse();

          task.outputProperties = createTaskOutputProperties(data.usage, response.headers);

          return data;
        },
        {
          name: "Chat Completion",
          params,
          properties: [
            {
              label: "model",
              text: params.model,
            },
          ],
        },
        handleOpenAIError
      );
    },

    backgroundCreate: (
      key: IntegrationTaskKey,
      params: Prettify<OpenAI.Chat.ChatCompletionCreateParamsNonStreaming>,
      options: OpenAIRequestOptions = {},
      fetchOptions: { retries?: FetchRetryOptions; timeout?: FetchTimeoutOptions } = {}
    ): Promise<OpenAI.Chat.ChatCompletion> => {
      return this.runTask(
        key,
        async (client, task, io) => {
          const url = createBackgroundFetchUrl(
            client,
            "/chat/completions",
            this.options.defaultQuery,
            options
          );

          const response = await io.backgroundFetchResponse<OpenAI.Chat.ChatCompletion>(
            "background",