{"version":3,"file":"chat_models.cjs","names":["ChatOpenAICompletions"],"sources":["../src/chat_models.ts"],"sourcesContent":["import type {\n  BaseChatModelParams,\n  LangSmithParams,\n} from \"@langchain/core/language_models/chat_models\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport { resolveLangSmithGatewayConfig } from \"@langchain/core/utils/gateway\";\nimport {\n  ChatOpenAICompletions,\n  type ChatOpenAICallOptions,\n  type OpenAIChatInput,\n  type OpenAIClient,\n  type OpenAICoreRequestOptions,\n} from \"@langchain/openai\";\n\nconst FIREWORKS_BASE_URL = \"https://api.fireworks.ai/inference/v1\";\nconst DEFAULT_FIREWORKS_CHAT_MODEL =\n  \"accounts/fireworks/models/llama-v3p1-8b-instruct\";\n\ntype FireworksUnsupportedArgs =\n  | \"frequencyPenalty\"\n  | \"presencePenalty\"\n  | \"logitBias\"\n  | \"functions\";\n\ntype FireworksUnsupportedCallOptions = \"functions\" | \"function_call\";\n\nexport interface ChatFireworksInput\n  extends\n    Partial<Omit<OpenAIChatInput, \"openAIApiKey\" | FireworksUnsupportedArgs>>,\n    BaseChatModelParams {\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 ChatFireworksCallOptions = Partial<\n  Omit<ChatOpenAICallOptions, FireworksUnsupportedCallOptions>\n>;\n\n/**\n * Fireworks chat model integration.\n *\n * The Fireworks chat API is OpenAI-compatible with a smaller supported\n * parameter surface than the upstream OpenAI endpoints.\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 { ChatFireworks } from \"@langchain/fireworks\";\n *\n * const model = new ChatFireworks({\n *   model: \"accounts/fireworks/models/firefunction-v2\",\n *   temperature: 0,\n * });\n *\n * const response = await model.invoke(\"Tell me a short joke about fireworks.\");\n * ```\n */\nexport class ChatFireworks extends ChatOpenAICompletions<ChatFireworksCallOptions> {\n  static lc_name() {\n    return \"ChatFireworks\";\n  }\n\n  lc_namespace = [\"langchain\", \"chat_models\", \"fireworks\"];\n\n  lc_serializable = true;\n\n  fireworksApiKey?: string;\n\n  apiKey?: string;\n\n  constructor(model: string, fields?: Omit<ChatFireworksInput, \"model\">);\n  constructor(fields?: ChatFireworksInput);\n  constructor(\n    modelOrFields?: string | ChatFireworksInput,\n    fieldsArg?: Omit<ChatFireworksInput, \"model\">\n  ) {\n    const fields =\n      typeof modelOrFields === \"string\"\n        ? { ...(fieldsArg ?? {}), model: modelOrFields }\n        : (modelOrFields ?? {});\n\n    const gatewayConfig = resolveLangSmithGatewayConfig({\n      baseURL:\n        fields.configuration?.baseURL ??\n        (getEnvironmentVariable(\"FIREWORKS_API_BASE\") ||\n          getEnvironmentVariable(\"FIREWORKS_BASE_URL\") ||\n          undefined),\n      providerPath: \"fireworks\",\n    });\n    const fireworksApiKey =\n      fields.apiKey ||\n      fields.fireworksApiKey ||\n      gatewayConfig.apiKey ||\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      model: fields.model ?? fields.modelName ?? DEFAULT_FIREWORKS_CHAT_MODEL,\n      apiKey: fireworksApiKey,\n      configuration: {\n        ...fields.configuration,\n        baseURL: gatewayConfig.baseURL ?? FIREWORKS_BASE_URL,\n      },\n      streamUsage: false,\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  getLsParams(options: this[\"ParsedCallOptions\"]): LangSmithParams {\n    const params = super.getLsParams(options);\n    params.ls_provider = \"fireworks\";\n    return params;\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.Chat.ChatCompletionCreateParamsStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<AsyncIterable<OpenAIClient.Chat.Completions.ChatCompletionChunk>>;\n\n  async completionWithRetry(\n    request: OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<OpenAIClient.Chat.Completions.ChatCompletion>;\n\n  async completionWithRetry(\n    request:\n      | OpenAIClient.Chat.ChatCompletionCreateParamsStreaming\n      | OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming,\n    options?: OpenAICoreRequestOptions\n  ): Promise<\n    | AsyncIterable<OpenAIClient.Chat.Completions.ChatCompletionChunk>\n    | OpenAIClient.Chat.Completions.ChatCompletion\n  > {\n    delete request.frequency_penalty;\n    delete request.presence_penalty;\n    delete request.logit_bias;\n    delete request.functions;\n\n    return super.completionWithRetry(request, options);\n  }\n\n  protected override get streamEventProvider(): string {\n    return \"fireworks\";\n  }\n}\n\nexport { FIREWORKS_BASE_URL, DEFAULT_FIREWORKS_CHAT_MODEL };\n"],"mappings":";;;;AAcA,MAAM,qBAAqB;AAC3B,MAAM,+BACJ;;;;;;;;;;;;;;;;;;;;;;;;;;AAsDF,IAAa,gBAAb,cAAmCA,kBAAAA,sBAAgD;CACjF,OAAO,UAAU;EACf,OAAO;CACT;CAEA,eAAe;EAAC;EAAa;EAAe;CAAW;CAEvD,kBAAkB;CAElB;CAEA;CAIA,YACE,eACA,WACA;EACA,MAAM,SACJ,OAAO,kBAAkB,WACrB;GAAE,GAAI,aAAa,CAAC;GAAI,OAAO;EAAc,IAC5C,iBAAiB,CAAC;EAEzB,MAAM,iBAAA,GAAA,8BAAA,8BAAA,CAA8C;GAClD,SACE,OAAO,eAAe,aAAA,GAAA,0BAAA,uBAAA,CACE,oBAAoB,MAAA,GAAA,0BAAA,uBAAA,CACnB,oBAAoB,KAC3C,KAAA;GACJ,cAAc;EAChB,CAAC;EACD,MAAM,kBACJ,OAAO,UACP,OAAO,mBACP,cAAc,WAAA,GAAA,0BAAA,uBAAA,CACS,mBAAmB;EAE5C,IAAI,CAAC,iBACH,MAAM,IAAI,MACR,4IACF;EAGF,MAAM;GACJ,GAAG;GACH,OAAO,OAAO,SAAS,OAAO,aAAA;GAC9B,QAAQ;GACR,eAAe;IACb,GAAG,OAAO;IACV,SAAS,cAAc,WAAA;GACzB;GACA,aAAa;EACf,CAAC;EAED,KAAK,kBAAkB;EACvB,KAAK,SAAS;EACd,KAAK,YAAY,wBAAA,OAAuC;CAC1D;CAEA,WAAW;EACT,OAAO;CACT;CAEA,IAAI,aAAoD;EACtD,OAAO;GACL,iBAAiB;GACjB,QAAQ;EACV;CACF;CAEA,YAAY,SAAqD;EAC/D,MAAM,SAAS,MAAM,YAAY,OAAO;EACxC,OAAO,cAAc;EACrB,OAAO;CACT;CAEA,SAAS;EACP,MAAM,SAAS,MAAM,OAAO;EAE5B,IACE,YAAY,UACZ,OAAO,OAAO,WAAW,YACzB,OAAO,UAAU,MACjB;GACA,OAAO,OAAO,OAAO;GACrB,OAAO,OAAO,OAAO;EACvB;EAEA,OAAO;CACT;CAYA,MAAM,oBACJ,SAGA,SAIA;EACA,OAAO,QAAQ;EACf,OAAO,QAAQ;EACf,OAAO,QAAQ;EACf,OAAO,QAAQ;EAEf,OAAO,MAAM,oBAAoB,SAAS,OAAO;CACnD;CAEA,IAAuB,sBAA8B;EACnD,OAAO;CACT;AACF"}