{"version":3,"sources":["../../../../src/adapters/vercel/backend/utils.ts"],"names":["encodeCustomMessage","msg","UserMessage","type","text","role","content","slice","decodeCustomMessage","value","_","id","split","join","unmaskCustomMessage","Role","USER","key","Array","isArray","map","val","filter","isTruthy","decoded","vercelFetcher","customFetch","url","options","isString","body","getProp","headers","JSON","parse","isPlainObject","messages","forEach","stringify","fetcher","fetch","parseHeadersFromEnv","env","parseEnv","z","preprocess","Object","fromEntries","String","pair","includes","record","string","extractTokenUsage","usage","promptTokens","inputTokens","completionTokens","outputTokens","totalTokens","reasoningTokens","cachedPromptTokens","cachedInputTokens","mergeTokenUsage","target","sources","source","isToolCallValid","obj","toolName","toolCallId","input","Boolean","parseBrokenJson"],"mappings":";;;;;;;;;;;AAcO,SAASA,oBAAoBC,GAAAA,EAAkB;AACpD,EAAA,OAAO,IAAIC,uBAAAA,CAAY;AACrB,IAAA;MACEC,IAAAA,EAAM,MAAA;MACNC,IAAAA,EAAM,CAAA,aAAA,EAAgBH,IAAII,IAAI,CAAA,CAAA;AAChC,KAAA;AACIJ,IAAAA,GAAAA,GAAAA,CAAIK,QAAQC,KAAAA;AACjB,GAAA,CAAA;AACH;AARgBP,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA;AAUT,SAASQ,oBAAoBC,KAAAA,EAAa;AAC/C,EAAA,MAAM,CAACC,GAAGC,EAAAA,EAAIN,IAAAA,EAAM,GAAGC,OAAAA,CAAAA,GAAWG,KAAAA,CAAMG,KAAAA,CAAM,GAAA,CAAA;AAC9C,EAAA,IAAID,OAAO,aAAA,EAAe;AACxB,IAAA;AACF,EAAA;AACA,EAAA,OAAO;AAAEN,IAAAA,IAAAA;IAAMC,OAAAA,EAASA,OAAAA,CAAQO,KAAK,GAAA;AAAK,GAAA;AAC5C;AANgBL,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA;AAQhB,SAASM,oBAAoBb,GAAAA,EAAwB;AACnD,EAAA,IAAIA,GAAAA,CAAII,IAAAA,KAASU,gBAAAA,CAAKC,IAAAA,EAAM;AAC1B,IAAA;AACF,EAAA;AAEA,EAAA,KAAA,MAAWC,GAAAA,IAAO;AAAC,IAAA,SAAA;AAAW,IAAA;AAAS,GAAA,EAAA;AACrC,IAAA,IAAIR,KAAAA,GAAQR,IAAIgB,GAAAA,CAAAA;AAChB,IAAA,IAAI,CAACR,KAAAA,EAAO;AACV,MAAA;AACF,IAAA;AAEA,IAAA,IAAIS,KAAAA,CAAMC,OAAAA,CAAQV,KAAAA,CAAAA,EAAQ;AACxBA,MAAAA,KAAAA,GAAQA,MACLW,GAAAA,CAAI,CAACC,GAAAA,KAASA,GAAAA,CAAIlB,SAAS,MAAA,GAASkB,GAAAA,CAAIjB,IAAAA,IAAQiB,GAAAA,CAAIf,UAAU,IAAA,CAAA,CAC9DgB,OAAOC,eAAAA,CAAAA,CACPV,KAAK,EAAA,CAAA;AACV,IAAA;AAEA,IAAA,MAAMW,OAAAA,GAAUhB,oBAAoBC,KAAAA,CAAAA;AACpC,IAAA,IAAIe,OAAAA,EAAS;AACXvB,MAAAA,GAAAA,CAAII,OAAOmB,OAAAA,CAAQnB,IAAAA;AACnBJ,MAAAA,GAAAA,CAAIgB,GAAAA,IAAOO,OAAAA,CAAQlB,OAAAA;AACnB,MAAA;AACF,IAAA;AACF,EAAA;AACF;AAzBSQ,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA;AA2BF,SAASW,cAAcC,WAAAA,EAA0B;AACtD,EAAA,OAAO,OAAOC,KAAKC,OAAAA,KAAAA;AACjB,IAAA,IACEA,WACAC,eAAAA,CAASD,OAAAA,CAAQE,IAAI,CAAA,KACpBC,kBAAAA,CAAQH,QAAQI,OAAAA,EAAS;AAAC,MAAA;KAAe,CAAA,KAAM,kBAAA,IAC9CD,kBAAAA,CAAQH,OAAAA,CAAQI,OAAAA,EAAS;AAAC,MAAA;AAAe,KAAA,CAAA,KAAM,kBAAA,CAAA,EACjD;AACA,MAAA,MAAMF,IAAAA,GAAOG,IAAAA,CAAKC,KAAAA,CAAMN,OAAAA,CAAQE,IAAI,CAAA;AACpC,MAAA,IAAIK,qBAAcL,IAAAA,CAAAA,IAASZ,MAAMC,OAAAA,CAAQW,IAAAA,CAAKM,QAAQ,CAAA,EAAG;AACvDN,QAAAA,IAAAA,CAAKM,QAAAA,CAASC,OAAAA,CAAQ,CAACpC,GAAAA,KAAAA;AACrB,UAAA,IAAI,CAACkC,oBAAAA,CAAclC,GAAAA,CAAAA,EAAM;AACvB,YAAA;AACF,UAAA;AACAa,UAAAA,mBAAAA,CAAoBb,GAAAA,CAAAA;QACtB,CAAA,CAAA;AACF,MAAA;AACA2B,MAAAA,OAAAA,CAAQE,IAAAA,GAAOG,IAAAA,CAAKK,SAAAA,CAAUR,IAAAA,CAAAA;AAChC,IAAA;AAEA,IAAA,MAAMS,UAAUb,WAAAA,IAAec,KAAAA;AAC/B,IAAA,OAAO,MAAMD,OAAAA,CAAQZ,GAAAA,EAAKC,OAAAA,CAAAA;AAC5B,EAAA,CAAA;AACF;AAvBgBH,MAAAA,CAAAA,aAAAA,EAAAA,eAAAA,CAAAA;AAyBT,SAASgB,oBAAoBC,GAAAA,EAAW;AAC7C,EAAA,OAAOC,gBAAAA,CACLD,GAAAA,EACAE,KAAAA,CAAEC,UAAAA,CAAW,CAACpC,KAAAA,KAAAA;AACZ,IAAA,OAAOqC,MAAAA,CAAOC,WAAAA,CACZC,MAAAA,CAAOvC,KAAAA,IAAS,EAAA,EACbG,KAAAA,CAAM,GAAA,CAAA,CACNU,MAAAA,CAAO,CAAC2B,IAAAA,KAASA,KAAKC,QAAAA,CAAS,GAAA,CAAA,CAAA,CAC/B9B,GAAAA,CAAI,CAAC6B,SAASA,IAAAA,CAAKrC,KAAAA,CAAM,GAAA,CAAA,CAAA,CAAA;AAEhC,EAAA,CAAA,EAAGgC,MAAEO,MAAAA,CAAOP,KAAAA,CAAEQ,MAAAA,EAAM,CAAA,CAAA,CAAA;AAExB;AAZgBX,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA;AAcT,SAASY,kBAAkBC,KAAAA,EAAyB;AACzD,EAAA,OAAO;AACLC,IAAAA,YAAAA,EAAcD,MAAME,WAAAA,IAAe,CAAA;AACnCC,IAAAA,gBAAAA,EAAkBH,MAAMI,YAAAA,IAAgB,CAAA;AACxCC,IAAAA,WAAAA,EAAaL,MAAMK,WAAAA,IAAe,CAAA;AAClCC,IAAAA,eAAAA,EAAiBN,KAAAA,CAAMM,eAAAA;AACvBC,IAAAA,kBAAAA,EAAoBP,KAAAA,CAAMQ;AAC5B,GAAA;AACF;AARgBT,MAAAA,CAAAA,iBAAAA,EAAAA,mBAAAA,CAAAA;AAUT,SAASU,eAAAA,CAAgBC,WAA2BC,OAAAA,EAAyB;AAClF,EAAA,KAAA,MAAWC,UAAUD,OAAAA,EAAS;AAC5BD,IAAAA,MAAAA,CAAOL,WAAAA,IAAeO,OAAOP,WAAAA,IAAe,CAAA;AAC5CK,IAAAA,MAAAA,CAAOT,YAAAA,IAAgBW,OAAOX,YAAAA,IAAgB,CAAA;AAC9CS,IAAAA,MAAAA,CAAOP,gBAAAA,IAAoBS,OAAOT,gBAAAA,IAAoB,CAAA;AACtD,IAAA,IAAIS,OAAON,eAAAA,EAAiB;AAC1BI,MAAAA,MAAAA,CAAOJ,eAAAA,GAAAA,CAAmBI,MAAAA,CAAOJ,eAAAA,IAAmB,CAAA,IAAKM,MAAAA,CAAON,eAAAA;AAClE,IAAA;AACA,IAAA,IAAIM,OAAOL,kBAAAA,EAAoB;AAC7BG,MAAAA,MAAAA,CAAOH,kBAAAA,GAAAA,CAAsBG,MAAAA,CAAOH,kBAAAA,IAAsB,CAAA,IAAKK,MAAAA,CAAOL,kBAAAA;AACxE,IAAA;AACF,EAAA;AACF;AAZgBE,MAAAA,CAAAA,eAAAA,EAAAA,iBAAAA,CAAAA;AAcT,SAASI,gBAAgBC,GAAAA,EAAiB;AAC/C,EAAA,IAAI,CAACA,IAAIC,QAAAA,IAAY,CAACD,IAAIE,UAAAA,IAAc,CAACF,IAAIG,KAAAA,EAAO;AAClD,IAAA,OAAO,KAAA;AACT,EAAA;AACA,EAAA,IAAI1C,eAAAA,CAASuC,GAAAA,CAAIG,KAAK,CAAA,EAAG;AACvB,IAAA,OAAOC,OAAAA,CAAQC,0BAAAA,CAAgBL,GAAAA,CAAIG,KAAAA,EAAO;MAAEtB,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;AAAK,KAAA,CAAA,CAAA;AAC/D,EAAA;AACA,EAAA,OAAOuB,OAAAA,CAAQJ,IAAIG,KAAK,CAAA;AAC1B;AARgBJ,MAAAA,CAAAA,eAAAA,EAAAA,iBAAAA,CAAAA","file":"utils.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { CustomMessage, Role, UserMessage } from \"@/backend/message.js\";\nimport { isPlainObject, isString, isTruthy } from \"remeda\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { TextPart, LanguageModelUsage, ToolCallPart } from \"ai\";\nimport { z } from \"zod\";\nimport { parseEnv } from \"@/internals/env.js\";\nimport { ChatModelUsage } from \"@/backend/chat.js\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\n\nexport function encodeCustomMessage(msg: CustomMessage): UserMessage {\n  return new UserMessage([\n    {\n      type: \"text\",\n      text: `#custom_role#${msg.role}#`,\n    },\n    ...(msg.content.slice() as TextPart[]),\n  ]);\n}\n\nexport function decodeCustomMessage(value: string) {\n  const [_, id, role, ...content] = value.split(\"#\");\n  if (id !== \"custom_role\") {\n    return;\n  }\n  return { role, content: content.join(\"#\") };\n}\n\nfunction unmaskCustomMessage(msg: Record<string, any>) {\n  if (msg.role !== Role.USER) {\n    return;\n  }\n\n  for (const key of [\"content\", \"text\"]) {\n    let value = msg[key];\n    if (!value) {\n      continue;\n    }\n\n    if (Array.isArray(value)) {\n      value = value\n        .map((val) => (val.type === \"text\" ? val.text || val.content : null))\n        .filter(isTruthy)\n        .join(\"\");\n    }\n\n    const decoded = decodeCustomMessage(value);\n    if (decoded) {\n      msg.role = decoded.role;\n      msg[key] = decoded.content;\n      break;\n    }\n  }\n}\n\nexport function vercelFetcher(customFetch?: typeof fetch): typeof fetch {\n  return async (url, options) => {\n    if (\n      options &&\n      isString(options.body) &&\n      (getProp(options.headers, [\"content-type\"]) === \"application/json\" ||\n        getProp(options.headers, [\"Content-Type\"]) === \"application/json\")\n    ) {\n      const body = JSON.parse(options.body);\n      if (isPlainObject(body) && Array.isArray(body.messages)) {\n        body.messages.forEach((msg) => {\n          if (!isPlainObject(msg)) {\n            return;\n          }\n          unmaskCustomMessage(msg);\n        });\n      }\n      options.body = JSON.stringify(body);\n    }\n\n    const fetcher = customFetch ?? fetch;\n    return await fetcher(url, options);\n  };\n}\n\nexport function parseHeadersFromEnv(env: string): Record<string, any> {\n  return parseEnv(\n    env,\n    z.preprocess((value) => {\n      return Object.fromEntries(\n        String(value || \"\")\n          .split(\",\")\n          .filter((pair) => pair.includes(\"=\"))\n          .map((pair) => pair.split(\"=\")),\n      );\n    }, z.record(z.string())),\n  );\n}\n\nexport function extractTokenUsage(usage: LanguageModelUsage): ChatModelUsage {\n  return {\n    promptTokens: usage.inputTokens ?? 0,\n    completionTokens: usage.outputTokens ?? 0,\n    totalTokens: usage.totalTokens ?? 0,\n    reasoningTokens: usage.reasoningTokens,\n    cachedPromptTokens: usage.cachedInputTokens,\n  };\n}\n\nexport function mergeTokenUsage(target: ChatModelUsage, ...sources: ChatModelUsage[]): void {\n  for (const source of sources) {\n    target.totalTokens += source.totalTokens ?? 0;\n    target.promptTokens += source.promptTokens ?? 0;\n    target.completionTokens += source.completionTokens ?? 0;\n    if (source.reasoningTokens) {\n      target.reasoningTokens = (target.reasoningTokens ?? 0) + source.reasoningTokens;\n    }\n    if (source.cachedPromptTokens) {\n      target.cachedPromptTokens = (target.cachedPromptTokens ?? 0) + source.cachedPromptTokens;\n    }\n  }\n}\n\nexport function isToolCallValid(obj: ToolCallPart) {\n  if (!obj.toolName || !obj.toolCallId || !obj.input) {\n    return false;\n  }\n  if (isString(obj.input)) {\n    return Boolean(parseBrokenJson(obj.input, { pair: [\"{\", \"}\"] }));\n  }\n  return Boolean(obj.input);\n}\n"]}