beeai-framework 0.1.29 → 0.1.30

package/dist/agents/requirement/runner.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/agents/requirement/runner.ts"],"names":["TEMP_MESSAGE_KEY","RequirementAgentRunner","state","reasoner","iterationErrorCounter","globalErrorCounter","llm","runConfig","tools","requirements","expectedOutput","toolCallCycleChecker","ctx","forceFinalAnswerAsTool","templates","answer","result","memory","UnconstrainedMemory","steps","iteration","usage","totalTokens","promptTokens","completionTokens","reasoningTokens","cachedPromptTokens","undefined","finalAnswer","FinalAnswerTool","RequirementsReasoner","maxRetriesPerIteration","maxRetriesPerStep","RetryCounter","AgentError","maxRetries","Math","max","totalMaxRetries","incrementIteration","maxIterations","runLLM","request","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","catch","err","ChatModelToolCallError","canStop","generatedContent","errorResponse","data","ChatModelOutput","AssistantMessage","finishReason","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","type","toolCallId","uuidv4","substring","toolName","name","inputSchema","parse","createRequest","extraRules","invokeToolCalls","toolCalls","toolResults","results","Promise","all","map","msg","runTool","toolCall","push","id","output","tool","error","toolError","render","reason","explain","isEmpty","getTextContent","toolNoResult","ToolMessage","value","use","addMessages","addMany","run","update","emitter","emit","runIteration","getToolCalls","length","textMessages","getTextMessages","text","m","join","finalAnswerToolCall","stream","add","updatedRequest","target","allowed","hidden","forced","preventStop","toolCallMsg","register","cycleFound","reset","tempMessages","filter","meta","deleteMany","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;;;AA+BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EApCb;;;;;;;;;;AAqCqBC,EAAAA,KAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,qBAAAA;AACAC,EAAAA,kBAAAA;EAEnB,WAAA,CACqBC,GAAAA,EACAC,WACnBC,KAAAA,EACmBC,YAAAA,EACnBC,gBACmBC,oBAAAA,EACAC,GAAAA,EACTC,wBACSC,SAAAA,EACnB;SATmBR,GAAAA,GAAAA,GAAAA;SACAC,SAAAA,GAAAA,SAAAA;SAEAE,YAAAA,GAAAA,YAAAA;SAEAE,oBAAAA,GAAAA,oBAAAA;SACAC,GAAAA,GAAAA,GAAAA;SACTC,sBAAAA,GAAAA,sBAAAA;SACSC,SAAAA,GAAAA,SAAAA;AAEnB,IAAA,IAAA,CAAKZ,KAAAA,GAAQ;MACXa,MAAAA,EAAQ,IAAA;MACRC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,MAAAA,EAAQ,IAAIC,mBAAAA,EAAAA;AACZC,MAAAA,KAAAA,EAAO,EAAA;MACPC,SAAAA,EAAW,CAAA;MACXC,KAAAA,EAAO;QACLC,WAAAA,EAAa,CAAA;QACbC,YAAAA,EAAc,CAAA;QACdC,gBAAAA,EAAkB,CAAA;QAClBC,eAAAA,EAAiB,CAAA;QACjBC,kBAAAA,EAAoBC;AACtB;AACF,KAAA;AAEA,IAAA,IAAA,CAAKlB,YAAAA,GAAeA,YAAAA;AACpB,IAAA,MAAMmB,WAAAA,GAAc,IAAIC,eAAAA,CAAgBnB,cAAAA,EAAgB,KAAKR,KAAK,CAAA;AAClE,IAAA,IAAA,CAAKC,QAAAA,GAAW,IAAI2B,oBAAAA,CAAqBtB,KAAAA,EAAOoB,aAAahB,GAAAA,CAAAA;AAE7D,IAAA,MAAMmB,sBAAAA,GAAyBxB,UAAUyB,iBAAAA,IAAqB,CAAA;AAC9D,IAAA,IAAA,CAAK5B,qBAAAA,GAAwB,IAAI6B,YAAAA,CAAaF,sBAAAA,EAAwBG,UAAAA,CAAAA;AAEtE,IAAA,MAAMC,aAAaC,IAAAA,CAAKC,GAAAA,CAAIN,sBAAAA,EAAwBxB,SAAAA,CAAU+B,mBAAmB,CAAA,CAAA;AACjF,IAAA,IAAA,CAAKjC,kBAAAA,GAAqB,IAAI4B,YAAAA,CAAaE,UAAAA,EAAYD,UAAAA,CAAAA;AACzD;EAEUK,kBAAAA,GAA2B;AACnC,IAAA,IAAA,CAAKrC,KAAAA,CAAMkB,SAAAA,EAAAA;AAEX,IAAA,IAAI,IAAA,CAAKb,UAAUiC,aAAAA,IAAiB,IAAA,CAAKtC,MAAMkB,SAAAA,GAAY,IAAA,CAAKb,UAAUiC,aAAAA,EAAe;AACvF,MAAA,MAAM,IAAIN,UAAAA,CACR,CAAA,0CAAA,EAA6C,IAAA,CAAKhC,KAAAA,CAAMkB,SAAS,CAAA,YAAA,CAAc,CAAA;AAEnF;AACF;AAEA,EAAA,MAAgBqB,OAAOC,OAAAA,EAA4D;AACjF,IAAA,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CACzB0C,MAAAA,CAAOH,KAAAA,CAAAA,CACPI,UAAAA,CAAWN,gBAAAA,CAAAA,CACXO,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,IAAIA,GAAAA,YAAeC,sBAAAA,IAA0BV,OAAAA,CAAQW,OAAAA,EAAS;AAC5D,UAAA,MAAM,EAAEC,gBAAAA,EAAkBP,QAAAA,EAAUQ,aAAAA,KAAkBJ,GAAAA,CAAIK,IAAAA;AAC1D,UAAA,IAAIF,gBAAAA,EAAkB;AACpB,YAAA,OAAO,IAAIG,eAAAA,CACT;AAAC,cAAA,IAAIC,iBAAiBJ,gBAAAA;eACtBC,aAAAA,EAAelC,KAAAA,EACfkC,eAAeI,YAAAA,CAAAA;AAEnB;AACF;AACA,QAAA,MAAMR,GAAAA;OACR,CAAA;AAEF,MAAA,IAAIJ,SAAS1B,KAAAA,EAAO;AAClBuC,QAAAA,eAAAA,CAAgB,IAAA,CAAK1D,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD;AAEA,MAAA,OAAO0B,QAAAA;KACT,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBkB,MAAAA,EAAM;AACzB;AACF;AAEA,EAAA,MAAgBf,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMoB,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,mBAAAA,CAAoB,IAAA,CAAKjD,SAAAA,CAAUkD,MAAAA,EAAQtB,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAO6C;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACAtD,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQuB,YAAAA;AACfC,MAAAA,UAAAA,EAAYxB,OAAAA,CAAQwB,UAAAA;MACpBC,sBAAAA,EAAwB,IAAA;MACxBhC,UAAAA,EAAY,IAAA,CAAK5B,UAAUyB,iBAAAA,IAAqB;AAClD,KAAA;AACF;AAEA,EAAA,MAAgBoC,0BAA0BC,QAAAA,EAAoD;AAE5F,IAAA,MAAMC,MAAAA,GAASC,gBAAgBF,QAAAA,EAAU;MAAEG,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAC5D,IAAA,IAAI,CAACF,MAAAA,EAAQ;AAEX,MAAA,IAAI,CAAC,IAAA,CAAKnE,QAAAA,CAASyB,WAAAA,CAAY6C,YAAAA,EAAc;AAC3C,QAAA,OAAO,IAAIf,gBAAAA,CAAiB;AAC1B,UAAA;YACEgB,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;YAC1CC,QAAAA,EAAU,IAAA,CAAK3E,SAASyB,WAAAA,CAAYmD,IAAAA;AACpClC,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAYoD,WAAAA,GAAcC,KAAAA,CAAM;cAAElC,QAAAA,EAAUsB;aAAS;AAC5E;AACD,SAAA,CAAA;AACH;AACA,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,IAAIX,gBAAAA,CAAiB;AAC1B,MAAA;QACEgB,IAAAA,EAAM,WAAA;AACNC,QAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;QAC1CC,QAAAA,EAAU,IAAA,CAAK3E,SAASyB,WAAAA,CAAYmD,IAAAA;QACpClC,KAAAA,EAAOyB;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBY,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKhF,QAAAA,CAAS+E,aAAAA,CAAc,KAAKhF,KAAAA,EAAO,IAAA,CAAKW,wBAAwBsE,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACd5E,OACA6E,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,OAAAA,CAAQpF,KAAAA,EAAOmF,GAAAA,EAAK;AAAEzF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAW2F,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAKrF,KAAAA,CAAMiB,MAAM2E,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,EAAAA,EAAAA;AACJxD,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOgD,QAAAA,CAAShD,KAAAA;AAChBmD,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIlF,MAAAA;AACJ,MAAA,IAAI6E,SAASK,KAAAA,EAAO;AAClBlF,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAUqF,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACLtF,QAAAA,MAAAA,GAAS,CAAC6E,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAK1F,SAAAA,CAAU2F,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAMpD,YAAAA,KAAAA,EAAOgD,QAAAA,CAAShD;AAAM;SACzD,CAAA;AACN;AACAyC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,WAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAO3F;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAI6E,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAK9F,qBAAAA,CAAsBwG,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK7F,kBAAAA,CAAmBuG,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAY/C,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAK5D,KAAAA,CAAMe,MAAAA,CAAO6F,OAAAA,CAAQhD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMiD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAK7G,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAAS6G,MAAAA,CAAO,IAAA,CAAKvG,YAAY,CAAA;AAE5C,IAAA,OAAO,CAAC,IAAA,CAAKP,KAAAA,CAAMa,MAAAA,EAAQ;AACzB,MAAA,IAAA,CAAKwB,kBAAAA,EAAkB;AAEvB,MAAA,MAAMG,OAAAA,GAAU,MAAM,IAAA,CAAKwC,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAKtE,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEhH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKoE,YAAAA,CAAazE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEhH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd;AAEA,EAAA,MAAgBiH,aAAazE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,IAAIK,QAAAA,CAASqE,YAAAA,EAAY,CAAGC,MAAAA,KAAW,CAAA,EAAG;AACxC,MAAA,MAAMC,YAAAA,GAAevE,SAASwE,eAAAA,EAAe;AAC7C,MAAA,MAAMC,IAAAA,GAAOF,aAAa5B,GAAAA,CAAI,CAAC+B,MAAMA,CAAAA,CAAED,IAAI,CAAA,CAAEE,IAAAA,CAAK,IAAA,CAAA;AAElD,MAAA,MAAMC,mBAAAA,GACJH,QAAQ9E,OAAAA,CAAQW,OAAAA,GAAU,MAAM,IAAA,CAAKe,yBAAAA,CAA0BoD,IAAAA,CAAAA,GAAQ,IAAA;AACzE,MAAA,IAAIG,mBAAAA,EAAqB;AACvB,QAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKhF,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AAC/D,QAAA,MAAMgG,MAAAA,CAAOC,GAAAA,CAAI,IAAIpE,eAAAA,CAAgB;AAACkE,UAAAA;SAAoB,CAAA,CAAA;OAC5D,MAAO;AACL,QAAA,MAAMxE,GAAAA,GAAM,IAAIjB,UAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsBwG,IAAIzD,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAK9C,kBAAAA,CAAmBuG,IAAIzD,GAAAA,CAAAA;AAE5B,QAAA,IAAI,CAACT,QAAQW,OAAAA,EAAS;AACpB,UAAA,OAAO,MAAM,IAAA,CAAK8D,YAAAA,CAAazE,OAAAA,CAAAA;AACjC;AAEA,QAAA,MAAM,IAAA,CAAKvC,QAAAA,CAAS6G,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMc,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;YACE6C,MAAAA,EAAQ,IAAA,CAAK5H,SAASyB,WAAAA,CAAYmD,IAAAA;YAClCiD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAKtH,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAKsG,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AAEA/E,MAAAA,QAAAA,CAASe,SAASuD,MAAAA,GAAS,CAAA;AAC3BtE,MAAAA,QAAAA,CAASe,QAAAA,CAASgC,KAAK6B,mBAAAA,CAAAA;AACzB;AAGA,IAAA,MAAMtC,SAAAA,GAAYtC,SAASqE,YAAAA,EAAY;AACvC,IAAA,KAAA,MAAWgB,eAAe/C,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAK1E,oBAAAA,CAAqB0H,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKzH,qBAAqB2H,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAK3H,qBAAqB4H,KAAAA,EAAK;AAC/B,QAAA,MAAMT,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;AACE6C,YAAAA,MAAAA,EAAQK,WAAAA,CAAYtD,QAAAA;YACpBkD,OAAAA,EAAS,KAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,IAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,OAAO,MAAM,IAAA,CAAKhB,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AACF;AAEA,IAAA,MAAMxC,cAAc,MAAM,IAAA,CAAKF,eAAAA,CAAgB1C,OAAAA,CAAQuB,cAAcoB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAKnF,KAAAA,CAAMe,MAAAA,CAAO6F,OAAAA,CAAQ;SAAI/D,QAAAA,CAASe,QAAAA;AAAawB,MAAAA,GAAAA;AAAY,KAAA,CAAA;AAGtE,IAAA,MAAMkD,YAAAA,GAAe,IAAA,CAAKtI,KAAAA,CAAMe,MAAAA,CAAO6C,QAAAA,CAAS2E,MAAAA,CAAO,CAAC9C,GAAAA,KAAQA,GAAAA,CAAI+C,IAAAA,CAAK1I,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAO0H,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOzF,QAAAA;AACT;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAI2F,wBAAAA,CAAyB;MAC9Cb,MAAAA,EAAQnG,WAAAA;MACRiH,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACA9F,IAAAA,UAAAA,CAAWgE,OAAAA,CAAQ+B,EAAAA,CAAG,QAAA,EAAU,OAAOxF,MAAMyF,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAKrI,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzChH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZ8F,QAAAA,MAAAA,EAAQxC,IAAAA,CAAKwC,MAAAA;AACbkD,QAAAA,KAAAA,EAAO1F,IAAAA,CAAK0F,KAAAA;QACZC,gBAAAA,EAAkBxH;OACpB,CAAA;KACF,CAAA;AACA,IAAA,OAAOsB,UAAAA;AACT;AACF","file":"runner.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { v4 as uuidv4 } from \"uuid\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { ChatModel, ChatModelInput, ChatModelOutput } from \"@/backend/chat.js\";\nimport { AssistantMessage, Message, ToolMessage } from \"@/backend/message.js\";\nimport { ToolCallPart } from \"ai\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { RunContext } from \"@/context.js\";\nimport {\n  RequirementAgentExecutionConfig,\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport { Requirement, Rule } from \"./requirements/requirement.js\";\nimport { RequirementsReasoner, createSystemMessage } from \"./utils/llm.js\";\nimport { FinalAnswerTool, runTool } from \"./utils/tool.js\";\nimport { ToolCallChecker } from \"./utils/toolCallChecker.js\";\nimport { z } from \"zod\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { RequirementAgent } from \"@/agents/requirement/agent.js\";\nimport { mergeTokenUsage } from \"@/adapters/vercel/backend/utils.js\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\nconst TEMP_MESSAGE_KEY = \"tempMessage\";\n\n/**\n * Runner for RequirementAgent execution\n */\nexport class RequirementAgentRunner {\n  protected readonly state: RequirementAgentRunState;\n  protected readonly reasoner: RequirementsReasoner;\n  protected readonly iterationErrorCounter: RetryCounter;\n  protected readonly globalErrorCounter: RetryCounter;\n\n  constructor(\n    protected readonly llm: ChatModel,\n    protected readonly runConfig: RequirementAgentExecutionConfig,\n    tools: AnyTool[],\n    protected readonly requirements: Requirement[],\n    expectedOutput: string | z.ZodSchema | null,\n    protected readonly toolCallCycleChecker: ToolCallChecker,\n    protected readonly ctx: RunContext<RequirementAgent>,\n    protected forceFinalAnswerAsTool: boolean,\n    protected readonly templates: RequirementAgentTemplates,\n  ) {\n    this.state = {\n      answer: null,\n      result: null,\n      memory: new UnconstrainedMemory(),\n      steps: [],\n      iteration: 0,\n      usage: {\n        totalTokens: 0,\n        promptTokens: 0,\n        completionTokens: 0,\n        reasoningTokens: 0,\n        cachedPromptTokens: undefined,\n      },\n    };\n\n    this.requirements = requirements;\n    const finalAnswer = new FinalAnswerTool(expectedOutput, this.state);\n    this.reasoner = new RequirementsReasoner(tools, finalAnswer, ctx);\n\n    const maxRetriesPerIteration = runConfig.maxRetriesPerStep ?? 0;\n    this.iterationErrorCounter = new RetryCounter(maxRetriesPerIteration, AgentError);\n\n    const maxRetries = Math.max(maxRetriesPerIteration, runConfig.totalMaxRetries ?? 0);\n    this.globalErrorCounter = new RetryCounter(maxRetries, AgentError);\n  }\n\n  protected incrementIteration(): void {\n    this.state.iteration++;\n\n    if (this.runConfig.maxIterations && this.state.iteration > this.runConfig.maxIterations) {\n      throw new AgentError(\n        `Agent was not able to resolve the task in ${this.state.iteration} iterations.`,\n      );\n    }\n  }\n\n  protected async runLLM(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm\n        .create(input)\n        .middleware(streamMiddleware)\n        .catch((err) => {\n          if (err instanceof ChatModelToolCallError && request.canStop) {\n            const { generatedContent, response: errorResponse } = err.data;\n            if (generatedContent) {\n              return new ChatModelOutput(\n                [new AssistantMessage(generatedContent)],\n                errorResponse?.usage,\n                errorResponse?.finishReason,\n              );\n            }\n          }\n          throw err;\n        });\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n      maxRetries: this.runConfig.maxRetriesPerStep ?? 0,\n    };\n  }\n\n  protected async createFinalAnswerToolCall(fullText: string): Promise<AssistantMessage | null> {\n    // Try to extract JSON from text\n    const parsed = parseBrokenJson(fullText, { pair: [\"{\", \"}\"] });\n    if (!parsed) {\n      // If no JSON and no custom schema, wrap in default schema\n      if (!this.reasoner.finalAnswer.customSchema) {\n        return new AssistantMessage([\n          {\n            type: \"tool-call\",\n            toolCallId: `call_${uuidv4().substring(0, 8)}`,\n            toolName: this.reasoner.finalAnswer.name,\n            input: this.reasoner.finalAnswer.inputSchema().parse({ response: fullText }),\n          },\n        ]);\n      }\n      return null;\n    }\n\n    return new AssistantMessage([\n      {\n        type: \"tool-call\",\n        toolCallId: `call_${uuidv4().substring(0, 8)}`,\n        toolName: this.reasoner.finalAnswer.name,\n        input: parsed,\n      },\n    ]);\n  }\n\n  protected async createRequest(extraRules: Rule[] = []): Promise<RequirementAgentRequest> {\n    return await this.reasoner.createRequest(this.state, this.forceFinalAnswerAsTool, extraRules);\n  }\n\n  protected async invokeToolCalls(\n    tools: AnyTool[],\n    toolCalls: ToolCallPart[],\n  ): Promise<ToolMessage[]> {\n    const toolResults: ToolMessage[] = [];\n\n    const results = await Promise.all(\n      toolCalls.map((msg) => runTool(tools, msg, { state: this.state })),\n    );\n\n    for (const toolCall of results) {\n      this.state.steps.push({\n        id: uuidv4(),\n        iteration: this.state.iteration,\n        input: toolCall.input,\n        output: toolCall.output,\n        tool: toolCall.tool,\n        error: toolCall.error,\n      });\n\n      let result: string;\n      if (toolCall.error) {\n        result = this.templates.toolError.render({\n          reason: toolCall.error.explain(),\n        });\n      } else {\n        result = !toolCall.output.isEmpty()\n          ? toolCall.output.getTextContent()\n          : this.templates.toolNoResult.render({\n              toolCall: { tool: toolCall.tool, input: toolCall.input },\n            });\n      }\n      toolResults.push(\n        new ToolMessage({\n          type: \"tool-result\",\n          toolName: toolCall.tool?.name || toolCall.msg.toolName,\n          toolCallId: toolCall.msg.toolCallId,\n          output: { type: \"text\", value: result },\n        }),\n      );\n\n      if (toolCall.error) {\n        this.iterationErrorCounter.use(toolCall.error);\n        this.globalErrorCounter.use(toolCall.error);\n      }\n    }\n\n    return toolResults;\n  }\n\n  async addMessages(messages: Message[]): Promise<void> {\n    await this.state.memory.addMany(messages);\n  }\n\n  async run(): Promise<RequirementAgentRunState> {\n    if (this.state.answer) {\n      return this.state;\n    }\n\n    // Initialize requirements\n    await this.reasoner.update(this.requirements);\n\n    while (!this.state.answer) {\n      this.incrementIteration();\n\n      const request = await this.createRequest();\n      await this.ctx.emitter.emit(\"start\", { state: this.state, request });\n      const response = await this.runIteration(request);\n      await this.ctx.emitter.emit(\"success\", { state: this.state, response });\n    }\n\n    return this.state;\n  }\n\n  protected async runIteration(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const response = await this.runLLM(request);\n\n    // Try to cast text message to final answer tool call if allowed\n    if (response.getToolCalls().length === 0) {\n      const textMessages = response.getTextMessages();\n      const text = textMessages.map((m) => m.text).join(\"\\n\");\n\n      const finalAnswerToolCall =\n        text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;\n      if (finalAnswerToolCall) {\n        const stream = this.createFinalAnswerStream(request.finalAnswer);\n        await stream.add(new ChatModelOutput([finalAnswerToolCall]));\n      } else {\n        const err = new AgentError(\"Model produced an invalid final answer tool call.\");\n        this.iterationErrorCounter.use(err);\n        this.globalErrorCounter.use(err);\n\n        if (!request.canStop) {\n          return await this.runIteration(request);\n        }\n\n        await this.reasoner.update([]);\n        const updatedRequest = await this.createRequest([\n          {\n            target: this.reasoner.finalAnswer.name,\n            allowed: true,\n            hidden: false,\n            forced: false,\n            preventStop: false,\n          },\n        ]);\n        this.forceFinalAnswerAsTool = true;\n        return await this.runIteration(updatedRequest);\n      }\n\n      response.messages.length = 0;\n      response.messages.push(finalAnswerToolCall);\n    }\n\n    // Check for cycles\n    const toolCalls = response.getToolCalls();\n    for (const toolCallMsg of toolCalls) {\n      this.toolCallCycleChecker.register(toolCallMsg);\n      if (this.toolCallCycleChecker.cycleFound) {\n        this.toolCallCycleChecker.reset();\n        const updatedRequest = await this.createRequest([\n          {\n            target: toolCallMsg.toolName,\n            allowed: false,\n            hidden: false,\n            forced: true,\n            preventStop: false,\n          },\n        ]);\n        return await this.runIteration(updatedRequest);\n      }\n    }\n\n    const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);\n    await this.state.memory.addMany([...response.messages, ...toolResults]);\n\n    // Delete temporary messages\n    const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);\n    await this.state.memory.deleteMany(tempMessages);\n\n    return response;\n  }\n\n  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/agents/requirement/runner.ts"],"names":["TEMP_MESSAGE_KEY","RequirementAgentRunner","state","reasoner","iterationErrorCounter","globalErrorCounter","llm","runConfig","tools","requirements","expectedOutput","toolCallCycleChecker","ctx","forceFinalAnswerAsTool","templates","answer","result","memory","UnconstrainedMemory","steps","iteration","usage","totalTokens","promptTokens","completionTokens","reasoningTokens","cachedPromptTokens","undefined","finalAnswer","FinalAnswerTool","RequirementsReasoner","maxRetriesPerIteration","maxRetriesPerStep","RetryCounter","AgentError","maxRetries","Math","max","totalMaxRetries","incrementIteration","maxIterations","runLLM","request","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","catch","err","ChatModelToolCallError","canStop","generatedContent","errorResponse","data","ChatModelOutput","AssistantMessage","finishReason","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","type","toolCallId","uuidv4","substring","toolName","name","inputSchema","parse","createRequest","extraRules","invokeToolCalls","toolCalls","toolResults","results","Promise","all","map","msg","runTool","toolCall","push","id","output","tool","error","toolError","render","reason","explain","isEmpty","getTextContent","toolNoResult","ToolMessage","value","use","addMessages","addMany","run","update","emitter","emit","runIteration","getToolCalls","length","textMessages","getTextMessages","text","m","join","finalAnswerToolCall","stream","add","updatedRequest","target","allowed","hidden","forced","preventStop","toolCallMsg","register","cycleFound","reset","tempMessages","filter","meta","deleteMany","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;;;AA+BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EApCb;;;;;;;;;;AAqCqBC,EAAAA,KAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,qBAAAA;AACAC,EAAAA,kBAAAA;EAEnB,WAAA,CACqBC,GAAAA,EACAC,WACnBC,KAAAA,EACmBC,YAAAA,EACnBC,gBACmBC,oBAAAA,EACAC,GAAAA,EACTC,wBACSC,SAAAA,EACnB;SATmBR,GAAAA,GAAAA,GAAAA;SACAC,SAAAA,GAAAA,SAAAA;SAEAE,YAAAA,GAAAA,YAAAA;SAEAE,oBAAAA,GAAAA,oBAAAA;SACAC,GAAAA,GAAAA,GAAAA;SACTC,sBAAAA,GAAAA,sBAAAA;SACSC,SAAAA,GAAAA,SAAAA;AAEnB,IAAA,IAAA,CAAKZ,KAAAA,GAAQ;MACXa,MAAAA,EAAQ,IAAA;MACRC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,MAAAA,EAAQ,IAAIC,mBAAAA,EAAAA;AACZC,MAAAA,KAAAA,EAAO,EAAA;MACPC,SAAAA,EAAW,CAAA;MACXC,KAAAA,EAAO;QACLC,WAAAA,EAAa,CAAA;QACbC,YAAAA,EAAc,CAAA;QACdC,gBAAAA,EAAkB,CAAA;QAClBC,eAAAA,EAAiB,CAAA;QACjBC,kBAAAA,EAAoBC;AACtB;AACF,KAAA;AAEA,IAAA,IAAA,CAAKlB,YAAAA,GAAeA,YAAAA;AACpB,IAAA,MAAMmB,WAAAA,GAAc,IAAIC,eAAAA,CAAgBnB,cAAAA,EAAgB,KAAKR,KAAK,CAAA;AAClE,IAAA,IAAA,CAAKC,QAAAA,GAAW,IAAI2B,oBAAAA,CAAqBtB,KAAAA,EAAOoB,aAAahB,GAAAA,CAAAA;AAE7D,IAAA,MAAMmB,sBAAAA,GAAyBxB,UAAUyB,iBAAAA,IAAqB,CAAA;AAC9D,IAAA,IAAA,CAAK5B,qBAAAA,GAAwB,IAAI6B,YAAAA,CAAaF,sBAAAA,EAAwBG,UAAAA,CAAAA;AAEtE,IAAA,MAAMC,aAAaC,IAAAA,CAAKC,GAAAA,CAAIN,sBAAAA,EAAwBxB,SAAAA,CAAU+B,mBAAmB,CAAA,CAAA;AACjF,IAAA,IAAA,CAAKjC,kBAAAA,GAAqB,IAAI4B,YAAAA,CAAaE,UAAAA,EAAYD,UAAAA,CAAAA;AACzD,EAAA;EAEUK,kBAAAA,GAA2B;AACnC,IAAA,IAAA,CAAKrC,KAAAA,CAAMkB,SAAAA,EAAAA;AAEX,IAAA,IAAI,IAAA,CAAKb,UAAUiC,aAAAA,IAAiB,IAAA,CAAKtC,MAAMkB,SAAAA,GAAY,IAAA,CAAKb,UAAUiC,aAAAA,EAAe;AACvF,MAAA,MAAM,IAAIN,UAAAA,CACR,CAAA,0CAAA,EAA6C,IAAA,CAAKhC,KAAAA,CAAMkB,SAAS,CAAA,YAAA,CAAc,CAAA;AAEnF,IAAA;AACF,EAAA;AAEA,EAAA,MAAgBqB,OAAOC,OAAAA,EAA4D;AACjF,IAAA,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CACzB0C,MAAAA,CAAOH,KAAAA,CAAAA,CACPI,UAAAA,CAAWN,gBAAAA,CAAAA,CACXO,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,IAAIA,GAAAA,YAAeC,sBAAAA,IAA0BV,OAAAA,CAAQW,OAAAA,EAAS;AAC5D,UAAA,MAAM,EAAEC,gBAAAA,EAAkBP,QAAAA,EAAUQ,aAAAA,KAAkBJ,GAAAA,CAAIK,IAAAA;AAC1D,UAAA,IAAIF,gBAAAA,EAAkB;AACpB,YAAA,OAAO,IAAIG,eAAAA,CACT;AAAC,cAAA,IAAIC,iBAAiBJ,gBAAAA;eACtBC,aAAAA,EAAelC,KAAAA,EACfkC,eAAeI,YAAAA,CAAAA;AAEnB,UAAA;AACF,QAAA;AACA,QAAA,MAAMR,GAAAA;MACR,CAAA,CAAA;AAEF,MAAA,IAAIJ,SAAS1B,KAAAA,EAAO;AAClBuC,QAAAA,eAAAA,CAAgB,IAAA,CAAK1D,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD,MAAA;AAEA,MAAA,OAAO0B,QAAAA;IACT,CAAA,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBkB,MAAAA,EAAM;AACzB,IAAA;AACF,EAAA;AAEA,EAAA,MAAgBf,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMoB,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,mBAAAA,CAAoB,IAAA,CAAKjD,SAAAA,CAAUkD,MAAAA,EAAQtB,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAO6C;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACAtD,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQuB,YAAAA;AACfC,MAAAA,UAAAA,EAAYxB,OAAAA,CAAQwB,UAAAA;MACpBC,sBAAAA,EAAwB,IAAA;MACxBhC,UAAAA,EAAY,IAAA,CAAK5B,UAAUyB,iBAAAA,IAAqB;AAClD,KAAA;AACF,EAAA;AAEA,EAAA,MAAgBoC,0BAA0BC,QAAAA,EAAoD;AAE5F,IAAA,MAAMC,MAAAA,GAASC,gBAAgBF,QAAAA,EAAU;MAAEG,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAC5D,IAAA,IAAI,CAACF,MAAAA,EAAQ;AAEX,MAAA,IAAI,CAAC,IAAA,CAAKnE,QAAAA,CAASyB,WAAAA,CAAY6C,YAAAA,EAAc;AAC3C,QAAA,OAAO,IAAIf,gBAAAA,CAAiB;AAC1B,UAAA;YACEgB,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;YAC1CC,QAAAA,EAAU,IAAA,CAAK3E,SAASyB,WAAAA,CAAYmD,IAAAA;AACpClC,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAYoD,WAAAA,GAAcC,KAAAA,CAAM;cAAElC,QAAAA,EAAUsB;aAAS;AAC5E;AACD,SAAA,CAAA;AACH,MAAA;AACA,MAAA,OAAO,IAAA;AACT,IAAA;AAEA,IAAA,OAAO,IAAIX,gBAAAA,CAAiB;AAC1B,MAAA;QACEgB,IAAAA,EAAM,WAAA;AACNC,QAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;QAC1CC,QAAAA,EAAU,IAAA,CAAK3E,SAASyB,WAAAA,CAAYmD,IAAAA;QACpClC,KAAAA,EAAOyB;AACT;AACD,KAAA,CAAA;AACH,EAAA;EAEA,MAAgBY,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKhF,QAAAA,CAAS+E,aAAAA,CAAc,KAAKhF,KAAAA,EAAO,IAAA,CAAKW,wBAAwBsE,UAAAA,CAAAA;AACpF,EAAA;EAEA,MAAgBC,eAAAA,CACd5E,OACA6E,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,OAAAA,CAAQpF,KAAAA,EAAOmF,GAAAA,EAAK;AAAEzF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAW2F,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAKrF,KAAAA,CAAMiB,MAAM2E,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,EAAAA,EAAAA;AACJxD,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOgD,QAAAA,CAAShD,KAAAA;AAChBmD,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIlF,MAAAA;AACJ,MAAA,IAAI6E,SAASK,KAAAA,EAAO;AAClBlF,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAUqF,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;MACF,CAAA,MAAO;AACLtF,QAAAA,MAAAA,GAAS,CAAC6E,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAK1F,SAAAA,CAAU2F,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAMpD,YAAAA,KAAAA,EAAOgD,QAAAA,CAAShD;AAAM;SACzD,CAAA;AACN,MAAA;AACAyC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,WAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAO3F;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAI6E,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAK9F,qBAAAA,CAAsBwG,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK7F,kBAAAA,CAAmBuG,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C,MAAA;AACF,IAAA;AAEA,IAAA,OAAOZ,WAAAA;AACT,EAAA;AAEA,EAAA,MAAMuB,YAAY/C,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAK5D,KAAAA,CAAMe,MAAAA,CAAO6F,OAAAA,CAAQhD,QAAAA,CAAAA;AAClC,EAAA;AAEA,EAAA,MAAMiD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAK7G,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd,IAAA;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAAS6G,MAAAA,CAAO,IAAA,CAAKvG,YAAY,CAAA;AAE5C,IAAA,OAAO,CAAC,IAAA,CAAKP,KAAAA,CAAMa,MAAAA,EAAQ;AACzB,MAAA,IAAA,CAAKwB,kBAAAA,EAAkB;AAEvB,MAAA,MAAMG,OAAAA,GAAU,MAAM,IAAA,CAAKwC,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAKtE,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEhH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKoE,YAAAA,CAAazE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEhH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE,IAAA;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd,EAAA;AAEA,EAAA,MAAgBiH,aAAazE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,IAAIK,QAAAA,CAASqE,YAAAA,EAAY,CAAGC,MAAAA,KAAW,CAAA,EAAG;AACxC,MAAA,MAAMC,YAAAA,GAAevE,SAASwE,eAAAA,EAAe;AAC7C,MAAA,MAAMC,IAAAA,GAAOF,aAAa5B,GAAAA,CAAI,CAAC+B,MAAMA,CAAAA,CAAED,IAAI,CAAA,CAAEE,IAAAA,CAAK,IAAA,CAAA;AAElD,MAAA,MAAMC,mBAAAA,GACJH,QAAQ9E,OAAAA,CAAQW,OAAAA,GAAU,MAAM,IAAA,CAAKe,yBAAAA,CAA0BoD,IAAAA,CAAAA,GAAQ,IAAA;AACzE,MAAA,IAAIG,mBAAAA,EAAqB;AACvB,QAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKhF,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AAC/D,QAAA,MAAMgG,MAAAA,CAAOC,GAAAA,CAAI,IAAIpE,eAAAA,CAAgB;AAACkE,UAAAA;SAAoB,CAAA,CAAA;MAC5D,CAAA,MAAO;AACL,QAAA,MAAMxE,GAAAA,GAAM,IAAIjB,UAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsBwG,IAAIzD,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAK9C,kBAAAA,CAAmBuG,IAAIzD,GAAAA,CAAAA;AAE5B,QAAA,IAAI,CAACT,QAAQW,OAAAA,EAAS;AACpB,UAAA,OAAO,MAAM,IAAA,CAAK8D,YAAAA,CAAazE,OAAAA,CAAAA;AACjC,QAAA;AAEA,QAAA,MAAM,IAAA,CAAKvC,QAAAA,CAAS6G,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMc,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;YACE6C,MAAAA,EAAQ,IAAA,CAAK5H,SAASyB,WAAAA,CAAYmD,IAAAA;YAClCiD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAKtH,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAKsG,YAAAA,CAAaW,cAAAA,CAAAA;AACjC,MAAA;AAEA/E,MAAAA,QAAAA,CAASe,SAASuD,MAAAA,GAAS,CAAA;AAC3BtE,MAAAA,QAAAA,CAASe,QAAAA,CAASgC,KAAK6B,mBAAAA,CAAAA;AACzB,IAAA;AAGA,IAAA,MAAMtC,SAAAA,GAAYtC,SAASqE,YAAAA,EAAY;AACvC,IAAA,KAAA,MAAWgB,eAAe/C,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAK1E,oBAAAA,CAAqB0H,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKzH,qBAAqB2H,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAK3H,qBAAqB4H,KAAAA,EAAK;AAC/B,QAAA,MAAMT,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;AACE6C,YAAAA,MAAAA,EAAQK,WAAAA,CAAYtD,QAAAA;YACpBkD,OAAAA,EAAS,KAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,IAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,OAAO,MAAM,IAAA,CAAKhB,YAAAA,CAAaW,cAAAA,CAAAA;AACjC,MAAA;AACF,IAAA;AAEA,IAAA,MAAMxC,cAAc,MAAM,IAAA,CAAKF,eAAAA,CAAgB1C,OAAAA,CAAQuB,cAAcoB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAKnF,KAAAA,CAAMe,MAAAA,CAAO6F,OAAAA,CAAQ;SAAI/D,QAAAA,CAASe,QAAAA;AAAawB,MAAAA,GAAAA;AAAY,KAAA,CAAA;AAGtE,IAAA,MAAMkD,YAAAA,GAAe,IAAA,CAAKtI,KAAAA,CAAMe,MAAAA,CAAO6C,QAAAA,CAAS2E,MAAAA,CAAO,CAAC9C,GAAAA,KAAQA,GAAAA,CAAI+C,IAAAA,CAAK1I,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAO0H,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOzF,QAAAA;AACT,EAAA;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAI2F,wBAAAA,CAAyB;MAC9Cb,MAAAA,EAAQnG,WAAAA;MACRiH,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACA9F,IAAAA,UAAAA,CAAWgE,OAAAA,CAAQ+B,EAAAA,CAAG,QAAA,EAAU,OAAOxF,MAAMyF,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAKrI,GAAAA,CAAIqG,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzChH,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZ8F,QAAAA,MAAAA,EAAQxC,IAAAA,CAAKwC,MAAAA;AACbkD,QAAAA,KAAAA,EAAO1F,IAAAA,CAAK0F,KAAAA;QACZC,gBAAAA,EAAkBxH;OACpB,CAAA;IACF,CAAA,CAAA;AACA,IAAA,OAAOsB,UAAAA;AACT,EAAA;AACF","file":"runner.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { v4 as uuidv4 } from \"uuid\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { ChatModel, ChatModelInput, ChatModelOutput } from \"@/backend/chat.js\";\nimport { AssistantMessage, Message, ToolMessage } from \"@/backend/message.js\";\nimport { ToolCallPart } from \"ai\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { RunContext } from \"@/context.js\";\nimport {\n  RequirementAgentExecutionConfig,\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport { Requirement, Rule } from \"./requirements/requirement.js\";\nimport { RequirementsReasoner, createSystemMessage } from \"./utils/llm.js\";\nimport { FinalAnswerTool, runTool } from \"./utils/tool.js\";\nimport { ToolCallChecker } from \"./utils/toolCallChecker.js\";\nimport { z } from \"zod\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { RequirementAgent } from \"@/agents/requirement/agent.js\";\nimport { mergeTokenUsage } from \"@/adapters/vercel/backend/utils.js\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\nconst TEMP_MESSAGE_KEY = \"tempMessage\";\n\n/**\n * Runner for RequirementAgent execution\n */\nexport class RequirementAgentRunner {\n  protected readonly state: RequirementAgentRunState;\n  protected readonly reasoner: RequirementsReasoner;\n  protected readonly iterationErrorCounter: RetryCounter;\n  protected readonly globalErrorCounter: RetryCounter;\n\n  constructor(\n    protected readonly llm: ChatModel,\n    protected readonly runConfig: RequirementAgentExecutionConfig,\n    tools: AnyTool[],\n    protected readonly requirements: Requirement[],\n    expectedOutput: string | z.ZodSchema | null,\n    protected readonly toolCallCycleChecker: ToolCallChecker,\n    protected readonly ctx: RunContext<RequirementAgent>,\n    protected forceFinalAnswerAsTool: boolean,\n    protected readonly templates: RequirementAgentTemplates,\n  ) {\n    this.state = {\n      answer: null,\n      result: null,\n      memory: new UnconstrainedMemory(),\n      steps: [],\n      iteration: 0,\n      usage: {\n        totalTokens: 0,\n        promptTokens: 0,\n        completionTokens: 0,\n        reasoningTokens: 0,\n        cachedPromptTokens: undefined,\n      },\n    };\n\n    this.requirements = requirements;\n    const finalAnswer = new FinalAnswerTool(expectedOutput, this.state);\n    this.reasoner = new RequirementsReasoner(tools, finalAnswer, ctx);\n\n    const maxRetriesPerIteration = runConfig.maxRetriesPerStep ?? 0;\n    this.iterationErrorCounter = new RetryCounter(maxRetriesPerIteration, AgentError);\n\n    const maxRetries = Math.max(maxRetriesPerIteration, runConfig.totalMaxRetries ?? 0);\n    this.globalErrorCounter = new RetryCounter(maxRetries, AgentError);\n  }\n\n  protected incrementIteration(): void {\n    this.state.iteration++;\n\n    if (this.runConfig.maxIterations && this.state.iteration > this.runConfig.maxIterations) {\n      throw new AgentError(\n        `Agent was not able to resolve the task in ${this.state.iteration} iterations.`,\n      );\n    }\n  }\n\n  protected async runLLM(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm\n        .create(input)\n        .middleware(streamMiddleware)\n        .catch((err) => {\n          if (err instanceof ChatModelToolCallError && request.canStop) {\n            const { generatedContent, response: errorResponse } = err.data;\n            if (generatedContent) {\n              return new ChatModelOutput(\n                [new AssistantMessage(generatedContent)],\n                errorResponse?.usage,\n                errorResponse?.finishReason,\n              );\n            }\n          }\n          throw err;\n        });\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n      maxRetries: this.runConfig.maxRetriesPerStep ?? 0,\n    };\n  }\n\n  protected async createFinalAnswerToolCall(fullText: string): Promise<AssistantMessage | null> {\n    // Try to extract JSON from text\n    const parsed = parseBrokenJson(fullText, { pair: [\"{\", \"}\"] });\n    if (!parsed) {\n      // If no JSON and no custom schema, wrap in default schema\n      if (!this.reasoner.finalAnswer.customSchema) {\n        return new AssistantMessage([\n          {\n            type: \"tool-call\",\n            toolCallId: `call_${uuidv4().substring(0, 8)}`,\n            toolName: this.reasoner.finalAnswer.name,\n            input: this.reasoner.finalAnswer.inputSchema().parse({ response: fullText }),\n          },\n        ]);\n      }\n      return null;\n    }\n\n    return new AssistantMessage([\n      {\n        type: \"tool-call\",\n        toolCallId: `call_${uuidv4().substring(0, 8)}`,\n        toolName: this.reasoner.finalAnswer.name,\n        input: parsed,\n      },\n    ]);\n  }\n\n  protected async createRequest(extraRules: Rule[] = []): Promise<RequirementAgentRequest> {\n    return await this.reasoner.createRequest(this.state, this.forceFinalAnswerAsTool, extraRules);\n  }\n\n  protected async invokeToolCalls(\n    tools: AnyTool[],\n    toolCalls: ToolCallPart[],\n  ): Promise<ToolMessage[]> {\n    const toolResults: ToolMessage[] = [];\n\n    const results = await Promise.all(\n      toolCalls.map((msg) => runTool(tools, msg, { state: this.state })),\n    );\n\n    for (const toolCall of results) {\n      this.state.steps.push({\n        id: uuidv4(),\n        iteration: this.state.iteration,\n        input: toolCall.input,\n        output: toolCall.output,\n        tool: toolCall.tool,\n        error: toolCall.error,\n      });\n\n      let result: string;\n      if (toolCall.error) {\n        result = this.templates.toolError.render({\n          reason: toolCall.error.explain(),\n        });\n      } else {\n        result = !toolCall.output.isEmpty()\n          ? toolCall.output.getTextContent()\n          : this.templates.toolNoResult.render({\n              toolCall: { tool: toolCall.tool, input: toolCall.input },\n            });\n      }\n      toolResults.push(\n        new ToolMessage({\n          type: \"tool-result\",\n          toolName: toolCall.tool?.name || toolCall.msg.toolName,\n          toolCallId: toolCall.msg.toolCallId,\n          output: { type: \"text\", value: result },\n        }),\n      );\n\n      if (toolCall.error) {\n        this.iterationErrorCounter.use(toolCall.error);\n        this.globalErrorCounter.use(toolCall.error);\n      }\n    }\n\n    return toolResults;\n  }\n\n  async addMessages(messages: Message[]): Promise<void> {\n    await this.state.memory.addMany(messages);\n  }\n\n  async run(): Promise<RequirementAgentRunState> {\n    if (this.state.answer) {\n      return this.state;\n    }\n\n    // Initialize requirements\n    await this.reasoner.update(this.requirements);\n\n    while (!this.state.answer) {\n      this.incrementIteration();\n\n      const request = await this.createRequest();\n      await this.ctx.emitter.emit(\"start\", { state: this.state, request });\n      const response = await this.runIteration(request);\n      await this.ctx.emitter.emit(\"success\", { state: this.state, response });\n    }\n\n    return this.state;\n  }\n\n  protected async runIteration(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const response = await this.runLLM(request);\n\n    // Try to cast text message to final answer tool call if allowed\n    if (response.getToolCalls().length === 0) {\n      const textMessages = response.getTextMessages();\n      const text = textMessages.map((m) => m.text).join(\"\\n\");\n\n      const finalAnswerToolCall =\n        text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;\n      if (finalAnswerToolCall) {\n        const stream = this.createFinalAnswerStream(request.finalAnswer);\n        await stream.add(new ChatModelOutput([finalAnswerToolCall]));\n      } else {\n        const err = new AgentError(\"Model produced an invalid final answer tool call.\");\n        this.iterationErrorCounter.use(err);\n        this.globalErrorCounter.use(err);\n\n        if (!request.canStop) {\n          return await this.runIteration(request);\n        }\n\n        await this.reasoner.update([]);\n        const updatedRequest = await this.createRequest([\n          {\n            target: this.reasoner.finalAnswer.name,\n            allowed: true,\n            hidden: false,\n            forced: false,\n            preventStop: false,\n          },\n        ]);\n        this.forceFinalAnswerAsTool = true;\n        return await this.runIteration(updatedRequest);\n      }\n\n      response.messages.length = 0;\n      response.messages.push(finalAnswerToolCall);\n    }\n\n    // Check for cycles\n    const toolCalls = response.getToolCalls();\n    for (const toolCallMsg of toolCalls) {\n      this.toolCallCycleChecker.register(toolCallMsg);\n      if (this.toolCallCycleChecker.cycleFound) {\n        this.toolCallCycleChecker.reset();\n        const updatedRequest = await this.createRequest([\n          {\n            target: toolCallMsg.toolName,\n            allowed: false,\n            hidden: false,\n            forced: true,\n            preventStop: false,\n          },\n        ]);\n        return await this.runIteration(updatedRequest);\n      }\n    }\n\n    const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);\n    await this.state.memory.addMany([...response.messages, ...toolResults]);\n\n    // Delete temporary messages\n    const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);\n    await this.state.memory.deleteMany(tempMessages);\n\n    return response;\n  }\n\n  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}

package/dist/agents/requirement/types.d.cts

@@ -3,17 +3,17 @@ import '../../backend/message.cjs';
 import '../../tools/base.cjs';
 import '../../errors.cjs';
 import '../../template.cjs';
-import '../../chat-BFl85cqt.cjs';
-import '../../base-XfWeXQ-S.cjs';
+import '../../chat-C20nYj46.cjs';
+import '../../base-s3ioVIE9.cjs';
 import 'zod';
-export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-CkEtV8R9.cjs';
+export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, i as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, j as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, k as RequirementAgentToolErrorPromptInputSchema, l as RequirementAgentToolNoResultPromptInputSchema } from '../../types-BNsa3W7Z.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import 'ai';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-jN55XZZq.cjs';
+import '../../emitter-D4OcelZ9.cjs';
 import '../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../cache/base.cjs';

