{"version":3,"file":"llms.cjs","names":["OpenAI"],"sources":["../src/llms.ts"],"sourcesContent":["import type { BaseLLMParams } from \"@langchain/core/language_models/llms\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport {\n  OpenAI,\n  type OpenAICallOptions,\n  type OpenAIClient,\n  type OpenAICoreRequestOptions,\n  type OpenAIInput,\n} from \"@langchain/openai\";\n\nconst FIREWORKS_BASE_URL = \"https://api.fireworks.ai/inference/v1\";\nconst DEFAULT_FIREWORKS_LLM_MODEL = \"accounts/fireworks/models/llama-v2-13b\";\n\ntype FireworksUnsupportedArgs =\n  | \"frequencyPenalty\"\n  | \"presencePenalty\"\n  | \"bestOf\"\n  | \"logitBias\";\n\ntype FireworksUnsupportedCallOptions = \"functions\" | \"function_call\" | \"tools\";\n\nexport interface FireworksInput\n  extends\n    Partial<Omit<OpenAIInput, \"openAIApiKey\" | FireworksUnsupportedArgs>>,\n    BaseLLMParams {\n  /**\n   * Prefer `apiKey`.\n   */\n  fireworksApiKey?: string;\n\n  /**\n   * The Fireworks API key to use.\n   */\n  apiKey?: string;\n}\n\nexport type FireworksCallOptions = Partial<\n  Omit<OpenAICallOptions, FireworksUnsupportedCallOptions>\n>;\n\n/**\n * Fireworks text completion LLM.\n *\n * Setup:\n *\n * ```bash\n * npm install @langchain/fireworks @langchain/core\n * export FIREWORKS_API_KEY=\"your-api-key\"\n * ```\n *\n * @example\n * ```typescript\n * import { Fireworks } from \"@langchain/fireworks\";\n *\n * const model = new Fireworks({\n *   temperature: 0,\n * });\n *\n * const response = await model.invoke(\"1 + 1 =\");\n * ```\n */\nexport class Fireworks extends OpenAI<FireworksCallOptions> {\n  static lc_name() {\n    return \"Fireworks\";\n  }\n\n  lc_namespace = [\"langchain\", \"llms\", \"fireworks\"];\n\n  lc_serializable = true;\n\n  fireworksApiKey?: string;\n\n  apiKey?: string;\n\n  constructor(\n    model: string,\n    fields?: Omit<FireworksInput, \"model\" | \"modelName\">\n  );\n  constructor(fields?: FireworksInput);\n  constructor(\n    modelOrFields?: string | FireworksInput,\n    fieldsArg?: Omit<FireworksInput, \"model\" | \"modelName\">\n  ) {\n    const fields =\n      typeof modelOrFields === \"string\"\n        ? { ...(fieldsArg ?? {}), model: modelOrFields }\n        : (modelOrFields ?? {});\n\n    const fireworksApiKey =\n      fields.apiKey ||\n      fields.fireworksApiKey ||\n      getEnvironmentVariable(\"FIREWORKS_API_KEY\");\n\n    if (!fireworksApiKey) {\n      throw new Error(\n        'Fireworks API key not found. Please set the FIREWORKS_API_KEY environment variable or pass the key into \"apiKey\" or \"fireworksApiKey\".'\n      );\n    }\n\n    super({\n      ...fields,\n      apiKey: fireworksApiKey,\n      model: fields.model ?? fields.modelName ?? DEFAULT_FIREWORKS_LLM_MODEL,\n      configuration: {\n        baseURL: FIREWORKS_BASE_URL,\n        ...fields.configuration,\n      },\n    });\n\n    this.fireworksApiKey = fireworksApiKey;\n    this.apiKey = fireworksApiKey;\n    this._addVersion(\"@langchain/fireworks\", __PKG_VERSION__);\n  }\n\n  _llmType() {\n    return \"fireworks\";\n  }\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      fireworksApiKey: \"FIREWORKS_API_KEY\",\n      apiKey: \"FIREWORKS_API_KEY\",\n    };\n  }\n\n  toJSON() {\n    const result = super.toJSON();\n\n    if (\n      \"kwargs\" in result &&\n      typeof result.kwargs === \"object\" &&\n      result.kwargs != null\n    ) {\n      delete result.kwargs.openai_api_key;\n      delete result.kwargs.configuration;\n    }\n\n    return result;\n  }\n\n  async completionWithRetry(\n    request: OpenAIClient.CompletionCreateParamsStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<AsyncIterable<OpenAIClient.Completion>>;\n\n  async completionWithRetry(\n    request: OpenAIClient.CompletionCreateParamsNonStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<OpenAIClient.Completions.Completion>;\n\n  async completionWithRetry(\n    request:\n      | OpenAIClient.CompletionCreateParamsStreaming\n      | OpenAIClient.CompletionCreateParamsNonStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<\n    AsyncIterable<OpenAIClient.Completion> | OpenAIClient.Completions.Completion\n  > {\n    if (Array.isArray(request.prompt)) {\n      if (request.prompt.length > 1) {\n        throw new Error(\"Multiple prompts are not supported by Fireworks\");\n      }\n\n      const prompt = request.prompt[0];\n      if (typeof prompt !== \"string\") {\n        throw new Error(\"Only string prompts are supported by Fireworks\");\n      }\n\n      request.prompt = prompt;\n    }\n\n    delete request.frequency_penalty;\n    delete request.presence_penalty;\n    delete request.best_of;\n    delete request.logit_bias;\n\n    return super.completionWithRetry(request, options);\n  }\n}\n"],"mappings":";;;AAUA,MAAM,qBAAqB;AAC3B,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;AAkDpC,IAAa,YAAb,cAA+BA,kBAAAA,OAA6B;CAC1D,OAAO,UAAU;AACf,SAAO;;CAGT,eAAe;EAAC;EAAa;EAAQ;EAAY;CAEjD,kBAAkB;CAElB;CAEA;CAOA,YACE,eACA,WACA;EACA,MAAM,SACJ,OAAO,kBAAkB,WACrB;GAAE,GAAI,aAAa,EAAE;GAAG,OAAO;GAAe,GAC7C,iBAAiB,EAAE;EAE1B,MAAM,kBACJ,OAAO,UACP,OAAO,oBAAA,GAAA,0BAAA,wBACgB,oBAAoB;AAE7C,MAAI,CAAC,gBACH,OAAM,IAAI,MACR,6IACD;AAGH,QAAM;GACJ,GAAG;GACH,QAAQ;GACR,OAAO,OAAO,SAAS,OAAO,aAAa;GAC3C,eAAe;IACb,SAAS;IACT,GAAG,OAAO;IACX;GACF,CAAC;AAEF,OAAK,kBAAkB;AACvB,OAAK,SAAS;AACd,OAAK,YAAY,wBAAA,QAAwC;;CAG3D,WAAW;AACT,SAAO;;CAGT,IAAI,aAAoD;AACtD,SAAO;GACL,iBAAiB;GACjB,QAAQ;GACT;;CAGH,SAAS;EACP,MAAM,SAAS,MAAM,QAAQ;AAE7B,MACE,YAAY,UACZ,OAAO,OAAO,WAAW,YACzB,OAAO,UAAU,MACjB;AACA,UAAO,OAAO,OAAO;AACrB,UAAO,OAAO,OAAO;;AAGvB,SAAO;;CAaT,MAAM,oBACJ,SAGA,SAGA;AACA,MAAI,MAAM,QAAQ,QAAQ,OAAO,EAAE;AACjC,OAAI,QAAQ,OAAO,SAAS,EAC1B,OAAM,IAAI,MAAM,kDAAkD;GAGpE,MAAM,SAAS,QAAQ,OAAO;AAC9B,OAAI,OAAO,WAAW,SACpB,OAAM,IAAI,MAAM,iDAAiD;AAGnE,WAAQ,SAAS;;AAGnB,SAAO,QAAQ;AACf,SAAO,QAAQ;AACf,SAAO,QAAQ;AACf,SAAO,QAAQ;AAEf,SAAO,MAAM,oBAAoB,SAAS,QAAQ"}