@langchain/google-common 2.1.17 → 2.1.19

package/dist/llms.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"llms.cjs","names":["AbstractGoogleLLMConnection","input: MessageContent","_parameters: GoogleAIModelParams","contents: GeminiContent[]","ChatGoogleBase","fields: ProxyChatInput<AuthOptions>","LLM","fields?: GoogleBaseLLMInput<AuthOptions>","ensureParams","copyAndValidateModelParamsInto","DefaultGeminiSafetyHandler","apiKey: string","ApiKeyGoogleAuth","fields?: GoogleAIBaseLLMInput<AuthOptions>","fields: GoogleBaseLLMInput<AuthOptions>","client: GoogleAbstractedClient","prompt: string","options: this[\"ParsedCallOptions\"]","copyAIModelParams","input: BaseLanguageModelInput","options?: BaseLanguageModelCallOptions","BaseLLM","CallbackManager","GenerationChunk","messages: BaseMessage[]","options?: string[] | BaseLanguageModelCallOptions","_callbacks?: Callbacks"],"sources":["../src/llms.ts"],"sourcesContent":["import { CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { BaseLLM, LLM } from \"@langchain/core/language_models/llms\";\nimport {\n  type BaseLanguageModelCallOptions,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { GenerationChunk } from \"@langchain/core/outputs\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\n\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport {\n  GoogleAIBaseLLMInput,\n  GoogleAIModelParams,\n  GoogleAISafetySetting,\n  GooglePlatformType,\n  GeminiContent,\n  GoogleAIResponseMimeType,\n} from \"./types.js\";\nimport {\n  copyAIModelParams,\n  copyAndValidateModelParamsInto,\n} from \"./utils/common.js\";\nimport { DefaultGeminiSafetyHandler } from \"./utils/gemini.js\";\nimport { ApiKeyGoogleAuth, GoogleAbstractedClient } from \"./auth.js\";\nimport { ensureParams } from \"./utils/failed_handler.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\n\nexport { GoogleBaseLLMInput };\n\nclass GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<\n  MessageContent,\n  AuthOptions\n> {\n  async formatContents(\n    input: MessageContent,\n    _parameters: GoogleAIModelParams\n  ): Promise<GeminiContent[]> {\n    const parts = await this.api.messageContentToParts!(input);\n    const contents: GeminiContent[] = [\n      {\n        role: \"user\", // Required by Vertex AI\n        parts,\n      },\n    ];\n    return contents;\n  }\n}\n\ntype ProxyChatInput<AuthOptions> = GoogleAIBaseLLMInput<AuthOptions> & {\n  connection: GoogleLLMConnection<AuthOptions>;\n};\n\nclass ProxyChatGoogle<AuthOptions> extends ChatGoogleBase<AuthOptions> {\n  constructor(fields: ProxyChatInput<AuthOptions>) {\n    super(fields);\n  }\n\n  buildAbstractedClient(\n    fields: ProxyChatInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    return fields.connection.client;\n  }\n}\n\n/**\n * Integration with an LLM.\n */\nexport abstract class GoogleBaseLLM<AuthOptions>\n  extends LLM<BaseLanguageModelCallOptions>\n  implements GoogleBaseLLMInput<AuthOptions>\n{\n  // Used for tracing, replace with the same name as your class\n  static lc_name() {\n    return \"GoogleLLM\";\n  }\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      authOptions: \"GOOGLE_AUTH_OPTIONS\",\n    };\n  }\n\n  originalFields?: GoogleBaseLLMInput<AuthOptions>;\n\n  lc_serializable = true;\n\n  modelName = \"gemini-pro\";\n\n  model = \"gemini-pro\";\n\n  temperature = 0.7;\n\n  maxOutputTokens = 1024;\n\n  topP = 0.8;\n\n  topK = 40;\n\n  stopSequences: string[] = [];\n\n  safetySettings: GoogleAISafetySetting[] = [];\n\n  safetyHandler: GoogleAISafetyHandler;\n\n  responseMimeType: GoogleAIResponseMimeType = \"text/plain\";\n\n  protected connection: GoogleLLMConnection<AuthOptions>;\n\n  protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n\n  constructor(fields?: GoogleBaseLLMInput<AuthOptions>) {\n    super(ensureParams(fields));\n    this.originalFields = fields;\n\n    copyAndValidateModelParamsInto(fields, this);\n    this.safetyHandler =\n      fields?.safetyHandler ?? new DefaultGeminiSafetyHandler();\n\n    const client = this.buildClient(fields);\n    this.buildConnection(fields ?? {}, client);\n  }\n\n  abstract buildAbstractedClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  buildConnection(\n    fields: GoogleBaseLLMInput<AuthOptions>,\n    client: GoogleAbstractedClient\n  ) {\n    this.connection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      false\n    );\n\n    this.streamedConnection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      true\n    );\n  }\n\n  get platform(): GooglePlatformType {\n    return this.connection.platform;\n  }\n\n  // Replace\n  _llmType() {\n    return \"googlellm\";\n  }\n\n  formatPrompt(prompt: string): MessageContent {\n    return prompt;\n  }\n\n  /**\n   * For some given input string and options, return a string output.\n   *\n   * Despite the fact that `invoke` is overridden below, we still need this\n   * in order to handle public APi calls to `generate()`.\n   */\n  async _call(\n    prompt: string,\n    options: this[\"ParsedCallOptions\"]\n  ): Promise<string> {\n    const parameters = copyAIModelParams(this, options);\n    const result = await this.connection.request(prompt, parameters, options);\n    const ret = this.connection.api.responseToString(result);\n    return ret;\n  }\n\n  // Normally, you should not override this method and instead should override\n  // _streamResponseChunks. We are doing so here to allow for multimodal inputs into\n  // the LLM.\n  async *_streamIterator(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): AsyncGenerator<string> {\n    // TODO: Refactor callback setup and teardown code into core\n    const prompt = BaseLLM._convertInputToPromptValue(input);\n    const [runnableConfig, callOptions] =\n      this._separateRunnableConfigFromCallOptions(options);\n    const callbackManager_ = await CallbackManager.configure(\n      runnableConfig.callbacks,\n      this.callbacks,\n      runnableConfig.tags,\n      this.tags,\n      runnableConfig.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const extra = {\n      options: callOptions,\n      invocation_params: this?.invocationParams(callOptions),\n      batch_size: 1,\n    };\n    const runManagers = await callbackManager_?.handleLLMStart(\n      this.toJSON(),\n      [prompt.toString()],\n      undefined,\n      undefined,\n      extra,\n      undefined,\n      undefined,\n      runnableConfig.runName\n    );\n    let generation = new GenerationChunk({\n      text: \"\",\n    });\n    const proxyChat = this.createProxyChat();\n    try {\n      for await (const chunk of proxyChat._streamIterator(input, options)) {\n        const stringValue = this.connection.api.chunkToString(chunk);\n        const generationChunk = new GenerationChunk({\n          text: stringValue,\n        });\n        generation = generation.concat(generationChunk);\n        yield stringValue;\n      }\n    } catch (err) {\n      await Promise.all(\n        (runManagers ?? []).map((runManager) => runManager?.handleLLMError(err))\n      );\n      throw err;\n    }\n    await Promise.all(\n      (runManagers ?? []).map((runManager) =>\n        runManager?.handleLLMEnd({\n          generations: [[generation]],\n        })\n      )\n    );\n  }\n\n  async predictMessages(\n    messages: BaseMessage[],\n    options?: string[] | BaseLanguageModelCallOptions,\n    _callbacks?: Callbacks\n  ): Promise<BaseMessage> {\n    const { content } = messages[0];\n    const result = await this.connection.request(\n      content,\n      {},\n      options as BaseLanguageModelCallOptions\n    );\n    const ret = this.connection.api.responseToBaseMessage(result);\n    return ret;\n  }\n\n  /**\n   * Internal implementation detail to allow Google LLMs to support\n   * multimodal input by delegating to the chat model implementation.\n   *\n   * TODO: Replace with something less hacky.\n   */\n  protected createProxyChat(): ChatGoogleBase<AuthOptions> {\n    return new ProxyChatGoogle<AuthOptions>({\n      ...this.originalFields,\n      connection: this.connection,\n    });\n  }\n\n  // TODO: Remove the need to override this - we are doing it to\n  // allow the LLM to handle multimodal types of input.\n  async invoke(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): Promise<string> {\n    const stream = await this._streamIterator(input, options);\n    let generatedOutput = \"\";\n    for await (const chunk of stream) {\n      generatedOutput += chunk;\n    }\n    return generatedOutput;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA+BA,IAAM,sBAAN,cAA+CA,+CAG7C;CACA,MAAM,eACJC,OACAC,aAC0B;EAC1B,MAAM,QAAQ,MAAM,KAAK,IAAI,sBAAuB,MAAM;EAC1D,MAAMC,WAA4B,CAChC;GACE,MAAM;GACN;EACD,CACF;AACD,SAAO;CACR;AACF;AAMD,IAAM,kBAAN,cAA2CC,mCAA4B;CACrE,YAAYC,QAAqC;EAC/C,MAAM,OAAO;CACd;CAED,sBACEA,QACwB;AACxB,SAAO,OAAO,WAAW;CAC1B;AACF;;;;AAKD,IAAsB,gBAAtB,cACUC,0CAEV;CAEE,OAAO,UAAU;AACf,SAAO;CACR;CAED,IAAI,aAAoD;AACtD,SAAO,EACL,aAAa,sBACd;CACF;CAED;CAEA,kBAAkB;CAElB,YAAY;CAEZ,QAAQ;CAER,cAAc;CAEd,kBAAkB;CAElB,OAAO;CAEP,OAAO;CAEP,gBAA0B,CAAE;CAE5B,iBAA0C,CAAE;CAE5C;CAEA,mBAA6C;CAE7C,AAAU;CAEV,AAAU;CAEV,YAAYC,QAA0C;EACpD,MAAMC,oCAAa,OAAO,CAAC;EAC3B,KAAK,iBAAiB;EAEtBC,8CAA+B,QAAQ,KAAK;EAC5C,KAAK,gBACH,QAAQ,iBAAiB,IAAIC;EAE/B,MAAM,SAAS,KAAK,YAAY,OAAO;EACvC,KAAK,gBAAgB,UAAU,CAAE,GAAE,OAAO;CAC3C;CAMD,kBAAkBC,QAAwC;AACxD,SAAO,IAAIC,8BAAiB;CAC7B;CAED,YAAYC,QAAgE;AAC1E,SAAO,QAAQ,iEAAiC,iBAAiB;CAClE;CAED,YACEA,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAErC,QAAO,KAAK,sBAAsB,OAAO;CAE5C;CAED,gBACEC,QACAC,QACA;EACA,KAAK,aAAa,IAAI,oBACpB;GAAE,GAAG;GAAQ,GAAG;EAAM,GACtB,KAAK,QACL,QACA;EAGF,KAAK,qBAAqB,IAAI,oBAC5B;GAAE,GAAG;GAAQ,GAAG;EAAM,GACtB,KAAK,QACL,QACA;CAEH;CAED,IAAI,WAA+B;AACjC,SAAO,KAAK,WAAW;CACxB;CAGD,WAAW;AACT,SAAO;CACR;CAED,aAAaC,QAAgC;AAC3C,SAAO;CACR;;;;;;;CAQD,MAAM,MACJA,QACAC,SACiB;EACjB,MAAM,aAAaC,iCAAkB,MAAM,QAAQ;EACnD,MAAM,SAAS,MAAM,KAAK,WAAW,QAAQ,QAAQ,YAAY,QAAQ;EACzE,MAAM,MAAM,KAAK,WAAW,IAAI,iBAAiB,OAAO;AACxD,SAAO;CACR;CAKD,OAAO,gBACLC,OACAC,SACwB;EAExB,MAAM,SAASC,8CAAQ,2BAA2B,MAAM;EACxD,MAAM,CAAC,gBAAgB,YAAY,GACjC,KAAK,uCAAuC,QAAQ;EACtD,MAAM,mBAAmB,MAAMC,mDAAgB,UAC7C,eAAe,WACf,KAAK,WACL,eAAe,MACf,KAAK,MACL,eAAe,UACf,KAAK,UACL,EAAE,SAAS,KAAK,QAAS,EAC1B;EACD,MAAM,QAAQ;GACZ,SAAS;GACT,mBAAmB,MAAM,iBAAiB,YAAY;GACtD,YAAY;EACb;EACD,MAAM,cAAc,MAAM,kBAAkB,eAC1C,KAAK,QAAQ,EACb,CAAC,OAAO,UAAU,AAAC,GACnB,QACA,QACA,OACA,QACA,QACA,eAAe,QAChB;EACD,IAAI,aAAa,IAAIC,yCAAgB,EACnC,MAAM,GACP;EACD,MAAM,YAAY,KAAK,iBAAiB;AACxC,MAAI;AACF,cAAW,MAAM,SAAS,UAAU,gBAAgB,OAAO,QAAQ,EAAE;IACnE,MAAM,cAAc,KAAK,WAAW,IAAI,cAAc,MAAM;IAC5D,MAAM,kBAAkB,IAAIA,yCAAgB,EAC1C,MAAM,YACP;IACD,aAAa,WAAW,OAAO,gBAAgB;IAC/C,MAAM;GACP;EACF,SAAQ,KAAK;GACZ,MAAM,QAAQ,KACX,eAAe,CAAE,GAAE,IAAI,CAAC,eAAe,YAAY,eAAe,IAAI,CAAC,CACzE;AACD,SAAM;EACP;EACD,MAAM,QAAQ,KACX,eAAe,CAAE,GAAE,IAAI,CAAC,eACvB,YAAY,aAAa,EACvB,aAAa,CAAC,CAAC,UAAW,CAAC,EAC5B,EAAC,CACH,CACF;CACF;CAED,MAAM,gBACJC,UACAC,SACAC,YACsB;EACtB,MAAM,EAAE,SAAS,GAAG,SAAS;EAC7B,MAAM,SAAS,MAAM,KAAK,WAAW,QACnC,SACA,CAAE,GACF,QACD;EACD,MAAM,MAAM,KAAK,WAAW,IAAI,sBAAsB,OAAO;AAC7D,SAAO;CACR;;;;;;;CAQD,AAAU,kBAA+C;AACvD,SAAO,IAAI,gBAA6B;GACtC,GAAG,KAAK;GACR,YAAY,KAAK;EAClB;CACF;CAID,MAAM,OACJP,OACAC,SACiB;EACjB,MAAM,SAAS,MAAM,KAAK,gBAAgB,OAAO,QAAQ;EACzD,IAAI,kBAAkB;AACtB,aAAW,MAAM,SAAS,QACxB,mBAAmB;AAErB,SAAO;CACR;AACF"}
+{"version":3,"file":"llms.cjs","names":["AbstractGoogleLLMConnection","ChatGoogleBase","LLM","ensureParams","DefaultGeminiSafetyHandler","ApiKeyGoogleAuth","copyAIModelParams","BaseLLM","CallbackManager","GenerationChunk"],"sources":["../src/llms.ts"],"sourcesContent":["import { CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { BaseLLM, LLM } from \"@langchain/core/language_models/llms\";\nimport {\n  type BaseLanguageModelCallOptions,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { GenerationChunk } from \"@langchain/core/outputs\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\n\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport {\n  GoogleAIBaseLLMInput,\n  GoogleAIModelParams,\n  GoogleAISafetySetting,\n  GooglePlatformType,\n  GeminiContent,\n  GoogleAIResponseMimeType,\n} from \"./types.js\";\nimport {\n  copyAIModelParams,\n  copyAndValidateModelParamsInto,\n} from \"./utils/common.js\";\nimport { DefaultGeminiSafetyHandler } from \"./utils/gemini.js\";\nimport { ApiKeyGoogleAuth, GoogleAbstractedClient } from \"./auth.js\";\nimport { ensureParams } from \"./utils/failed_handler.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\n\nexport { GoogleBaseLLMInput };\n\nclass GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<\n  MessageContent,\n  AuthOptions\n> {\n  async formatContents(\n    input: MessageContent,\n    _parameters: GoogleAIModelParams\n  ): Promise<GeminiContent[]> {\n    const parts = await this.api.messageContentToParts!(input);\n    const contents: GeminiContent[] = [\n      {\n        role: \"user\", // Required by Vertex AI\n        parts,\n      },\n    ];\n    return contents;\n  }\n}\n\ntype ProxyChatInput<AuthOptions> = GoogleAIBaseLLMInput<AuthOptions> & {\n  connection: GoogleLLMConnection<AuthOptions>;\n};\n\nclass ProxyChatGoogle<AuthOptions> extends ChatGoogleBase<AuthOptions> {\n  constructor(fields: ProxyChatInput<AuthOptions>) {\n    super(fields);\n  }\n\n  buildAbstractedClient(\n    fields: ProxyChatInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    return fields.connection.client;\n  }\n}\n\n/**\n * Integration with an LLM.\n */\nexport abstract class GoogleBaseLLM<AuthOptions>\n  extends LLM<BaseLanguageModelCallOptions>\n  implements GoogleBaseLLMInput<AuthOptions>\n{\n  // Used for tracing, replace with the same name as your class\n  static lc_name() {\n    return \"GoogleLLM\";\n  }\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      authOptions: \"GOOGLE_AUTH_OPTIONS\",\n    };\n  }\n\n  originalFields?: GoogleBaseLLMInput<AuthOptions>;\n\n  lc_serializable = true;\n\n  modelName = \"gemini-pro\";\n\n  model = \"gemini-pro\";\n\n  temperature = 0.7;\n\n  maxOutputTokens = 1024;\n\n  topP = 0.8;\n\n  topK = 40;\n\n  stopSequences: string[] = [];\n\n  safetySettings: GoogleAISafetySetting[] = [];\n\n  safetyHandler: GoogleAISafetyHandler;\n\n  responseMimeType: GoogleAIResponseMimeType = \"text/plain\";\n\n  protected connection: GoogleLLMConnection<AuthOptions>;\n\n  protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n\n  constructor(fields?: GoogleBaseLLMInput<AuthOptions>) {\n    super(ensureParams(fields));\n    this.originalFields = fields;\n\n    copyAndValidateModelParamsInto(fields, this);\n    this.safetyHandler =\n      fields?.safetyHandler ?? new DefaultGeminiSafetyHandler();\n\n    const client = this.buildClient(fields);\n    this.buildConnection(fields ?? {}, client);\n  }\n\n  abstract buildAbstractedClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  buildConnection(\n    fields: GoogleBaseLLMInput<AuthOptions>,\n    client: GoogleAbstractedClient\n  ) {\n    this.connection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      false\n    );\n\n    this.streamedConnection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      true\n    );\n  }\n\n  get platform(): GooglePlatformType {\n    return this.connection.platform;\n  }\n\n  // Replace\n  _llmType() {\n    return \"googlellm\";\n  }\n\n  formatPrompt(prompt: string): MessageContent {\n    return prompt;\n  }\n\n  /**\n   * For some given input string and options, return a string output.\n   *\n   * Despite the fact that `invoke` is overridden below, we still need this\n   * in order to handle public APi calls to `generate()`.\n   */\n  async _call(\n    prompt: string,\n    options: this[\"ParsedCallOptions\"]\n  ): Promise<string> {\n    const parameters = copyAIModelParams(this, options);\n    const result = await this.connection.request(prompt, parameters, options);\n    const ret = this.connection.api.responseToString(result);\n    return ret;\n  }\n\n  // Normally, you should not override this method and instead should override\n  // _streamResponseChunks. We are doing so here to allow for multimodal inputs into\n  // the LLM.\n  async *_streamIterator(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): AsyncGenerator<string> {\n    // TODO: Refactor callback setup and teardown code into core\n    const prompt = BaseLLM._convertInputToPromptValue(input);\n    const [runnableConfig, callOptions] =\n      this._separateRunnableConfigFromCallOptions(options);\n    const callbackManager_ = await CallbackManager.configure(\n      runnableConfig.callbacks,\n      this.callbacks,\n      runnableConfig.tags,\n      this.tags,\n      runnableConfig.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const extra = {\n      options: callOptions,\n      invocation_params: this?.invocationParams(callOptions),\n      batch_size: 1,\n    };\n    const runManagers = await callbackManager_?.handleLLMStart(\n      this.toJSON(),\n      [prompt.toString()],\n      undefined,\n      undefined,\n      extra,\n      undefined,\n      undefined,\n      runnableConfig.runName\n    );\n    let generation = new GenerationChunk({\n      text: \"\",\n    });\n    const proxyChat = this.createProxyChat();\n    try {\n      for await (const chunk of proxyChat._streamIterator(input, options)) {\n        const stringValue = this.connection.api.chunkToString(chunk);\n        const generationChunk = new GenerationChunk({\n          text: stringValue,\n        });\n        generation = generation.concat(generationChunk);\n        yield stringValue;\n      }\n    } catch (err) {\n      await Promise.all(\n        (runManagers ?? []).map((runManager) => runManager?.handleLLMError(err))\n      );\n      throw err;\n    }\n    await Promise.all(\n      (runManagers ?? []).map((runManager) =>\n        runManager?.handleLLMEnd({\n          generations: [[generation]],\n        })\n      )\n    );\n  }\n\n  async predictMessages(\n    messages: BaseMessage[],\n    options?: string[] | BaseLanguageModelCallOptions,\n    _callbacks?: Callbacks\n  ): Promise<BaseMessage> {\n    const { content } = messages[0];\n    const result = await this.connection.request(\n      content,\n      {},\n      options as BaseLanguageModelCallOptions\n    );\n    const ret = this.connection.api.responseToBaseMessage(result);\n    return ret;\n  }\n\n  /**\n   * Internal implementation detail to allow Google LLMs to support\n   * multimodal input by delegating to the chat model implementation.\n   *\n   * TODO: Replace with something less hacky.\n   */\n  protected createProxyChat(): ChatGoogleBase<AuthOptions> {\n    return new ProxyChatGoogle<AuthOptions>({\n      ...this.originalFields,\n      connection: this.connection,\n    });\n  }\n\n  // TODO: Remove the need to override this - we are doing it to\n  // allow the LLM to handle multimodal types of input.\n  async invoke(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): Promise<string> {\n    const stream = await this._streamIterator(input, options);\n    let generatedOutput = \"\";\n    for await (const chunk of stream) {\n      generatedOutput += chunk;\n    }\n    return generatedOutput;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA+BA,IAAM,sBAAN,cAA+CA,+CAG7C;CACA,MAAM,eACJ,OACA,aAC0B;AAQ1B,SANkC,CAChC;GACE,MAAM;GACN,OAJU,MAAM,KAAK,IAAI,sBAAuB,MAAM;GAKvD,CACF;;;AASL,IAAM,kBAAN,cAA2CC,mCAA4B;CACrE,YAAY,QAAqC;AAC/C,QAAM,OAAO;;CAGf,sBACE,QACwB;AACxB,SAAO,OAAO,WAAW;;;;;;AAO7B,IAAsB,gBAAtB,cACUC,yCAEV;CAEE,OAAO,UAAU;AACf,SAAO;;CAGT,IAAI,aAAoD;AACtD,SAAO,EACL,aAAa,uBACd;;CAGH;CAEA,kBAAkB;CAElB,YAAY;CAEZ,QAAQ;CAER,cAAc;CAEd,kBAAkB;CAElB,OAAO;CAEP,OAAO;CAEP,gBAA0B,EAAE;CAE5B,iBAA0C,EAAE;CAE5C;CAEA,mBAA6C;CAE7C,AAAU;CAEV,AAAU;CAEV,YAAY,QAA0C;AACpD,QAAMC,oCAAa,OAAO,CAAC;AAC3B,OAAK,iBAAiB;AAEtB,gDAA+B,QAAQ,KAAK;AAC5C,OAAK,gBACH,QAAQ,iBAAiB,IAAIC,2CAA4B;EAE3D,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,OAAK,gBAAgB,UAAU,EAAE,EAAE,OAAO;;CAO5C,kBAAkB,QAAwC;AACxD,SAAO,IAAIC,8BAAiB,OAAO;;CAGrC,YAAY,QAAgE;AAC1E,SAAO,QAAQ,gEAAiC,iBAAiB;;CAGnE,YACE,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAErC,QAAO,KAAK,sBAAsB,OAAO;;CAI7C,gBACE,QACA,QACA;AACA,OAAK,aAAa,IAAI,oBACpB;GAAE,GAAG;GAAQ,GAAG;GAAM,EACtB,KAAK,QACL,QACA,MACD;AAED,OAAK,qBAAqB,IAAI,oBAC5B;GAAE,GAAG;GAAQ,GAAG;GAAM,EACtB,KAAK,QACL,QACA,KACD;;CAGH,IAAI,WAA+B;AACjC,SAAO,KAAK,WAAW;;CAIzB,WAAW;AACT,SAAO;;CAGT,aAAa,QAAgC;AAC3C,SAAO;;;;;;;;CAST,MAAM,MACJ,QACA,SACiB;EACjB,MAAM,aAAaC,iCAAkB,MAAM,QAAQ;EACnD,MAAM,SAAS,MAAM,KAAK,WAAW,QAAQ,QAAQ,YAAY,QAAQ;AAEzE,SADY,KAAK,WAAW,IAAI,iBAAiB,OAAO;;CAO1D,OAAO,gBACL,OACA,SACwB;EAExB,MAAM,SAASC,6CAAQ,2BAA2B,MAAM;EACxD,MAAM,CAAC,gBAAgB,eACrB,KAAK,uCAAuC,QAAQ;EACtD,MAAM,mBAAmB,MAAMC,kDAAgB,UAC7C,eAAe,WACf,KAAK,WACL,eAAe,MACf,KAAK,MACL,eAAe,UACf,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B;EACD,MAAM,QAAQ;GACZ,SAAS;GACT,mBAAmB,MAAM,iBAAiB,YAAY;GACtD,YAAY;GACb;EACD,MAAM,cAAc,MAAM,kBAAkB,eAC1C,KAAK,QAAQ,EACb,CAAC,OAAO,UAAU,CAAC,EACnB,QACA,QACA,OACA,QACA,QACA,eAAe,QAChB;EACD,IAAI,aAAa,IAAIC,wCAAgB,EACnC,MAAM,IACP,CAAC;EACF,MAAM,YAAY,KAAK,iBAAiB;AACxC,MAAI;AACF,cAAW,MAAM,SAAS,UAAU,gBAAgB,OAAO,QAAQ,EAAE;IACnE,MAAM,cAAc,KAAK,WAAW,IAAI,cAAc,MAAM;IAC5D,MAAM,kBAAkB,IAAIA,wCAAgB,EAC1C,MAAM,aACP,CAAC;AACF,iBAAa,WAAW,OAAO,gBAAgB;AAC/C,UAAM;;WAED,KAAK;AACZ,SAAM,QAAQ,KACX,eAAe,EAAE,EAAE,KAAK,eAAe,YAAY,eAAe,IAAI,CAAC,CACzE;AACD,SAAM;;AAER,QAAM,QAAQ,KACX,eAAe,EAAE,EAAE,KAAK,eACvB,YAAY,aAAa,EACvB,aAAa,CAAC,CAAC,WAAW,CAAC,EAC5B,CAAC,CACH,CACF;;CAGH,MAAM,gBACJ,UACA,SACA,YACsB;EACtB,MAAM,EAAE,YAAY,SAAS;EAC7B,MAAM,SAAS,MAAM,KAAK,WAAW,QACnC,SACA,EAAE,EACF,QACD;AAED,SADY,KAAK,WAAW,IAAI,sBAAsB,OAAO;;;;;;;;CAU/D,AAAU,kBAA+C;AACvD,SAAO,IAAI,gBAA6B;GACtC,GAAG,KAAK;GACR,YAAY,KAAK;GAClB,CAAC;;CAKJ,MAAM,OACJ,OACA,SACiB;EACjB,MAAM,SAAS,MAAM,KAAK,gBAAgB,OAAO,QAAQ;EACzD,IAAI,kBAAkB;AACtB,aAAW,MAAM,SAAS,OACxB,oBAAmB;AAErB,SAAO"}

package/dist/llms.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"llms.d.cts","names":["Callbacks","LLM","BaseLanguageModelCallOptions","BaseLanguageModelInput","BaseMessage","MessageContent","AbstractGoogleLLMConnection","GoogleAIBaseLLMInput","GoogleAIModelParams","GoogleAISafetySetting","GooglePlatformType","GeminiContent","GoogleAIResponseMimeType","GoogleAbstractedClient","ChatGoogleBase","GoogleBaseLLMInput","GoogleAISafetyHandler","GoogleLLMConnection","AuthOptions","Promise","GoogleBaseLLM","AsyncGenerator"],"sources":["../src/llms.d.ts"],"sourcesContent":["import { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { LLM } from \"@langchain/core/language_models/llms\";\nimport { type BaseLanguageModelCallOptions, BaseLanguageModelInput } from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GooglePlatformType, GeminiContent, GoogleAIResponseMimeType } from \"./types.js\";\nimport { GoogleAbstractedClient } from \"./auth.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\nexport { GoogleBaseLLMInput };\ndeclare class GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<MessageContent, AuthOptions> {\n    formatContents(input: MessageContent, _parameters: GoogleAIModelParams): Promise<GeminiContent[]>;\n}\n/**\n * Integration with an LLM.\n */\nexport declare abstract class GoogleBaseLLM<AuthOptions> extends LLM<BaseLanguageModelCallOptions> implements GoogleBaseLLMInput<AuthOptions> {\n    static lc_name(): string;\n    get lc_secrets(): {\n        [key: string]: string;\n    } | undefined;\n    originalFields?: GoogleBaseLLMInput<AuthOptions>;\n    lc_serializable: boolean;\n    modelName: string;\n    model: string;\n    temperature: number;\n    maxOutputTokens: number;\n    topP: number;\n    topK: number;\n    stopSequences: string[];\n    safetySettings: GoogleAISafetySetting[];\n    safetyHandler: GoogleAISafetyHandler;\n    responseMimeType: GoogleAIResponseMimeType;\n    protected connection: GoogleLLMConnection<AuthOptions>;\n    protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n    constructor(fields?: GoogleBaseLLMInput<AuthOptions>);\n    abstract buildAbstractedClient(fields?: GoogleAIBaseLLMInput<AuthOptions>): GoogleAbstractedClient;\n    buildApiKeyClient(apiKey: string): GoogleAbstractedClient;\n    buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined;\n    buildClient(fields?: GoogleAIBaseLLMInput<AuthOptions>): GoogleAbstractedClient;\n    buildConnection(fields: GoogleBaseLLMInput<AuthOptions>, client: GoogleAbstractedClient): void;\n    get platform(): GooglePlatformType;\n    _llmType(): string;\n    formatPrompt(prompt: string): MessageContent;\n    /**\n     * For some given input string and options, return a string output.\n     *\n     * Despite the fact that `invoke` is overridden below, we still need this\n     * in order to handle public APi calls to `generate()`.\n     */\n    _call(prompt: string, options: this[\"ParsedCallOptions\"]): Promise<string>;\n    _streamIterator(input: BaseLanguageModelInput, options?: BaseLanguageModelCallOptions): AsyncGenerator<string>;\n    predictMessages(messages: BaseMessage[], options?: string[] | BaseLanguageModelCallOptions, _callbacks?: Callbacks): Promise<BaseMessage>;\n    /**\n     * Internal implementation detail to allow Google LLMs to support\n     * multimodal input by delegating to the chat model implementation.\n     *\n     * TODO: Replace with something less hacky.\n     */\n    protected createProxyChat(): ChatGoogleBase<AuthOptions>;\n    invoke(input: BaseLanguageModelInput, options?: BaseLanguageModelCallOptions): Promise<string>;\n}\n//# sourceMappingURL=llms.d.ts.map"],"mappings":";;;;;;;;;;AAS8B,cAChBiB,mBAAmBC,CAAAA,WAAAA,CAAAA,SAAsBZ,2BAAtB,CAAkDD,cAAlD,EAAkEa,WAAlE,CAAA,CAAA;EAAkDb,cAAAA,CAAAA,KAAAA,EACzDA,cADyDA,EAAAA,WAAAA,EAC5BG,mBAD4BH,CAAAA,EACNc,OADMd,CACEM,aADFN,EAAAA,CAAAA;;;;;AACNc,uBAK/CC,aAL+CD,CAAAA,WAAAA,CAAAA,SAKZlB,GALYkB,CAKRjB,4BALQiB,CAAAA,YAKiCJ,kBALjCI,CAKoDD,WALpDC,CAAAA,CAAAA;EADtBb,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA2B,IAAA,UAAA,CAAA,CAAA,EAAA;IAMpDc,CAAAA,GAAAA,EAAAA,MAAa,CAAA,EAAA,MAAAF;EAA0BhB,CAAAA,GAAAA,SAAAA;EAA4DgB,cAAAA,CAAAA,EAK5GH,kBAL4GG,CAKzFA,WALyFA,CAAAA;EAKzFA,eAAAA,EAAAA,OAAAA;EAAnBH,SAAAA,EAAAA,MAAAA;EASDN,KAAAA,EAAAA,MAAAA;EACDO,WAAAA,EAAAA,MAAAA;EACGJ,eAAAA,EAAAA,MAAAA;EACwBM,IAAAA,EAAAA,MAAAA;EAApBD,IAAAA,EAAAA,MAAAA;EAC4BC,aAAAA,EAAAA,MAAAA,EAAAA;EAApBD,cAAAA,EAJdR,qBAIcQ,EAAAA;EACUC,aAAAA,EAJzBF,qBAIyBE;EAAnBH,gBAAAA,EAHHH,wBAGGG;EACwCG,UAAAA,UAAAA,EAHvCD,mBAGuCC,CAHnBA,WAGmBA,CAAAA;EAArBX,UAAAA,kBAAAA,EAFVU,mBAEUV,CAFUW,WAEVX,CAAAA;EAAoCM,WAAAA,CAAAA,MAAAA,CAAAA,EADvDE,kBACuDF,CADpCK,WACoCL,CAAAA;EACzCA,SAAAA,qBAAAA,CAAAA,MAAAA,CAAAA,EADKN,oBACLM,CAD0BK,WAC1BL,CAAAA,CAAAA,EADyCA,sBACzCA;EACOK,iBAAAA,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EADPL,sBACOK;EAArBX,WAAAA,CAAAA,MAAAA,CAAAA,EAAAA,oBAAAA,CAAqBW,WAArBX,CAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACqBW,WAAAA,CAAAA,MAAAA,CAAAA,EAArBX,oBAAqBW,CAAAA,WAAAA,CAAAA,CAAAA,EAAeL,sBAAfK;EAArBX,eAAAA,CAAAA,MAAAA,EACGQ,kBADHR,CACsBW,WADtBX,CAAAA,EAAAA,MAAAA,EAC4CM,sBAD5CN,CAAAA,EAAAA,IAAAA;EAAoCM,IAAAA,QAAAA,CAAAA,CAAAA,EAEzCH,kBAFyCG;EACdK,QAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAnBH,YAAAA,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAGMV,cAHNU;EAAyCF;;;;;;EAWuBQ,KAAAA,CAAAA,MAAAA,EAAAA,MAAAA,EAAAA,OAAAA,EAAAA,IAAAA,CAAAA,mBAAAA,CAAAA,CAAAA,EAD7BF,OAC6BE,CAAAA,MAAAA,CAAAA;EAC9DjB,eAAAA,CAAAA,KAAAA,EADHD,sBACGC,EAAAA,OAAAA,CAAAA,EAD+BF,4BAC/BE,CAAAA,EAD8DiB,cAC9DjB,CAAAA,MAAAA,CAAAA;EAAoCF,eAAAA,CAAAA,QAAAA,EAApCE,WAAoCF,EAAAA,EAAAA,OAAAA,CAAAA,EAAAA,MAAAA,EAAAA,GAAAA,4BAAAA,EAAAA,UAAAA,CAAAA,EAA2CF,SAA3CE,CAAAA,EAAuDiB,OAAvDjB,CAA+DE,WAA/DF,CAAAA;EAA2CF;;;;;;EAQzDE,UAAAA,eAAAA,CAAAA,CAAAA,EADnBY,cACmBZ,CADJgB,WACIhB,CAAAA;EAA+BiB,MAAAA,CAAAA,KAAAA,EAAjEhB,sBAAiEgB,EAAAA,OAAAA,CAAAA,EAA/BjB,4BAA+BiB,CAAAA,EAAAA,OAAAA,CAAAA,MAAAA,CAAAA"}
+{"version":3,"file":"llms.d.cts","names":[],"sources":["../src/llms.ts"],"mappings":";;;;;;;;;;cA+BM,mBAAA,sBAAyC,2BAAA,CAC7C,cAAA,EACA,WAAA;EAEM,cAAA,CACJ,KAAA,EAAO,cAAA,EACP,WAAA,EAAa,mBAAA,GACZ,OAAA,CAAQ,aAAA;AAAA;;;;uBA+BS,aAAA,sBACZ,GAAA,CAAI,4BAAA,aACD,kBAAA,CAAmB,WAAA;EAAA,OAGvB,OAAA,CAAA;EAAA,IAIH,UAAA,CAAA;IAAA,CAAiB,GAAA;EAAA;EAMrB,cAAA,GAAiB,kBAAA,CAAmB,WAAA;EAEpC,eAAA;EAEA,SAAA;EAEA,KAAA;EAEA,WAAA;EAEA,eAAA;EAEA,IAAA;EAEA,IAAA;EAEA,aAAA;EAEA,cAAA,EAAgB,qBAAA;EAEhB,aAAA,EAAe,qBAAA;EAEf,gBAAA,EAAkB,wBAAA;EAAA,UAER,UAAA,EAAY,mBAAA,CAAoB,WAAA;EAAA,UAEhC,kBAAA,EAAoB,mBAAA,CAAoB,WAAA;EAElD,WAAA,CAAY,MAAA,GAAS,kBAAA,CAAmB,WAAA;EAAA,SAY/B,qBAAA,CACP,MAAA,GAAS,oBAAA,CAAqB,WAAA,IAC7B,sBAAA;EAEH,iBAAA,CAAkB,MAAA,WAAiB,sBAAA;EAInC,WAAA,CAAY,MAAA,GAAS,oBAAA,CAAqB,WAAA;EAI1C,WAAA,CACE,MAAA,GAAS,oBAAA,CAAqB,WAAA,IAC7B,sBAAA;EASH,eAAA,CACE,MAAA,EAAQ,kBAAA,CAAmB,WAAA,GAC3B,MAAA,EAAQ,sBAAA;EAAA,IAiBN,QAAA,CAAA,GAAY,kBAAA;EAKhB,QAAA,CAAA;EAIA,YAAA,CAAa,MAAA,WAAiB,cAAA;EArEZ;;;;;;EA+EZ,KAAA,CACJ,MAAA,UACA,OAAA,8BACC,OAAA;EAUI,eAAA,CACL,KAAA,EAAO,sBAAA,EACP,OAAA,GAAU,4BAAA,GACT,cAAA;EAyDG,eAAA,CACJ,QAAA,EAAU,WAAA,IACV,OAAA,cAAqB,4BAAA,EACrB,UAAA,GAAa,SAAA,GACZ,OAAA,CAAQ,WAAA;EAxIR;;;;;;EAAA,UAyJO,eAAA,CAAA,GAAmB,cAAA,CAAe,WAAA;EAStC,MAAA,CACJ,KAAA,EAAO,sBAAA,EACP,OAAA,GAAU,4BAAA,GACT,OAAA;AAAA"}