package/dist/agents/requirement/types.d.ts

@@ -3,17 +3,17 @@ import '../../backend/message.js';
 import '../../tools/base.js';
 import '../../errors.js';
 import '../../template.js';
-import '../../chat-BRuyDeKR.js';
-import '../../base-CyFx7BRL.js';
+import '../../chat-CfTNfR60.js';
+import '../../base-BjzyQ8k6.js';
 import 'zod';
-export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-BQ-3LdkE.js';
+export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, i as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, j as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, k as RequirementAgentToolErrorPromptInputSchema, l as RequirementAgentToolNoResultPromptInputSchema } from '../../types-DYtzF6ya.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import 'ai';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-36-9MnvA.js';
+import '../../emitter-kHxkUxco.js';
 import '../../internals/helpers/promise.js';
 import 'promise-based-task';
 import '../../cache/base.js';

package/dist/agents/requirement/utils/llm.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,kBAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC;AACF;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F;AACA,IAAA,OAAOL,IAAAA;AACT;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B;AACF;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC;AACF;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB;AACF;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd;AACF;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB;AACF;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B;AACF;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC;AACF;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,oBACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,eAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,yBAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB,EAAA;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB,IAAA;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,kBAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC,IAAA;AACF,EAAA;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F,IAAA;AACA,IAAA,OAAOL,IAAAA;AACT,EAAA;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B,IAAA;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B,MAAA;AACF,IAAA;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD,MAAA;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF,IAAA;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd,QAAA;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb,QAAA;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb,QAAA;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB,QAAA;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC,QAAA;AACF,MAAA;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd,MAAA;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf,QAAA;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB,QAAA;AACF,MAAA;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd,QAAA;AACF,MAAA;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB,MAAA;AACF,IAAA;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf,MAAA;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B,MAAA;AACF,IAAA;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC,MAAA;AACF,IAAA;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,oBACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C,IAAA;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB,IAAA;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,eAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf,IAAA;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF,EAAA;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,yBAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}

