ai 3.1.0-canary.2 → 3.1.0-canary.4

package/{provider → openai}/dist/index.mjs

@@ -1,5 +1,5 @@
 // ai-model-specification/errors/api-call-error.ts
-var ApiCallError = class extends Error {
+var APICallError = class extends Error {
   constructor({
     message,
     url,
@@ -7,11 +7,15 @@ var ApiCallError = class extends Error {
     statusCode,
     responseBody,
     cause,
-    isRetryable = statusCode != null && (statusCode === 429 || statusCode >= 500),
+    isRetryable = statusCode != null && (statusCode === 408 || // request timeout
+    statusCode === 409 || // conflict
+    statusCode === 429 || // too many requests
+    statusCode >= 500),
+    // server error
     data
   }) {
     super(message);
-    this.name = "ApiCallError";
+    this.name = "AI_APICallError";
     this.url = url;
     this.requestBodyValues = requestBodyValues;
     this.statusCode = statusCode;
@@ -20,6 +24,9 @@ var ApiCallError = class extends Error {
     this.isRetryable = isRetryable;
     this.data = data;
   }
+  static isAPICallError(error) {
+    return error instanceof Error && error.name === "AI_APICallError" && typeof error.url === "string" && typeof error.requestBodyValues === "object" && (error.statusCode == null || typeof error.statusCode === "number") && (error.responseBody == null || typeof error.responseBody === "string") && (error.cause == null || typeof error.cause === "object") && typeof error.isRetryable === "boolean" && (error.data == null || typeof error.data === "object");
+  }
   toJSON() {
     return {
       name: this.name,
@@ -35,6 +42,26 @@ var ApiCallError = class extends Error {
   }
 };
 
+// ai-model-specification/errors/invalid-prompt-error.ts
+var InvalidPromptError = class extends Error {
+  constructor({ prompt: prompt2, message }) {
+    super(`Invalid prompt: ${message}`);
+    this.name = "AI_InvalidPromptError";
+    this.prompt = prompt2;
+  }
+  static isInvalidPromptError(error) {
+    return error instanceof Error && error.name === "AI_InvalidPromptError" && prompt != null;
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      prompt: this.prompt
+    };
+  }
+};
+
 // ai-model-specification/util/get-error-message.ts
 function getErrorMessage(error) {
   if (error == null) {
@@ -49,6 +76,50 @@ function getErrorMessage(error) {
   return JSON.stringify(error);
 }
 
+// ai-model-specification/errors/load-api-key-error.ts
+var LoadAPIKeyError = class extends Error {
+  constructor({ message }) {
+    super(message);
+    this.name = "AI_LoadAPIKeyError";
+  }
+  static isLoadAPIKeyError(error) {
+    return error instanceof Error && error.name === "AI_LoadAPIKeyError";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message
+    };
+  }
+};
+
+// ai-model-specification/util/load-api-key.ts
+function loadApiKey({
+  apiKey,
+  environmentVariableName,
+  apiKeyParameterName = "apiKey",
+  description
+}) {
+  if (apiKey != null) {
+    return apiKey;
+  }
+  if (typeof process === "undefined") {
+    throw new LoadAPIKeyError({
+      message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter. Environment variables is not supported in this environment.`
+    });
+  }
+  apiKey = process.env[environmentVariableName];
+  if (apiKey == null) {
+    throw new LoadAPIKeyError({
+      message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter or the ${environmentVariableName} environment variable.`
+    });
+  }
+  return apiKey;
+}
+
+// ai-model-specification/util/parse-json.ts
+import SecureJSON from "secure-json-parse";
+
 // ai-model-specification/errors/json-parse-error.ts
 var JSONParseError = class extends Error {
   constructor({ text, cause }) {
@@ -56,10 +127,13 @@ var JSONParseError = class extends Error {
       `JSON parsing failed: Text: ${text}.
 Error message: ${getErrorMessage(cause)}`
     );
-    this.name = "JSONParseError";
+    this.name = "AI_JSONParseError";
     this.cause = cause;
     this.text = text;
   }
+  static isJSONParseError(error) {
+    return error instanceof Error && error.name === "AI_JSONParseError" && typeof error.text === "string" && typeof error.cause === "string";
+  }
   toJSON() {
     return {
       name: this.name,
@@ -71,20 +145,6 @@ Error message: ${getErrorMessage(cause)}`
   }
 };
 
-// ai-model-specification/errors/load-api-key-error.ts
-var LoadAPIKeyError = class extends Error {
-  constructor({ message }) {
-    super(message);
-    this.name = "LoadAPIKeyError";
-  }
-  toJSON() {
-    return {
-      name: this.name,
-      message: this.message
-    };
-  }
-};
-
 // ai-model-specification/errors/type-validation-error.ts
 var TypeValidationError = class extends Error {
   constructor({ value, cause }) {
@@ -92,10 +152,13 @@ var TypeValidationError = class extends Error {
       `Type validation failed: Value: ${JSON.stringify(value)}.
 Error message: ${getErrorMessage(cause)}`
     );
-    this.name = "TypeValidationError";
+    this.name = "AI_TypeValidationError";
     this.cause = cause;
     this.value = value;
   }
+  static isTypeValidationError(error) {
+    return error instanceof Error && error.name === "AI_TypeValidationError" && typeof error.value === "string" && typeof error.cause === "string";
+  }
   toJSON() {
     return {
       name: this.name,
@@ -107,58 +170,6 @@ Error message: ${getErrorMessage(cause)}`
   }
 };
 
-// ai-model-specification/errors/unsupported-functionality-error.ts
-var UnsupportedFunctionalityError = class extends Error {
-  constructor({
-    provider,
-    functionality
-  }) {
-    super(
-      `Functionality not supported by the provider. Provider: ${provider}.
-Functionality: ${functionality}`
-    );
-    this.name = "UnsupportedFunctionalityError";
-    this.provider = provider;
-    this.functionality = functionality;
-  }
-  toJSON() {
-    return {
-      name: this.name,
-      message: this.message,
-      stack: this.stack,
-      provider: this.provider,
-      functionality: this.functionality
-    };
-  }
-};
-
-// ai-model-specification/util/load-api-key.ts
-function loadApiKey({
-  apiKey,
-  environmentVariableName,
-  apiKeyParameterName = "apiKey",
-  description
-}) {
-  if (apiKey != null) {
-    return apiKey;
-  }
-  if (typeof process === "undefined") {
-    throw new LoadAPIKeyError({
-      message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter. Environment variables is not supported in this environment.`
-    });
-  }
-  apiKey = process.env[environmentVariableName];
-  if (apiKey == null) {
-    throw new LoadAPIKeyError({
-      message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter or the ${environmentVariableName} environment variable.`
-    });
-  }
-  return apiKey;
-}
-
-// ai-model-specification/util/parse-json.ts
-import SecureJSON from "secure-json-parse";
-
 // ai-model-specification/util/validate-types.ts
 function validateTypes({
   value,
@@ -192,7 +203,7 @@ function safeValidateTypes({
   } catch (error) {
     return {
       success: false,
-      error: error instanceof TypeValidationError ? error : new TypeValidationError({ value, cause: error })
+      error: TypeValidationError.isTypeValidationError(error) ? error : new TypeValidationError({ value, cause: error })
     };
   }
 }
@@ -209,7 +220,7 @@ function parseJSON({
     }
     return validateTypes({ value, schema });
   } catch (error) {
-    if (error instanceof JSONParseError || error instanceof TypeValidationError) {
+    if (JSONParseError.isJSONParseError(error) || TypeValidationError.isTypeValidationError(error)) {
       throw error;
     }
     throw new JSONParseError({ text, cause: error });
@@ -231,7 +242,7 @@ function safeParseJSON({
   } catch (error) {
     return {
       success: false,
-      error: error instanceof JSONParseError ? error : new JSONParseError({ text, cause: error })
+      error: JSONParseError.isJSONParseError(error) ? error : new JSONParseError({ text, cause: error })
     };
   }
 }
@@ -255,8 +266,8 @@ var postJsonToApi = async ({
 }) => postToApi({
   url,
   headers: {
-    "Content-Type": "application/json",
-    ...headers
+    ...headers,
+    "Content-Type": "application/json"
   },
   body: {
     content: JSON.stringify(body),
@@ -275,9 +286,12 @@ var postToApi = async ({
   abortSignal
 }) => {
   try {
+    const definedHeaders = Object.fromEntries(
+      Object.entries(headers).filter(([_key, value]) => value != null)
+    );
     const response = await fetch(url, {
       method: "POST",
-      headers,
+      headers: definedHeaders,
       body: body.content,
       signal: abortSignal
     });
@@ -290,11 +304,11 @@ var postToApi = async ({
         });
       } catch (error) {
         if (error instanceof Error) {
-          if (error.name === "AbortError" || error instanceof ApiCallError) {
+          if (error.name === "AbortError" || APICallError.isAPICallError(error)) {
             throw error;
           }
         }
-        throw new ApiCallError({
+        throw new APICallError({
           message: "Failed to process error response",
           cause: error,
           statusCode: response.status,
@@ -311,11 +325,11 @@ var postToApi = async ({
       });
     } catch (error) {
       if (error instanceof Error) {
-        if (error.name === "AbortError" || error instanceof ApiCallError) {
+        if (error.name === "AbortError" || APICallError.isAPICallError(error)) {
           throw error;
         }
       }
-      throw new ApiCallError({
+      throw new APICallError({
         message: "Failed to process successful response",
         cause: error,
         statusCode: response.status,
@@ -332,12 +346,13 @@ var postToApi = async ({
     if (error instanceof TypeError && error.message === "fetch failed") {
       const cause = error.cause;
       if (cause != null) {
-        throw new ApiCallError({
+        throw new APICallError({
           message: `Cannot connect to API: ${cause.message}`,
           cause,
           url,
           requestBodyValues: body.values,
           isRetryable: true
+          // retry when network error
         });
       }
     }
@@ -356,7 +371,7 @@ var createJsonErrorResponseHandler = ({
 }) => async ({ response, url, requestBodyValues }) => {
   const responseBody = await response.text();
   if (responseBody.trim() === "") {
-    return new ApiCallError({
+    return new APICallError({
       message: response.statusText,
       url,
       requestBodyValues,
@@ -370,7 +385,7 @@ var createJsonErrorResponseHandler = ({
       text: responseBody,
       schema: errorSchema
     });
-    return new ApiCallError({
+    return new APICallError({
       message: errorToMessage(parsedError),
       url,
       requestBodyValues,
@@ -380,7 +395,7 @@ var createJsonErrorResponseHandler = ({
       isRetryable: isRetryable == null ? void 0 : isRetryable(response, parsedError)
     });
   } catch (parseError) {
-    return new ApiCallError({
+    return new APICallError({
       message: response.statusText,
       url,
       requestBodyValues,
@@ -418,7 +433,7 @@ var createJsonResponseHandler = (responseSchema) => async ({ response, url, requ
     schema: responseSchema
   });
   if (!parsedResult.success) {
-    throw new ApiCallError({
+    throw new APICallError({
       message: "Invalid JSON response",
       cause: parsedResult.error,
       statusCode: response.status,
@@ -455,14 +470,42 @@ function convertUint8ArrayToBase64(array) {
   return globalThis.btoa(latin1string);
 }
 
-// provider/openai/openai-chat-language-model.ts
+// ai-model-specification/errors/unsupported-functionality-error.ts
+var UnsupportedFunctionalityError = class extends Error {
+  constructor({
+    provider,
+    functionality
+  }) {
+    super(
+      `Functionality not supported by the provider. Provider: ${provider}.
+Functionality: ${functionality}`
+    );
+    this.name = "AI_UnsupportedFunctionalityError";
+    this.provider = provider;
+    this.functionality = functionality;
+  }
+  static isUnsupportedFunctionalityError(error) {
+    return error instanceof Error && error.name === "AI_UnsupportedFunctionalityError" && typeof error.provider === "string" && typeof error.functionality === "string";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      provider: this.provider,
+      functionality: this.functionality
+    };
+  }
+};
+
+// openai/openai-chat-language-model.ts
 import { nanoid } from "nanoid";
 import { z as z2 } from "zod";
 
-// provider/openai/convert-to-openai-chat-messages.ts
-function convertToOpenAIChatMessages(prompt) {
+// openai/convert-to-openai-chat-messages.ts
+function convertToOpenAIChatMessages(prompt2) {
   const messages = [];
-  for (const { role, content } of prompt) {
+  for (const { role, content } of prompt2) {
     switch (role) {
       case "system": {
         messages.push({ role: "system", content });
@@ -481,7 +524,7 @@ function convertToOpenAIChatMessages(prompt) {
                 return {
                   type: "image_url",
                   image_url: {
-                    url: `data:${(_a = part.mimeType) != null ? _a : "image/jpeg"};base64,${convertUint8ArrayToBase64(part.image)}`
+                    url: part.image instanceof URL ? part.image.toString() : `data:${(_a = part.mimeType) != null ? _a : "image/jpeg"};base64,${convertUint8ArrayToBase64(part.image)}`
                   }
                 };
               }
@@ -542,7 +585,24 @@ function convertToOpenAIChatMessages(prompt) {
   return messages;
 }
 
-// provider/openai/openai-error.ts
+// openai/map-openai-finish-reason.ts
+function mapOpenAIFinishReason(finishReason) {
+  switch (finishReason) {
+    case "stop":
+      return "stop";
+    case "length":
+      return "length";
+    case "content-filter":
+      return "content-filter";
+    case "function_call":
+    case "tool-calls":
+      return "tool-calls";
+    default:
+      return "other";
+  }
+}
+
+// openai/openai-error.ts
 import { z } from "zod";
 var openAIErrorDataSchema = z.object({
   error: z.object({
@@ -554,28 +614,24 @@ var openAIErrorDataSchema = z.object({
 });
 var openaiFailedResponseHandler = createJsonErrorResponseHandler({
   errorSchema: openAIErrorDataSchema,
-  errorToMessage: (data) => data.error.message,
-  isRetryable: (response, error) => response.status >= 500 || response.status === 429 && // insufficient_quota is also reported as a 429, but it's not retryable:
-  (error == null ? void 0 : error.error.type) !== "insufficient_quota"
+  errorToMessage: (data) => data.error.message
 });
 
-// provider/openai/openai-chat-language-model.ts
+// openai/openai-chat-language-model.ts
 var OpenAIChatLanguageModel = class {
-  constructor(settings, config) {
+  constructor(modelId, settings, config) {
     this.specificationVersion = "v1";
     this.defaultObjectGenerationMode = "tool";
+    this.modelId = modelId;
     this.settings = settings;
     this.config = config;
   }
   get provider() {
     return this.config.provider;
   }
-  get modelId() {
-    return this.settings.id;
-  }
   getArgs({
     mode,
-    prompt,
+    prompt: prompt2,
     maxTokens,
     temperature,
     topP,
@@ -586,8 +642,11 @@ var OpenAIChatLanguageModel = class {
     var _a;
     const type = mode.type;
     const baseArgs = {
+      // model id:
+      model: this.modelId,
       // model specific settings:
-      ...this.config.mapSettings(this.settings),
+      logit_bias: this.settings.logitBias,
+      user: this.settings.user,
       // standardized settings:
       max_tokens: maxTokens,
       temperature: scale({
@@ -612,7 +671,7 @@ var OpenAIChatLanguageModel = class {
       }),
       seed,
       // messages:
-      messages: convertToOpenAIChatMessages(prompt)
+      messages: convertToOpenAIChatMessages(prompt2)
     };
     switch (type) {
       case "regular": {
@@ -656,49 +715,54 @@ var OpenAIChatLanguageModel = class {
   }
   async doGenerate(options) {
     var _a, _b;
+    const args = this.getArgs(options);
     const response = await postJsonToApi({
       url: `${this.config.baseUrl}/chat/completions`,
-      headers: {
-        Authorization: `Bearer ${this.config.apiKey()}`
-      },
-      body: {
-        ...this.getArgs(options)
-      },
+      headers: this.config.headers(),
+      body: args,
       failedResponseHandler: openaiFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
         openAIChatResponseSchema
-      )
+      ),
+      abortSignal: options.abortSignal
     });
-    const message = response.choices[0].message;
+    const { messages: rawPrompt, ...rawSettings } = args;
+    const choice = response.choices[0];
     return {
-      text: (_a = message.content) != null ? _a : void 0,
-      toolCalls: (_b = message.tool_calls) == null ? void 0 : _b.map((toolCall) => ({
+      text: (_a = choice.message.content) != null ? _a : void 0,
+      toolCalls: (_b = choice.message.tool_calls) == null ? void 0 : _b.map((toolCall) => ({
         toolCallType: "function",
         toolCallId: toolCall.id,
         toolName: toolCall.function.name,
         args: toolCall.function.arguments
       })),
+      finishReason: mapOpenAIFinishReason(choice.finish_reason),
+      usage: {
+        promptTokens: response.usage.prompt_tokens,
+        completionTokens: response.usage.completion_tokens
+      },
+      rawCall: { rawPrompt, rawSettings },
       warnings: []
     };
   }
   async doStream(options) {
+    const args = this.getArgs(options);
     const response = await postJsonToApi({
       url: `${this.config.baseUrl}/chat/completions`,
-      headers: {
-        Authorization: `Bearer ${this.config.apiKey()}`
-      },
+      headers: this.config.headers(),
       body: {
-        ...this.getArgs(options),
+        ...args,
         stream: true
       },
       failedResponseHandler: openaiFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
         openaiChatChunkSchema
-      )
+      ),
+      abortSignal: options.abortSignal
     });
+    const { messages: rawPrompt, ...rawSettings } = args;
     const toolCalls = [];
     return {
-      warnings: [],
       stream: response.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
@@ -752,7 +816,9 @@ var OpenAIChatLanguageModel = class {
             }
           }
         })
-      )
+      ),
+      rawCall: { rawPrompt, rawSettings },
+      warnings: []
     };
   }
 };
@@ -808,32 +874,34 @@ var openaiChatChunkSchema = z2.object({
   )
 });
 
-// provider/openai/openai-completion-language-model.ts
+// openai/openai-completion-language-model.ts
 import { z as z3 } from "zod";
 
-// provider/openai/convert-to-openai-completion-prompt.ts
+// openai/convert-to-openai-completion-prompt.ts
 function convertToOpenAICompletionPrompt({
-  prompt,
+  prompt: prompt2,
   inputFormat,
   provider,
   user = "user",
   assistant = "assistant"
 }) {
-  if (inputFormat === "prompt" && prompt.length === 1 && prompt[0].role === "user" && prompt[0].content.length === 1 && prompt[0].content[0].type === "text") {
-    return { prompt: prompt[0].content[0].text };
+  if (inputFormat === "prompt" && prompt2.length === 1 && prompt2[0].role === "user" && prompt2[0].content.length === 1 && prompt2[0].content[0].type === "text") {
+    return { prompt: prompt2[0].content[0].text };
   }
   let text = "";
-  if (prompt[0].role === "system") {
-    text += `${prompt[0].content}
+  if (prompt2[0].role === "system") {
+    text += `${prompt2[0].content}
 
 `;
-    prompt = prompt.slice(1);
+    prompt2 = prompt2.slice(1);
   }
-  for (const { role, content } of prompt) {
+  for (const { role, content } of prompt2) {
     switch (role) {
       case "system": {
-        throw new Error(`Unexpected system message in prompt: ${content}`);
-        break;
+        throw new InvalidPromptError({
+          message: "Unexpected system message in prompt: ${content}",
+          prompt: prompt2
+        });
       }
       case "user": {
         const userMessage = content.map((part) => {
@@ -896,24 +964,22 @@ ${user}:`]
   };
 }
 
-// provider/openai/openai-completion-language-model.ts
+// openai/openai-completion-language-model.ts
 var OpenAICompletionLanguageModel = class {
-  constructor(settings, config) {
+  constructor(modelId, settings, config) {
     this.specificationVersion = "v1";
     this.defaultObjectGenerationMode = void 0;
+    this.modelId = modelId;
     this.settings = settings;
     this.config = config;
   }
   get provider() {
     return this.config.provider;
   }
-  get modelId() {
-    return this.settings.id;
-  }
   getArgs({
     mode,
     inputFormat,
-    prompt,
+    prompt: prompt2,
     maxTokens,
     temperature,
     topP,
@@ -924,13 +990,18 @@ var OpenAICompletionLanguageModel = class {
     var _a;
     const type = mode.type;
     const { prompt: completionPrompt, stopSequences } = convertToOpenAICompletionPrompt({
-      prompt,
+      prompt: prompt2,
       inputFormat,
       provider: this.provider
     });
     const baseArgs = {
+      // model id:
+      model: this.modelId,
       // model specific settings:
-      ...this.config.mapSettings(this.settings),
+      echo: this.settings.echo,
+      logit_bias: this.settings.logitBias,
+      suffix: this.settings.suffix,
+      user: this.settings.user,
       // standardized settings:
       max_tokens: maxTokens,
       temperature: scale({
@@ -994,30 +1065,35 @@ var OpenAICompletionLanguageModel = class {
     }
   }
   async doGenerate(options) {
+    const args = this.getArgs(options);
     const response = await postJsonToApi({
       url: `${this.config.baseUrl}/completions`,
-      headers: {
-        Authorization: `Bearer ${this.config.apiKey()}`
-      },
-      body: {
-        ...this.getArgs(options)
-      },
+      headers: this.config.headers(),
+      body: args,
       failedResponseHandler: openaiFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
         openAICompletionResponseSchema
-      )
+      ),
+      abortSignal: options.abortSignal
     });
+    const { prompt: rawPrompt, ...rawSettings } = args;
+    const choice = response.choices[0];
     return {
-      text: response.choices[0].text,
+      text: choice.text,
+      usage: {
+        promptTokens: response.usage.prompt_tokens,
+        completionTokens: response.usage.completion_tokens
+      },
+      finishReason: mapOpenAIFinishReason(choice.finish_reason),
+      rawCall: { rawPrompt, rawSettings },
       warnings: []
     };
   }
   async doStream(options) {
+    const args = this.getArgs(options);
     const response = await postJsonToApi({
       url: `${this.config.baseUrl}/completions`,
-      headers: {
-        Authorization: `Bearer ${this.config.apiKey()}`
-      },
+      headers: this.config.headers(),
       body: {
         ...this.getArgs(options),
         stream: true
@@ -1025,10 +1101,11 @@ var OpenAICompletionLanguageModel = class {
       failedResponseHandler: openaiFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
         openaiCompletionChunkSchema
-      )
+      ),
+      abortSignal: options.abortSignal
     });
+    const { prompt: rawPrompt, ...rawSettings } = args;
     return {
-      warnings: [],
       stream: response.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
@@ -1046,7 +1123,9 @@ var OpenAICompletionLanguageModel = class {
             }
           }
         })
-      )
+      ),
+      rawCall: { rawPrompt, rawSettings },
+      warnings: []
     };
   }
 };
@@ -1073,49 +1152,44 @@ var openaiCompletionChunkSchema = z3.object({
   )
 });
 
-// provider/openai/openai-facade.ts
+// openai/openai-facade.ts
 var OpenAI = class {
-  constructor({ baseUrl, apiKey } = {}) {
-    this.baseUrl = baseUrl;
-    this.apiKey = apiKey;
+  constructor(options = {}) {
+    this.baseUrl = options.baseUrl;
+    this.apiKey = options.apiKey;
+    this.organization = options.organization;
   }
-  chat(settings) {
+  get baseConfig() {
     var _a;
-    return new OpenAIChatLanguageModel(settings, {
-      provider: "openai.chat",
+    return {
+      organization: this.organization,
       baseUrl: (_a = this.baseUrl) != null ? _a : "https://api.openai.com/v1",
-      apiKey: () => loadApiKey({
-        apiKey: this.apiKey,
-        environmentVariableName: "OPENAI_API_KEY",
-        description: "OpenAI"
-      }),
-      mapSettings: (settings2) => ({
-        model: settings2.id,
-        logit_bias: settings2.logitBias
-      })
-    });
-  }
-  completion(settings) {
-    var _a;
-    return new OpenAICompletionLanguageModel(
-      settings,
-      {
-        provider: "openai.completion",
-        baseUrl: (_a = this.baseUrl) != null ? _a : "https://api.openai.com/v1",
-        apiKey: () => loadApiKey({
+      headers: () => ({
+        Authorization: `Bearer ${loadApiKey({
           apiKey: this.apiKey,
           environmentVariableName: "OPENAI_API_KEY",
           description: "OpenAI"
-        }),
-        mapSettings: (settings2) => ({
-          model: settings2.id,
-          logit_bias: settings2.logitBias
-        })
-      }
-    );
+        })}`,
+        "OpenAI-Organization": this.organization
+      })
+    };
+  }
+  chat(modelId, settings = {}) {
+    return new OpenAIChatLanguageModel(modelId, settings, {
+      provider: "openai.chat",
+      ...this.baseConfig
+    });
+  }
+  completion(modelId, settings = {}) {
+    return new OpenAICompletionLanguageModel(modelId, settings, {
+      provider: "openai.completion",
+      ...this.baseConfig
+    });
   }
 };
+var openai = new OpenAI();
 export {
-  OpenAI
+  OpenAI,
+  openai
 };
 //# sourceMappingURL=index.mjs.map