Class WatsonXApi

packages/openai-adapters/src/apis/WatsonX.ts:22–353

} from "./base.js";

export class WatsonXApi implements BaseLlmApi {
  apiBase: string;
  apiVersion: string = "2023-05-29";
  projectId?: string;
  deploymentId?: string;

  constructor(protected config: WatsonXConfig) {
    this.apiBase = config.apiBase ?? "https://us-south.ml.cloud.ibm.com";
    if (!this.apiBase.endsWith("/")) {
      this.apiBase += "/";
    }
    this.apiVersion = config.env.apiVersion ?? this.apiVersion;
    this.projectId = config.env.projectId;
    this.deploymentId = config.env.deploymentId;
  }

  async getBearerToken(): Promise<{ token: string; expiration: number }> {
    if (this.apiBase?.includes("cloud.ibm.com")) {
      // watsonx SaaS
      const wxToken = (await (
        await customFetch(this.config.requestOptions)(
          `https://iam.cloud.ibm.com/identity/token?apikey=${this.config.apiKey}&grant_type=urn:ibm:params:oauth:grant-type:apikey`,
          {
            method: "POST",
            headers: {
              "Content-Type": "application/x-www-form-urlencoded",
              Accept: "application/json",
            },
          },
        )
      ).json()) as any;
      return {
        token: wxToken["access_token"],
        expiration: wxToken["expiration"],
      };
    } else {
      // watsonx Software
      // if (this.config.env.bearerTokenRequired) {
      // In certain WatsonX environments, ZenApiKey authentication is disabled,
      // and it's necessary to call this endpoint with username+api_key to get a bearer token.
      // See the docs: https://www.ibm.com/docs/en/watsonx/w-and-w/2.1.0?topic=keys-generating-bearer-token
      // Ask @sestinj why the rest is commented out.
      const base64Decoded = Buffer.from(
        this.config.apiKey ?? "",
        "base64",
      ).toString();
      const [username, api_key] = base64Decoded.split(":");

      const wxToken = (await (
        await customFetch(this.config.requestOptions)(
          new URL("icp4d-api/v1/authorize", this.apiBase),
          {
            method: "POST",
            headers: {
              "Content-Type": "application/json",
              Accept: "application/json",
            },
            body: JSON.stringify({