package/dist/agents/requirement/utils/llm.d.cts

@@ -2,7 +2,7 @@ import { AnyTool } from '../../../tools/base.cjs';
 import { RunContext } from '../../../context.cjs';
 import { SystemMessage } from '../../../backend/message.cjs';
 import { PromptTemplate } from '../../../template.cjs';
-import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-CkEtV8R9.cjs';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-BNsa3W7Z.cjs';
 import { Requirement, Rule } from '../requirements/requirement.cjs';
 import 'ajv';
 import '../../../errors.cjs';
@@ -14,15 +14,15 @@ import '../../../cache/base.cjs';
 import '../../../internals/helpers/schema.cjs';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../emitter-jN55XZZq.cjs';
+import '../../../emitter-D4OcelZ9.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'ai';
 import '../../../memory/base.cjs';
-import '../../../chat-BFl85cqt.cjs';
+import '../../../chat-C20nYj46.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-s3ioVIE9.cjs';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

package/dist/agents/requirement/utils/llm.d.ts

@@ -2,7 +2,7 @@ import { AnyTool } from '../../../tools/base.js';
 import { RunContext } from '../../../context.js';
 import { SystemMessage } from '../../../backend/message.js';
 import { PromptTemplate } from '../../../template.js';
-import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-BQ-3LdkE.js';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-DYtzF6ya.js';
 import { Requirement, Rule } from '../requirements/requirement.js';
 import 'ajv';
 import '../../../errors.js';
