@adhd/agent-mcp 0.0.3 → 0.0.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/drizzle/0000_nifty_sasquatch.sql +55 -0
- package/drizzle/0001_task_usage.sql +16 -0
- package/drizzle/meta/0000_snapshot.json +387 -0
- package/drizzle/meta/0001_snapshot.json +495 -0
- package/drizzle/meta/_journal.json +20 -0
- package/package.json +5 -1
- package/src/db/schema.d.ts +223 -0
- package/src/db/schema.d.ts.map +1 -1
- package/src/db/schema.js +31 -1
- package/src/db/schema.js.map +1 -1
- package/src/engine/orchestrator.d.ts.map +1 -1
- package/src/engine/orchestrator.js +4 -3
- package/src/engine/orchestrator.js.map +1 -1
- package/src/index.js +7 -0
- package/src/index.js.map +1 -1
- package/src/plugins/index.d.ts +2 -0
- package/src/plugins/index.d.ts.map +1 -0
- package/src/plugins/index.js +2 -0
- package/src/plugins/index.js.map +1 -0
- package/src/plugins/usage-plugin.d.ts +27 -0
- package/src/plugins/usage-plugin.d.ts.map +1 -0
- package/src/plugins/usage-plugin.js +125 -0
- package/src/plugins/usage-plugin.js.map +1 -0
- package/src/providers/anthropic.d.ts.map +1 -1
- package/src/providers/anthropic.js +5 -0
- package/src/providers/anthropic.js.map +1 -1
- package/src/providers/openai.d.ts.map +1 -1
- package/src/providers/openai.js +4 -0
- package/src/providers/openai.js.map +1 -1
- package/src/providers/types.d.ts +3 -2
- package/src/providers/types.d.ts.map +1 -1
- package/src/server.d.ts +3 -0
- package/src/server.d.ts.map +1 -1
- package/src/server.js +35 -5
- package/src/server.js.map +1 -1
- package/src/tools/task.d.ts +10 -2
- package/src/tools/task.d.ts.map +1 -1
- package/src/tools/task.js +18 -1
- package/src/tools/task.js.map +1 -1
- package/src/tools/usage.d.ts +54 -0
- package/src/tools/usage.d.ts.map +1 -0
- package/src/tools/usage.js +117 -0
- package/src/tools/usage.js.map +1 -0
- package/src/validation/agent.d.ts.map +1 -1
- package/src/validation/agent.js +11 -2
- package/src/validation/agent.js.map +1 -1
- package/src/validation/index.d.ts +1 -0
- package/src/validation/index.d.ts.map +1 -1
- package/src/validation/index.js +1 -0
- package/src/validation/index.js.map +1 -1
- package/src/validation/task.d.ts +17 -0
- package/src/validation/task.d.ts.map +1 -1
- package/src/validation/task.js +5 -0
- package/src/validation/task.js.map +1 -1
- package/src/validation/usage.d.ts +53 -0
- package/src/validation/usage.d.ts.map +1 -0
- package/src/validation/usage.js +50 -0
- package/src/validation/usage.js.map +1 -0
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAE1C,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AACjC,OAAO,MAAM,MAAM,SAAS,CAAC;AAE7B,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAWhD,iFAAiF;AAEjF,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;AACnD,MAAM,eAAe,GAAK,4CAA4C,CAAC;AACvE,MAAM,eAAe,GAAK,sCAAsC,CAAC;AACjE,uEAAuE;AACvE,MAAM,iBAAiB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,QAAQ;AAQjD,KAAK,UAAU,iBAAiB;IAC5B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,aAAa,CAClC,UAAU,EACV,CAAC,uBAAuB,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,CAAC,EACvD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAwC,CAAC;IAChF,OAAO,MAAM,CAAC,aAAa,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,KAAuB;IACrD,wEAAwE;IACxE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,aAAa,CAClC,UAAU,EACV,CAAC,uBAAuB,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,CAAC,EACvD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAA4B,CAAC;IACtE,QAAQ,CAAC,eAAe,CAAC,GAAG,KAAK,CAAC;IAElC,yCAAyC;IACzC,MAAM,aAAa,CACf,UAAU,EACV;QACI,sBAAsB;QACtB,IAAI;QACJ,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;KACjC,EACD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;AACN,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,YAAoB;IACjD,oFAAoF;IACpF,qFAAqF;IACrF,0EAA0E;IAC1E,mFAAmF;IACnF,wDAAwD;IACxD,EAAE;IACF,qCAAqC;IACrC,8EAA8E;IAC9E,2EAA2E;IAC3E,6DAA6D;IAC7D,mFAAmF;IACnF,8EAA8E;IAC9E,EAAE;IACF,+EAA+E;IAC/E,kFAAkF;IAClF,EAAE;IACF,sEAAsE;IACtE,EAAE;IACF,kFAAkF;IAClF,kFAAkF;IAClF,gEAAgE;IAEhE,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;QAC7B,UAAU,EAAE,eAAe;QAC3B,aAAa,EAAE,YAAY;QAC3B,SAAS,EAAE,eAAe;KAC7B,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,eAAe,EAAE;QACrC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,mCAAmC,EAAE;QAChE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE;KACxB,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IAEjC,IAAI,IAAI,GAA4F,EAAE,CAAC;IACvG,IAAI,CAAC;QACD,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;IAC9C,CAAC;IAAC,MAAM,CAAC;QACL,MAAM,IAAI,KAAK,CACX,+CAA+C;YAC/C,aAAa,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,IAAI;YAC7C,WAAW,OAAO,EAAE,CACvB,CAAC;IACN,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CACX,wBAAwB;YACxB,aAAa,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,IAAI;YAC7C,WAAW,OAAO,EAAE,CACvB,CAAC;IACN,CAAC;IAED,MAAM,KAAK,GAAqB;QAC5B,WAAW,EAAG,IAAI,CAAC,YAAY;QAC/B,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,YAAY,EAAE,2BAA2B;QAC7E,SAAS,EAAK,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,IAAI;KAC9D,CAAC;IAEF,gFAAgF;IAChF,MAAM,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAEhC,OAAO,KAAK,CAAC;AACjB,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,cAAc;IACzB,MAAM,KAAK,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAExC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,GAAG,iBAAiB,EAAE,CAAC;QACnD,OAAO,KAAK,CAAC,WAAW,CAAC,CAAC,aAAa;IAC3C,CAAC;IAED,0CAA0C;IAC1C,IAAI,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,KAAK,CAAC,WAAW,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACL,6EAA6E;QAC7E,OAAO,KAAK,CAAC,WAAW,CAAC;IAC7B,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAwB;IAC9C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;QACnC,YAAY,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI;YAC/B,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACjB,CAAyB;KAC7B,CAAC,CAAC,CAAC;AACR,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAmB;IAC5C,+EAA+E;IAC/E,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IAE5D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,EAAgB,EAAE;QAC3C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACxD,CAAC;YACD,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,aAAa;wBACnB,WAAW,EAAE,UAAU,CAAC,UAAU;wBAClC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;wBAC1C,QAAQ,EAAE,UAAU,CAAC,OAAO;qBAC/B;iBACJ;aACJ,CAAC;QACN,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YAC5D,MAAM,aAAa,GAAwB,EAAE,CAAC;YAC9C,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClB,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACjC,aAAa,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,KAAK,EAAE,EAAE,CAAC,SAAoC;iBACjD,CAAC,CAAC;YACP,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACzD,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QACjE,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC5D,CAAC;QAED,0DAA0D;QAC1D,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,OAAO,iBAAiB;IAClB,MAAM,CAAY;IACT,MAAM,CAAiD;IAExE,YAAY,MAAsD;QAC9D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YACxB,gFAAgF;YAChF,gEAAgE;YAChE,IAAI,CAAC,MAAM,GAAG,IAA4B,CAAC;YAC3C,OAAO;QACX,CAAC;QAED,yCAAyC;QACzC,yDAAyD;QACzD,uFAAuF;QACvF,2FAA2F;QAC3F,oEAAoE;QACpE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,mBAAmB,CAAC,IAAI,SAAS,CAAC;QACjF,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,sBAAsB,CAAC,IAAI,SAAS,CAAC;QAE1F,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5C,CAAC;aAAM,IAAI,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACJ,4EAA4E;YAC5E,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAClC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;QACnC,uEAAuE;QACvE,sEAAsE;QACtE,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,MAAM,cAAc,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QAE5C,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;QAEtF,MAAM,GAAG,GAAG,KAAK,IAAmC,EAAE;YAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAC9C;gBACI,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,MAAM,EAAE,YAAY;gBACpB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI;gBACzC,QAAQ,EAAE,mBAAmB,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAC/C,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;aACzC,EACD,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAC7B,CAAC;YAEF,MAAM,SAAS,GAAe,EAAE,CAAC;YACjC,MAAM,YAAY,GAAa,EAAE,CAAC;YAElC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACnC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAClC,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5B,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChD,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,KAAK,CAAC,8CAA8C,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBAChF,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACX,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC;wBAC3C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC;wBAC1C,SAAS,EAAE,KAAK,CAAC,KAAK;qBACzB,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,EAAE;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBACtE,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACvD,SAAS,EAAE,MAAM,EAAE;aACtB,CAAC;YAEF,OAAO;gBACH,OAAO;gBACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW;
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAE1C,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AACjC,OAAO,MAAM,MAAM,SAAS,CAAC;AAE7B,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAWhD,iFAAiF;AAEjF,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;AACnD,MAAM,eAAe,GAAK,4CAA4C,CAAC;AACvE,MAAM,eAAe,GAAK,sCAAsC,CAAC;AACjE,uEAAuE;AACvE,MAAM,iBAAiB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,QAAQ;AAQjD,KAAK,UAAU,iBAAiB;IAC5B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,aAAa,CAClC,UAAU,EACV,CAAC,uBAAuB,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,CAAC,EACvD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAwC,CAAC;IAChF,OAAO,MAAM,CAAC,aAAa,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,KAAuB;IACrD,wEAAwE;IACxE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,aAAa,CAClC,UAAU,EACV,CAAC,uBAAuB,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,CAAC,EACvD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAA4B,CAAC;IACtE,QAAQ,CAAC,eAAe,CAAC,GAAG,KAAK,CAAC;IAElC,yCAAyC;IACzC,MAAM,aAAa,CACf,UAAU,EACV;QACI,sBAAsB;QACtB,IAAI;QACJ,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;KACjC,EACD,EAAE,QAAQ,EAAE,MAAM,EAAE,CACvB,CAAC;AACN,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,YAAoB;IACjD,oFAAoF;IACpF,qFAAqF;IACrF,0EAA0E;IAC1E,mFAAmF;IACnF,wDAAwD;IACxD,EAAE;IACF,qCAAqC;IACrC,8EAA8E;IAC9E,2EAA2E;IAC3E,6DAA6D;IAC7D,mFAAmF;IACnF,8EAA8E;IAC9E,EAAE;IACF,+EAA+E;IAC/E,kFAAkF;IAClF,EAAE;IACF,sEAAsE;IACtE,EAAE;IACF,kFAAkF;IAClF,kFAAkF;IAClF,gEAAgE;IAEhE,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;QAC7B,UAAU,EAAE,eAAe;QAC3B,aAAa,EAAE,YAAY;QAC3B,SAAS,EAAE,eAAe;KAC7B,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,eAAe,EAAE;QACrC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,mCAAmC,EAAE;QAChE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE;KACxB,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IAEjC,IAAI,IAAI,GAA4F,EAAE,CAAC;IACvG,IAAI,CAAC;QACD,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;IAC9C,CAAC;IAAC,MAAM,CAAC;QACL,MAAM,IAAI,KAAK,CACX,+CAA+C;YAC/C,aAAa,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,IAAI;YAC7C,WAAW,OAAO,EAAE,CACvB,CAAC;IACN,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CACX,wBAAwB;YACxB,aAAa,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,IAAI;YAC7C,WAAW,OAAO,EAAE,CACvB,CAAC;IACN,CAAC;IAED,MAAM,KAAK,GAAqB;QAC5B,WAAW,EAAG,IAAI,CAAC,YAAY;QAC/B,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,YAAY,EAAE,2BAA2B;QAC7E,SAAS,EAAK,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,IAAI;KAC9D,CAAC;IAEF,gFAAgF;IAChF,MAAM,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAEhC,OAAO,KAAK,CAAC;AACjB,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,cAAc;IACzB,MAAM,KAAK,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAExC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,GAAG,iBAAiB,EAAE,CAAC;QACnD,OAAO,KAAK,CAAC,WAAW,CAAC,CAAC,aAAa;IAC3C,CAAC;IAED,0CAA0C;IAC1C,IAAI,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,KAAK,CAAC,WAAW,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACL,6EAA6E;QAC7E,OAAO,KAAK,CAAC,WAAW,CAAC;IAC7B,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAwB;IAC9C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;QACnC,YAAY,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI;YAC/B,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACjB,CAAyB;KAC7B,CAAC,CAAC,CAAC;AACR,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAmB;IAC5C,+EAA+E;IAC/E,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IAE5D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,EAAgB,EAAE;QAC3C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACxD,CAAC;YACD,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,aAAa;wBACnB,WAAW,EAAE,UAAU,CAAC,UAAU;wBAClC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;wBAC1C,QAAQ,EAAE,UAAU,CAAC,OAAO;qBAC/B;iBACJ;aACJ,CAAC;QACN,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YAC5D,MAAM,aAAa,GAAwB,EAAE,CAAC;YAC9C,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClB,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACjC,aAAa,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,KAAK,EAAE,EAAE,CAAC,SAAoC;iBACjD,CAAC,CAAC;YACP,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACzD,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QACjE,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC5D,CAAC;QAED,0DAA0D;QAC1D,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,OAAO,iBAAiB;IAClB,MAAM,CAAY;IACT,MAAM,CAAiD;IAExE,YAAY,MAAsD;QAC9D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YACxB,gFAAgF;YAChF,gEAAgE;YAChE,IAAI,CAAC,MAAM,GAAG,IAA4B,CAAC;YAC3C,OAAO;QACX,CAAC;QAED,yCAAyC;QACzC,yDAAyD;QACzD,uFAAuF;QACvF,2FAA2F;QAC3F,oEAAoE;QACpE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,mBAAmB,CAAC,IAAI,SAAS,CAAC;QACjF,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,sBAAsB,CAAC,IAAI,SAAS,CAAC;QAE1F,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5C,CAAC;aAAM,IAAI,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACJ,4EAA4E;YAC5E,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAClC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;QACnC,uEAAuE;QACvE,sEAAsE;QACtE,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,MAAM,cAAc,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QAE5C,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;QAEtF,MAAM,GAAG,GAAG,KAAK,IAAmC,EAAE;YAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAC9C;gBACI,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,MAAM,EAAE,YAAY;gBACpB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI;gBACzC,QAAQ,EAAE,mBAAmB,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAC/C,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;aACzC,EACD,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAC7B,CAAC;YAEF,MAAM,SAAS,GAAe,EAAE,CAAC;YACjC,MAAM,YAAY,GAAa,EAAE,CAAC;YAElC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACnC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAClC,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5B,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChD,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,KAAK,CAAC,8CAA8C,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBAChF,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACX,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC;wBAC3C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC;wBAC1C,SAAS,EAAE,KAAK,CAAC,KAAK;qBACzB,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,EAAE;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBACtE,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACvD,SAAS,EAAE,MAAM,EAAE;aACtB,CAAC;YAEF,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;YAChC,OAAO;gBACH,OAAO;gBACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW;gBAC7D,KAAK,EAAE;oBACH,WAAW,EAAE,QAAQ,CAAC,YAAY;oBAClC,YAAY,EAAE,QAAQ,CAAC,aAAa;iBACvC;aACJ,CAAC;QACN,CAAC,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YACd,OAAO,MAAM,CAAC,GAAG,EAAE;gBACf,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,MAAM,EAAE,WAAW,CAAC,MAAM;gBAC1B,eAAe,EAAE,KAAK,CAAC,EAAE;oBACrB,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;wBAC1B,MAAM,KAAK,CAAC,CAAC,8BAA8B;oBAC/C,CAAC;gBACL,CAAC;aACJ,CAAC,CAAC;QACP,CAAC;QAED,OAAO,GAAG,EAAE,CAAC;IACjB,CAAC;CACJ"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAO5B,OAAO,KAAK,EAAE,cAAc,EAAqB,MAAM,wBAAwB,CAAC;AAEhF,OAAO,KAAK,EACR,WAAW,EACX,mBAAmB,EACnB,oBAAoB,EAEvB,MAAM,YAAY,CAAC;AA6DpB,qBAAa,cAAe,YAAW,WAAW;IAC9C,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8C;gBAEzD,MAAM,EAAE,OAAO,CAAC,cAAc,EAAE;QAAE,IAAI,EAAE,QAAQ,CAAA;KAAE,CAAC;IAQzD,IAAI,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAO5B,OAAO,KAAK,EAAE,cAAc,EAAqB,MAAM,wBAAwB,CAAC;AAEhF,OAAO,KAAK,EACR,WAAW,EACX,mBAAmB,EACnB,oBAAoB,EAEvB,MAAM,YAAY,CAAC;AA6DpB,qBAAa,cAAe,YAAW,WAAW;IAC9C,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8C;gBAEzD,MAAM,EAAE,OAAO,CAAC,cAAc,EAAE;QAAE,IAAI,EAAE,QAAQ,CAAA;KAAE,CAAC;IAQzD,IAAI,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;CAuE1E"}
|
package/src/providers/openai.js
CHANGED
|
@@ -105,9 +105,13 @@ export class OpenAIProvider {
|
|
|
105
105
|
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
106
106
|
createdAt: nowIso(),
|
|
107
107
|
};
|
|
108
|
+
const sdkUsage = response.usage;
|
|
108
109
|
return {
|
|
109
110
|
message,
|
|
110
111
|
stopReason: toolCalls.length > 0 ? "tool_calls" : "completed",
|
|
112
|
+
usage: sdkUsage
|
|
113
|
+
? { inputTokens: sdkUsage.prompt_tokens, outputTokens: sdkUsage.completion_tokens }
|
|
114
|
+
: undefined,
|
|
111
115
|
};
|
|
112
116
|
};
|
|
113
117
|
if (retryConfig) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,MAAM,MAAM,SAAS,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAWhD,SAAS,gBAAgB,CAAC,QAAmB;IACzC,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;QAC1B,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,QAAQ;gBACT,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM;gBACP,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC5D,KAAK,WAAW,CAAC,CAAC,CAAC;gBACf,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpD,OAAO;wBACH,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;wBAChC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;4BACrC,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE;gCACN,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,KAAK,EAAE,CAAC,IAAI,EAAE;gCAChC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;6BAC1C;yBACJ,CAAC,CAAC;qBACN,CAAC;gBACN,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YACjE,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACV,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;oBACd,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;gBACxD,CAAC;gBACD,OAAO;oBACH,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,UAAU,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;iBAC7C,CAAC;YACN,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACN,MAAM,UAAU,GAAU,OAAO,CAAC,IAAI,CAAC;gBACvC,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,aAAa,CAAC,KAAwB;IAC3C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,EAAE,UAAmB;QACzB,QAAQ,EAAE;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,UAAU,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI;gBAC7B,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACjB,CAA4B;SAChC;KACJ,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,OAAO,cAAc;IACJ,MAAM,CAAS;IACjB,MAAM,CAA8C;IAErE,YAAY,MAAmD;QAC3D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACrB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,gBAAgB,CAAC;YACzD,OAAO,EAAE,MAAM,CAAC,OAAO;SAC1B,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;QACnC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QAE5C,MAAM,GAAG,GAAG,KAAK,IAAmC,EAAE;YAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACtD;gBACI,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBACjC,QAAQ,EAAE,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAC5C,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC;aACtC,EACD,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,IAAI,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC7B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;oBACzC,IAAI,EAAE,CAAC,IAAI,KAAK,UAAU;wBAAE,SAAS;oBAErC,MAAM,cAAc,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBACtD,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,KAAK,CAAC,8CAA8C,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;oBACtF,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACX,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC;wBACjD,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC;wBAChD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;qBAC/C,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,EAAE;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,IAAI,SAAS;gBAC7C,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACvD,SAAS,EAAE,MAAM,EAAE;aACtB,CAAC;YAEF,OAAO;gBACH,OAAO;gBACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW;
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,MAAM,MAAM,SAAS,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAWhD,SAAS,gBAAgB,CAAC,QAAmB;IACzC,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;QAC1B,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,QAAQ;gBACT,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM;gBACP,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC5D,KAAK,WAAW,CAAC,CAAC,CAAC;gBACf,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpD,OAAO;wBACH,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;wBAChC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;4BACrC,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE;gCACN,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,KAAK,EAAE,CAAC,IAAI,EAAE;gCAChC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;6BAC1C;yBACJ,CAAC,CAAC;qBACN,CAAC;gBACN,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YACjE,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACV,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;oBACd,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;gBACxD,CAAC;gBACD,OAAO;oBACH,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,UAAU,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;iBAC7C,CAAC;YACN,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACN,MAAM,UAAU,GAAU,OAAO,CAAC,IAAI,CAAC;gBACvC,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,aAAa,CAAC,KAAwB;IAC3C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,EAAE,UAAmB;QACzB,QAAQ,EAAE;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,UAAU,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI;gBAC7B,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACjB,CAA4B;SAChC;KACJ,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,OAAO,cAAc;IACJ,MAAM,CAAS;IACjB,MAAM,CAA8C;IAErE,YAAY,MAAmD;QAC3D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACrB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,gBAAgB,CAAC;YACzD,OAAO,EAAE,MAAM,CAAC,OAAO;SAC1B,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;QACnC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QAE5C,MAAM,GAAG,GAAG,KAAK,IAAmC,EAAE;YAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACtD;gBACI,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;gBACjC,QAAQ,EAAE,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAC5C,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC;aACtC,EACD,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,IAAI,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC7B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;oBACzC,IAAI,EAAE,CAAC,IAAI,KAAK,UAAU;wBAAE,SAAS;oBAErC,MAAM,cAAc,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBACtD,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,KAAK,CAAC,8CAA8C,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;oBACtF,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACX,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC;wBACjD,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC;wBAChD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;qBAC/C,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,EAAE;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,IAAI,SAAS;gBAC7C,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACvD,SAAS,EAAE,MAAM,EAAE;aACtB,CAAC;YAEF,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;YAChC,OAAO;gBACH,OAAO;gBACP,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW;gBAC7D,KAAK,EAAE,QAAQ;oBACX,CAAC,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,aAAa,EAAE,YAAY,EAAE,QAAQ,CAAC,iBAAiB,EAAE;oBACnF,CAAC,CAAC,SAAS;aAClB,CAAC;QACN,CAAC,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YACd,OAAO,MAAM,CAAC,GAAG,EAAE;gBACf,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,MAAM,EAAE,WAAW,CAAC,MAAM;gBAC1B,eAAe,EAAE,KAAK,CAAC,EAAE;oBACrB,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;wBAC1B,MAAM,KAAK,CAAC;oBAChB,CAAC;gBACL,CAAC;aACJ,CAAC,CAAC;QACP,CAAC;QAED,OAAO,GAAG,EAAE,CAAC;IACjB,CAAC;CACJ"}
|
package/src/providers/types.d.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import type { Message } from "../validation/index.js";
|
|
2
|
-
import type { ToolDefinition } from "@adhd/agent-mcp-types";
|
|
3
|
-
export type { ToolDefinition };
|
|
2
|
+
import type { ToolDefinition, TokenUsage } from "@adhd/agent-mcp-types";
|
|
3
|
+
export type { ToolDefinition, TokenUsage };
|
|
4
4
|
export interface ProviderChatRequest {
|
|
5
5
|
messages: Message[];
|
|
6
6
|
tools?: ToolDefinition[];
|
|
@@ -28,6 +28,7 @@ export interface ProviderChatRequest {
|
|
|
28
28
|
export interface ProviderChatResponse {
|
|
29
29
|
message: Message;
|
|
30
30
|
stopReason: "completed" | "tool_calls";
|
|
31
|
+
usage?: TokenUsage;
|
|
31
32
|
}
|
|
32
33
|
export interface LLMProvider {
|
|
33
34
|
chat(request: ProviderChatRequest): Promise<ProviderChatResponse>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/providers/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC;AAE3C,MAAM,WAAW,mBAAmB;IAChC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB;;;;;;OAMG;IACH,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB;;;;;;;OAOG;IACH,WAAW,CAAC,EAAE,CACV,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,OAAO,KACZ,OAAO,CAAC;QAAE,MAAM,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,oBAAoB;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,WAAW,GAAG,YAAY,CAAC;IACvC,KAAK,CAAC,EAAE,UAAU,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IACxB,IAAI,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;CACrE"}
|
package/src/server.d.ts
CHANGED
|
@@ -7,6 +7,7 @@ import { BackgroundQueue } from "./engine/queue.js";
|
|
|
7
7
|
import { Orchestrator } from "./engine/orchestrator.js";
|
|
8
8
|
import type { PolicyEngine } from "./engine/policy.js";
|
|
9
9
|
import type { IHookRegistry } from "@adhd/agent-mcp-types";
|
|
10
|
+
import { type Database } from "./tools/usage.js";
|
|
10
11
|
export interface ServerDeps {
|
|
11
12
|
agentStore: AgentStore;
|
|
12
13
|
sessionStore: SessionStore;
|
|
@@ -15,6 +16,8 @@ export interface ServerDeps {
|
|
|
15
16
|
policy: PolicyEngine;
|
|
16
17
|
orchestrator: Orchestrator;
|
|
17
18
|
hooks: IHookRegistry;
|
|
19
|
+
/** Drizzle DB handle — used by usage_query and to enrich task/result responses. */
|
|
20
|
+
db: Database;
|
|
18
21
|
selfUrl?: string;
|
|
19
22
|
}
|
|
20
23
|
export declare function createServer(deps: ServerDeps): Server;
|
package/src/server.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAQnE,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAQnE,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAW3D,OAAO,EAAc,KAAK,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAkB7D,MAAM,WAAW,UAAU;IACvB,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,EAAE,YAAY,CAAC;IAC3B,SAAS,EAAE,SAAS,CAAC;IACrB,KAAK,EAAE,eAAe,CAAC;IACvB,MAAM,EAAE,YAAY,CAAC;IACrB,YAAY,EAAE,YAAY,CAAC;IAC3B,KAAK,EAAE,aAAa,CAAC;IACrB,mFAAmF;IACnF,EAAE,EAAE,QAAQ,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAiTD,wBAAgB,YAAY,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,CA0TrD;AAED,wBAAsB,WAAW,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC;IACzD,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3B,UAAU,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC;CAC5B,CAAC,CAsED"}
|
package/src/server.js
CHANGED
|
@@ -7,8 +7,9 @@ import { logger } from "./logger.js";
|
|
|
7
7
|
import { agentCreate, agentRead, agentUpdate, agentDelete, agentList, } from "./tools/agent-crud.js";
|
|
8
8
|
import { agentTool, sessionList, sessionClose, sessionClear } from "./tools/session.js";
|
|
9
9
|
import { taskTool, taskList, taskCancel, resultTool } from "./tools/task.js";
|
|
10
|
+
import { usageQuery } from "./tools/usage.js";
|
|
10
11
|
import { ToolError } from "./validation/errors.js";
|
|
11
|
-
import { agentCreateInputSchema, agentReadInputSchema, agentUpdateInputSchema, agentDeleteInputSchema, agentToolInputSchema, sessionListInputSchema, sessionCloseInputSchema, sessionClearInputSchema, taskToolInputSchema, taskListInputSchema, taskCancelInputSchema, resultInputSchema, } from "./validation/index.js";
|
|
12
|
+
import { agentCreateInputSchema, agentReadInputSchema, agentUpdateInputSchema, agentDeleteInputSchema, agentToolInputSchema, sessionListInputSchema, sessionCloseInputSchema, sessionClearInputSchema, taskToolInputSchema, taskListInputSchema, taskCancelInputSchema, resultInputSchema, taskUsageInputSchema, } from "./validation/index.js";
|
|
12
13
|
function toMcpErrorContent(error) {
|
|
13
14
|
let message;
|
|
14
15
|
if (error instanceof ToolError) {
|
|
@@ -350,6 +351,16 @@ export function createServer(deps) {
|
|
|
350
351
|
description: "Clear all messages from a session's context without closing it",
|
|
351
352
|
inputSchema: toMcpInputSchema(sessionClearInputSchema),
|
|
352
353
|
},
|
|
354
|
+
{
|
|
355
|
+
name: "usage_query",
|
|
356
|
+
description: "Query recorded token usage from task_usage by task_id (returns the full delegation subtree), root_task_id, agent_name, or time window",
|
|
357
|
+
inputSchema: toMcpInputSchema(taskUsageInputSchema),
|
|
358
|
+
},
|
|
359
|
+
{
|
|
360
|
+
name: "guide",
|
|
361
|
+
description: "Returns a complete guide explaining how to use this server — call this first if you are unsure what to do",
|
|
362
|
+
inputSchema: { type: "object", properties: {} },
|
|
363
|
+
},
|
|
353
364
|
];
|
|
354
365
|
// In-process handler that routes tool calls to the local handlers
|
|
355
366
|
const inProcessHandler = async (toolName, args, ctx) => {
|
|
@@ -368,9 +379,17 @@ export function createServer(deps) {
|
|
|
368
379
|
selfUrl: deps.selfUrl,
|
|
369
380
|
inProcessDescriptors,
|
|
370
381
|
inProcessHandler,
|
|
382
|
+
db: deps.db,
|
|
371
383
|
}, ctx);
|
|
372
384
|
case "result":
|
|
373
|
-
return resultTool(resultInputSchema.parse(args), {
|
|
385
|
+
return resultTool(resultInputSchema.parse(args), {
|
|
386
|
+
taskStore: deps.taskStore,
|
|
387
|
+
db: deps.db,
|
|
388
|
+
});
|
|
389
|
+
case "usage_query":
|
|
390
|
+
return usageQuery(deps.db, taskUsageInputSchema.parse(args ?? {}));
|
|
391
|
+
case "guide":
|
|
392
|
+
return USAGE_GUIDE;
|
|
374
393
|
case "task_list":
|
|
375
394
|
return taskList(taskListInputSchema.parse(args), { taskStore: deps.taskStore });
|
|
376
395
|
case "task_cancel":
|
|
@@ -466,7 +485,12 @@ export function createServer(deps) {
|
|
|
466
485
|
inputSchema: toMcpInputSchema(resultInputSchema),
|
|
467
486
|
},
|
|
468
487
|
{
|
|
469
|
-
name: "
|
|
488
|
+
name: "usage_query",
|
|
489
|
+
description: "Query recorded token usage from task_usage by task_id (returns the full delegation subtree), root_task_id, agent_name, or time window. Bare {} returns the most recent rows.",
|
|
490
|
+
inputSchema: toMcpInputSchema(taskUsageInputSchema),
|
|
491
|
+
},
|
|
492
|
+
{
|
|
493
|
+
name: "guide",
|
|
470
494
|
description: "Returns a complete guide explaining how to use this server — call this first if you are unsure what to do",
|
|
471
495
|
inputSchema: { type: "object", properties: {} },
|
|
472
496
|
},
|
|
@@ -525,6 +549,7 @@ export function createServer(deps) {
|
|
|
525
549
|
selfUrl: deps.selfUrl,
|
|
526
550
|
inProcessDescriptors,
|
|
527
551
|
inProcessHandler,
|
|
552
|
+
db: deps.db,
|
|
528
553
|
})
|
|
529
554
|
// No callerContext — top-level call
|
|
530
555
|
);
|
|
@@ -533,8 +558,13 @@ export function createServer(deps) {
|
|
|
533
558
|
case "task_cancel":
|
|
534
559
|
return toMcpContent(taskCancel(taskCancelInputSchema.parse(args), { taskStore: deps.taskStore }));
|
|
535
560
|
case "result":
|
|
536
|
-
return toMcpContent(resultTool(resultInputSchema.parse(args), {
|
|
537
|
-
|
|
561
|
+
return toMcpContent(resultTool(resultInputSchema.parse(args), {
|
|
562
|
+
taskStore: deps.taskStore,
|
|
563
|
+
db: deps.db,
|
|
564
|
+
}));
|
|
565
|
+
case "usage_query":
|
|
566
|
+
return toMcpContent(usageQuery(deps.db, taskUsageInputSchema.parse(args ?? {})));
|
|
567
|
+
case "guide":
|
|
538
568
|
return toMcpContent(USAGE_GUIDE);
|
|
539
569
|
default:
|
|
540
570
|
return toMcpErrorContent(new ToolError("VALIDATION_ERROR", `Unknown tool: ${name}`));
|
package/src/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EACH,qBAAqB,EACrB,sBAAsB,GACzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAUrC,OAAO,EACH,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,SAAS,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7E,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EACH,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,qBAAqB,EACrB,iBAAiB,GACpB,MAAM,uBAAuB,CAAC;AAa/B,SAAS,iBAAiB,CAAC,KAAc;IACrC,IAAI,OAAe,CAAC;IACpB,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACjD,CAAC;SAAM,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAChC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC5B,CAAC;SAAM,CAAC;QACJ,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO;QACH,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;KACtD,CAAC;AACN,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAChC,OAAO;QACH,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;aACvC;SACJ;KACJ,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,SAAS,gBAAgB,CAAC,MAAoB;IAC1C,MAAM,UAAU,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,CAA4B,CAAC;IAErE,4EAA4E;IAC5E,4CAA4C;IAC5C,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC;QAC/C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qEAAqE;IACrE,oEAAoE;IACpE,MAAM,QAAQ,GACT,UAAU,CAAC,OAAO,CAA2C;QAC7D,UAAU,CAAC,OAAO,CAA2C,CAAC;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,gBAAgB,GAA4B,EAAE,CAAC;QACrD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;YACpC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,KAAgC,CAAC,CAAC;YACtE,CAAC;QACL,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IAC5D,CAAC;IAED,2CAA2C;IAC3C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsOnB,CAAC,IAAI,EAAE,CAAC;AAET,MAAM,UAAU,YAAY,CAAC,IAAgB;IACzC,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,EACvC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAClC,CAAC;IAEF,wEAAwE;IACxE,8DAA8D;IAC9D,6CAA6C;IAC7C,MAAM,oBAAoB,GAA8B;QACpD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,gCAAgC;YAC7C,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,0BAA0B;YACvC,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;SACnD;QACD;YACI,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,uBAAuB;YACpC,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;SACvD;QACD;YACI,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;SACxD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;KACJ,CAAC;IAEF,kEAAkE;IAClE,MAAM,gBAAgB,GAAyB,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACzE,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,OAAO;gBACR,OAAO,SAAS,CACZ,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAChC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,EACrF,GAAG,CACN,CAAC;YACN,KAAK,MAAM;gBACP,OAAO,QAAQ,CACX,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC/B;oBACI,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,oBAAoB;oBACpB,gBAAgB;iBACnB,EACD,GAAG,CACN,CAAC;YACN,KAAK,QAAQ;gBACT,OAAO,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,WAAW;gBACZ,OAAO,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACxF,KAAK,cAAc;gBACf,OAAO,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACnD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP;gBACI,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,4BAA4B,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC;IACL,CAAC,CAAC;IAEF,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC1D,KAAK,EAAE;YACH;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,sCAAsC;gBACnD,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,wCAAwC;gBACrD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,mCAAmC;gBAChD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,kDAAkD;gBAC/D,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,eAAe;gBAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,yBAAyB;gBACtC,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gEAAgE;gBAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,wKAAwK;gBACrL,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,YAAY;gBACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;aACvD;YACD;gBACI,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;gBACzD,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;aACnD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,2GAA2G;gBACxH,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;SACJ;KACJ,CAAC,CAAC,CAAC;IAEJ,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;QAC5D,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACD,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1E,KAAK,OAAO;oBACR,OAAO,YAAY,CACf,MAAM,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC;oBACF,uCAAuC;qBAC1C,CAAC;gBAEN,KAAK,cAAc;oBACf,OAAO,YAAY,CACf,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,MAAM;oBACP,OAAO,YAAY,CACf,MAAM,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;wBACrB,oBAAoB;wBACpB,gBAAgB;qBACnB,CAAC;oBACF,oCAAoC;qBACvC,CAAC;gBAEN,KAAK,WAAW;oBACZ,OAAO,YAAY,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,QAAQ;oBACT,OAAO,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,OAAO;oBACR,OAAO,YAAY,CAAC,WAAW,CAAC,CAAC;gBAErC;oBACI,OAAO,iBAAiB,CAAC,IAAI,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7F,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB;IAI9C,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC;IACtD,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;IAEzD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACxB,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAE1D,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACvB,8DAA8D;QAC9D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAErE,MAAM,aAAa,GAAG,IAAI,6BAA6B,CAAC;YACpD,kBAAkB,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACnD,MAAM,aAAa,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEpC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;gBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;gBAC/D,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,oBAAoB,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACxC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;YACP,CAAC;YACD,UAAU;SACb,CAAC;IACN,CAAC;IAED,gBAAgB;IAChB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACtB,mFAAmF;QACnF,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,qEAAqE,CAAC,CAAC;QAClG,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;AACvD,CAAC"}
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/ai/agent-mcp/src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EACH,qBAAqB,EACrB,sBAAsB,GACzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAUrC,OAAO,EACH,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,SAAS,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAiB,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EACH,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,qBAAqB,EACrB,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,uBAAuB,CAAC;AAe/B,SAAS,iBAAiB,CAAC,KAAc;IACrC,IAAI,OAAe,CAAC;IACpB,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACjD,CAAC;SAAM,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAChC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC5B,CAAC;SAAM,CAAC;QACJ,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO;QACH,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;KACtD,CAAC;AACN,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAChC,OAAO;QACH,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;aACvC;SACJ;KACJ,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,SAAS,gBAAgB,CAAC,MAAoB;IAC1C,MAAM,UAAU,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,CAA4B,CAAC;IAErE,4EAA4E;IAC5E,4CAA4C;IAC5C,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC;QAC/C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qEAAqE;IACrE,oEAAoE;IACpE,MAAM,QAAQ,GACT,UAAU,CAAC,OAAO,CAA2C;QAC7D,UAAU,CAAC,OAAO,CAA2C,CAAC;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,gBAAgB,GAA4B,EAAE,CAAC;QACrD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;YACpC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,KAAgC,CAAC,CAAC;YACtE,CAAC;QACL,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IAC5D,CAAC;IAED,2CAA2C;IAC3C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsOnB,CAAC,IAAI,EAAE,CAAC;AAET,MAAM,UAAU,YAAY,CAAC,IAAgB;IACzC,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,EACvC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAClC,CAAC;IAEF,wEAAwE;IACxE,8DAA8D;IAC9D,6CAA6C;IAC7C,MAAM,oBAAoB,GAA8B;QACpD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,gCAAgC;YAC7C,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,0BAA0B;YACvC,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;SACnD;QACD;YACI,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;SACrD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,uBAAuB;YACpC,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;SACvD;QACD;YACI,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;SACxD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;SACzD;QACD;YACI,IAAI,EAAE,aAAa;YACnB,WAAW,EACP,uIAAuI;YAC3I,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SACtD;QACD;YACI,IAAI,EAAE,OAAO;YACb,WAAW,EACP,2GAA2G;YAC/G,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;SAClD;KACJ,CAAC;IAEF,kEAAkE;IAClE,MAAM,gBAAgB,GAAyB,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACzE,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,OAAO;gBACR,OAAO,SAAS,CACZ,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAChC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,EACrF,GAAG,CACN,CAAC;YACN,KAAK,MAAM;gBACP,OAAO,QAAQ,CACX,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC/B;oBACI,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,oBAAoB;oBACpB,gBAAgB;oBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;iBACd,EACD,GAAG,CACN,CAAC;YACN,KAAK,QAAQ;gBACT,OAAO,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC7C,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,EAAE,EAAE,IAAI,CAAC,EAAE;iBACd,CAAC,CAAC;YACP,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,oBAAoB,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;YACvE,KAAK,OAAO;gBACR,OAAO,WAAW,CAAC;YACvB,KAAK,WAAW;gBACZ,OAAO,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACpF,KAAK,aAAa;gBACd,OAAO,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACxF,KAAK,cAAc;gBACf,OAAO,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACnD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP,KAAK,eAAe;gBAChB,OAAO,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrD,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;YACP;gBACI,MAAM,IAAI,SAAS,CAAC,kBAAkB,EAAE,4BAA4B,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC;IACL,CAAC,CAAC;IAEF,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC1D,KAAK,EAAE;YACH;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,sCAAsC;gBACnD,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,wCAAwC;gBACrD,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,mCAAmC;gBAChD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,kDAAkD;gBAC/D,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,eAAe;gBAC5B,WAAW,EAAE,gBAAgB,CAAC,sBAAsB,CAAC;aACxD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,yBAAyB;gBACtC,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gEAAgE;gBAC7E,WAAW,EAAE,gBAAgB,CAAC,uBAAuB,CAAC;aACzD;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,wKAAwK;gBACrL,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,YAAY;gBACzB,WAAW,EAAE,gBAAgB,CAAC,mBAAmB,CAAC;aACrD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,kCAAkC;gBAC/C,WAAW,EAAE,gBAAgB,CAAC,qBAAqB,CAAC;aACvD;YACD;gBACI,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4CAA4C;gBACzD,WAAW,EAAE,gBAAgB,CAAC,iBAAiB,CAAC;aACnD;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,WAAW,EACP,8KAA8K;gBAClL,WAAW,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;aACtD;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,2GAA2G;gBACxH,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAClD;SACJ;KACJ,CAAC,CAAC,CAAC;IAEJ,yEAAyE;IACzE,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;QAC5D,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACD,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,cAAc;oBACf,OAAO,YAAY,CAAC,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1G,KAAK,YAAY;oBACb,OAAO,YAAY,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE1E,KAAK,OAAO;oBACR,OAAO,YAAY,CACf,MAAM,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC;oBACF,uCAAuC;qBAC1C,CAAC;gBAEN,KAAK,cAAc;oBACf,OAAO,YAAY,CACf,WAAW,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,eAAe;oBAChB,OAAO,YAAY,CACf,YAAY,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC9C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;qBACtB,CAAC,CACL,CAAC;gBAEN,KAAK,MAAM;oBACP,OAAO,YAAY,CACf,MAAM,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;wBACrB,oBAAoB;wBACpB,gBAAgB;wBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;qBACd,CAAC;oBACF,oCAAoC;qBACvC,CAAC;gBAEN,KAAK,WAAW;oBACZ,OAAO,YAAY,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAElG,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;gBAEtG,KAAK,QAAQ;oBACT,OAAO,YAAY,CACf,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;wBACtC,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,EAAE,EAAE,IAAI,CAAC,EAAE;qBACd,CAAC,CACL,CAAC;gBAEN,KAAK,aAAa;oBACd,OAAO,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,oBAAoB,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;gBAErF,KAAK,OAAO;oBACR,OAAO,YAAY,CAAC,WAAW,CAAC,CAAC;gBAErC;oBACI,OAAO,iBAAiB,CAAC,IAAI,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7F,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB;IAI9C,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC;IACtD,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;IAEzD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACxB,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAE1D,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACvB,8DAA8D;QAC9D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAErE,MAAM,aAAa,GAAG,IAAI,6BAA6B,CAAC;YACpD,kBAAkB,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACnD,MAAM,aAAa,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEpC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;gBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;gBAC/D,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,oBAAoB,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACxC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;YACP,CAAC;YACD,UAAU;SACb,CAAC;IACN,CAAC;IAED,gBAAgB;IAChB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACtB,mFAAmF;QACnF,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,qEAAqE,CAAC,CAAC;QAClG,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAClD,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAErC,OAAO;YACH,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;SACJ,CAAC;IACN,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;AACvD,CAAC"}
|
package/src/tools/task.d.ts
CHANGED
|
@@ -5,8 +5,9 @@ import type { InProcessToolDescriptor, InProcessToolHandler } from "../clients/i
|
|
|
5
5
|
import type { AgentStore } from "../store/agent-store.js";
|
|
6
6
|
import type { SessionStore } from "../store/session-store.js";
|
|
7
7
|
import type { TaskStore } from "../store/task-store.js";
|
|
8
|
-
import type { ExecutionContext, ResultInput, Task, TaskCancelInput, TaskListInput, TaskToolInput, TaskToolOutput } from "../validation/index.js";
|
|
8
|
+
import type { ExecutionContext, ResultInput, Task, TaskCancelInput, TaskListInput, TaskToolInput, TaskToolOutput, TaskUsageReport } from "../validation/index.js";
|
|
9
9
|
import type { IHookRegistry } from "@adhd/agent-mcp-types";
|
|
10
|
+
import { type Database } from "./usage.js";
|
|
10
11
|
export interface TaskDeps {
|
|
11
12
|
agentStore: AgentStore;
|
|
12
13
|
sessionStore: SessionStore;
|
|
@@ -18,6 +19,11 @@ export interface TaskDeps {
|
|
|
18
19
|
selfUrl: string | undefined;
|
|
19
20
|
inProcessDescriptors: InProcessToolDescriptor[];
|
|
20
21
|
inProcessHandler: InProcessToolHandler;
|
|
22
|
+
/**
|
|
23
|
+
* Drizzle DB handle — used to enrich `task` / `result` responses with the
|
|
24
|
+
* task's token-usage rollup (direct + subtree). See [dod.2].
|
|
25
|
+
*/
|
|
26
|
+
db: Database;
|
|
21
27
|
}
|
|
22
28
|
/**
|
|
23
29
|
* `task` tool — runs a prompt against a session's agent (session mode) or runs a
|
|
@@ -39,5 +45,7 @@ export declare function taskList(input: TaskListInput, deps: Pick<TaskDeps, "tas
|
|
|
39
45
|
export declare function taskCancel(input: TaskCancelInput, deps: Pick<TaskDeps, "taskStore">): {
|
|
40
46
|
success: true;
|
|
41
47
|
};
|
|
42
|
-
export declare function resultTool(input: ResultInput, deps: Pick<TaskDeps, "taskStore">): Task
|
|
48
|
+
export declare function resultTool(input: ResultInput, deps: Pick<TaskDeps, "taskStore" | "db">): Task & {
|
|
49
|
+
usage?: TaskUsageReport;
|
|
50
|
+
};
|
|
43
51
|
//# sourceMappingURL=task.d.ts.map
|
package/src/tools/task.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"task.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAE9F,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,KAAK,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,EACJ,eAAe,EACf,aAAa,EAEb,aAAa,EACb,cAAc,
|
|
1
|
+
{"version":3,"file":"task.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAE9F,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,KAAK,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,EACJ,eAAe,EACf,aAAa,EAEb,aAAa,EACb,cAAc,EACd,eAAe,EAClB,MAAM,wBAAwB,CAAC;AAIhC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAwB,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,WAAW,QAAQ;IACrB,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,EAAE,YAAY,CAAC;IAC3B,SAAS,EAAE,SAAS,CAAC;IACrB,YAAY,EAAE,YAAY,CAAC;IAC3B,KAAK,EAAE,eAAe,CAAC;IACvB,MAAM,EAAE,YAAY,CAAC;IACrB,KAAK,EAAE,aAAa,CAAC;IACrB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,oBAAoB,EAAE,uBAAuB,EAAE,CAAC;IAChD,gBAAgB,EAAE,oBAAoB,CAAC;IACvC;;;OAGG;IACH,EAAE,EAAE,QAAQ,CAAC;CAChB;AAgHD;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAC1B,KAAK,EAAE,aAAa,EACpB,IAAI,EAAE,QAAQ,EACd,aAAa,CAAC,EAAE,gBAAgB,GACjC,OAAO,CAAC,cAAc,CAAC,CAqIzB;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,IAAI,EAAE,CAExF;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG;IAAE,OAAO,EAAE,IAAI,CAAA;CAAE,CAavG;AAED,wBAAgB,UAAU,CACtB,KAAK,EAAE,WAAW,EAClB,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI,CAAC,GACzC,IAAI,GAAG;IAAE,KAAK,CAAC,EAAE,eAAe,CAAA;CAAE,CAIpC"}
|
package/src/tools/task.js
CHANGED
|
@@ -4,6 +4,7 @@ import { McpClientRegistry as McpClientRegistryCtor } from "../clients/registry.
|
|
|
4
4
|
import { ToolError } from "../validation/errors.js";
|
|
5
5
|
import { generateId } from "../utils/ids.js";
|
|
6
6
|
import { nowIso } from "../utils/timestamps.js";
|
|
7
|
+
import { buildTaskUsageReport } from "./usage.js";
|
|
7
8
|
/**
|
|
8
9
|
* One-shot ephemeral execution: loads the agent definition but creates no session
|
|
9
10
|
* row and persists no messages. Always synchronous.
|
|
@@ -12,6 +13,9 @@ async function runEphemeralTask(input, deps, callerContext) {
|
|
|
12
13
|
const agentDefinition = deps.agentStore.read(input.agent_name);
|
|
13
14
|
const taskId = generateId();
|
|
14
15
|
const ephemeralSessionId = generateId();
|
|
16
|
+
const rootTaskId = callerContext
|
|
17
|
+
? (callerContext.rootTaskId ?? callerContext.taskId)
|
|
18
|
+
: undefined;
|
|
15
19
|
const executionContext = {
|
|
16
20
|
taskId,
|
|
17
21
|
sessionId: ephemeralSessionId,
|
|
@@ -19,6 +23,7 @@ async function runEphemeralTask(input, deps, callerContext) {
|
|
|
19
23
|
agentDefinition,
|
|
20
24
|
callingAgentName: callerContext?.agentName,
|
|
21
25
|
parentTaskId: callerContext?.taskId,
|
|
26
|
+
rootTaskId: rootTaskId ?? undefined,
|
|
22
27
|
recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
|
|
23
28
|
toolCallCount: 0,
|
|
24
29
|
};
|
|
@@ -82,10 +87,12 @@ async function runEphemeralTask(input, deps, callerContext) {
|
|
|
82
87
|
catch {
|
|
83
88
|
// Orchestrator already captured status via captureTaskStore
|
|
84
89
|
}
|
|
90
|
+
const usage = buildTaskUsageReport(deps.db, taskId);
|
|
85
91
|
return {
|
|
86
92
|
task_id: taskId,
|
|
87
93
|
status: capturedStatus,
|
|
88
94
|
result: capturedResult,
|
|
95
|
+
usage,
|
|
89
96
|
};
|
|
90
97
|
}
|
|
91
98
|
/**
|
|
@@ -121,6 +128,11 @@ export async function taskTool(input, deps, callerContext) {
|
|
|
121
128
|
parentTaskId: callerContext?.taskId,
|
|
122
129
|
recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
|
|
123
130
|
});
|
|
131
|
+
// Derive rootTaskId at creation time from the in-memory callerContext chain.
|
|
132
|
+
// A DB walk is deliberately avoided — ephemeral tasks have no tasks row.
|
|
133
|
+
const rootTaskId = callerContext
|
|
134
|
+
? (callerContext.rootTaskId ?? callerContext.taskId)
|
|
135
|
+
: undefined;
|
|
124
136
|
// 4. Build execution context
|
|
125
137
|
const executionContext = {
|
|
126
138
|
taskId: task.id,
|
|
@@ -129,6 +141,7 @@ export async function taskTool(input, deps, callerContext) {
|
|
|
129
141
|
agentDefinition,
|
|
130
142
|
callingAgentName: callerContext?.agentName,
|
|
131
143
|
parentTaskId: callerContext?.taskId,
|
|
144
|
+
rootTaskId: rootTaskId ?? undefined,
|
|
132
145
|
recursionDepth: (callerContext?.recursionDepth ?? -1) + 1,
|
|
133
146
|
toolCallCount: 0,
|
|
134
147
|
};
|
|
@@ -200,10 +213,12 @@ export async function taskTool(input, deps, callerContext) {
|
|
|
200
213
|
// Return the current task state
|
|
201
214
|
}
|
|
202
215
|
const finalTask = deps.taskStore.read(task.id);
|
|
216
|
+
const usage = buildTaskUsageReport(deps.db, finalTask.id);
|
|
203
217
|
return {
|
|
204
218
|
task_id: finalTask.id,
|
|
205
219
|
status: finalTask.status,
|
|
206
220
|
result: finalTask.result,
|
|
221
|
+
usage,
|
|
207
222
|
};
|
|
208
223
|
}
|
|
209
224
|
}
|
|
@@ -220,6 +235,8 @@ export function taskCancel(input, deps) {
|
|
|
220
235
|
return { success: true };
|
|
221
236
|
}
|
|
222
237
|
export function resultTool(input, deps) {
|
|
223
|
-
|
|
238
|
+
const task = deps.taskStore.read(input.task_id); // throws TASK_NOT_FOUND
|
|
239
|
+
const usage = buildTaskUsageReport(deps.db, task.id);
|
|
240
|
+
return { ...task, usage };
|
|
224
241
|
}
|
|
225
242
|
//# sourceMappingURL=task.js.map
|
package/src/tools/task.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"task.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAKtC,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAIzD,OAAO,EAAE,iBAAiB,IAAI,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;
|
|
1
|
+
{"version":3,"file":"task.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/task.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAKtC,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAIzD,OAAO,EAAE,iBAAiB,IAAI,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAYpF,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAEhD,OAAO,EAAE,oBAAoB,EAAiB,MAAM,YAAY,CAAC;AAoBjE;;;GAGG;AACH,KAAK,UAAU,gBAAgB,CAC3B,KAA6C,EAC7C,IAAc,EACd,aAAgC;IAEhC,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,kBAAkB,GAAG,UAAU,EAAE,CAAC;IACxC,MAAM,UAAU,GAAG,aAAa;QAC5B,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,gBAAgB,GAAqB;QACvC,MAAM;QACN,SAAS,EAAE,kBAAkB;QAC7B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,UAAU,EAAE,UAAU,IAAI,SAAS;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,kBAAkB;QAC7B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY;QACzC,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,kBAAkB;gBAC7B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,WAAW;SACd;QACH,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAEpB,sDAAsD;IACtD,IAAI,cAAc,GAAe,SAAS,CAAC;IAC3C,IAAI,cAAkC,CAAC;IAEvC,MAAM,gBAAgB,GAAG;QACrB,YAAY,EAAE,CAAC,GAAW,EAAE,MAAkB,EAAE,MAA4B,EAAE,EAAE;YAC5E,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,MAAM,EAAE,MAAM,KAAK,SAAS;gBAAE,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;YACjE,OAAO,EAAU,CAAC;QACtB,CAAC;QACD,gEAAgE;QAChE,WAAW,EAAE,GAAG,EAAE,GAAE,CAAC;QACrB,gEAAgE;QAChE,sBAAsB,EAAE,GAAG,EAAE,GAAE,CAAC;KACX,CAAC;IAE1B,MAAM,gBAAgB,GAAG;QACrB,gEAAgE;QAChE,aAAa,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;KACL,CAAC;IAE7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,IAAI,CAAC;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ;YACR,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,gBAAgB;YAC3B,YAAY,EAAE,gBAAgB;YAC9B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC;QACL,4DAA4D;IAChE,CAAC;IAED,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAEpD,OAAO;QACH,OAAO,EAAE,MAAM;QACf,MAAM,EAAE,cAAc;QACtB,MAAM,EAAE,cAAc;QACtB,KAAK;KACR,CAAC;AACN,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC1B,KAAoB,EACpB,IAAc,EACd,aAAgC;IAEhC,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;QACxB,OAAO,gBAAgB,CAAC,KAAK,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;IACxD,CAAC;IAED,sBAAsB;IACtB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzD,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,SAAS,CAAC,gBAAgB,EAAE,YAAY,KAAK,CAAC,UAAU,aAAa,CAAC,CAAC;IACrF,CAAC;IAED,uCAAuC;IACvC,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE/E,qBAAqB;IACrB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAC/B,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;KAC5D,CAAC,CAAC;IAEH,6EAA6E;IAC7E,yEAAyE;IACzE,MAAM,UAAU,GAAG,aAAa;QAC5B,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEhB,6BAA6B;IAC7B,MAAM,gBAAgB,GAAqB;QACvC,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,SAAS,EAAE,eAAe,CAAC,IAAI;QAC/B,eAAe;QACf,gBAAgB,EAAE,aAAa,EAAE,SAAS;QAC1C,YAAY,EAAE,aAAa,EAAE,MAAM;QACnC,UAAU,EAAE,UAAU,IAAI,SAAS;QACnC,cAAc,EAAE,CAAC,aAAa,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;QACzD,aAAa,EAAE,CAAC;KACnB,CAAC;IAEF,4BAA4B;IAC5B,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,2BAA2B;IAC3B,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAEzD,oBAAoB;IACpB,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtF,yBAAyB;IACzB,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG;QAChB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,KAAK,CAAC,UAAU;QAC3B,IAAI,EAAE,MAAe;QACrB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,SAAS,EAAE,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,CAAC,GAAG,gBAAgB,EAAE,WAAW,CAAC,CAAC;IAEpD,0CAA0C;IAC1C,MAAM,WAAW,GAAG,eAAe,CAAC,YAAY;QAC5C,CAAC,CAAC;YACI;gBACI,EAAE,EAAE,UAAU,EAAE;gBAChB,SAAS,EAAE,KAAK,CAAC,UAAU;gBAC3B,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,eAAe,CAAC,YAAY;gBACrC,SAAS,EAAE,MAAM,EAAE;aACtB;YACD,GAAG,QAAQ;SACd;QACH,CAAC,CAAC,QAAQ,CAAC;IAEf,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,IAAI,qBAAqB;IACtC,8DAA8D;IAC9D,eAAe,CAAC,UAAiB,EACjC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,gBAAgB,EACrB,gBAAgB,CACnB,CAAC;IAEF,sBAAsB;IACtB,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACtC,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;YACxB,gBAAgB;YAChB,QAAQ,EAAE,WAAW;YACrB,QAAQ;YACR,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;SACpB,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,wDAAwD;QACxD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAErC,MAAM,CAAC,IAAI,CACP,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE,EAChD,wCAAwC,CAC3C,CAAC;QAEF,OAAO;YACH,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,MAAM,EAAE,SAAS;SACpB,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,8CAA8C;QAC9C,IAAI,CAAC;YACD,MAAM,OAAO,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,+CAA+C;YAC/C,gCAAgC;QACpC,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1D,OAAO;YACH,OAAO,EAAE,SAAS,CAAC,EAAE;YACrB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,KAAK;SACR,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,KAAoB,EAAE,IAAiC;IAC5E,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAsB,EAAE,IAAiC;IAChF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;IAEzE,MAAM,mBAAmB,GAAG,CAAC,SAAS,EAAE,SAAS,CAAU,CAAC;IAC5D,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,MAA4C,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,SAAS,CACf,sBAAsB,EACtB,SAAS,KAAK,CAAC,OAAO,iBAAiB,IAAI,CAAC,MAAM,2BAA2B,CAChF,CAAC;IACN,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,UAAU,CACtB,KAAkB,EAClB,IAAwC;IAExC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;IACzE,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACrD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;AAC9B,CAAC"}
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
|
|
2
|
+
import { taskUsageTable } from "../db/schema.js";
|
|
3
|
+
import type { TaskUsageInput, TaskUsageReport } from "../validation/usage.js";
|
|
4
|
+
/**
|
|
5
|
+
* Drizzle database handle. Identical to what the stores and UsagePlugin receive.
|
|
6
|
+
*/
|
|
7
|
+
export type Database = BetterSQLite3Database<any>;
|
|
8
|
+
/** A single row from the `task_usage` table, as returned by drizzle select. */
|
|
9
|
+
export type TaskUsageRow = typeof taskUsageTable.$inferSelect;
|
|
10
|
+
export interface UsageQueryResult {
|
|
11
|
+
rows: TaskUsageRow[];
|
|
12
|
+
summary: {
|
|
13
|
+
totalInputTokens: number;
|
|
14
|
+
totalOutputTokens: number;
|
|
15
|
+
totalToolCalls: number;
|
|
16
|
+
totalModelCalls: number;
|
|
17
|
+
taskCount: number;
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* `usage_query` tool implementation.
|
|
22
|
+
*
|
|
23
|
+
* Queries `task_usage` with optional filters. Every filter is optional; a bare
|
|
24
|
+
* `{}` (or `undefined`) returns the most recent rows, capped at `limit`.
|
|
25
|
+
*
|
|
26
|
+
* - `task_id`: returns that row AND the full delegation subtree
|
|
27
|
+
* (`task_id = ? OR root_task_id = ?`).
|
|
28
|
+
* - `root_task_id`: returns the root row plus every descendant
|
|
29
|
+
* (`task_id = ? OR root_task_id = ?`).
|
|
30
|
+
* - `agent_name`, `since`: plain equality / lower-bound filters.
|
|
31
|
+
* - `include_incomplete` (default false): when false, only `is_complete = 1`
|
|
32
|
+
* rows are returned.
|
|
33
|
+
*
|
|
34
|
+
* Returns `{ rows: [], summary: { ...zeros, taskCount: 0 } }` when the table
|
|
35
|
+
* is empty or nothing matches.
|
|
36
|
+
*/
|
|
37
|
+
export declare function usageQuery(db: Database, input: TaskUsageInput): UsageQueryResult;
|
|
38
|
+
/**
|
|
39
|
+
* Builds a {@link TaskUsageReport} for enriching a `task` / `result` response.
|
|
40
|
+
*
|
|
41
|
+
* - `direct`: aggregates `task_usage WHERE task_id = ?` (this task's own model
|
|
42
|
+
* calls only).
|
|
43
|
+
* - `subtree`: aggregates `task_usage WHERE task_id = ? OR root_task_id = ?`
|
|
44
|
+
* (this task plus every sub-task that recorded it as its root).
|
|
45
|
+
* - `taskCount`: number of distinct rows folded into the subtree.
|
|
46
|
+
*
|
|
47
|
+
* Returns `undefined` when no `task_usage` row exists for the task — i.e. the
|
|
48
|
+
* task ran zero model calls (still running, or cancelled before its first
|
|
49
|
+
* model call). See `[shape:TaskUsageReport]`.
|
|
50
|
+
*
|
|
51
|
+
* claudecli rows are present with zero token counts and are returned as-is.
|
|
52
|
+
*/
|
|
53
|
+
export declare function buildTaskUsageReport(db: Database, taskId: string): TaskUsageReport | undefined;
|
|
54
|
+
//# sourceMappingURL=usage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/usage.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EACV,cAAc,EACd,eAAe,EAEhB,MAAM,wBAAwB,CAAC;AAEhC;;GAEG;AAEH,MAAM,MAAM,QAAQ,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC;AAElD,+EAA+E;AAC/E,MAAM,MAAM,YAAY,GAAG,OAAO,cAAc,CAAC,YAAY,CAAC;AAE9D,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,YAAY,EAAE,CAAC;IACrB,OAAO,EAAE;QACP,gBAAgB,EAAE,MAAM,CAAC;QACzB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AA6BD;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,UAAU,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,cAAc,GAAG,gBAAgB,CA4ChF;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,QAAQ,EACZ,MAAM,EAAE,MAAM,GACb,eAAe,GAAG,SAAS,CAkB7B"}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
import { and, desc, eq, gte, or } from "drizzle-orm";
|
|
2
|
+
import { taskUsageTable } from "../db/schema.js";
|
|
3
|
+
/**
|
|
4
|
+
* Folds a set of `task_usage` rows into a single {@link UsageSummary}.
|
|
5
|
+
*
|
|
6
|
+
* claudecli rows carry zeros for token counts — they are summed as-is, so a
|
|
7
|
+
* caller seeing `inputTokens === 0` must not conclude "zero tokens consumed";
|
|
8
|
+
* it may mean the provider does not report token counts. See
|
|
9
|
+
* `[inv:claudecli-undefined]`.
|
|
10
|
+
*/
|
|
11
|
+
function summarise(rows) {
|
|
12
|
+
return rows.reduce((acc, row) => ({
|
|
13
|
+
inputTokens: acc.inputTokens + (row.inputTokens ?? 0),
|
|
14
|
+
outputTokens: acc.outputTokens + (row.outputTokens ?? 0),
|
|
15
|
+
modelCalls: acc.modelCalls + (row.modelCalls ?? 0),
|
|
16
|
+
toolCallCount: acc.toolCallCount + (row.toolCallCount ?? 0),
|
|
17
|
+
latencyMs: acc.latencyMs + (row.latencyMs ?? 0),
|
|
18
|
+
}), {
|
|
19
|
+
inputTokens: 0,
|
|
20
|
+
outputTokens: 0,
|
|
21
|
+
modelCalls: 0,
|
|
22
|
+
toolCallCount: 0,
|
|
23
|
+
latencyMs: 0,
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* `usage_query` tool implementation.
|
|
28
|
+
*
|
|
29
|
+
* Queries `task_usage` with optional filters. Every filter is optional; a bare
|
|
30
|
+
* `{}` (or `undefined`) returns the most recent rows, capped at `limit`.
|
|
31
|
+
*
|
|
32
|
+
* - `task_id`: returns that row AND the full delegation subtree
|
|
33
|
+
* (`task_id = ? OR root_task_id = ?`).
|
|
34
|
+
* - `root_task_id`: returns the root row plus every descendant
|
|
35
|
+
* (`task_id = ? OR root_task_id = ?`).
|
|
36
|
+
* - `agent_name`, `since`: plain equality / lower-bound filters.
|
|
37
|
+
* - `include_incomplete` (default false): when false, only `is_complete = 1`
|
|
38
|
+
* rows are returned.
|
|
39
|
+
*
|
|
40
|
+
* Returns `{ rows: [], summary: { ...zeros, taskCount: 0 } }` when the table
|
|
41
|
+
* is empty or nothing matches.
|
|
42
|
+
*/
|
|
43
|
+
export function usageQuery(db, input) {
|
|
44
|
+
const filters = [];
|
|
45
|
+
// task_id and root_task_id both expand to a subtree match.
|
|
46
|
+
if (input?.task_id) {
|
|
47
|
+
const id = input.task_id;
|
|
48
|
+
const subtree = or(eq(taskUsageTable.taskId, id), eq(taskUsageTable.rootTaskId, id));
|
|
49
|
+
if (subtree)
|
|
50
|
+
filters.push(subtree);
|
|
51
|
+
}
|
|
52
|
+
if (input?.root_task_id) {
|
|
53
|
+
const id = input.root_task_id;
|
|
54
|
+
const subtree = or(eq(taskUsageTable.taskId, id), eq(taskUsageTable.rootTaskId, id));
|
|
55
|
+
if (subtree)
|
|
56
|
+
filters.push(subtree);
|
|
57
|
+
}
|
|
58
|
+
if (input?.agent_name) {
|
|
59
|
+
filters.push(eq(taskUsageTable.agentName, input.agent_name));
|
|
60
|
+
}
|
|
61
|
+
if (input?.since) {
|
|
62
|
+
filters.push(gte(taskUsageTable.createdAt, input.since));
|
|
63
|
+
}
|
|
64
|
+
if (!input?.include_incomplete) {
|
|
65
|
+
filters.push(eq(taskUsageTable.isComplete, 1));
|
|
66
|
+
}
|
|
67
|
+
const limit = input?.limit ?? 50;
|
|
68
|
+
const rows = db
|
|
69
|
+
.select()
|
|
70
|
+
.from(taskUsageTable)
|
|
71
|
+
.where(filters.length > 0 ? and(...filters) : undefined)
|
|
72
|
+
.orderBy(desc(taskUsageTable.createdAt))
|
|
73
|
+
.limit(limit)
|
|
74
|
+
.all();
|
|
75
|
+
return {
|
|
76
|
+
rows,
|
|
77
|
+
summary: {
|
|
78
|
+
totalInputTokens: rows.reduce((n, r) => n + (r.inputTokens ?? 0), 0),
|
|
79
|
+
totalOutputTokens: rows.reduce((n, r) => n + (r.outputTokens ?? 0), 0),
|
|
80
|
+
totalToolCalls: rows.reduce((n, r) => n + (r.toolCallCount ?? 0), 0),
|
|
81
|
+
totalModelCalls: rows.reduce((n, r) => n + (r.modelCalls ?? 0), 0),
|
|
82
|
+
taskCount: rows.length,
|
|
83
|
+
},
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
/**
|
|
87
|
+
* Builds a {@link TaskUsageReport} for enriching a `task` / `result` response.
|
|
88
|
+
*
|
|
89
|
+
* - `direct`: aggregates `task_usage WHERE task_id = ?` (this task's own model
|
|
90
|
+
* calls only).
|
|
91
|
+
* - `subtree`: aggregates `task_usage WHERE task_id = ? OR root_task_id = ?`
|
|
92
|
+
* (this task plus every sub-task that recorded it as its root).
|
|
93
|
+
* - `taskCount`: number of distinct rows folded into the subtree.
|
|
94
|
+
*
|
|
95
|
+
* Returns `undefined` when no `task_usage` row exists for the task — i.e. the
|
|
96
|
+
* task ran zero model calls (still running, or cancelled before its first
|
|
97
|
+
* model call). See `[shape:TaskUsageReport]`.
|
|
98
|
+
*
|
|
99
|
+
* claudecli rows are present with zero token counts and are returned as-is.
|
|
100
|
+
*/
|
|
101
|
+
export function buildTaskUsageReport(db, taskId) {
|
|
102
|
+
const subtreeRows = db
|
|
103
|
+
.select()
|
|
104
|
+
.from(taskUsageTable)
|
|
105
|
+
.where(or(eq(taskUsageTable.taskId, taskId), eq(taskUsageTable.rootTaskId, taskId)))
|
|
106
|
+
.all();
|
|
107
|
+
if (subtreeRows.length === 0) {
|
|
108
|
+
return undefined;
|
|
109
|
+
}
|
|
110
|
+
const directRows = subtreeRows.filter(row => row.taskId === taskId);
|
|
111
|
+
return {
|
|
112
|
+
direct: summarise(directRows),
|
|
113
|
+
subtree: summarise(subtreeRows),
|
|
114
|
+
taskCount: subtreeRows.length,
|
|
115
|
+
};
|
|
116
|
+
}
|
|
117
|
+
//# sourceMappingURL=usage.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../../../../packages/ai/agent-mcp/src/tools/usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAY,MAAM,aAAa,CAAC;AAI/D,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AA2BjD;;;;;;;GAOG;AACH,SAAS,SAAS,CAAC,IAAoB;IACrC,OAAO,IAAI,CAAC,MAAM,CAChB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACb,WAAW,EAAE,GAAG,CAAC,WAAW,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;QACrD,YAAY,EAAE,GAAG,CAAC,YAAY,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;QACxD,UAAU,EAAE,GAAG,CAAC,UAAU,GAAG,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC;QAClD,aAAa,EAAE,GAAG,CAAC,aAAa,GAAG,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,CAAC;QAC3D,SAAS,EAAE,GAAG,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC;KAChD,CAAC,EACF;QACE,WAAW,EAAE,CAAC;QACd,YAAY,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;QACb,aAAa,EAAE,CAAC;QAChB,SAAS,EAAE,CAAC;KACb,CACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,UAAU,UAAU,CAAC,EAAY,EAAE,KAAqB;IAC5D,MAAM,OAAO,GAAU,EAAE,CAAC;IAE1B,2DAA2D;IAC3D,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;QACnB,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;QACzB,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,IAAI,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,KAAK,EAAE,YAAY,EAAE,CAAC;QACxB,MAAM,EAAE,GAAG,KAAK,CAAC,YAAY,CAAC;QAC9B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,IAAI,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;IAC/D,CAAC;IACD,IAAI,KAAK,EAAE,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,CAAC,KAAK,EAAE,kBAAkB,EAAE,CAAC;QAC/B,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC;IAEjC,MAAM,IAAI,GAAG,EAAE;SACZ,MAAM,EAAE;SACR,IAAI,CAAC,cAAc,CAAC;SACpB,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;SACvD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;SACvC,KAAK,CAAC,KAAK,CAAC;SACZ,GAAG,EAAE,CAAC;IAET,OAAO;QACL,IAAI;QACJ,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACpE,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACtE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YACpE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YAClE,SAAS,EAAE,IAAI,CAAC,MAAM;SACvB;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,oBAAoB,CAClC,EAAY,EACZ,MAAc;IAEd,MAAM,WAAW,GAAG,EAAE;SACnB,MAAM,EAAE;SACR,IAAI,CAAC,cAAc,CAAC;SACpB,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;SACnF,GAAG,EAAE,CAAC;IAET,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;IAEpE,OAAO;QACL,MAAM,EAAE,SAAS,CAAC,UAAU,CAAC;QAC7B,OAAO,EAAE,SAAS,CAAC,WAAW,CAAC;QAC/B,SAAS,EAAE,WAAW,CAAC,MAAM;KAC9B,CAAC;AACJ,CAAC"}
|