{"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 {\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>,\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\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 {\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);\n constructor(fields?: ChatFireworksInput);\n constructor(\n modelOrFields?: string | ChatFireworksInput,\n fieldsArg?: Omit\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 model: fields.model ?? fields.modelName ?? DEFAULT_FIREWORKS_CHAT_MODEL,\n apiKey: fireworksApiKey,\n configuration: {\n baseURL: FIREWORKS_BASE_URL,\n ...fields.configuration,\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>;\n\n async completionWithRetry(\n request: OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming,\n options?: OpenAICoreRequestOptions\n ): Promise;\n\n async completionWithRetry(\n request:\n | OpenAIClient.Chat.ChatCompletionCreateParamsStreaming\n | OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming,\n options?: OpenAICoreRequestOptions\n ): Promise<\n | AsyncIterable\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":";;;AAaA,MAAM,qBAAqB;AAC3B,MAAM,+BACJ;;;;;;;;;;;;;;;;;;;;;;;;;;AAsDF,IAAa,gBAAb,cAAmCA,kBAAAA,sBAAgD;CACjF,OAAO,UAAU;AACf,SAAO;;CAGT,eAAe;EAAC;EAAa;EAAe;EAAY;CAExD,kBAAkB;CAElB;CAEA;CAIA,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,OAAO,OAAO,SAAS,OAAO,aAAA;GAC9B,QAAQ;GACR,eAAe;IACb,SAAS;IACT,GAAG,OAAO;IACX;GACD,aAAa;GACd,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,YAAY,SAAqD;EAC/D,MAAM,SAAS,MAAM,YAAY,QAAQ;AACzC,SAAO,cAAc;AACrB,SAAO;;CAGT,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,SAIA;AACA,SAAO,QAAQ;AACf,SAAO,QAAQ;AACf,SAAO,QAAQ;AACf,SAAO,QAAQ;AAEf,SAAO,MAAM,oBAAoB,SAAS,QAAQ;;CAGpD,IAAuB,sBAA8B;AACnD,SAAO"}