@@ -14,15 +14,15 @@ import '../../../cache/base.js';
 import '../../../internals/helpers/schema.js';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../emitter-36-9MnvA.js';
+import '../../../emitter-kHxkUxco.js';
 import '../../../internals/helpers/promise.js';
 import 'ai';
 import '../../../memory/base.js';
-import '../../../chat-BRuyDeKR.js';
+import '../../../chat-CfTNfR60.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BjzyQ8k6.js';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

package/dist/agents/requirement/utils/llm.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,WAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC;AACF;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F;AACA,IAAA,OAAOL,IAAAA;AACT;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B;AACF;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC;AACF;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB;AACF;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd;AACF;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB;AACF;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B;AACF;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC;AACF;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,WACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,IAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,aAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB,EAAA;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB,IAAA;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,WAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC,IAAA;AACF,EAAA;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F,IAAA;AACA,IAAA,OAAOL,IAAAA;AACT,EAAA;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B,IAAA;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B,MAAA;AACF,IAAA;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD,MAAA;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF,IAAA;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd,QAAA;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb,QAAA;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb,QAAA;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB,QAAA;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC,QAAA;AACF,MAAA;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd,MAAA;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf,QAAA;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB,QAAA;AACF,MAAA;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd,QAAA;AACF,MAAA;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB,MAAA;AACF,IAAA;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf,MAAA;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B,MAAA;AACF,IAAA;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC,MAAA;AACF,IAAA;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,WACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C,IAAA;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB,IAAA;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,IAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf,IAAA;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF,EAAA;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,aAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}