package/dist/llms.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llms.d.ts","names":["Callbacks","LLM","BaseLanguageModelCallOptions","BaseLanguageModelInput","BaseMessage","MessageContent","AbstractGoogleLLMConnection","GoogleAIBaseLLMInput","GoogleAIModelParams","GoogleAISafetySetting","GooglePlatformType","GeminiContent","GoogleAIResponseMimeType","GoogleAbstractedClient","ChatGoogleBase","GoogleBaseLLMInput","GoogleAISafetyHandler","GoogleLLMConnection","AuthOptions","Promise","GoogleBaseLLM","AsyncGenerator"],"sources":["../src/llms.d.ts"],"sourcesContent":["import { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { LLM } from \"@langchain/core/language_models/llms\";\nimport { type BaseLanguageModelCallOptions, BaseLanguageModelInput } from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GooglePlatformType, GeminiContent, GoogleAIResponseMimeType } from \"./types.js\";\nimport { GoogleAbstractedClient } from \"./auth.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\nexport { GoogleBaseLLMInput };\ndeclare class GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<MessageContent, AuthOptions> {\n    formatContents(input: MessageContent, _parameters: GoogleAIModelParams): Promise<GeminiContent[]>;\n}\n/**\n * Integration with an LLM.\n */\nexport declare abstract class GoogleBaseLLM<AuthOptions> extends LLM<BaseLanguageModelCallOptions> implements GoogleBaseLLMInput<AuthOptions> {\n    static lc_name(): string;\n    get lc_secrets(): {\n        [key: string]: string;\n    } | undefined;\n    originalFields?: GoogleBaseLLMInput<AuthOptions>;\n    lc_serializable: boolean;\n    modelName: string;\n    model: string;\n    temperature: number;\n    maxOutputTokens: number;\n    topP: number;\n    topK: number;\n    stopSequences: string[];\n    safetySettings: GoogleAISafetySetting[];\n    safetyHandler: GoogleAISafetyHandler;\n    responseMimeType: GoogleAIResponseMimeType;\n    protected connection: GoogleLLMConnection<AuthOptions>;\n    protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n    constructor(fields?: GoogleBaseLLMInput<AuthOptions>);\n    abstract buildAbstractedClient(fields?: GoogleAIBaseLLMInput<AuthOptions>): GoogleAbstractedClient;\n    buildApiKeyClient(apiKey: string): GoogleAbstractedClient;\n    buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined;\n    buildClient(fields?: GoogleAIBaseLLMInput<AuthOptions>): GoogleAbstractedClient;\n    buildConnection(fields: GoogleBaseLLMInput<AuthOptions>, client: GoogleAbstractedClient): void;\n    get platform(): GooglePlatformType;\n    _llmType(): string;\n    formatPrompt(prompt: string): MessageContent;\n    /**\n     * For some given input string and options, return a string output.\n     *\n     * Despite the fact that `invoke` is overridden below, we still need this\n     * in order to handle public APi calls to `generate()`.\n     */\n    _call(prompt: string, options: this[\"ParsedCallOptions\"]): Promise<string>;\n    _streamIterator(input: BaseLanguageModelInput, options?: BaseLanguageModelCallOptions): AsyncGenerator<string>;\n    predictMessages(messages: BaseMessage[], options?: string[] | BaseLanguageModelCallOptions, _callbacks?: Callbacks): Promise<BaseMessage>;\n    /**\n     * Internal implementation detail to allow Google LLMs to support\n     * multimodal input by delegating to the chat model implementation.\n     *\n     * TODO: Replace with something less hacky.\n     */\n    protected createProxyChat(): ChatGoogleBase<AuthOptions>;\n    invoke(input: BaseLanguageModelInput, options?: BaseLanguageModelCallOptions): Promise<string>;\n}\n//# sourceMappingURL=llms.d.ts.map"],"mappings":";;;;;;;;;;AAS8B,cAChBiB,mBAAmBC,CAAAA,WAAAA,CAAAA,SAAsBZ,2BAAtB,CAAkDD,cAAlD,EAAkEa,WAAlE,CAAA,CAAA;EAAkDb,cAAAA,CAAAA,KAAAA,EACzDA,cADyDA,EAAAA,WAAAA,EAC5BG,mBAD4BH,CAAAA,EACNc,OADMd,CACEM,aADFN,EAAAA,CAAAA;;;;;AACNc,uBAK/CC,aAL+CD,CAAAA,WAAAA,CAAAA,SAKZlB,GALYkB,CAKRjB,4BALQiB,CAAAA,YAKiCJ,kBALjCI,CAKoDD,WALpDC,CAAAA,CAAAA;EADtBb,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA2B,IAAA,UAAA,CAAA,CAAA,EAAA;IAMpDc,CAAAA,GAAAA,EAAAA,MAAa,CAAA,EAAA,MAAAF;EAA0BhB,CAAAA,GAAAA,SAAAA;EAA4DgB,cAAAA,CAAAA,EAK5GH,kBAL4GG,CAKzFA,WALyFA,CAAAA;EAKzFA,eAAAA,EAAAA,OAAAA;EAAnBH,SAAAA,EAAAA,MAAAA;EASDN,KAAAA,EAAAA,MAAAA;EACDO,WAAAA,EAAAA,MAAAA;EACGJ,eAAAA,EAAAA,MAAAA;EACwBM,IAAAA,EAAAA,MAAAA;EAApBD,IAAAA,EAAAA,MAAAA;EAC4BC,aAAAA,EAAAA,MAAAA,EAAAA;EAApBD,cAAAA,EAJdR,qBAIcQ,EAAAA;EACUC,aAAAA,EAJzBF,qBAIyBE;EAAnBH,gBAAAA,EAHHH,wBAGGG;EACwCG,UAAAA,UAAAA,EAHvCD,mBAGuCC,CAHnBA,WAGmBA,CAAAA;EAArBX,UAAAA,kBAAAA,EAFVU,mBAEUV,CAFUW,WAEVX,CAAAA;EAAoCM,WAAAA,CAAAA,MAAAA,CAAAA,EADvDE,kBACuDF,CADpCK,WACoCL,CAAAA;EACzCA,SAAAA,qBAAAA,CAAAA,MAAAA,CAAAA,EADKN,oBACLM,CAD0BK,WAC1BL,CAAAA,CAAAA,EADyCA,sBACzCA;EACOK,iBAAAA,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EADPL,sBACOK;EAArBX,WAAAA,CAAAA,MAAAA,CAAAA,EAAAA,oBAAAA,CAAqBW,WAArBX,CAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACqBW,WAAAA,CAAAA,MAAAA,CAAAA,EAArBX,oBAAqBW,CAAAA,WAAAA,CAAAA,CAAAA,EAAeL,sBAAfK;EAArBX,eAAAA,CAAAA,MAAAA,EACGQ,kBADHR,CACsBW,WADtBX,CAAAA,EAAAA,MAAAA,EAC4CM,sBAD5CN,CAAAA,EAAAA,IAAAA;EAAoCM,IAAAA,QAAAA,CAAAA,CAAAA,EAEzCH,kBAFyCG;EACdK,QAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAnBH,YAAAA,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAGMV,cAHNU;EAAyCF;;;;;;EAWuBQ,KAAAA,CAAAA,MAAAA,EAAAA,MAAAA,EAAAA,OAAAA,EAAAA,IAAAA,CAAAA,mBAAAA,CAAAA,CAAAA,EAD7BF,OAC6BE,CAAAA,MAAAA,CAAAA;EAC9DjB,eAAAA,CAAAA,KAAAA,EADHD,sBACGC,EAAAA,OAAAA,CAAAA,EAD+BF,4BAC/BE,CAAAA,EAD8DiB,cAC9DjB,CAAAA,MAAAA,CAAAA;EAAoCF,eAAAA,CAAAA,QAAAA,EAApCE,WAAoCF,EAAAA,EAAAA,OAAAA,CAAAA,EAAAA,MAAAA,EAAAA,GAAAA,4BAAAA,EAAAA,UAAAA,CAAAA,EAA2CF,SAA3CE,CAAAA,EAAuDiB,OAAvDjB,CAA+DE,WAA/DF,CAAAA;EAA2CF;;;;;;EAQzDE,UAAAA,eAAAA,CAAAA,CAAAA,EADnBY,cACmBZ,CADJgB,WACIhB,CAAAA;EAA+BiB,MAAAA,CAAAA,KAAAA,EAAjEhB,sBAAiEgB,EAAAA,OAAAA,CAAAA,EAA/BjB,4BAA+BiB,CAAAA,EAAAA,OAAAA,CAAAA,MAAAA,CAAAA"}
+{"version":3,"file":"llms.d.ts","names":[],"sources":["../src/llms.ts"],"mappings":";;;;;;;;;;cA+BM,mBAAA,sBAAyC,2BAAA,CAC7C,cAAA,EACA,WAAA;EAEM,cAAA,CACJ,KAAA,EAAO,cAAA,EACP,WAAA,EAAa,mBAAA,GACZ,OAAA,CAAQ,aAAA;AAAA;;;;uBA+BS,aAAA,sBACZ,GAAA,CAAI,4BAAA,aACD,kBAAA,CAAmB,WAAA;EAAA,OAGvB,OAAA,CAAA;EAAA,IAIH,UAAA,CAAA;IAAA,CAAiB,GAAA;EAAA;EAMrB,cAAA,GAAiB,kBAAA,CAAmB,WAAA;EAEpC,eAAA;EAEA,SAAA;EAEA,KAAA;EAEA,WAAA;EAEA,eAAA;EAEA,IAAA;EAEA,IAAA;EAEA,aAAA;EAEA,cAAA,EAAgB,qBAAA;EAEhB,aAAA,EAAe,qBAAA;EAEf,gBAAA,EAAkB,wBAAA;EAAA,UAER,UAAA,EAAY,mBAAA,CAAoB,WAAA;EAAA,UAEhC,kBAAA,EAAoB,mBAAA,CAAoB,WAAA;EAElD,WAAA,CAAY,MAAA,GAAS,kBAAA,CAAmB,WAAA;EAAA,SAY/B,qBAAA,CACP,MAAA,GAAS,oBAAA,CAAqB,WAAA,IAC7B,sBAAA;EAEH,iBAAA,CAAkB,MAAA,WAAiB,sBAAA;EAInC,WAAA,CAAY,MAAA,GAAS,oBAAA,CAAqB,WAAA;EAI1C,WAAA,CACE,MAAA,GAAS,oBAAA,CAAqB,WAAA,IAC7B,sBAAA;EASH,eAAA,CACE,MAAA,EAAQ,kBAAA,CAAmB,WAAA,GAC3B,MAAA,EAAQ,sBAAA;EAAA,IAiBN,QAAA,CAAA,GAAY,kBAAA;EAKhB,QAAA,CAAA;EAIA,YAAA,CAAa,MAAA,WAAiB,cAAA;EArEZ;;;;;;EA+EZ,KAAA,CACJ,MAAA,UACA,OAAA,8BACC,OAAA;EAUI,eAAA,CACL,KAAA,EAAO,sBAAA,EACP,OAAA,GAAU,4BAAA,GACT,cAAA;EAyDG,eAAA,CACJ,QAAA,EAAU,WAAA,IACV,OAAA,cAAqB,4BAAA,EACrB,UAAA,GAAa,SAAA,GACZ,OAAA,CAAQ,WAAA;EAxIR;;;;;;EAAA,UAyJO,eAAA,CAAA,GAAmB,cAAA,CAAe,WAAA;EAStC,MAAA,CACJ,KAAA,EAAO,sBAAA,EACP,OAAA,GAAU,4BAAA,GACT,OAAA;AAAA"}