package/dist/agents/requirement/utils/tool.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,yBAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,iCAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,0BAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,aAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,kBAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,2BAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,MAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,aAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,aAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,aAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,MAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,aAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,6BAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,4BAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAIjB,0BAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,yBAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,iCAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF,EAAA;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,0BAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D,IAAA;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,aAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,EAAA,CAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,kBAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,2BAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;IACvC,CAAA,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR,IAAA;AACF,EAAA;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,MAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,aAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,aAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,IAAA,CAAA,MAAA,IAAWW,0BAA0BE,aAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;IACT,CAAA,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,MAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;IACF,CAAA,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT,IAAA;AACF,EAAA;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,aAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,6BAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;IAC1D,CAAA,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,4BAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD,IAAA;AAEA,IAAA,OAAO,IAAIjB,0BAAiB,uBAAA,CAAA;AAC9B,EAAA;AACF","file":"tool.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

package/dist/agents/requirement/utils/tool.d.cts

@@ -1,9 +1,9 @@
 import '../../../tools/base.cjs';
 import 'ai';
 import '../../../errors.cjs';
-import '../../../emitter-jN55XZZq.cjs';
+import '../../../emitter-D4OcelZ9.cjs';
 import 'zod';
-export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-CkEtV8R9.cjs';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-BNsa3W7Z.cjs';
 import '../../../context.cjs';
 import 'ajv';
 import '../../../internals/serializable.cjs';