package/dist/llms.js

@@ -12,12 +12,10 @@ import { BaseLLM, LLM } from "@langchain/core/language_models/llms";
 //#region src/llms.ts
 var GoogleLLMConnection = class extends AbstractGoogleLLMConnection {
 	async formatContents(input, _parameters) {
-		const parts = await this.api.messageContentToParts(input);
-		const contents = [{
+		return [{
 			role: "user",
-			parts
+			parts: await this.api.messageContentToParts(input)
 		}];
-		return contents;
 	}
 };
 var ProxyChatGoogle = class extends ChatGoogleBase {
@@ -99,8 +97,7 @@ var GoogleBaseLLM = class extends LLM {
 	async _call(prompt, options) {
 		const parameters = copyAIModelParams(this, options);
 		const result = await this.connection.request(prompt, parameters, options);
-		const ret = this.connection.api.responseToString(result);
-		return ret;
+		return this.connection.api.responseToString(result);
 	}
 	async *_streamIterator(input, options) {
 		const prompt = BaseLLM._convertInputToPromptValue(input);
@@ -130,8 +127,7 @@ var GoogleBaseLLM = class extends LLM {
 	async predictMessages(messages, options, _callbacks) {
 		const { content } = messages[0];
 		const result = await this.connection.request(content, {}, options);
-		const ret = this.connection.api.responseToBaseMessage(result);
-		return ret;
+		return this.connection.api.responseToBaseMessage(result);
 	}
 	/**
 	* Internal implementation detail to allow Google LLMs to support

package/dist/llms.js.map

@@ -1 +1 @@
-{"version":3,"file":"llms.js","names":["input: MessageContent","_parameters: GoogleAIModelParams","contents: GeminiContent[]","fields: ProxyChatInput<AuthOptions>","fields?: GoogleBaseLLMInput<AuthOptions>","apiKey: string","fields?: GoogleAIBaseLLMInput<AuthOptions>","fields: GoogleBaseLLMInput<AuthOptions>","client: GoogleAbstractedClient","prompt: string","options: this[\"ParsedCallOptions\"]","input: BaseLanguageModelInput","options?: BaseLanguageModelCallOptions","messages: BaseMessage[]","options?: string[] | BaseLanguageModelCallOptions","_callbacks?: Callbacks"],"sources":["../src/llms.ts"],"sourcesContent":["import { CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { BaseLLM, LLM } from \"@langchain/core/language_models/llms\";\nimport {\n  type BaseLanguageModelCallOptions,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { GenerationChunk } from \"@langchain/core/outputs\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\n\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport {\n  GoogleAIBaseLLMInput,\n  GoogleAIModelParams,\n  GoogleAISafetySetting,\n  GooglePlatformType,\n  GeminiContent,\n  GoogleAIResponseMimeType,\n} from \"./types.js\";\nimport {\n  copyAIModelParams,\n  copyAndValidateModelParamsInto,\n} from \"./utils/common.js\";\nimport { DefaultGeminiSafetyHandler } from \"./utils/gemini.js\";\nimport { ApiKeyGoogleAuth, GoogleAbstractedClient } from \"./auth.js\";\nimport { ensureParams } from \"./utils/failed_handler.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\n\nexport { GoogleBaseLLMInput };\n\nclass GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<\n  MessageContent,\n  AuthOptions\n> {\n  async formatContents(\n    input: MessageContent,\n    _parameters: GoogleAIModelParams\n  ): Promise<GeminiContent[]> {\n    const parts = await this.api.messageContentToParts!(input);\n    const contents: GeminiContent[] = [\n      {\n        role: \"user\", // Required by Vertex AI\n        parts,\n      },\n    ];\n    return contents;\n  }\n}\n\ntype ProxyChatInput<AuthOptions> = GoogleAIBaseLLMInput<AuthOptions> & {\n  connection: GoogleLLMConnection<AuthOptions>;\n};\n\nclass ProxyChatGoogle<AuthOptions> extends ChatGoogleBase<AuthOptions> {\n  constructor(fields: ProxyChatInput<AuthOptions>) {\n    super(fields);\n  }\n\n  buildAbstractedClient(\n    fields: ProxyChatInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    return fields.connection.client;\n  }\n}\n\n/**\n * Integration with an LLM.\n */\nexport abstract class GoogleBaseLLM<AuthOptions>\n  extends LLM<BaseLanguageModelCallOptions>\n  implements GoogleBaseLLMInput<AuthOptions>\n{\n  // Used for tracing, replace with the same name as your class\n  static lc_name() {\n    return \"GoogleLLM\";\n  }\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      authOptions: \"GOOGLE_AUTH_OPTIONS\",\n    };\n  }\n\n  originalFields?: GoogleBaseLLMInput<AuthOptions>;\n\n  lc_serializable = true;\n\n  modelName = \"gemini-pro\";\n\n  model = \"gemini-pro\";\n\n  temperature = 0.7;\n\n  maxOutputTokens = 1024;\n\n  topP = 0.8;\n\n  topK = 40;\n\n  stopSequences: string[] = [];\n\n  safetySettings: GoogleAISafetySetting[] = [];\n\n  safetyHandler: GoogleAISafetyHandler;\n\n  responseMimeType: GoogleAIResponseMimeType = \"text/plain\";\n\n  protected connection: GoogleLLMConnection<AuthOptions>;\n\n  protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n\n  constructor(fields?: GoogleBaseLLMInput<AuthOptions>) {\n    super(ensureParams(fields));\n    this.originalFields = fields;\n\n    copyAndValidateModelParamsInto(fields, this);\n    this.safetyHandler =\n      fields?.safetyHandler ?? new DefaultGeminiSafetyHandler();\n\n    const client = this.buildClient(fields);\n    this.buildConnection(fields ?? {}, client);\n  }\n\n  abstract buildAbstractedClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  buildConnection(\n    fields: GoogleBaseLLMInput<AuthOptions>,\n    client: GoogleAbstractedClient\n  ) {\n    this.connection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      false\n    );\n\n    this.streamedConnection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      true\n    );\n  }\n\n  get platform(): GooglePlatformType {\n    return this.connection.platform;\n  }\n\n  // Replace\n  _llmType() {\n    return \"googlellm\";\n  }\n\n  formatPrompt(prompt: string): MessageContent {\n    return prompt;\n  }\n\n  /**\n   * For some given input string and options, return a string output.\n   *\n   * Despite the fact that `invoke` is overridden below, we still need this\n   * in order to handle public APi calls to `generate()`.\n   */\n  async _call(\n    prompt: string,\n    options: this[\"ParsedCallOptions\"]\n  ): Promise<string> {\n    const parameters = copyAIModelParams(this, options);\n    const result = await this.connection.request(prompt, parameters, options);\n    const ret = this.connection.api.responseToString(result);\n    return ret;\n  }\n\n  // Normally, you should not override this method and instead should override\n  // _streamResponseChunks. We are doing so here to allow for multimodal inputs into\n  // the LLM.\n  async *_streamIterator(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): AsyncGenerator<string> {\n    // TODO: Refactor callback setup and teardown code into core\n    const prompt = BaseLLM._convertInputToPromptValue(input);\n    const [runnableConfig, callOptions] =\n      this._separateRunnableConfigFromCallOptions(options);\n    const callbackManager_ = await CallbackManager.configure(\n      runnableConfig.callbacks,\n      this.callbacks,\n      runnableConfig.tags,\n      this.tags,\n      runnableConfig.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const extra = {\n      options: callOptions,\n      invocation_params: this?.invocationParams(callOptions),\n      batch_size: 1,\n    };\n    const runManagers = await callbackManager_?.handleLLMStart(\n      this.toJSON(),\n      [prompt.toString()],\n      undefined,\n      undefined,\n      extra,\n      undefined,\n      undefined,\n      runnableConfig.runName\n    );\n    let generation = new GenerationChunk({\n      text: \"\",\n    });\n    const proxyChat = this.createProxyChat();\n    try {\n      for await (const chunk of proxyChat._streamIterator(input, options)) {\n        const stringValue = this.connection.api.chunkToString(chunk);\n        const generationChunk = new GenerationChunk({\n          text: stringValue,\n        });\n        generation = generation.concat(generationChunk);\n        yield stringValue;\n      }\n    } catch (err) {\n      await Promise.all(\n        (runManagers ?? []).map((runManager) => runManager?.handleLLMError(err))\n      );\n      throw err;\n    }\n    await Promise.all(\n      (runManagers ?? []).map((runManager) =>\n        runManager?.handleLLMEnd({\n          generations: [[generation]],\n        })\n      )\n    );\n  }\n\n  async predictMessages(\n    messages: BaseMessage[],\n    options?: string[] | BaseLanguageModelCallOptions,\n    _callbacks?: Callbacks\n  ): Promise<BaseMessage> {\n    const { content } = messages[0];\n    const result = await this.connection.request(\n      content,\n      {},\n      options as BaseLanguageModelCallOptions\n    );\n    const ret = this.connection.api.responseToBaseMessage(result);\n    return ret;\n  }\n\n  /**\n   * Internal implementation detail to allow Google LLMs to support\n   * multimodal input by delegating to the chat model implementation.\n   *\n   * TODO: Replace with something less hacky.\n   */\n  protected createProxyChat(): ChatGoogleBase<AuthOptions> {\n    return new ProxyChatGoogle<AuthOptions>({\n      ...this.originalFields,\n      connection: this.connection,\n    });\n  }\n\n  // TODO: Remove the need to override this - we are doing it to\n  // allow the LLM to handle multimodal types of input.\n  async invoke(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): Promise<string> {\n    const stream = await this._streamIterator(input, options);\n    let generatedOutput = \"\";\n    for await (const chunk of stream) {\n      generatedOutput += chunk;\n    }\n    return generatedOutput;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA+BA,IAAM,sBAAN,cAA+C,4BAG7C;CACA,MAAM,eACJA,OACAC,aAC0B;EAC1B,MAAM,QAAQ,MAAM,KAAK,IAAI,sBAAuB,MAAM;EAC1D,MAAMC,WAA4B,CAChC;GACE,MAAM;GACN;EACD,CACF;AACD,SAAO;CACR;AACF;AAMD,IAAM,kBAAN,cAA2C,eAA4B;CACrE,YAAYC,QAAqC;EAC/C,MAAM,OAAO;CACd;CAED,sBACEA,QACwB;AACxB,SAAO,OAAO,WAAW;CAC1B;AACF;;;;AAKD,IAAsB,gBAAtB,cACU,IAEV;CAEE,OAAO,UAAU;AACf,SAAO;CACR;CAED,IAAI,aAAoD;AACtD,SAAO,EACL,aAAa,sBACd;CACF;CAED;CAEA,kBAAkB;CAElB,YAAY;CAEZ,QAAQ;CAER,cAAc;CAEd,kBAAkB;CAElB,OAAO;CAEP,OAAO;CAEP,gBAA0B,CAAE;CAE5B,iBAA0C,CAAE;CAE5C;CAEA,mBAA6C;CAE7C,AAAU;CAEV,AAAU;CAEV,YAAYC,QAA0C;EACpD,MAAM,aAAa,OAAO,CAAC;EAC3B,KAAK,iBAAiB;EAEtB,+BAA+B,QAAQ,KAAK;EAC5C,KAAK,gBACH,QAAQ,iBAAiB,IAAI;EAE/B,MAAM,SAAS,KAAK,YAAY,OAAO;EACvC,KAAK,gBAAgB,UAAU,CAAE,GAAE,OAAO;CAC3C;CAMD,kBAAkBC,QAAwC;AACxD,SAAO,IAAI,iBAAiB;CAC7B;CAED,YAAYC,QAAgE;AAC1E,SAAO,QAAQ,UAAU,uBAAuB,iBAAiB;CAClE;CAED,YACEA,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAErC,QAAO,KAAK,sBAAsB,OAAO;CAE5C;CAED,gBACEC,QACAC,QACA;EACA,KAAK,aAAa,IAAI,oBACpB;GAAE,GAAG;GAAQ,GAAG;EAAM,GACtB,KAAK,QACL,QACA;EAGF,KAAK,qBAAqB,IAAI,oBAC5B;GAAE,GAAG;GAAQ,GAAG;EAAM,GACtB,KAAK,QACL,QACA;CAEH;CAED,IAAI,WAA+B;AACjC,SAAO,KAAK,WAAW;CACxB;CAGD,WAAW;AACT,SAAO;CACR;CAED,aAAaC,QAAgC;AAC3C,SAAO;CACR;;;;;;;CAQD,MAAM,MACJA,QACAC,SACiB;EACjB,MAAM,aAAa,kBAAkB,MAAM,QAAQ;EACnD,MAAM,SAAS,MAAM,KAAK,WAAW,QAAQ,QAAQ,YAAY,QAAQ;EACzE,MAAM,MAAM,KAAK,WAAW,IAAI,iBAAiB,OAAO;AACxD,SAAO;CACR;CAKD,OAAO,gBACLC,OACAC,SACwB;EAExB,MAAM,SAAS,QAAQ,2BAA2B,MAAM;EACxD,MAAM,CAAC,gBAAgB,YAAY,GACjC,KAAK,uCAAuC,QAAQ;EACtD,MAAM,mBAAmB,MAAM,gBAAgB,UAC7C,eAAe,WACf,KAAK,WACL,eAAe,MACf,KAAK,MACL,eAAe,UACf,KAAK,UACL,EAAE,SAAS,KAAK,QAAS,EAC1B;EACD,MAAM,QAAQ;GACZ,SAAS;GACT,mBAAmB,MAAM,iBAAiB,YAAY;GACtD,YAAY;EACb;EACD,MAAM,cAAc,MAAM,kBAAkB,eAC1C,KAAK,QAAQ,EACb,CAAC,OAAO,UAAU,AAAC,GACnB,QACA,QACA,OACA,QACA,QACA,eAAe,QAChB;EACD,IAAI,aAAa,IAAI,gBAAgB,EACnC,MAAM,GACP;EACD,MAAM,YAAY,KAAK,iBAAiB;AACxC,MAAI;AACF,cAAW,MAAM,SAAS,UAAU,gBAAgB,OAAO,QAAQ,EAAE;IACnE,MAAM,cAAc,KAAK,WAAW,IAAI,cAAc,MAAM;IAC5D,MAAM,kBAAkB,IAAI,gBAAgB,EAC1C,MAAM,YACP;IACD,aAAa,WAAW,OAAO,gBAAgB;IAC/C,MAAM;GACP;EACF,SAAQ,KAAK;GACZ,MAAM,QAAQ,KACX,eAAe,CAAE,GAAE,IAAI,CAAC,eAAe,YAAY,eAAe,IAAI,CAAC,CACzE;AACD,SAAM;EACP;EACD,MAAM,QAAQ,KACX,eAAe,CAAE,GAAE,IAAI,CAAC,eACvB,YAAY,aAAa,EACvB,aAAa,CAAC,CAAC,UAAW,CAAC,EAC5B,EAAC,CACH,CACF;CACF;CAED,MAAM,gBACJC,UACAC,SACAC,YACsB;EACtB,MAAM,EAAE,SAAS,GAAG,SAAS;EAC7B,MAAM,SAAS,MAAM,KAAK,WAAW,QACnC,SACA,CAAE,GACF,QACD;EACD,MAAM,MAAM,KAAK,WAAW,IAAI,sBAAsB,OAAO;AAC7D,SAAO;CACR;;;;;;;CAQD,AAAU,kBAA+C;AACvD,SAAO,IAAI,gBAA6B;GACtC,GAAG,KAAK;GACR,YAAY,KAAK;EAClB;CACF;CAID,MAAM,OACJJ,OACAC,SACiB;EACjB,MAAM,SAAS,MAAM,KAAK,gBAAgB,OAAO,QAAQ;EACzD,IAAI,kBAAkB;AACtB,aAAW,MAAM,SAAS,QACxB,mBAAmB;AAErB,SAAO;CACR;AACF"}
+{"version":3,"file":"llms.js","names":[],"sources":["../src/llms.ts"],"sourcesContent":["import { CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { BaseLLM, LLM } from \"@langchain/core/language_models/llms\";\nimport {\n  type BaseLanguageModelCallOptions,\n  BaseLanguageModelInput,\n} from \"@langchain/core/language_models/base\";\nimport { BaseMessage, MessageContent } from \"@langchain/core/messages\";\nimport { GenerationChunk } from \"@langchain/core/outputs\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\n\nimport { AbstractGoogleLLMConnection } from \"./connection.js\";\nimport {\n  GoogleAIBaseLLMInput,\n  GoogleAIModelParams,\n  GoogleAISafetySetting,\n  GooglePlatformType,\n  GeminiContent,\n  GoogleAIResponseMimeType,\n} from \"./types.js\";\nimport {\n  copyAIModelParams,\n  copyAndValidateModelParamsInto,\n} from \"./utils/common.js\";\nimport { DefaultGeminiSafetyHandler } from \"./utils/gemini.js\";\nimport { ApiKeyGoogleAuth, GoogleAbstractedClient } from \"./auth.js\";\nimport { ensureParams } from \"./utils/failed_handler.js\";\nimport { ChatGoogleBase } from \"./chat_models.js\";\nimport type { GoogleBaseLLMInput, GoogleAISafetyHandler } from \"./types.js\";\n\nexport { GoogleBaseLLMInput };\n\nclass GoogleLLMConnection<AuthOptions> extends AbstractGoogleLLMConnection<\n  MessageContent,\n  AuthOptions\n> {\n  async formatContents(\n    input: MessageContent,\n    _parameters: GoogleAIModelParams\n  ): Promise<GeminiContent[]> {\n    const parts = await this.api.messageContentToParts!(input);\n    const contents: GeminiContent[] = [\n      {\n        role: \"user\", // Required by Vertex AI\n        parts,\n      },\n    ];\n    return contents;\n  }\n}\n\ntype ProxyChatInput<AuthOptions> = GoogleAIBaseLLMInput<AuthOptions> & {\n  connection: GoogleLLMConnection<AuthOptions>;\n};\n\nclass ProxyChatGoogle<AuthOptions> extends ChatGoogleBase<AuthOptions> {\n  constructor(fields: ProxyChatInput<AuthOptions>) {\n    super(fields);\n  }\n\n  buildAbstractedClient(\n    fields: ProxyChatInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    return fields.connection.client;\n  }\n}\n\n/**\n * Integration with an LLM.\n */\nexport abstract class GoogleBaseLLM<AuthOptions>\n  extends LLM<BaseLanguageModelCallOptions>\n  implements GoogleBaseLLMInput<AuthOptions>\n{\n  // Used for tracing, replace with the same name as your class\n  static lc_name() {\n    return \"GoogleLLM\";\n  }\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      authOptions: \"GOOGLE_AUTH_OPTIONS\",\n    };\n  }\n\n  originalFields?: GoogleBaseLLMInput<AuthOptions>;\n\n  lc_serializable = true;\n\n  modelName = \"gemini-pro\";\n\n  model = \"gemini-pro\";\n\n  temperature = 0.7;\n\n  maxOutputTokens = 1024;\n\n  topP = 0.8;\n\n  topK = 40;\n\n  stopSequences: string[] = [];\n\n  safetySettings: GoogleAISafetySetting[] = [];\n\n  safetyHandler: GoogleAISafetyHandler;\n\n  responseMimeType: GoogleAIResponseMimeType = \"text/plain\";\n\n  protected connection: GoogleLLMConnection<AuthOptions>;\n\n  protected streamedConnection: GoogleLLMConnection<AuthOptions>;\n\n  constructor(fields?: GoogleBaseLLMInput<AuthOptions>) {\n    super(ensureParams(fields));\n    this.originalFields = fields;\n\n    copyAndValidateModelParamsInto(fields, this);\n    this.safetyHandler =\n      fields?.safetyHandler ?? new DefaultGeminiSafetyHandler();\n\n    const client = this.buildClient(fields);\n    this.buildConnection(fields ?? {}, client);\n  }\n\n  abstract buildAbstractedClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: GoogleAIBaseLLMInput<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: GoogleAIBaseLLMInput<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  buildConnection(\n    fields: GoogleBaseLLMInput<AuthOptions>,\n    client: GoogleAbstractedClient\n  ) {\n    this.connection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      false\n    );\n\n    this.streamedConnection = new GoogleLLMConnection(\n      { ...fields, ...this },\n      this.caller,\n      client,\n      true\n    );\n  }\n\n  get platform(): GooglePlatformType {\n    return this.connection.platform;\n  }\n\n  // Replace\n  _llmType() {\n    return \"googlellm\";\n  }\n\n  formatPrompt(prompt: string): MessageContent {\n    return prompt;\n  }\n\n  /**\n   * For some given input string and options, return a string output.\n   *\n   * Despite the fact that `invoke` is overridden below, we still need this\n   * in order to handle public APi calls to `generate()`.\n   */\n  async _call(\n    prompt: string,\n    options: this[\"ParsedCallOptions\"]\n  ): Promise<string> {\n    const parameters = copyAIModelParams(this, options);\n    const result = await this.connection.request(prompt, parameters, options);\n    const ret = this.connection.api.responseToString(result);\n    return ret;\n  }\n\n  // Normally, you should not override this method and instead should override\n  // _streamResponseChunks. We are doing so here to allow for multimodal inputs into\n  // the LLM.\n  async *_streamIterator(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): AsyncGenerator<string> {\n    // TODO: Refactor callback setup and teardown code into core\n    const prompt = BaseLLM._convertInputToPromptValue(input);\n    const [runnableConfig, callOptions] =\n      this._separateRunnableConfigFromCallOptions(options);\n    const callbackManager_ = await CallbackManager.configure(\n      runnableConfig.callbacks,\n      this.callbacks,\n      runnableConfig.tags,\n      this.tags,\n      runnableConfig.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const extra = {\n      options: callOptions,\n      invocation_params: this?.invocationParams(callOptions),\n      batch_size: 1,\n    };\n    const runManagers = await callbackManager_?.handleLLMStart(\n      this.toJSON(),\n      [prompt.toString()],\n      undefined,\n      undefined,\n      extra,\n      undefined,\n      undefined,\n      runnableConfig.runName\n    );\n    let generation = new GenerationChunk({\n      text: \"\",\n    });\n    const proxyChat = this.createProxyChat();\n    try {\n      for await (const chunk of proxyChat._streamIterator(input, options)) {\n        const stringValue = this.connection.api.chunkToString(chunk);\n        const generationChunk = new GenerationChunk({\n          text: stringValue,\n        });\n        generation = generation.concat(generationChunk);\n        yield stringValue;\n      }\n    } catch (err) {\n      await Promise.all(\n        (runManagers ?? []).map((runManager) => runManager?.handleLLMError(err))\n      );\n      throw err;\n    }\n    await Promise.all(\n      (runManagers ?? []).map((runManager) =>\n        runManager?.handleLLMEnd({\n          generations: [[generation]],\n        })\n      )\n    );\n  }\n\n  async predictMessages(\n    messages: BaseMessage[],\n    options?: string[] | BaseLanguageModelCallOptions,\n    _callbacks?: Callbacks\n  ): Promise<BaseMessage> {\n    const { content } = messages[0];\n    const result = await this.connection.request(\n      content,\n      {},\n      options as BaseLanguageModelCallOptions\n    );\n    const ret = this.connection.api.responseToBaseMessage(result);\n    return ret;\n  }\n\n  /**\n   * Internal implementation detail to allow Google LLMs to support\n   * multimodal input by delegating to the chat model implementation.\n   *\n   * TODO: Replace with something less hacky.\n   */\n  protected createProxyChat(): ChatGoogleBase<AuthOptions> {\n    return new ProxyChatGoogle<AuthOptions>({\n      ...this.originalFields,\n      connection: this.connection,\n    });\n  }\n\n  // TODO: Remove the need to override this - we are doing it to\n  // allow the LLM to handle multimodal types of input.\n  async invoke(\n    input: BaseLanguageModelInput,\n    options?: BaseLanguageModelCallOptions\n  ): Promise<string> {\n    const stream = await this._streamIterator(input, options);\n    let generatedOutput = \"\";\n    for await (const chunk of stream) {\n      generatedOutput += chunk;\n    }\n    return generatedOutput;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA+BA,IAAM,sBAAN,cAA+C,4BAG7C;CACA,MAAM,eACJ,OACA,aAC0B;AAQ1B,SANkC,CAChC;GACE,MAAM;GACN,OAJU,MAAM,KAAK,IAAI,sBAAuB,MAAM;GAKvD,CACF;;;AASL,IAAM,kBAAN,cAA2C,eAA4B;CACrE,YAAY,QAAqC;AAC/C,QAAM,OAAO;;CAGf,sBACE,QACwB;AACxB,SAAO,OAAO,WAAW;;;;;;AAO7B,IAAsB,gBAAtB,cACU,IAEV;CAEE,OAAO,UAAU;AACf,SAAO;;CAGT,IAAI,aAAoD;AACtD,SAAO,EACL,aAAa,uBACd;;CAGH;CAEA,kBAAkB;CAElB,YAAY;CAEZ,QAAQ;CAER,cAAc;CAEd,kBAAkB;CAElB,OAAO;CAEP,OAAO;CAEP,gBAA0B,EAAE;CAE5B,iBAA0C,EAAE;CAE5C;CAEA,mBAA6C;CAE7C,AAAU;CAEV,AAAU;CAEV,YAAY,QAA0C;AACpD,QAAM,aAAa,OAAO,CAAC;AAC3B,OAAK,iBAAiB;AAEtB,iCAA+B,QAAQ,KAAK;AAC5C,OAAK,gBACH,QAAQ,iBAAiB,IAAI,4BAA4B;EAE3D,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,OAAK,gBAAgB,UAAU,EAAE,EAAE,OAAO;;CAO5C,kBAAkB,QAAwC;AACxD,SAAO,IAAI,iBAAiB,OAAO;;CAGrC,YAAY,QAAgE;AAC1E,SAAO,QAAQ,UAAU,uBAAuB,iBAAiB;;CAGnE,YACE,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAErC,QAAO,KAAK,sBAAsB,OAAO;;CAI7C,gBACE,QACA,QACA;AACA,OAAK,aAAa,IAAI,oBACpB;GAAE,GAAG;GAAQ,GAAG;GAAM,EACtB,KAAK,QACL,QACA,MACD;AAED,OAAK,qBAAqB,IAAI,oBAC5B;GAAE,GAAG;GAAQ,GAAG;GAAM,EACtB,KAAK,QACL,QACA,KACD;;CAGH,IAAI,WAA+B;AACjC,SAAO,KAAK,WAAW;;CAIzB,WAAW;AACT,SAAO;;CAGT,aAAa,QAAgC;AAC3C,SAAO;;;;;;;;CAST,MAAM,MACJ,QACA,SACiB;EACjB,MAAM,aAAa,kBAAkB,MAAM,QAAQ;EACnD,MAAM,SAAS,MAAM,KAAK,WAAW,QAAQ,QAAQ,YAAY,QAAQ;AAEzE,SADY,KAAK,WAAW,IAAI,iBAAiB,OAAO;;CAO1D,OAAO,gBACL,OACA,SACwB;EAExB,MAAM,SAAS,QAAQ,2BAA2B,MAAM;EACxD,MAAM,CAAC,gBAAgB,eACrB,KAAK,uCAAuC,QAAQ;EACtD,MAAM,mBAAmB,MAAM,gBAAgB,UAC7C,eAAe,WACf,KAAK,WACL,eAAe,MACf,KAAK,MACL,eAAe,UACf,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B;EACD,MAAM,QAAQ;GACZ,SAAS;GACT,mBAAmB,MAAM,iBAAiB,YAAY;GACtD,YAAY;GACb;EACD,MAAM,cAAc,MAAM,kBAAkB,eAC1C,KAAK,QAAQ,EACb,CAAC,OAAO,UAAU,CAAC,EACnB,QACA,QACA,OACA,QACA,QACA,eAAe,QAChB;EACD,IAAI,aAAa,IAAI,gBAAgB,EACnC,MAAM,IACP,CAAC;EACF,MAAM,YAAY,KAAK,iBAAiB;AACxC,MAAI;AACF,cAAW,MAAM,SAAS,UAAU,gBAAgB,OAAO,QAAQ,EAAE;IACnE,MAAM,cAAc,KAAK,WAAW,IAAI,cAAc,MAAM;IAC5D,MAAM,kBAAkB,IAAI,gBAAgB,EAC1C,MAAM,aACP,CAAC;AACF,iBAAa,WAAW,OAAO,gBAAgB;AAC/C,UAAM;;WAED,KAAK;AACZ,SAAM,QAAQ,KACX,eAAe,EAAE,EAAE,KAAK,eAAe,YAAY,eAAe,IAAI,CAAC,CACzE;AACD,SAAM;;AAER,QAAM,QAAQ,KACX,eAAe,EAAE,EAAE,KAAK,eACvB,YAAY,aAAa,EACvB,aAAa,CAAC,CAAC,WAAW,CAAC,EAC5B,CAAC,CACH,CACF;;CAGH,MAAM,gBACJ,UACA,SACA,YACsB;EACtB,MAAM,EAAE,YAAY,SAAS;EAC7B,MAAM,SAAS,MAAM,KAAK,WAAW,QACnC,SACA,EAAE,EACF,QACD;AAED,SADY,KAAK,WAAW,IAAI,sBAAsB,OAAO;;;;;;;;CAU/D,AAAU,kBAA+C;AACvD,SAAO,IAAI,gBAA6B;GACtC,GAAG,KAAK;GACR,YAAY,KAAK;GAClB,CAAC;;CAKJ,MAAM,OACJ,OACA,SACiB;EACjB,MAAM,SAAS,MAAM,KAAK,gBAAgB,OAAO,QAAQ;EACzD,IAAI,kBAAkB;AACtB,aAAW,MAAM,SAAS,OACxB,oBAAmB;AAErB,SAAO"}

package/dist/output_parsers.cjs

@@ -1,8 +1,7 @@
-const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
-const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
+let _langchain_core_output_parsers = require("@langchain/core/output_parsers");
 
 //#region src/output_parsers.ts
-var BaseGoogleSearchOutputParser = class extends __langchain_core_output_parsers.BaseLLMOutputParser {
+var BaseGoogleSearchOutputParser = class extends _langchain_core_output_parsers.BaseLLMOutputParser {
 	lc_namespace = ["google_common", "output_parsers"];
 	generationToGroundingInfo(generation) {
 		if ("message" in generation) {
@@ -14,14 +13,12 @@ var BaseGoogleSearchOutputParser = class extends __langchain_core_output_parsers
 				supports
 			};
 		}
-		return void 0;
 	}
 	generationsToGroundingInfo(generations) {
 		for (const generation of generations) {
 			const info = this.generationToGroundingInfo(generation);
 			if (info !== void 0) return info;
 		}
-		return void 0;
 	}
 	generationToString(generation) {
 		if ("message" in generation) {
@@ -43,9 +40,7 @@ var BaseGoogleSearchOutputParser = class extends __langchain_core_output_parsers
 		const textBefore = text.substring(0, start);
 		const textSegment = text.substring(start, end);
 		const textAfter = text.substring(end);
-		const textPrefix = this.segmentPrefix(grounding, support, index) ?? "";
-		const textSuffix = this.segmentSuffix(grounding, support, index) ?? "";
-		return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;
+		return `${textBefore}${this.segmentPrefix(grounding, support, index) ?? ""}${textSegment}${this.segmentSuffix(grounding, support, index) ?? ""}${textAfter}`;
 	}
 	annotateTextSegments(text, grounding) {
 		let ret = text;
@@ -69,8 +64,7 @@ var BaseGoogleSearchOutputParser = class extends __langchain_core_output_parsers
 	annotateText(text, grounding) {
 		const prefix = this.textPrefix(text, grounding) ?? "";
 		const suffix = this.textSuffix(text, grounding) ?? "";
-		const body = this.annotateTextSegments(text, grounding);
-		return `${prefix}${body}${suffix}`;
+		return `${prefix}${this.annotateTextSegments(text, grounding)}${suffix}`;
 	}
 	async parseResult(generations, _callbacks) {
 		const text = this.generationsToString(generations);
@@ -80,12 +74,9 @@ var BaseGoogleSearchOutputParser = class extends __langchain_core_output_parsers
 	}
 };
 var SimpleGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser {
-	segmentPrefix(_grounding, _support, _index) {
-		return void 0;
-	}
+	segmentPrefix(_grounding, _support, _index) {}
 	segmentSuffix(_grounding, support, _index) {
-		const indices = support.groundingChunkIndices.map((i) => i + 1);
-		return ` [${indices.join(", ")}]`;
+		return ` [${support.groundingChunkIndices.map((i) => i + 1).join(", ")}]`;
 	}
 	textPrefix(_text, _grounding) {
 		return "Google Says:\n";
@@ -96,22 +87,18 @@ var SimpleGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser
 	}
 	textSuffix(_text, grounding) {
 		let ret = "\n";
-		const chunks = grounding?.metadata?.groundingChunks ?? [];
-		chunks.forEach((chunk, index) => {
+		(grounding?.metadata?.groundingChunks ?? []).forEach((chunk, index) => {
 			ret = `${ret}${this.chunkToString(chunk, index)}\n`;
 		});
 		return ret;
 	}
 };
 var MarkdownGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser {
-	segmentPrefix(_grounding, _support, _index) {
-		return void 0;
-	}
+	segmentPrefix(_grounding, _support, _index) {}
 	chunkLink(grounding, index) {
 		const chunk = grounding.metadata.groundingChunks[index];
 		const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;
-		const num = index + 1;
-		return `[[${num}](${url})]`;
+		return `[[${index + 1}](${url})]`;
 	}
 	segmentSuffix(grounding, support, _index) {
 		let ret = "";
@@ -121,20 +108,16 @@ var MarkdownGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParse
 		});
 		return ret;
 	}
-	textPrefix(_text, _grounding) {
-		return void 0;
-	}
+	textPrefix(_text, _grounding) {}
 	chunkSuffixLink(chunk, index) {
 		const num = index + 1;
 		const info = chunk.retrievedContext ?? chunk.web;
 		const url = info.uri;
-		const site = info.title;
-		return `${num}. [${site}](${url})`;
+		return `${num}. [${info.title}](${url})`;
 	}
 	textSuffix(_text, grounding) {
 		let ret = "\n**Search Sources**\n";
-		const chunks = grounding.metadata.groundingChunks;
-		chunks.forEach((chunk, index) => {
+		grounding.metadata.groundingChunks.forEach((chunk, index) => {
 			ret = `${ret}${this.chunkSuffixLink(chunk, index)}\n`;
 		});
 		const search = this.searchSuggestion(grounding);

package/dist/output_parsers.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"output_parsers.cjs","names":["BaseLLMOutputParser","generation: Generation | ChatGeneration","generations: Generations","content: MessageContent","text: string","grounding: GroundingInfo","support: GeminiGroundingSupport","index: number","_callbacks?: Callbacks","_grounding: GroundingInfo","_support: GeminiGroundingSupport","_index: number","indices: number[]","_text: string","chunk: GeminiGroundingChunk","chunks: GeminiGroundingChunk[]"],"sources":["../src/output_parsers.ts"],"sourcesContent":["import { BaseLLMOutputParser } from \"@langchain/core/output_parsers\";\nimport { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { ChatGeneration, Generation } from \"@langchain/core/outputs\";\nimport { MessageContent } from \"@langchain/core/messages\";\nimport {\n  GeminiGroundingChunk,\n  GeminiGroundingMetadata,\n  GeminiGroundingSupport,\n} from \"./types.js\";\n\ntype Generations = Generation[] | ChatGeneration[];\n\ntype GroundingInfo = {\n  metadata: GeminiGroundingMetadata;\n  supports: GeminiGroundingSupport[];\n};\n\nexport abstract class BaseGoogleSearchOutputParser extends BaseLLMOutputParser<string> {\n  lc_namespace: string[] = [\"google_common\", \"output_parsers\"];\n\n  protected generationToGroundingInfo(\n    generation: Generation | ChatGeneration\n  ): GroundingInfo | undefined {\n    if (\"message\" in generation) {\n      const responseMetadata = generation?.message?.response_metadata;\n      const metadata = responseMetadata?.groundingMetadata;\n      const supports =\n        responseMetadata?.groundingSupport ?? metadata?.groundingSupports ?? [];\n      if (metadata) {\n        return {\n          metadata,\n          supports,\n        };\n      }\n    }\n    return undefined;\n  }\n\n  protected generationsToGroundingInfo(\n    generations: Generations\n  ): GroundingInfo | undefined {\n    for (const generation of generations) {\n      const info = this.generationToGroundingInfo(generation);\n      if (info !== undefined) {\n        return info;\n      }\n    }\n    return undefined;\n  }\n\n  protected generationToString(\n    generation: Generation | ChatGeneration\n  ): string {\n    if (\"message\" in generation) {\n      const content: MessageContent = generation?.message?.content;\n      if (typeof content === \"string\") {\n        return content;\n      } else {\n        return content\n          .map((c) => {\n            if (c?.type === \"text\") {\n              return c?.text ?? \"\";\n            } else {\n              return \"\";\n            }\n          })\n          .reduce(\n            (previousValue, currentValue) => `${previousValue}${currentValue}`\n          );\n      }\n    }\n    return generation.text;\n  }\n\n  protected generationsToString(generations: Generations): string {\n    return generations\n      .map((generation) => this.generationToString(generation))\n      .reduce(\n        (previousValue, currentValue) => `${previousValue}${currentValue}`\n      );\n  }\n\n  protected abstract segmentPrefix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string | undefined;\n\n  protected abstract segmentSuffix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string | undefined;\n\n  protected annotateSegment(\n    text: string,\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string {\n    const start = support.segment.startIndex ?? 0;\n    const end = support.segment.endIndex;\n\n    const textBefore = text.substring(0, start);\n    const textSegment = text.substring(start, end);\n    const textAfter = text.substring(end);\n\n    const textPrefix = this.segmentPrefix(grounding, support, index) ?? \"\";\n    const textSuffix = this.segmentSuffix(grounding, support, index) ?? \"\";\n\n    return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;\n  }\n\n  protected annotateTextSegments(\n    text: string,\n    grounding: GroundingInfo\n  ): string {\n    // Go through each support info in reverse, since the segment info\n    // is sorted, and we won't need to adjust string indexes this way.\n    let ret = text;\n    for (let co = grounding.supports.length - 1; co >= 0; co -= 1) {\n      const support = grounding.supports[co];\n      ret = this.annotateSegment(ret, grounding, support, co);\n    }\n    return ret;\n  }\n\n  protected abstract textPrefix(\n    text: string,\n    grounding: GroundingInfo\n  ): string | undefined;\n\n  protected abstract textSuffix(\n    text: string,\n    grounding: GroundingInfo\n  ): string | undefined;\n\n  /**\n   * Google requires us to\n   * \"Display the Search Suggestion exactly as provided without any modifications\"\n   * So this will typically be called from the textSuffix() method to get\n   * a string that renders HTML.\n   * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions\n   * @param grounding\n   */\n  protected searchSuggestion(grounding: GroundingInfo): string {\n    return grounding?.metadata?.searchEntryPoint?.renderedContent ?? \"\";\n  }\n\n  protected annotateText(text: string, grounding: GroundingInfo): string {\n    const prefix = this.textPrefix(text, grounding) ?? \"\";\n    const suffix = this.textSuffix(text, grounding) ?? \"\";\n    const body = this.annotateTextSegments(text, grounding);\n    return `${prefix}${body}${suffix}`;\n  }\n\n  async parseResult(\n    generations: Generations,\n    _callbacks?: Callbacks\n  ): Promise<string> {\n    const text = this.generationsToString(generations);\n\n    const grounding = this.generationsToGroundingInfo(generations);\n    if (!grounding) {\n      return text;\n    }\n\n    return this.annotateText(text, grounding);\n  }\n}\n\nexport class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n  protected segmentPrefix(\n    _grounding: GroundingInfo,\n    _support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected segmentSuffix(\n    _grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    const indices: number[] = support.groundingChunkIndices.map((i) => i + 1);\n    return ` [${indices.join(\", \")}]`;\n  }\n\n  protected textPrefix(_text: string, _grounding: GroundingInfo): string {\n    return \"Google Says:\\n\";\n  }\n\n  protected chunkToString(chunk: GeminiGroundingChunk, index: number): string {\n    const info = chunk.retrievedContext ?? chunk.web;\n    return `${index + 1}. ${info.title} - ${info.uri}`;\n  }\n\n  protected textSuffix(_text: string, grounding: GroundingInfo): string {\n    let ret = \"\\n\";\n    const chunks: GeminiGroundingChunk[] =\n      grounding?.metadata?.groundingChunks ?? [];\n    chunks.forEach((chunk, index) => {\n      ret = `${ret}${this.chunkToString(chunk, index)}\\n`;\n    });\n    return ret;\n  }\n}\n\nexport class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n  protected segmentPrefix(\n    _grounding: GroundingInfo,\n    _support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected chunkLink(grounding: GroundingInfo, index: number): string {\n    const chunk = grounding.metadata.groundingChunks[index];\n    const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;\n    const num = index + 1;\n    return `[[${num}](${url})]`;\n  }\n\n  protected segmentSuffix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    let ret = \"\";\n    support.groundingChunkIndices.forEach((chunkIndex) => {\n      const link = this.chunkLink(grounding, chunkIndex);\n      ret = `${ret}${link}`;\n    });\n    return ret;\n  }\n\n  protected textPrefix(\n    _text: string,\n    _grounding: GroundingInfo\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected chunkSuffixLink(\n    chunk: GeminiGroundingChunk,\n    index: number\n  ): string {\n    const num = index + 1;\n    const info = chunk.retrievedContext ?? chunk.web;\n    const url = info.uri;\n    const site = info.title;\n    return `${num}. [${site}](${url})`;\n  }\n\n  protected textSuffix(\n    _text: string,\n    grounding: GroundingInfo\n  ): string | undefined {\n    let ret = \"\\n**Search Sources**\\n\";\n    const chunks: GeminiGroundingChunk[] = grounding.metadata.groundingChunks;\n    chunks.forEach((chunk, index) => {\n      ret = `${ret}${this.chunkSuffixLink(chunk, index)}\\n`;\n    });\n\n    const search = this.searchSuggestion(grounding);\n    ret = `${ret}\\n${search}`;\n\n    return ret;\n  }\n}\n"],"mappings":";;;;AAiBA,IAAsB,+BAAtB,cAA2DA,oDAA4B;CACrF,eAAyB,CAAC,iBAAiB,gBAAiB;CAE5D,AAAU,0BACRC,YAC2B;AAC3B,MAAI,aAAa,YAAY;GAC3B,MAAM,mBAAmB,YAAY,SAAS;GAC9C,MAAM,WAAW,kBAAkB;GACnC,MAAM,WACJ,kBAAkB,oBAAoB,UAAU,qBAAqB,CAAE;AACzE,OAAI,SACF,QAAO;IACL;IACA;GACD;EAEJ;AACD,SAAO;CACR;CAED,AAAU,2BACRC,aAC2B;AAC3B,OAAK,MAAM,cAAc,aAAa;GACpC,MAAM,OAAO,KAAK,0BAA0B,WAAW;AACvD,OAAI,SAAS,OACX,QAAO;EAEV;AACD,SAAO;CACR;CAED,AAAU,mBACRD,YACQ;AACR,MAAI,aAAa,YAAY;GAC3B,MAAME,UAA0B,YAAY,SAAS;AACrD,OAAI,OAAO,YAAY,SACrB,QAAO;OAEP,QAAO,QACJ,IAAI,CAAC,MAAM;AACV,QAAI,GAAG,SAAS,OACd,QAAO,GAAG,QAAQ;QAElB,QAAO;GAEV,EAAC,CACD,OACC,CAAC,eAAe,iBAAiB,GAAG,gBAAgB,cAAc,CACnE;EAEN;AACD,SAAO,WAAW;CACnB;CAED,AAAU,oBAAoBD,aAAkC;AAC9D,SAAO,YACJ,IAAI,CAAC,eAAe,KAAK,mBAAmB,WAAW,CAAC,CACxD,OACC,CAAC,eAAe,iBAAiB,GAAG,gBAAgB,cAAc,CACnE;CACJ;CAcD,AAAU,gBACRE,MACAC,WACAC,SACAC,OACQ;EACR,MAAM,QAAQ,QAAQ,QAAQ,cAAc;EAC5C,MAAM,MAAM,QAAQ,QAAQ;EAE5B,MAAM,aAAa,KAAK,UAAU,GAAG,MAAM;EAC3C,MAAM,cAAc,KAAK,UAAU,OAAO,IAAI;EAC9C,MAAM,YAAY,KAAK,UAAU,IAAI;EAErC,MAAM,aAAa,KAAK,cAAc,WAAW,SAAS,MAAM,IAAI;EACpE,MAAM,aAAa,KAAK,cAAc,WAAW,SAAS,MAAM,IAAI;AAEpE,SAAO,GAAG,aAAa,aAAa,cAAc,aAAa,WAAW;CAC3E;CAED,AAAU,qBACRH,MACAC,WACQ;EAGR,IAAI,MAAM;AACV,OAAK,IAAI,KAAK,UAAU,SAAS,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG;GAC7D,MAAM,UAAU,UAAU,SAAS;GACnC,MAAM,KAAK,gBAAgB,KAAK,WAAW,SAAS,GAAG;EACxD;AACD,SAAO;CACR;;;;;;;;;CAoBD,AAAU,iBAAiBA,WAAkC;AAC3D,SAAO,WAAW,UAAU,kBAAkB,mBAAmB;CAClE;CAED,AAAU,aAAaD,MAAcC,WAAkC;EACrE,MAAM,SAAS,KAAK,WAAW,MAAM,UAAU,IAAI;EACnD,MAAM,SAAS,KAAK,WAAW,MAAM,UAAU,IAAI;EACnD,MAAM,OAAO,KAAK,qBAAqB,MAAM,UAAU;AACvD,SAAO,GAAG,SAAS,OAAO,QAAQ;CACnC;CAED,MAAM,YACJH,aACAM,YACiB;EACjB,MAAM,OAAO,KAAK,oBAAoB,YAAY;EAElD,MAAM,YAAY,KAAK,2BAA2B,YAAY;AAC9D,MAAI,CAAC,UACH,QAAO;AAGT,SAAO,KAAK,aAAa,MAAM,UAAU;CAC1C;AACF;AAED,IAAa,iCAAb,cAAoD,6BAA6B;CAC/E,AAAU,cACRC,YACAC,UACAC,QACoB;AACpB,SAAO;CACR;CAED,AAAU,cACRF,YACAH,SACAK,QACoB;EACpB,MAAMC,UAAoB,QAAQ,sBAAsB,IAAI,CAAC,MAAM,IAAI,EAAE;AACzE,SAAO,CAAC,EAAE,EAAE,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC;CAClC;CAED,AAAU,WAAWC,OAAeJ,YAAmC;AACrE,SAAO;CACR;CAED,AAAU,cAAcK,OAA6BP,OAAuB;EAC1E,MAAM,OAAO,MAAM,oBAAoB,MAAM;AAC7C,SAAO,GAAG,QAAQ,EAAE,EAAE,EAAE,KAAK,MAAM,GAAG,EAAE,KAAK,KAAK;CACnD;CAED,AAAU,WAAWM,OAAeR,WAAkC;EACpE,IAAI,MAAM;EACV,MAAMU,SACJ,WAAW,UAAU,mBAAmB,CAAE;EAC5C,OAAO,QAAQ,CAAC,OAAO,UAAU;GAC/B,MAAM,GAAG,MAAM,KAAK,cAAc,OAAO,MAAM,CAAC,EAAE,CAAC;EACpD,EAAC;AACF,SAAO;CACR;AACF;AAED,IAAa,mCAAb,cAAsD,6BAA6B;CACjF,AAAU,cACRN,YACAC,UACAC,QACoB;AACpB,SAAO;CACR;CAED,AAAU,UAAUN,WAA0BE,OAAuB;EACnE,MAAM,QAAQ,UAAU,SAAS,gBAAgB;EACjD,MAAM,MAAM,MAAM,kBAAkB,OAAO,MAAM,KAAK;EACtD,MAAM,MAAM,QAAQ;AACpB,SAAO,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC;CAC5B;CAED,AAAU,cACRF,WACAC,SACAK,QACoB;EACpB,IAAI,MAAM;EACV,QAAQ,sBAAsB,QAAQ,CAAC,eAAe;GACpD,MAAM,OAAO,KAAK,UAAU,WAAW,WAAW;GAClD,MAAM,GAAG,MAAM,MAAM;EACtB,EAAC;AACF,SAAO;CACR;CAED,AAAU,WACRE,OACAJ,YACoB;AACpB,SAAO;CACR;CAED,AAAU,gBACRK,OACAP,OACQ;EACR,MAAM,MAAM,QAAQ;EACpB,MAAM,OAAO,MAAM,oBAAoB,MAAM;EAC7C,MAAM,MAAM,KAAK;EACjB,MAAM,OAAO,KAAK;AAClB,SAAO,GAAG,IAAI,GAAG,EAAE,KAAK,EAAE,EAAE,IAAI,CAAC,CAAC;CACnC;CAED,AAAU,WACRM,OACAR,WACoB;EACpB,IAAI,MAAM;EACV,MAAMU,SAAiC,UAAU,SAAS;EAC1D,OAAO,QAAQ,CAAC,OAAO,UAAU;GAC/B,MAAM,GAAG,MAAM,KAAK,gBAAgB,OAAO,MAAM,CAAC,EAAE,CAAC;EACtD,EAAC;EAEF,MAAM,SAAS,KAAK,iBAAiB,UAAU;EAC/C,MAAM,GAAG,IAAI,EAAE,EAAE,QAAQ;AAEzB,SAAO;CACR;AACF"}
+{"version":3,"file":"output_parsers.cjs","names":["BaseLLMOutputParser"],"sources":["../src/output_parsers.ts"],"sourcesContent":["import { BaseLLMOutputParser } from \"@langchain/core/output_parsers\";\nimport { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { ChatGeneration, Generation } from \"@langchain/core/outputs\";\nimport { MessageContent } from \"@langchain/core/messages\";\nimport {\n  GeminiGroundingChunk,\n  GeminiGroundingMetadata,\n  GeminiGroundingSupport,\n} from \"./types.js\";\n\ntype Generations = Generation[] | ChatGeneration[];\n\ntype GroundingInfo = {\n  metadata: GeminiGroundingMetadata;\n  supports: GeminiGroundingSupport[];\n};\n\nexport abstract class BaseGoogleSearchOutputParser extends BaseLLMOutputParser<string> {\n  lc_namespace: string[] = [\"google_common\", \"output_parsers\"];\n\n  protected generationToGroundingInfo(\n    generation: Generation | ChatGeneration\n  ): GroundingInfo | undefined {\n    if (\"message\" in generation) {\n      const responseMetadata = generation?.message?.response_metadata;\n      const metadata = responseMetadata?.groundingMetadata;\n      const supports =\n        responseMetadata?.groundingSupport ?? metadata?.groundingSupports ?? [];\n      if (metadata) {\n        return {\n          metadata,\n          supports,\n        };\n      }\n    }\n    return undefined;\n  }\n\n  protected generationsToGroundingInfo(\n    generations: Generations\n  ): GroundingInfo | undefined {\n    for (const generation of generations) {\n      const info = this.generationToGroundingInfo(generation);\n      if (info !== undefined) {\n        return info;\n      }\n    }\n    return undefined;\n  }\n\n  protected generationToString(\n    generation: Generation | ChatGeneration\n  ): string {\n    if (\"message\" in generation) {\n      const content: MessageContent = generation?.message?.content;\n      if (typeof content === \"string\") {\n        return content;\n      } else {\n        return content\n          .map((c) => {\n            if (c?.type === \"text\") {\n              return c?.text ?? \"\";\n            } else {\n              return \"\";\n            }\n          })\n          .reduce(\n            (previousValue, currentValue) => `${previousValue}${currentValue}`\n          );\n      }\n    }\n    return generation.text;\n  }\n\n  protected generationsToString(generations: Generations): string {\n    return generations\n      .map((generation) => this.generationToString(generation))\n      .reduce(\n        (previousValue, currentValue) => `${previousValue}${currentValue}`\n      );\n  }\n\n  protected abstract segmentPrefix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string | undefined;\n\n  protected abstract segmentSuffix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string | undefined;\n\n  protected annotateSegment(\n    text: string,\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    index: number\n  ): string {\n    const start = support.segment.startIndex ?? 0;\n    const end = support.segment.endIndex;\n\n    const textBefore = text.substring(0, start);\n    const textSegment = text.substring(start, end);\n    const textAfter = text.substring(end);\n\n    const textPrefix = this.segmentPrefix(grounding, support, index) ?? \"\";\n    const textSuffix = this.segmentSuffix(grounding, support, index) ?? \"\";\n\n    return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;\n  }\n\n  protected annotateTextSegments(\n    text: string,\n    grounding: GroundingInfo\n  ): string {\n    // Go through each support info in reverse, since the segment info\n    // is sorted, and we won't need to adjust string indexes this way.\n    let ret = text;\n    for (let co = grounding.supports.length - 1; co >= 0; co -= 1) {\n      const support = grounding.supports[co];\n      ret = this.annotateSegment(ret, grounding, support, co);\n    }\n    return ret;\n  }\n\n  protected abstract textPrefix(\n    text: string,\n    grounding: GroundingInfo\n  ): string | undefined;\n\n  protected abstract textSuffix(\n    text: string,\n    grounding: GroundingInfo\n  ): string | undefined;\n\n  /**\n   * Google requires us to\n   * \"Display the Search Suggestion exactly as provided without any modifications\"\n   * So this will typically be called from the textSuffix() method to get\n   * a string that renders HTML.\n   * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions\n   * @param grounding\n   */\n  protected searchSuggestion(grounding: GroundingInfo): string {\n    return grounding?.metadata?.searchEntryPoint?.renderedContent ?? \"\";\n  }\n\n  protected annotateText(text: string, grounding: GroundingInfo): string {\n    const prefix = this.textPrefix(text, grounding) ?? \"\";\n    const suffix = this.textSuffix(text, grounding) ?? \"\";\n    const body = this.annotateTextSegments(text, grounding);\n    return `${prefix}${body}${suffix}`;\n  }\n\n  async parseResult(\n    generations: Generations,\n    _callbacks?: Callbacks\n  ): Promise<string> {\n    const text = this.generationsToString(generations);\n\n    const grounding = this.generationsToGroundingInfo(generations);\n    if (!grounding) {\n      return text;\n    }\n\n    return this.annotateText(text, grounding);\n  }\n}\n\nexport class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n  protected segmentPrefix(\n    _grounding: GroundingInfo,\n    _support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected segmentSuffix(\n    _grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    const indices: number[] = support.groundingChunkIndices.map((i) => i + 1);\n    return ` [${indices.join(\", \")}]`;\n  }\n\n  protected textPrefix(_text: string, _grounding: GroundingInfo): string {\n    return \"Google Says:\\n\";\n  }\n\n  protected chunkToString(chunk: GeminiGroundingChunk, index: number): string {\n    const info = chunk.retrievedContext ?? chunk.web;\n    return `${index + 1}. ${info.title} - ${info.uri}`;\n  }\n\n  protected textSuffix(_text: string, grounding: GroundingInfo): string {\n    let ret = \"\\n\";\n    const chunks: GeminiGroundingChunk[] =\n      grounding?.metadata?.groundingChunks ?? [];\n    chunks.forEach((chunk, index) => {\n      ret = `${ret}${this.chunkToString(chunk, index)}\\n`;\n    });\n    return ret;\n  }\n}\n\nexport class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n  protected segmentPrefix(\n    _grounding: GroundingInfo,\n    _support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected chunkLink(grounding: GroundingInfo, index: number): string {\n    const chunk = grounding.metadata.groundingChunks[index];\n    const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;\n    const num = index + 1;\n    return `[[${num}](${url})]`;\n  }\n\n  protected segmentSuffix(\n    grounding: GroundingInfo,\n    support: GeminiGroundingSupport,\n    _index: number\n  ): string | undefined {\n    let ret = \"\";\n    support.groundingChunkIndices.forEach((chunkIndex) => {\n      const link = this.chunkLink(grounding, chunkIndex);\n      ret = `${ret}${link}`;\n    });\n    return ret;\n  }\n\n  protected textPrefix(\n    _text: string,\n    _grounding: GroundingInfo\n  ): string | undefined {\n    return undefined;\n  }\n\n  protected chunkSuffixLink(\n    chunk: GeminiGroundingChunk,\n    index: number\n  ): string {\n    const num = index + 1;\n    const info = chunk.retrievedContext ?? chunk.web;\n    const url = info.uri;\n    const site = info.title;\n    return `${num}. [${site}](${url})`;\n  }\n\n  protected textSuffix(\n    _text: string,\n    grounding: GroundingInfo\n  ): string | undefined {\n    let ret = \"\\n**Search Sources**\\n\";\n    const chunks: GeminiGroundingChunk[] = grounding.metadata.groundingChunks;\n    chunks.forEach((chunk, index) => {\n      ret = `${ret}${this.chunkSuffixLink(chunk, index)}\\n`;\n    });\n\n    const search = this.searchSuggestion(grounding);\n    ret = `${ret}\\n${search}`;\n\n    return ret;\n  }\n}\n"],"mappings":";;;AAiBA,IAAsB,+BAAtB,cAA2DA,mDAA4B;CACrF,eAAyB,CAAC,iBAAiB,iBAAiB;CAE5D,AAAU,0BACR,YAC2B;AAC3B,MAAI,aAAa,YAAY;GAC3B,MAAM,mBAAmB,YAAY,SAAS;GAC9C,MAAM,WAAW,kBAAkB;GACnC,MAAM,WACJ,kBAAkB,oBAAoB,UAAU,qBAAqB,EAAE;AACzE,OAAI,SACF,QAAO;IACL;IACA;IACD;;;CAMP,AAAU,2BACR,aAC2B;AAC3B,OAAK,MAAM,cAAc,aAAa;GACpC,MAAM,OAAO,KAAK,0BAA0B,WAAW;AACvD,OAAI,SAAS,OACX,QAAO;;;CAMb,AAAU,mBACR,YACQ;AACR,MAAI,aAAa,YAAY;GAC3B,MAAM,UAA0B,YAAY,SAAS;AACrD,OAAI,OAAO,YAAY,SACrB,QAAO;OAEP,QAAO,QACJ,KAAK,MAAM;AACV,QAAI,GAAG,SAAS,OACd,QAAO,GAAG,QAAQ;QAElB,QAAO;KAET,CACD,QACE,eAAe,iBAAiB,GAAG,gBAAgB,eACrD;;AAGP,SAAO,WAAW;;CAGpB,AAAU,oBAAoB,aAAkC;AAC9D,SAAO,YACJ,KAAK,eAAe,KAAK,mBAAmB,WAAW,CAAC,CACxD,QACE,eAAe,iBAAiB,GAAG,gBAAgB,eACrD;;CAeL,AAAU,gBACR,MACA,WACA,SACA,OACQ;EACR,MAAM,QAAQ,QAAQ,QAAQ,cAAc;EAC5C,MAAM,MAAM,QAAQ,QAAQ;EAE5B,MAAM,aAAa,KAAK,UAAU,GAAG,MAAM;EAC3C,MAAM,cAAc,KAAK,UAAU,OAAO,IAAI;EAC9C,MAAM,YAAY,KAAK,UAAU,IAAI;AAKrC,SAAO,GAAG,aAHS,KAAK,cAAc,WAAW,SAAS,MAAM,IAAI,KAGhC,cAFjB,KAAK,cAAc,WAAW,SAAS,MAAM,IAAI,KAEL;;CAGjE,AAAU,qBACR,MACA,WACQ;EAGR,IAAI,MAAM;AACV,OAAK,IAAI,KAAK,UAAU,SAAS,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG;GAC7D,MAAM,UAAU,UAAU,SAAS;AACnC,SAAM,KAAK,gBAAgB,KAAK,WAAW,SAAS,GAAG;;AAEzD,SAAO;;;;;;;;;;CAqBT,AAAU,iBAAiB,WAAkC;AAC3D,SAAO,WAAW,UAAU,kBAAkB,mBAAmB;;CAGnE,AAAU,aAAa,MAAc,WAAkC;EACrE,MAAM,SAAS,KAAK,WAAW,MAAM,UAAU,IAAI;EACnD,MAAM,SAAS,KAAK,WAAW,MAAM,UAAU,IAAI;AAEnD,SAAO,GAAG,SADG,KAAK,qBAAqB,MAAM,UAAU,GAC7B;;CAG5B,MAAM,YACJ,aACA,YACiB;EACjB,MAAM,OAAO,KAAK,oBAAoB,YAAY;EAElD,MAAM,YAAY,KAAK,2BAA2B,YAAY;AAC9D,MAAI,CAAC,UACH,QAAO;AAGT,SAAO,KAAK,aAAa,MAAM,UAAU;;;AAI7C,IAAa,iCAAb,cAAoD,6BAA6B;CAC/E,AAAU,cACR,YACA,UACA,QACoB;CAItB,AAAU,cACR,YACA,SACA,QACoB;AAEpB,SAAO,KADmB,QAAQ,sBAAsB,KAAK,MAAM,IAAI,EAAE,CACrD,KAAK,KAAK,CAAC;;CAGjC,AAAU,WAAW,OAAe,YAAmC;AACrE,SAAO;;CAGT,AAAU,cAAc,OAA6B,OAAuB;EAC1E,MAAM,OAAO,MAAM,oBAAoB,MAAM;AAC7C,SAAO,GAAG,QAAQ,EAAE,IAAI,KAAK,MAAM,KAAK,KAAK;;CAG/C,AAAU,WAAW,OAAe,WAAkC;EACpE,IAAI,MAAM;AAGV,GADE,WAAW,UAAU,mBAAmB,EAAE,EACrC,SAAS,OAAO,UAAU;AAC/B,SAAM,GAAG,MAAM,KAAK,cAAc,OAAO,MAAM,CAAC;IAChD;AACF,SAAO;;;AAIX,IAAa,mCAAb,cAAsD,6BAA6B;CACjF,AAAU,cACR,YACA,UACA,QACoB;CAItB,AAAU,UAAU,WAA0B,OAAuB;EACnE,MAAM,QAAQ,UAAU,SAAS,gBAAgB;EACjD,MAAM,MAAM,MAAM,kBAAkB,OAAO,MAAM,KAAK;AAEtD,SAAO,KADK,QAAQ,EACJ,IAAI,IAAI;;CAG1B,AAAU,cACR,WACA,SACA,QACoB;EACpB,IAAI,MAAM;AACV,UAAQ,sBAAsB,SAAS,eAAe;GACpD,MAAM,OAAO,KAAK,UAAU,WAAW,WAAW;AAClD,SAAM,GAAG,MAAM;IACf;AACF,SAAO;;CAGT,AAAU,WACR,OACA,YACoB;CAItB,AAAU,gBACR,OACA,OACQ;EACR,MAAM,MAAM,QAAQ;EACpB,MAAM,OAAO,MAAM,oBAAoB,MAAM;EAC7C,MAAM,MAAM,KAAK;AAEjB,SAAO,GAAG,IAAI,KADD,KAAK,MACM,IAAI,IAAI;;CAGlC,AAAU,WACR,OACA,WACoB;EACpB,IAAI,MAAM;AAEV,EADuC,UAAU,SAAS,gBACnD,SAAS,OAAO,UAAU;AAC/B,SAAM,GAAG,MAAM,KAAK,gBAAgB,OAAO,MAAM,CAAC;IAClD;EAEF,MAAM,SAAS,KAAK,iBAAiB,UAAU;AAC/C,QAAM,GAAG,IAAI,IAAI;AAEjB,SAAO"}

package/dist/output_parsers.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"output_parsers.d.cts","names":["BaseLLMOutputParser","Callbacks","ChatGeneration","Generation","GeminiGroundingChunk","GeminiGroundingMetadata","GeminiGroundingSupport","Generations","GroundingInfo","BaseGoogleSearchOutputParser","Promise","SimpleGoogleSearchOutputParser","MarkdownGoogleSearchOutputParser"],"sources":["../src/output_parsers.d.ts"],"sourcesContent":["import { BaseLLMOutputParser } from \"@langchain/core/output_parsers\";\nimport { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { ChatGeneration, Generation } from \"@langchain/core/outputs\";\nimport { GeminiGroundingChunk, GeminiGroundingMetadata, GeminiGroundingSupport } from \"./types.js\";\ntype Generations = Generation[] | ChatGeneration[];\ntype GroundingInfo = {\n    metadata: GeminiGroundingMetadata;\n    supports: GeminiGroundingSupport[];\n};\nexport declare abstract class BaseGoogleSearchOutputParser extends BaseLLMOutputParser<string> {\n    lc_namespace: string[];\n    protected generationToGroundingInfo(generation: Generation | ChatGeneration): GroundingInfo | undefined;\n    protected generationsToGroundingInfo(generations: Generations): GroundingInfo | undefined;\n    protected generationToString(generation: Generation | ChatGeneration): string;\n    protected generationsToString(generations: Generations): string;\n    protected abstract segmentPrefix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;\n    protected abstract segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;\n    protected annotateSegment(text: string, grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string;\n    protected annotateTextSegments(text: string, grounding: GroundingInfo): string;\n    protected abstract textPrefix(text: string, grounding: GroundingInfo): string | undefined;\n    protected abstract textSuffix(text: string, grounding: GroundingInfo): string | undefined;\n    /**\n     * Google requires us to\n     * \"Display the Search Suggestion exactly as provided without any modifications\"\n     * So this will typically be called from the textSuffix() method to get\n     * a string that renders HTML.\n     * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions\n     * @param grounding\n     */\n    protected searchSuggestion(grounding: GroundingInfo): string;\n    protected annotateText(text: string, grounding: GroundingInfo): string;\n    parseResult(generations: Generations, _callbacks?: Callbacks): Promise<string>;\n}\nexport declare class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected segmentSuffix(_grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected textPrefix(_text: string, _grounding: GroundingInfo): string;\n    protected chunkToString(chunk: GeminiGroundingChunk, index: number): string;\n    protected textSuffix(_text: string, grounding: GroundingInfo): string;\n}\nexport declare class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected chunkLink(grounding: GroundingInfo, index: number): string;\n    protected segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected textPrefix(_text: string, _grounding: GroundingInfo): string | undefined;\n    protected chunkSuffixLink(chunk: GeminiGroundingChunk, index: number): string;\n    protected textSuffix(_text: string, grounding: GroundingInfo): string | undefined;\n}\nexport {};\n//# sourceMappingURL=output_parsers.d.ts.map"],"mappings":";;;;;;KAIKO,WAAAA,GAAcJ,eAAeD;KAC7BM,aAAAA;EADAD,QAAAA,EAESF,uBAFKF;EACdK,QAAAA,EAESF,sBADAD,EAAAA;AAGd,CAAA;AAEoDF,uBAFtBM,4BAAAA,SAAqCT,mBAEfG,CAAAA,MAAAA,CAAAA,CAAAA;EAAaD,YAAAA,EAAAA,MAAAA,EAAAA;EAAiBM,UAAAA,yBAAAA,CAAAA,UAAAA,EAA9BL,UAA8BK,GAAjBN,cAAiBM,CAAAA,EAAAA,aAAAA,GAAAA,SAAAA;EAC5BD,UAAAA,0BAAAA,CAAAA,WAAAA,EAAAA,WAAAA,CAAAA,EAAcC,aAAdD,GAAAA,SAAAA;EAAcC,UAAAA,kBAAAA,CAAAA,UAAAA,EACvBL,UADuBK,GACVN,cADUM,CAAAA,EAAAA,MAAAA;EACvBL,UAAAA,mBAAAA,CAAAA,WAAAA,EACEI,WADFJ,CAAAA,EAAAA,MAAAA;EAAaD,mBAAAA,aAAAA,CAAAA,SAAAA,EAEVM,aAFUN,EAAAA,OAAAA,EAEcI,sBAFdJ,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACXK,mBAAAA,aAAAA,CAAAA,SAAAA,EAECC,aAFDD,EAAAA,OAAAA,EAEyBD,sBAFzBC,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACCC,UAAAA,eAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAEOA,aAFPA,EAAAA,OAAAA,EAE+BF,sBAF/BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAAwBF,UAAAA,oBAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAGZE,aAHYF,CAAAA,EAAAA,MAAAA;EACxBE,mBAAAA,UAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAGWA,aAHXA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAwBF,mBAAAA,UAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAIbE,aAJaF,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACjBE;;;;;;;;EAcAP,UAAAA,gBAAAA,CAAAA,SAAAA,EAFbO,aAEaP,CAAAA,EAAAA,MAAAA;EAAYS,UAAAA,YAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EADfF,aACeE,CAAAA,EAAAA,MAAAA;EAtBAV,WAAAA,CAAAA,WAAAA,EAsBtCO,WAtBsCP,EAAAA,UAAAA,CAAAA,EAsBZC,SAtBYD,CAAAA,EAsBAU,OAtBAV,CAAAA,MAAAA,CAAAA;AAAmB;AAwBjEW,cAAAA,8BAAAA,SAAuCF,4BAAAA,CAAT;EACXD,UAAAA,aAAAA,CAAAA,UAAAA,EAAAA,aAAAA,EAAAA,QAAAA,EAAyBF,sBAAzBE,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAyBF,UAAAA,aAAAA,CAAAA,UAAAA,EACzBE,aADyBF,EAAAA,OAAAA,EACDA,sBADCA,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACzBE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,UAAAA,EACYA,aADZA,CAAAA,EAAAA,MAAAA;EAAwBF,UAAAA,aAAAA,CAAAA,KAAAA,EAE7BF,oBAF6BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EACZE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAEDA,aAFCA,CAAAA,EAAAA,MAAAA;;AAEDA,cAE9BI,gCAAAA,SAAyCH,4BAAAA,CAFXD;EALSC,UAAAA,aAAAA,CAAAA,UAAAA,EAQpBD,aARoBC,EAAAA,QAAAA,EAQKH,sBARLG,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAA4B,UAAA,SAAA,CAAA,SAAA,EASrDD,aATqD,EAAA,KAAA,EAAA,MAAA,CAAA,EAAA,MAAA;EAOnEI,UAAAA,aAAAA,CAAAA,SAAAA,EAGkBJ,aAHc,EAAA,OAAA,EAGUF,sBAHV,EAAA,MAAA,EAAA,MAAA,CAAA,EAAA,MAAA,GAAA,SAAA;EACbE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,UAAAA,EAGYA,aAHZA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAyBF,UAAAA,eAAAA,CAAAA,KAAAA,EAI5BF,oBAJ4BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAC9BE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAIgBA,aAJhBA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA"}
+{"version":3,"file":"output_parsers.d.cts","names":[],"sources":["../src/output_parsers.ts"],"mappings":";;;;;;KAUK,WAAA,GAAc,UAAA,KAAe,cAAA;AAAA,KAE7B,aAAA;EACH,QAAA,EAAU,uBAAA;EACV,QAAA,EAAU,sBAAA;AAAA;AAAA,uBAGU,4BAAA,SAAqC,mBAAA;EACzD,YAAA;EAAA,UAEU,yBAAA,CACR,UAAA,EAAY,UAAA,GAAa,cAAA,GACxB,aAAA;EAAA,UAgBO,0BAAA,CACR,WAAA,EAAa,WAAA,GACZ,aAAA;EAAA,UAUO,kBAAA,CACR,UAAA,EAAY,UAAA,GAAa,cAAA;EAAA,UAuBjB,mBAAA,CAAoB,WAAA,EAAa,WAAA;EAAA,mBAQxB,aAAA,CACjB,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,mBAGiB,aAAA,CACjB,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,UAGQ,eAAA,CACR,IAAA,UACA,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,UAeQ,oBAAA,CACR,IAAA,UACA,SAAA,EAAW,aAAA;EAAA,mBAYM,UAAA,CACjB,IAAA,UACA,SAAA,EAAW,aAAA;EAAA,mBAGM,UAAA,CACjB,IAAA,UACA,SAAA,EAAW,aAAA;EArHO;;;;;;;;EAAA,UAgIV,gBAAA,CAAiB,SAAA,EAAW,aAAA;EAAA,UAI5B,YAAA,CAAa,IAAA,UAAc,SAAA,EAAW,aAAA;EAO1C,WAAA,CACJ,WAAA,EAAa,WAAA,EACb,UAAA,GAAa,SAAA,GACZ,OAAA;AAAA;AAAA,cAYQ,8BAAA,SAAuC,4BAAA;EAAA,UACxC,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,QAAA,EAAU,sBAAA,EACV,MAAA;EAAA,UAKQ,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,OAAA,EAAS,sBAAA,EACT,MAAA;EAAA,UAMQ,UAAA,CAAW,KAAA,UAAe,UAAA,EAAY,aAAA;EAAA,UAItC,aAAA,CAAc,KAAA,EAAO,oBAAA,EAAsB,KAAA;EAAA,UAK3C,UAAA,CAAW,KAAA,UAAe,SAAA,EAAW,aAAA;AAAA;AAAA,cAWpC,gCAAA,SAAyC,4BAAA;EAAA,UAC1C,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,QAAA,EAAU,sBAAA,EACV,MAAA;EAAA,UAKQ,SAAA,CAAU,SAAA,EAAW,aAAA,EAAe,KAAA;EAAA,UAOpC,aAAA,CACR,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,MAAA;EAAA,UAUQ,UAAA,CACR,KAAA,UACA,UAAA,EAAY,aAAA;EAAA,UAKJ,eAAA,CACR,KAAA,EAAO,oBAAA,EACP,KAAA;EAAA,UASQ,UAAA,CACR,KAAA,UACA,SAAA,EAAW,aAAA;AAAA"}

package/dist/output_parsers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"output_parsers.d.ts","names":["BaseLLMOutputParser","Callbacks","ChatGeneration","Generation","GeminiGroundingChunk","GeminiGroundingMetadata","GeminiGroundingSupport","Generations","GroundingInfo","BaseGoogleSearchOutputParser","Promise","SimpleGoogleSearchOutputParser","MarkdownGoogleSearchOutputParser"],"sources":["../src/output_parsers.d.ts"],"sourcesContent":["import { BaseLLMOutputParser } from \"@langchain/core/output_parsers\";\nimport { Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { ChatGeneration, Generation } from \"@langchain/core/outputs\";\nimport { GeminiGroundingChunk, GeminiGroundingMetadata, GeminiGroundingSupport } from \"./types.js\";\ntype Generations = Generation[] | ChatGeneration[];\ntype GroundingInfo = {\n    metadata: GeminiGroundingMetadata;\n    supports: GeminiGroundingSupport[];\n};\nexport declare abstract class BaseGoogleSearchOutputParser extends BaseLLMOutputParser<string> {\n    lc_namespace: string[];\n    protected generationToGroundingInfo(generation: Generation | ChatGeneration): GroundingInfo | undefined;\n    protected generationsToGroundingInfo(generations: Generations): GroundingInfo | undefined;\n    protected generationToString(generation: Generation | ChatGeneration): string;\n    protected generationsToString(generations: Generations): string;\n    protected abstract segmentPrefix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;\n    protected abstract segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;\n    protected annotateSegment(text: string, grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string;\n    protected annotateTextSegments(text: string, grounding: GroundingInfo): string;\n    protected abstract textPrefix(text: string, grounding: GroundingInfo): string | undefined;\n    protected abstract textSuffix(text: string, grounding: GroundingInfo): string | undefined;\n    /**\n     * Google requires us to\n     * \"Display the Search Suggestion exactly as provided without any modifications\"\n     * So this will typically be called from the textSuffix() method to get\n     * a string that renders HTML.\n     * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions\n     * @param grounding\n     */\n    protected searchSuggestion(grounding: GroundingInfo): string;\n    protected annotateText(text: string, grounding: GroundingInfo): string;\n    parseResult(generations: Generations, _callbacks?: Callbacks): Promise<string>;\n}\nexport declare class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected segmentSuffix(_grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected textPrefix(_text: string, _grounding: GroundingInfo): string;\n    protected chunkToString(chunk: GeminiGroundingChunk, index: number): string;\n    protected textSuffix(_text: string, grounding: GroundingInfo): string;\n}\nexport declare class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {\n    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected chunkLink(grounding: GroundingInfo, index: number): string;\n    protected segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;\n    protected textPrefix(_text: string, _grounding: GroundingInfo): string | undefined;\n    protected chunkSuffixLink(chunk: GeminiGroundingChunk, index: number): string;\n    protected textSuffix(_text: string, grounding: GroundingInfo): string | undefined;\n}\nexport {};\n//# sourceMappingURL=output_parsers.d.ts.map"],"mappings":";;;;;;KAIKO,WAAAA,GAAcJ,eAAeD;KAC7BM,aAAAA;EADAD,QAAAA,EAESF,uBAFKF;EACdK,QAAAA,EAESF,sBADAD,EAAAA;AAGd,CAAA;AAEoDF,uBAFtBM,4BAAAA,SAAqCT,mBAEfG,CAAAA,MAAAA,CAAAA,CAAAA;EAAaD,YAAAA,EAAAA,MAAAA,EAAAA;EAAiBM,UAAAA,yBAAAA,CAAAA,UAAAA,EAA9BL,UAA8BK,GAAjBN,cAAiBM,CAAAA,EAAAA,aAAAA,GAAAA,SAAAA;EAC5BD,UAAAA,0BAAAA,CAAAA,WAAAA,EAAAA,WAAAA,CAAAA,EAAcC,aAAdD,GAAAA,SAAAA;EAAcC,UAAAA,kBAAAA,CAAAA,UAAAA,EACvBL,UADuBK,GACVN,cADUM,CAAAA,EAAAA,MAAAA;EACvBL,UAAAA,mBAAAA,CAAAA,WAAAA,EACEI,WADFJ,CAAAA,EAAAA,MAAAA;EAAaD,mBAAAA,aAAAA,CAAAA,SAAAA,EAEVM,aAFUN,EAAAA,OAAAA,EAEcI,sBAFdJ,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACXK,mBAAAA,aAAAA,CAAAA,SAAAA,EAECC,aAFDD,EAAAA,OAAAA,EAEyBD,sBAFzBC,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACCC,UAAAA,eAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAEOA,aAFPA,EAAAA,OAAAA,EAE+BF,sBAF/BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAAwBF,UAAAA,oBAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAGZE,aAHYF,CAAAA,EAAAA,MAAAA;EACxBE,mBAAAA,UAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAGWA,aAHXA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAwBF,mBAAAA,UAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAIbE,aAJaF,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACjBE;;;;;;;;EAcAP,UAAAA,gBAAAA,CAAAA,SAAAA,EAFbO,aAEaP,CAAAA,EAAAA,MAAAA;EAAYS,UAAAA,YAAAA,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EADfF,aACeE,CAAAA,EAAAA,MAAAA;EAtBAV,WAAAA,CAAAA,WAAAA,EAsBtCO,WAtBsCP,EAAAA,UAAAA,CAAAA,EAsBZC,SAtBYD,CAAAA,EAsBAU,OAtBAV,CAAAA,MAAAA,CAAAA;AAAmB;AAwBjEW,cAAAA,8BAAAA,SAAuCF,4BAAAA,CAAT;EACXD,UAAAA,aAAAA,CAAAA,UAAAA,EAAAA,aAAAA,EAAAA,QAAAA,EAAyBF,sBAAzBE,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAyBF,UAAAA,aAAAA,CAAAA,UAAAA,EACzBE,aADyBF,EAAAA,OAAAA,EACDA,sBADCA,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EACzBE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,UAAAA,EACYA,aADZA,CAAAA,EAAAA,MAAAA;EAAwBF,UAAAA,aAAAA,CAAAA,KAAAA,EAE7BF,oBAF6BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EACZE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAEDA,aAFCA,CAAAA,EAAAA,MAAAA;;AAEDA,cAE9BI,gCAAAA,SAAyCH,4BAAAA,CAFXD;EALSC,UAAAA,aAAAA,CAAAA,UAAAA,EAQpBD,aARoBC,EAAAA,QAAAA,EAQKH,sBARLG,EAAAA,MAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAA4B,UAAA,SAAA,CAAA,SAAA,EASrDD,aATqD,EAAA,KAAA,EAAA,MAAA,CAAA,EAAA,MAAA;EAOnEI,UAAAA,aAAAA,CAAAA,SAAAA,EAGkBJ,aAHc,EAAA,OAAA,EAGUF,sBAHV,EAAA,MAAA,EAAA,MAAA,CAAA,EAAA,MAAA,GAAA,SAAA;EACbE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,UAAAA,EAGYA,aAHZA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAyBF,UAAAA,eAAAA,CAAAA,KAAAA,EAI5BF,oBAJ4BE,EAAAA,KAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAC9BE,UAAAA,UAAAA,CAAAA,KAAAA,EAAAA,MAAAA,EAAAA,SAAAA,EAIgBA,aAJhBA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA"}
+{"version":3,"file":"output_parsers.d.ts","names":[],"sources":["../src/output_parsers.ts"],"mappings":";;;;;;KAUK,WAAA,GAAc,UAAA,KAAe,cAAA;AAAA,KAE7B,aAAA;EACH,QAAA,EAAU,uBAAA;EACV,QAAA,EAAU,sBAAA;AAAA;AAAA,uBAGU,4BAAA,SAAqC,mBAAA;EACzD,YAAA;EAAA,UAEU,yBAAA,CACR,UAAA,EAAY,UAAA,GAAa,cAAA,GACxB,aAAA;EAAA,UAgBO,0BAAA,CACR,WAAA,EAAa,WAAA,GACZ,aAAA;EAAA,UAUO,kBAAA,CACR,UAAA,EAAY,UAAA,GAAa,cAAA;EAAA,UAuBjB,mBAAA,CAAoB,WAAA,EAAa,WAAA;EAAA,mBAQxB,aAAA,CACjB,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,mBAGiB,aAAA,CACjB,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,UAGQ,eAAA,CACR,IAAA,UACA,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,KAAA;EAAA,UAeQ,oBAAA,CACR,IAAA,UACA,SAAA,EAAW,aAAA;EAAA,mBAYM,UAAA,CACjB,IAAA,UACA,SAAA,EAAW,aAAA;EAAA,mBAGM,UAAA,CACjB,IAAA,UACA,SAAA,EAAW,aAAA;EArHO;;;;;;;;EAAA,UAgIV,gBAAA,CAAiB,SAAA,EAAW,aAAA;EAAA,UAI5B,YAAA,CAAa,IAAA,UAAc,SAAA,EAAW,aAAA;EAO1C,WAAA,CACJ,WAAA,EAAa,WAAA,EACb,UAAA,GAAa,SAAA,GACZ,OAAA;AAAA;AAAA,cAYQ,8BAAA,SAAuC,4BAAA;EAAA,UACxC,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,QAAA,EAAU,sBAAA,EACV,MAAA;EAAA,UAKQ,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,OAAA,EAAS,sBAAA,EACT,MAAA;EAAA,UAMQ,UAAA,CAAW,KAAA,UAAe,UAAA,EAAY,aAAA;EAAA,UAItC,aAAA,CAAc,KAAA,EAAO,oBAAA,EAAsB,KAAA;EAAA,UAK3C,UAAA,CAAW,KAAA,UAAe,SAAA,EAAW,aAAA;AAAA;AAAA,cAWpC,gCAAA,SAAyC,4BAAA;EAAA,UAC1C,aAAA,CACR,UAAA,EAAY,aAAA,EACZ,QAAA,EAAU,sBAAA,EACV,MAAA;EAAA,UAKQ,SAAA,CAAU,SAAA,EAAW,aAAA,EAAe,KAAA;EAAA,UAOpC,aAAA,CACR,SAAA,EAAW,aAAA,EACX,OAAA,EAAS,sBAAA,EACT,MAAA;EAAA,UAUQ,UAAA,CACR,KAAA,UACA,UAAA,EAAY,aAAA;EAAA,UAKJ,eAAA,CACR,KAAA,EAAO,oBAAA,EACP,KAAA;EAAA,UASQ,UAAA,CACR,KAAA,UACA,SAAA,EAAW,aAAA;AAAA"}

package/dist/output_parsers.js

@@ -13,14 +13,12 @@ var BaseGoogleSearchOutputParser = class extends BaseLLMOutputParser {
 				supports
 			};
 		}
-		return void 0;
 	}
 	generationsToGroundingInfo(generations) {
 		for (const generation of generations) {
 			const info = this.generationToGroundingInfo(generation);
 			if (info !== void 0) return info;
 		}
-		return void 0;
 	}
 	generationToString(generation) {
 		if ("message" in generation) {
@@ -42,9 +40,7 @@ var BaseGoogleSearchOutputParser = class extends BaseLLMOutputParser {
 		const textBefore = text.substring(0, start);
 		const textSegment = text.substring(start, end);
 		const textAfter = text.substring(end);
-		const textPrefix = this.segmentPrefix(grounding, support, index) ?? "";
-		const textSuffix = this.segmentSuffix(grounding, support, index) ?? "";
-		return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;
+		return `${textBefore}${this.segmentPrefix(grounding, support, index) ?? ""}${textSegment}${this.segmentSuffix(grounding, support, index) ?? ""}${textAfter}`;
 	}
 	annotateTextSegments(text, grounding) {
 		let ret = text;
@@ -68,8 +64,7 @@ var BaseGoogleSearchOutputParser = class extends BaseLLMOutputParser {
 	annotateText(text, grounding) {
 		const prefix = this.textPrefix(text, grounding) ?? "";
 		const suffix = this.textSuffix(text, grounding) ?? "";
-		const body = this.annotateTextSegments(text, grounding);
-		return `${prefix}${body}${suffix}`;
+		return `${prefix}${this.annotateTextSegments(text, grounding)}${suffix}`;
 	}
 	async parseResult(generations, _callbacks) {
 		const text = this.generationsToString(generations);
@@ -79,12 +74,9 @@ var BaseGoogleSearchOutputParser = class extends BaseLLMOutputParser {
 	}
 };
 var SimpleGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser {
-	segmentPrefix(_grounding, _support, _index) {
-		return void 0;
-	}
+	segmentPrefix(_grounding, _support, _index) {}
 	segmentSuffix(_grounding, support, _index) {
-		const indices = support.groundingChunkIndices.map((i) => i + 1);
-		return ` [${indices.join(", ")}]`;
+		return ` [${support.groundingChunkIndices.map((i) => i + 1).join(", ")}]`;
 	}
 	textPrefix(_text, _grounding) {
 		return "Google Says:\n";
@@ -95,22 +87,18 @@ var SimpleGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser
 	}
 	textSuffix(_text, grounding) {
 		let ret = "\n";
-		const chunks = grounding?.metadata?.groundingChunks ?? [];
-		chunks.forEach((chunk, index) => {
+		(grounding?.metadata?.groundingChunks ?? []).forEach((chunk, index) => {
 			ret = `${ret}${this.chunkToString(chunk, index)}\n`;
 		});
 		return ret;
 	}
 };
 var MarkdownGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParser {
-	segmentPrefix(_grounding, _support, _index) {
-		return void 0;
-	}
+	segmentPrefix(_grounding, _support, _index) {}
 	chunkLink(grounding, index) {
 		const chunk = grounding.metadata.groundingChunks[index];
 		const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;
-		const num = index + 1;
-		return `[[${num}](${url})]`;
+		return `[[${index + 1}](${url})]`;
 	}
 	segmentSuffix(grounding, support, _index) {
 		let ret = "";
@@ -120,20 +108,16 @@ var MarkdownGoogleSearchOutputParser = class extends BaseGoogleSearchOutputParse
 		});
 		return ret;
 	}
-	textPrefix(_text, _grounding) {
-		return void 0;
-	}
+	textPrefix(_text, _grounding) {}
 	chunkSuffixLink(chunk, index) {
 		const num = index + 1;
 		const info = chunk.retrievedContext ?? chunk.web;
 		const url = info.uri;
-		const site = info.title;
-		return `${num}. [${site}](${url})`;
+		return `${num}. [${info.title}](${url})`;
 	}
 	textSuffix(_text, grounding) {
 		let ret = "\n**Search Sources**\n";
-		const chunks = grounding.metadata.groundingChunks;
-		chunks.forEach((chunk, index) => {
+		grounding.metadata.groundingChunks.forEach((chunk, index) => {
 			ret = `${ret}${this.chunkSuffixLink(chunk, index)}\n`;
 		});
 		const search = this.searchSuggestion(grounding);