@@ -17,8 +17,8 @@ import '../../../internals/helpers/promise.cjs';
 import '../../../memory/base.cjs';
 import '../../../backend/message.cjs';
 import '../../../template.cjs';
-import '../../../chat-BFl85cqt.cjs';
+import '../../../chat-C20nYj46.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-s3ioVIE9.cjs';

package/dist/agents/requirement/utils/tool.d.ts

@@ -1,9 +1,9 @@
 import '../../../tools/base.js';
 import 'ai';
 import '../../../errors.js';
-import '../../../emitter-36-9MnvA.js';
+import '../../../emitter-kHxkUxco.js';
 import 'zod';
-export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-BQ-3LdkE.js';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-DYtzF6ya.js';
 import '../../../context.js';
 import 'ajv';
 import '../../../internals/serializable.js';
@@ -17,8 +17,8 @@ import '../../../internals/helpers/promise.js';
 import '../../../memory/base.js';
 import '../../../backend/message.js';
 import '../../../template.js';
-import '../../../chat-BRuyDeKR.js';
+import '../../../chat-CfTNfR60.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BjzyQ8k6.js';

package/dist/agents/requirement/utils/tool.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,eAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,sBAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAIjB,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,eAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,sBAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF,EAAA;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D,IAAA;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,EAAA,CAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;IACvC,CAAA,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR,IAAA;AACF,EAAA;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,IAAA,CAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;IACT,CAAA,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;IACF,CAAA,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT,IAAA;AACF,EAAA;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;IAC1D,CAAA,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD,IAAA;AAEA,IAAA,OAAO,IAAIjB,iBAAiB,uBAAA,CAAA;AAC9B,EAAA;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

package/dist/agents/requirement/utils/toolCallChecker.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;;;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB;KACF,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB;AACF;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF;AACF","file":"toolCallChecker.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;;;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF,EAAA;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF,IAAA;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB,MAAA;IACF,CAAA,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB,IAAA;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B,IAAA;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB,IAAA;AACF,EAAA;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB,EAAA;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF,EAAA;AACF","file":"toolCallChecker.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}

package/dist/agents/requirement/utils/toolCallChecker.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB;KACF,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB;AACF;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF;AACF","file":"toolCallChecker.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF,EAAA;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF,IAAA;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB,MAAA;IACF,CAAA,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB,IAAA;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B,IAAA;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB,IAAA;AACF,EAAA;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB,EAAA;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF,EAAA;AACF","file":"toolCallChecker.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}