@multiplayer-app/agent 0.1.9 → 0.1.11
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/README.md +106 -0
- package/dist/agent/definition.d.ts +22 -0
- package/dist/agent/definition.d.ts.map +1 -0
- package/dist/agent/definition.js +3 -0
- package/dist/agent/definition.js.map +1 -0
- package/dist/agent/definition.test.d.ts +2 -0
- package/dist/agent/definition.test.d.ts.map +1 -0
- package/dist/agent/definition.test.js +40 -0
- package/dist/agent/definition.test.js.map +1 -0
- package/dist/agent/model-failure.d.ts +10 -0
- package/dist/agent/model-failure.d.ts.map +1 -0
- package/dist/agent/model-failure.js +50 -0
- package/dist/agent/model-failure.js.map +1 -0
- package/dist/agent/profiles/coding.d.ts +4 -0
- package/dist/agent/profiles/coding.d.ts.map +1 -0
- package/dist/agent/profiles/coding.js +17 -0
- package/dist/agent/profiles/coding.js.map +1 -0
- package/dist/agent/profiles/index.d.ts +4 -0
- package/dist/agent/profiles/index.d.ts.map +1 -0
- package/dist/agent/profiles/index.js +9 -0
- package/dist/agent/profiles/index.js.map +1 -0
- package/dist/agent/profiles/research.d.ts +4 -0
- package/dist/agent/profiles/research.d.ts.map +1 -0
- package/dist/agent/profiles/research.js +53 -0
- package/dist/agent/profiles/research.js.map +1 -0
- package/dist/agent/progress-reminder.test.d.ts +2 -0
- package/dist/agent/progress-reminder.test.d.ts.map +1 -0
- package/dist/agent/progress-reminder.test.js +20 -0
- package/dist/agent/progress-reminder.test.js.map +1 -0
- package/dist/agent/turn-runner.d.ts +54 -0
- package/dist/agent/turn-runner.d.ts.map +1 -0
- package/dist/agent/turn-runner.js +147 -0
- package/dist/agent/turn-runner.js.map +1 -0
- package/dist/agent/turn-runner.test.d.ts +2 -0
- package/dist/agent/turn-runner.test.d.ts.map +1 -0
- package/dist/agent/turn-runner.test.js +181 -0
- package/dist/agent/turn-runner.test.js.map +1 -0
- package/dist/context-pruning.d.ts +13 -0
- package/dist/context-pruning.d.ts.map +1 -0
- package/dist/context-pruning.js +61 -0
- package/dist/context-pruning.js.map +1 -0
- package/dist/context-pruning.test.d.ts +2 -0
- package/dist/context-pruning.test.d.ts.map +1 -0
- package/dist/context-pruning.test.js +48 -0
- package/dist/context-pruning.test.js.map +1 -0
- package/dist/index.d.ts +11 -6
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +18 -10
- package/dist/index.js.map +1 -1
- package/dist/models/attribution.d.ts.map +1 -0
- package/dist/models/attribution.js.map +1 -0
- package/dist/{model.d.ts → models/resolve-model.d.ts} +2 -2
- package/dist/models/resolve-model.d.ts.map +1 -0
- package/dist/{model.js → models/resolve-model.js} +4 -2
- package/dist/models/resolve-model.js.map +1 -0
- package/dist/models/usage.d.ts +11 -0
- package/dist/models/usage.d.ts.map +1 -0
- package/dist/models/usage.js +19 -0
- package/dist/models/usage.js.map +1 -0
- package/dist/{git-sync.d.ts → persistence/git-sync.d.ts} +1 -1
- package/dist/persistence/git-sync.d.ts.map +1 -0
- package/dist/persistence/git-sync.js.map +1 -0
- package/dist/persistence/git-sync.test.d.ts.map +1 -0
- package/dist/{git-sync.test.js → persistence/git-sync.test.js} +2 -2
- package/dist/persistence/git-sync.test.js.map +1 -0
- package/dist/{session-storage-sync.d.ts → persistence/snapshot-sync.d.ts} +2 -2
- package/dist/persistence/snapshot-sync.d.ts.map +1 -0
- package/dist/{session-storage-sync.js → persistence/snapshot-sync.js} +1 -1
- package/dist/persistence/snapshot-sync.js.map +1 -0
- package/dist/persistence/snapshot-sync.test.d.ts +2 -0
- package/dist/persistence/snapshot-sync.test.d.ts.map +1 -0
- package/dist/{session-storage-sync.test.js → persistence/snapshot-sync.test.js} +3 -3
- package/dist/persistence/snapshot-sync.test.js.map +1 -0
- package/dist/persistence/workspace.d.ts +17 -0
- package/dist/persistence/workspace.d.ts.map +1 -0
- package/dist/persistence/workspace.js +26 -0
- package/dist/persistence/workspace.js.map +1 -0
- package/dist/persistence/workspace.test.d.ts +2 -0
- package/dist/persistence/workspace.test.d.ts.map +1 -0
- package/dist/persistence/workspace.test.js +38 -0
- package/dist/persistence/workspace.test.js.map +1 -0
- package/dist/research/extract-texts.d.ts +41 -0
- package/dist/research/extract-texts.d.ts.map +1 -0
- package/dist/research/extract-texts.js +239 -0
- package/dist/research/extract-texts.js.map +1 -0
- package/dist/research/extract-texts.test.d.ts +2 -0
- package/dist/research/extract-texts.test.d.ts.map +1 -0
- package/dist/research/extract-texts.test.js +120 -0
- package/dist/research/extract-texts.test.js.map +1 -0
- package/dist/research/extraction-tool.d.ts +6 -0
- package/dist/research/extraction-tool.d.ts.map +1 -0
- package/dist/research/extraction-tool.js +41 -0
- package/dist/research/extraction-tool.js.map +1 -0
- package/dist/research/extraction-tool.test.d.ts +2 -0
- package/dist/research/extraction-tool.test.d.ts.map +1 -0
- package/dist/research/extraction-tool.test.js +102 -0
- package/dist/research/extraction-tool.test.js.map +1 -0
- package/dist/research/filingramp/attachment-tools.d.ts +4 -0
- package/dist/research/filingramp/attachment-tools.d.ts.map +1 -0
- package/dist/research/filingramp/attachment-tools.js +198 -0
- package/dist/research/filingramp/attachment-tools.js.map +1 -0
- package/dist/research/filingramp/document-search-tool.d.ts +4 -0
- package/dist/research/filingramp/document-search-tool.d.ts.map +1 -0
- package/dist/research/filingramp/document-search-tool.js +231 -0
- package/dist/research/filingramp/document-search-tool.js.map +1 -0
- package/dist/research/filingramp/filing-detail-tool.d.ts +8 -0
- package/dist/research/filingramp/filing-detail-tool.d.ts.map +1 -0
- package/dist/research/filingramp/filing-detail-tool.js +161 -0
- package/dist/research/filingramp/filing-detail-tool.js.map +1 -0
- package/dist/research/filingramp/filing-search-tools.d.ts +4 -0
- package/dist/research/filingramp/filing-search-tools.d.ts.map +1 -0
- package/dist/research/filingramp/filing-search-tools.js +207 -0
- package/dist/research/filingramp/filing-search-tools.js.map +1 -0
- package/dist/research/filingramp/shared.d.ts +64 -0
- package/dist/research/filingramp/shared.d.ts.map +1 -0
- package/dist/research/filingramp/shared.js +130 -0
- package/dist/research/filingramp/shared.js.map +1 -0
- package/dist/research/filingramp-client.d.ts +1 -1
- package/dist/research/filingramp-client.d.ts.map +1 -1
- package/dist/research/filingramp-contract.d.ts.map +1 -1
- package/dist/research/filingramp-contract.js +8 -2
- package/dist/research/filingramp-contract.js.map +1 -1
- package/dist/research/filingramp-tools.d.ts +4 -8
- package/dist/research/filingramp-tools.d.ts.map +1 -1
- package/dist/research/filingramp-tools.js +19 -597
- package/dist/research/filingramp-tools.js.map +1 -1
- package/dist/research/filingramp-tools.test.js +138 -2
- package/dist/research/filingramp-tools.test.js.map +1 -1
- package/dist/research/prompt.d.ts +1 -1
- package/dist/research/prompt.d.ts.map +1 -1
- package/dist/research/prompt.js +16 -7
- package/dist/research/prompt.js.map +1 -1
- package/dist/research/scope.d.ts +1 -1
- package/dist/research/scope.d.ts.map +1 -1
- package/dist/runtime/cli.d.ts.map +1 -0
- package/dist/{cli.js → runtime/cli.js} +3 -3
- package/dist/runtime/cli.js.map +1 -0
- package/dist/{config.d.ts → runtime/config.d.ts} +1 -1
- package/dist/runtime/config.d.ts.map +1 -0
- package/dist/{config.js → runtime/config.js} +1 -1
- package/dist/runtime/config.js.map +1 -0
- package/dist/{runtime.d.ts → runtime/index.d.ts} +2 -2
- package/dist/runtime/index.d.ts.map +1 -0
- package/dist/{runtime.js → runtime/index.js} +7 -7
- package/dist/runtime/index.js.map +1 -0
- package/dist/runtime/lambda-microvm.d.ts +3 -0
- package/dist/runtime/lambda-microvm.d.ts.map +1 -0
- package/dist/{lambda-microvm-entrypoint.js → runtime/lambda-microvm.js} +4 -4
- package/dist/runtime/lambda-microvm.js.map +1 -0
- package/dist/runtime/lifecycle.test.d.ts.map +1 -0
- package/dist/{lifecycle.test.js → runtime/lifecycle.test.js} +5 -5
- package/dist/runtime/lifecycle.test.js.map +1 -0
- package/dist/session/checkpoint-timer.test.d.ts.map +1 -0
- package/dist/{checkpoint-timer.test.js → session/checkpoint-timer.test.js} +8 -8
- package/dist/session/checkpoint-timer.test.js.map +1 -0
- package/dist/session/conversation.d.ts +15 -0
- package/dist/session/conversation.d.ts.map +1 -0
- package/dist/session/conversation.js +61 -0
- package/dist/session/conversation.js.map +1 -0
- package/dist/session/conversation.test.d.ts +2 -0
- package/dist/session/conversation.test.d.ts.map +1 -0
- package/dist/session/conversation.test.js +53 -0
- package/dist/session/conversation.test.js.map +1 -0
- package/dist/session/idle-checkpoint.test.d.ts.map +1 -0
- package/dist/{idle-checkpoint.test.js → session/idle-checkpoint.test.js} +8 -4
- package/dist/session/idle-checkpoint.test.js.map +1 -0
- package/dist/session/index.d.ts +7 -0
- package/dist/session/index.d.ts.map +1 -0
- package/dist/session/index.js +16 -0
- package/dist/session/index.js.map +1 -0
- package/dist/session/runtime.d.ts +58 -0
- package/dist/session/runtime.d.ts.map +1 -0
- package/dist/session/runtime.js +292 -0
- package/dist/session/runtime.js.map +1 -0
- package/dist/session/runtime.test.d.ts +2 -0
- package/dist/session/runtime.test.d.ts.map +1 -0
- package/dist/session/runtime.test.js +201 -0
- package/dist/session/runtime.test.js.map +1 -0
- package/dist/session/tool-context.d.ts +12 -0
- package/dist/session/tool-context.d.ts.map +1 -0
- package/dist/session/tool-context.js +3 -0
- package/dist/session/tool-context.js.map +1 -0
- package/dist/session/tools.d.ts +6 -0
- package/dist/session/tools.d.ts.map +1 -0
- package/dist/session/tools.js +97 -0
- package/dist/session/tools.js.map +1 -0
- package/dist/session/tools.test.d.ts +2 -0
- package/dist/session/tools.test.d.ts.map +1 -0
- package/dist/session/tools.test.js +55 -0
- package/dist/session/tools.test.js.map +1 -0
- package/dist/task/loop.d.ts +6 -0
- package/dist/task/loop.d.ts.map +1 -0
- package/dist/{loop.js → task/loop.js} +6 -6
- package/dist/task/loop.js.map +1 -0
- package/dist/tool-call-repair.d.ts +11 -0
- package/dist/tool-call-repair.d.ts.map +1 -0
- package/dist/tool-call-repair.js +91 -0
- package/dist/tool-call-repair.js.map +1 -0
- package/dist/tool-call-repair.test.d.ts +2 -0
- package/dist/tool-call-repair.test.d.ts.map +1 -0
- package/dist/tool-call-repair.test.js +50 -0
- package/dist/tool-call-repair.test.js.map +1 -0
- package/dist/tools/exec-tools.d.ts +1 -0
- package/dist/tools/exec-tools.d.ts.map +1 -1
- package/dist/tools/exec-tools.js +23 -3
- package/dist/tools/exec-tools.js.map +1 -1
- package/dist/transport/link.d.ts.map +1 -0
- package/dist/transport/link.js.map +1 -0
- package/dist/{protocol.d.ts → transport/protocol.d.ts} +4 -0
- package/dist/transport/protocol.d.ts.map +1 -0
- package/dist/transport/protocol.js.map +1 -0
- package/dist/transport/protocol.test.d.ts.map +1 -0
- package/dist/transport/protocol.test.js.map +1 -0
- package/dist/{report.d.ts → transport/reporter.d.ts} +1 -1
- package/dist/transport/reporter.d.ts.map +1 -0
- package/dist/{report.js → transport/reporter.js} +1 -1
- package/dist/transport/reporter.js.map +1 -0
- package/package.json +11 -11
- package/dist/attribution.d.ts.map +0 -1
- package/dist/attribution.js.map +0 -1
- package/dist/checkpoint-timer.test.d.ts.map +0 -1
- package/dist/checkpoint-timer.test.js.map +0 -1
- package/dist/cli.d.ts.map +0 -1
- package/dist/cli.js.map +0 -1
- package/dist/config.d.ts.map +0 -1
- package/dist/config.js.map +0 -1
- package/dist/git-sync.d.ts.map +0 -1
- package/dist/git-sync.js.map +0 -1
- package/dist/git-sync.test.d.ts.map +0 -1
- package/dist/git-sync.test.js.map +0 -1
- package/dist/idle-checkpoint.test.d.ts.map +0 -1
- package/dist/idle-checkpoint.test.js.map +0 -1
- package/dist/lambda-microvm-entrypoint.d.ts +0 -3
- package/dist/lambda-microvm-entrypoint.d.ts.map +0 -1
- package/dist/lambda-microvm-entrypoint.js.map +0 -1
- package/dist/lifecycle.test.d.ts.map +0 -1
- package/dist/lifecycle.test.js.map +0 -1
- package/dist/link.d.ts.map +0 -1
- package/dist/link.js.map +0 -1
- package/dist/loop.d.ts +0 -6
- package/dist/loop.d.ts.map +0 -1
- package/dist/loop.js.map +0 -1
- package/dist/model.d.ts.map +0 -1
- package/dist/model.js.map +0 -1
- package/dist/protocol.d.ts.map +0 -1
- package/dist/protocol.js.map +0 -1
- package/dist/protocol.test.d.ts.map +0 -1
- package/dist/protocol.test.js.map +0 -1
- package/dist/report.d.ts.map +0 -1
- package/dist/report.js.map +0 -1
- package/dist/runtime.d.ts.map +0 -1
- package/dist/runtime.js.map +0 -1
- package/dist/session-storage-sync.d.ts.map +0 -1
- package/dist/session-storage-sync.js.map +0 -1
- package/dist/session-storage-sync.test.d.ts +0 -2
- package/dist/session-storage-sync.test.d.ts.map +0 -1
- package/dist/session-storage-sync.test.js.map +0 -1
- package/dist/session.d.ts +0 -46
- package/dist/session.d.ts.map +0 -1
- package/dist/session.js +0 -512
- package/dist/session.js.map +0 -1
- /package/dist/{attribution.d.ts → models/attribution.d.ts} +0 -0
- /package/dist/{attribution.js → models/attribution.js} +0 -0
- /package/dist/{git-sync.js → persistence/git-sync.js} +0 -0
- /package/dist/{git-sync.test.d.ts → persistence/git-sync.test.d.ts} +0 -0
- /package/dist/{cli.d.ts → runtime/cli.d.ts} +0 -0
- /package/dist/{lifecycle.test.d.ts → runtime/lifecycle.test.d.ts} +0 -0
- /package/dist/{checkpoint-timer.test.d.ts → session/checkpoint-timer.test.d.ts} +0 -0
- /package/dist/{idle-checkpoint.test.d.ts → session/idle-checkpoint.test.d.ts} +0 -0
- /package/dist/{link.d.ts → transport/link.d.ts} +0 -0
- /package/dist/{link.js → transport/link.js} +0 -0
- /package/dist/{protocol.js → transport/protocol.js} +0 -0
- /package/dist/{protocol.test.d.ts → transport/protocol.test.d.ts} +0 -0
- /package/dist/{protocol.test.js → transport/protocol.test.js} +0 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extract-texts.js","sourceRoot":"","sources":["../../src/research/extract-texts.ts"],"names":[],"mappings":";;;;;;AAoEA,8BAeC;AASD,oCAQC;AAmBD,4CA0HC;AAjPD,8FAA8F;AAC9F,6FAA6F;AAC7F,uEAAuE;AACvE,EAAE;AACF,8FAA8F;AAC9F,iGAAiG;AACjG,uFAAuF;AACvF,2BAAmC;AACnC,gDAAuB;AACvB,2BAAqD;AACrD,yDAAiD;AAEpC,QAAA,sBAAsB,GAAG,KAAM,CAAA;AAC5C,MAAM,qBAAqB,GAAG,IAAK,CAAA;AACnC,MAAM,mBAAmB,GAAG,CAAC,CAAA;AAC7B,MAAM,mBAAmB,GAAG,CAAC,CAAA;AAC7B,mGAAmG;AACnG,MAAM,oBAAoB,GAAG,GAAG,CAAA;AAChC,MAAM,aAAa,GAAG,GAAG,CAAA;AAEzB,MAAM,UAAU,GACd,oRAAoR,CAAA;AACtR,MAAM,WAAW,GACf,2aAA2a,CAAA;AAwB7a,SAAS,MAAM,CAAC,YAAoB,EAAE,QAAgB;IACpD,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAA;IACnD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,cAAI,CAAC,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,QAAQ,EAAE,CAAC,CAAA;IACzH,OAAO,MAAM,CAAA;AACf,CAAC;AAED,kGAAkG;AAClG,KAAK,UAAU,SAAS,CAAC,YAAoB,EAAE,OAA0B;;IACvE,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,MAAA,OAAO,CAAC,KAAK,mCAAI,EAAE,CAAC,CAAC,CAAA;IAC1C,IAAI,WAAW,GAAG,CAAC,CAAA;IACnB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,KAAK,MAAM,GAAG,IAAI,MAAM,IAAA,+BAAY,EAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrE,MAAM,SAAS,GAAG,OAAO,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAA;YACpJ,IAAI,SAAS;gBAAE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;;gBACjC,WAAW,IAAI,CAAC,CAAA;QACvB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,WAAW,EAAE,CAAA;AACxD,CAAC;AAED,kGAAkG;AAClG,SAAgB,SAAS,CAAC,IAAY,EAAE,QAAkB,EAAE,MAAM,GAAG,qBAAqB,EAAE,UAAU,GAAG,mBAAmB;IAC1H,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAA;IACrF,MAAM,KAAK,GAA4B,EAAE,CAAA;IACzC,IAAI,OAAO,GAAG,CAAC,CAAA;IACf,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACzC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC;YAAE,MAAK;QAChC,OAAO,IAAI,CAAC,CAAA;QACZ,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,CAAA;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,MAAM,CAAC,CAAA;QACzE,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QACpC,IAAI,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC;YAAE,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;;YACzD,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAA;IAC/B,CAAC;IACD,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,UAAU,KAAK,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,CAAA;IACvH,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAA;AAC9B,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,MAAM,MAAM,GAAa,EAAE,CAAA;IAC3B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,8BAAsB;QAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,8BAAsB,CAAC,CAAC,CAAA;IACxI,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;AACtC,CAAC;AAED,8GAA8G;AAC9G,SAAgB,YAAY,CAAC,GAA4B,EAAE,MAAgB,EAAE,OAAkC;IAC7G,MAAM,GAAG,GAA4B,EAAE,GAAG,GAAG,EAAE,CAAA;IAC/C,KAAK,MAAM,KAAK,IAAI,MAAM;QAAE,IAAI,CAAC,CAAC,KAAK,IAAI,GAAG,CAAC;YAAE,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,CAAA;IAClE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,CAAC;SAC1C,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;SACpH,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAA;IAC1B,IAAI,OAAO,CAAC,MAAM;QAAE,GAAG,CAAC,OAAO,GAAG,OAAO,CAAA;IACzC,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,IAAa;IAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAA;IACvC,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAA;IACzE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAA4B,CAAA;IAC9D,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,MAAM,CAAC,CAAA;IAC1B,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAE,MAAM,CAAC,IAAkC,CAAC,CAAC,CAAC,EAAE,CAAA;AACrF,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAc;IACvC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,aAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC7C,OAAO,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAA0B,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;IAC1H,CAAC;IAAC,WAAM,CAAC;QACP,OAAO,IAAI,GAAG,EAAE,CAAA;IAClB,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,gBAAgB,CAAC,OAA0B,EAAE,IAAoB;;IACrF,IAAI,CAAC,CAAA,MAAA,OAAO,CAAC,KAAK,0CAAE,MAAM,CAAA,IAAI,CAAC,OAAO,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;IAC7F,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAA;IAC5E,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,CAAC,CAAA;IAC5D,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAA;IACzC,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAA;IAChE,MAAM,IAAI,GAAG,MAAA,OAAO,CAAC,IAAI,mCAAI,KAAK,CAAA;IAClC,MAAM,OAAO,GAAG,KAAK,EAAE,MAAc,EAAE,EAAE;;QACvC,MAAM,IAAI,GAAG,MAAM,aAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,CAAC,CAAA;QACzE,IAAI,CAAC,CAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,QAAQ,CAAC,MAAM,CAAA;YAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,SAA+B,EAAE,CAAA;QACjH,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;QAC3H,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,EAAE,CAAA;IAC5E,CAAC,CAAA;IAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAA;QACb,IAAI,KAAK,GAAG,CAAC,CAAA;QACb,IAAI,OAAO,GAAG,CAAC,CAAA;QACf,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAc,EAAE,CAAC,CAAC,CAAA;YAChF,IAAI,CAAC,KAAK,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC,CAAA;YAC/B,KAAK,IAAI,KAAK,CAAC,MAAM,CAAA;YACrB,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAC5D,CAAC;QACD,OAAO;YACL,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,OAAO,CAAC,MAAM;YACzB,WAAW,EAAE,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM;YAC5C,uBAAuB,EAAE,WAAW;YACpC,OAAO;YACP,KAAK;YACL,UAAU,EAAE,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,oBAAoB,GAAG,aAAa,CAAC;YACvF,KAAK,EAAE,IAAI,CAAC,OAAO;SACpB,CAAA;IACH,CAAC;IAED,MAAM,aAAE,CAAC,KAAK,CAAC,cAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IACzD,MAAM,QAAQ,GAA4C,EAAE,CAAA;IAC5D,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,IAAI,GAAG,CAAC,CAAA;IACZ,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,OAAO,GAAG,CAAC,CAAA;IACf,IAAI,UAAU,GAAG,CAAC,CAAA;IAClB,IAAI,IAAI,GAAG,CAAC,CAAA;IACZ,MAAM,MAAM,GAAG,KAAK,EAAE,GAA4B,EAAE,EAAE;QACpD,MAAM,aAAE,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACvD,IAAI,IAAI,CAAC,CAAA;QACT,IAAI,GAAG,CAAC,KAAK,KAAK,IAAI;YAAE,KAAK,IAAI,CAAC,CAAA;IACpC,CAAC,CAAA;IAED,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,CAAA;IAC1B,MAAM,MAAM,GAAG,KAAK,IAAI,EAAE;;QACxB,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC;YACpB,IAAI,MAAA,IAAI,CAAC,MAAM,0CAAE,OAAO;gBAAE,OAAM;YAChC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAY,CAAA;YACtC,IAAI,CAAC;gBACH,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAA;gBAChD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;oBAClB,0FAA0F;oBAC1F,OAAO,IAAI,CAAC,CAAA;oBACZ,MAAM,MAAM,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAA;oBAC7G,SAAQ;gBACV,CAAC;gBACD,MAAM,OAAO,GAA8B,EAAE,CAAA;gBAC7C,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;oBAC5C,KAAK,IAAI,CAAC,CAAA;oBACV,UAAU,IAAI,IAAI,CAAC,MAAM,CAAA;oBACzB,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC;wBAChC,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,WAAW,EAAE,IAAI,CAAC,MAAM;wBACxB,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU;wBACvC,MAAM,EAAE,aAAa,OAAO,CAAC,QAAQ,aAAa,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GACzE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAC7E,iBAAiB,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,KAAK,GAAG,CAAC,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,OAAO,oBAAoB,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE;qBACxK,CAAC,CAAA;oBACF,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAA;oBAC1B,IAAI,KAAK,EAAE,CAAC;wBACV,MAAM,WAAW,GAAG,MAAA,KAAK,CAAC,WAAW,mCAAI,CAAC,CAAA;wBAC1C,MAAM,YAAY,GAAG,MAAA,KAAK,CAAC,YAAY,mCAAI,CAAC,CAAA;wBAC5C,IAAI,CAAC,WAAW,CAAC;4BACf,WAAW;4BACX,YAAY;4BACZ,WAAW,EAAE,MAAA,KAAK,CAAC,WAAW,mCAAI,WAAW,GAAG,YAAY;4BAC5D,eAAe,EAAE,MAAA,KAAK,CAAC,iBAAiB,0CAAE,eAAe;4BACzD,gBAAgB,EAAE,MAAA,KAAK,CAAC,iBAAiB,0CAAE,gBAAgB;yBAC5D,CAAC,CAAA;oBACJ,CAAC;oBACD,KAAK,MAAM,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC;wBAAE,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAA;gBACtI,CAAC;gBACD,0FAA0F;gBAC1F,yFAAyF;gBACzF,kEAAkE;gBAClE,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAA,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,KAAK,IAAI,CAAC,mCAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;gBACvG,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;gBAC1E,KAAK,MAAM,GAAG,IAAI,KAAK;oBAAE,MAAM,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;YAC3F,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC,CAAA;YACnE,CAAC;oBAAS,CAAC;gBACT,IAAI,IAAI,CAAC,CAAA;gBACT,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,OAAO,CAAC,MAAM;oBAAE,MAAA,IAAI,CAAC,UAAU,qDAAG,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;YACxF,CAAC;QACH,CAAC;IACH,CAAC,CAAA;IACD,MAAA,IAAI,CAAC,UAAU,qDAAG,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;IACpC,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,MAAA,OAAO,CAAC,WAAW,mCAAI,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAA;IAEpI,OAAO;QACL,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,SAAS,EAAE,OAAO,CAAC,MAAM;QACzB,WAAW,EAAE,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM;QAC5C,uBAAuB,EAAE,WAAW;QACpC,OAAO;QACP,KAAK;QACL,IAAI;QACJ,KAAK;QACL,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9B,UAAU;QACV,KAAK,EAAE,IAAI,CAAC,OAAO;QACnB,IAAI,EAAE,yFAAyF;KAChG,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extract-texts.test.d.ts","sourceRoot":"","sources":["../../src/research/extract-texts.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const strict_1 = __importDefault(require("node:assert/strict"));
|
|
7
|
+
const node_fs_1 = require("node:fs");
|
|
8
|
+
const node_os_1 = __importDefault(require("node:os"));
|
|
9
|
+
const node_path_1 = __importDefault(require("node:path"));
|
|
10
|
+
const node_test_1 = require("node:test");
|
|
11
|
+
const test_1 = require("ai/test");
|
|
12
|
+
const extract_texts_1 = require("./extract-texts");
|
|
13
|
+
function answer(text) {
|
|
14
|
+
return {
|
|
15
|
+
content: [{ type: 'text', text }],
|
|
16
|
+
finishReason: { unified: 'stop', raw: 'stop' },
|
|
17
|
+
usage: {
|
|
18
|
+
inputTokens: { total: 100, noCache: 20, cacheRead: 80, cacheWrite: 0 },
|
|
19
|
+
outputTokens: { total: 10, text: 10, reasoning: 0 },
|
|
20
|
+
},
|
|
21
|
+
warnings: [],
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
async function workspace(files) {
|
|
25
|
+
const dir = await node_fs_1.promises.mkdtemp(node_path_1.default.join(node_os_1.default.tmpdir(), 'extract-'));
|
|
26
|
+
for (const [name, content] of Object.entries(files)) {
|
|
27
|
+
await node_fs_1.promises.mkdir(node_path_1.default.dirname(node_path_1.default.join(dir, name)), { recursive: true });
|
|
28
|
+
await node_fs_1.promises.writeFile(node_path_1.default.join(dir, name), content);
|
|
29
|
+
}
|
|
30
|
+
return dir;
|
|
31
|
+
}
|
|
32
|
+
(0, node_test_1.describe)('windowsOf', () => {
|
|
33
|
+
(0, node_test_1.it)('merges overlapping passages and labels their offsets', () => {
|
|
34
|
+
const text = `${'x'.repeat(5000)}cancel within 30 days${'y'.repeat(100)}nonrenewal 45 days${'z'.repeat(5000)}`;
|
|
35
|
+
const { passages, matches } = (0, extract_texts_1.windowsOf)(text, ['cancel', 'non-?renew'], 500);
|
|
36
|
+
strict_1.default.equal(matches, 2);
|
|
37
|
+
strict_1.default.equal(passages.length, 1);
|
|
38
|
+
strict_1.default.match(passages[0], /^\[chars 4500–/);
|
|
39
|
+
strict_1.default.match(passages[0], /cancel within 30 days[\s\S]*nonrenewal 45 days/);
|
|
40
|
+
});
|
|
41
|
+
(0, node_test_1.it)('finds nothing in a document without a match', () => {
|
|
42
|
+
strict_1.default.deepEqual((0, extract_texts_1.windowsOf)('homeowners policy', ['water shut-?off']).passages, []);
|
|
43
|
+
});
|
|
44
|
+
});
|
|
45
|
+
(0, node_test_1.describe)('normalizeRow', () => {
|
|
46
|
+
(0, node_test_1.it)('fills missing fields and flags values outside the codebook', () => {
|
|
47
|
+
const row = (0, extract_texts_1.normalizeRow)({ category: 'surcharge', credit: 5 }, ['category', 'credit', 'appliesTo'], { category: ['credit', 'eligibility'] });
|
|
48
|
+
strict_1.default.equal(row.appliesTo, null);
|
|
49
|
+
strict_1.default.deepEqual(row.invalid, ['category']);
|
|
50
|
+
});
|
|
51
|
+
});
|
|
52
|
+
(0, node_test_1.describe)('extractFromTexts', () => {
|
|
53
|
+
(0, node_test_1.it)('sends only the windows, skips documents without a match and resumes', async () => {
|
|
54
|
+
const dir = await workspace({
|
|
55
|
+
'data/text/a.txt': `${'filler '.repeat(20000)}Notice of cancellation: 30 days. Nonrenewal: 45 days.`,
|
|
56
|
+
'data/text/b.txt': 'Nothing relevant here.',
|
|
57
|
+
'data/manifest.jsonl': '{"textPath":"data/text/a.txt"}\n{"textPath":"data/text/b.txt"}\n{"name":"scan.pdf"}\n',
|
|
58
|
+
});
|
|
59
|
+
const prompts = [];
|
|
60
|
+
const model = new test_1.MockLanguageModelV3({
|
|
61
|
+
doGenerate: async (options) => {
|
|
62
|
+
prompts.push(JSON.stringify(options.prompt));
|
|
63
|
+
return answer('{"rows":[{"noticeType":"cancellation","days":30,"quote":"30 days","found":true},{"noticeType":"nonrenewal","days":45,"quote":"45 days","found":true}]}');
|
|
64
|
+
},
|
|
65
|
+
});
|
|
66
|
+
const usage = [];
|
|
67
|
+
const request = {
|
|
68
|
+
manifest: 'data/manifest.jsonl',
|
|
69
|
+
question: 'Notice periods',
|
|
70
|
+
fields: ['noticeType', 'days'],
|
|
71
|
+
outputPath: 'data/extracted/notices.jsonl',
|
|
72
|
+
windows: { patterns: ['cancel', 'non-?renew'] },
|
|
73
|
+
many: true,
|
|
74
|
+
};
|
|
75
|
+
const deps = { workspaceDir: dir, model, modelId: 'mock', recordUsage: (u) => usage.push(u.totalTokens) };
|
|
76
|
+
const result = await (0, extract_texts_1.extractFromTexts)(request, deps);
|
|
77
|
+
strict_1.default.equal(result.documents, 2);
|
|
78
|
+
strict_1.default.equal(result.manifestRowsWithoutText, 1);
|
|
79
|
+
strict_1.default.equal(result.noMatch, 1);
|
|
80
|
+
strict_1.default.equal(result.calls, 1);
|
|
81
|
+
strict_1.default.equal(result.found, 2);
|
|
82
|
+
strict_1.default.ok(prompts[0].length < 10000, 'only the passages reach the model');
|
|
83
|
+
strict_1.default.deepEqual(usage, [110]);
|
|
84
|
+
const rows = (await node_fs_1.promises.readFile(node_path_1.default.join(dir, 'data/extracted/notices.jsonl'), 'utf8')).trim().split('\n').map((line) => JSON.parse(line));
|
|
85
|
+
strict_1.default.deepEqual(rows.map((row) => { var _a; return [row.source, (_a = row.days) !== null && _a !== void 0 ? _a : null]; }).sort(), [['data/text/a.txt', 30], ['data/text/a.txt', 45], ['data/text/b.txt', null]]);
|
|
86
|
+
const again = await (0, extract_texts_1.extractFromTexts)(request, deps);
|
|
87
|
+
strict_1.default.equal(again.documents, 0);
|
|
88
|
+
strict_1.default.equal(again.alreadyDone, 2);
|
|
89
|
+
strict_1.default.equal(prompts.length, 1);
|
|
90
|
+
});
|
|
91
|
+
(0, node_test_1.it)('reads a long document in chunks instead of cutting it, and dryRun calls no model', async () => {
|
|
92
|
+
const dir = await workspace({ 'data/text/manual.txt': `${'a'.repeat(extract_texts_1.EXTRACTION_CHUNK_CHARS)}water shut-off credit 5%` });
|
|
93
|
+
let calls = 0;
|
|
94
|
+
const model = new test_1.MockLanguageModelV3({
|
|
95
|
+
doGenerate: async (options) => {
|
|
96
|
+
calls += 1;
|
|
97
|
+
const found = JSON.stringify(options.prompt).includes('credit 5%');
|
|
98
|
+
return answer(found ? '{"credit":"5%","quote":"credit 5%","found":true}' : '{"credit":null,"quote":null,"found":false}');
|
|
99
|
+
},
|
|
100
|
+
});
|
|
101
|
+
const request = { paths: ['data/text/manual.txt'], question: 'Water device credit', fields: ['credit'], outputPath: 'data/out.jsonl' };
|
|
102
|
+
const deps = { workspaceDir: dir, model, modelId: 'mock', recordUsage: () => undefined };
|
|
103
|
+
const estimate = await (0, extract_texts_1.extractFromTexts)({ ...request, dryRun: true }, deps);
|
|
104
|
+
strict_1.default.equal(estimate.calls, 2);
|
|
105
|
+
strict_1.default.equal(calls, 0);
|
|
106
|
+
const result = await (0, extract_texts_1.extractFromTexts)(request, deps);
|
|
107
|
+
strict_1.default.equal(result.calls, 2);
|
|
108
|
+
const [row] = (await node_fs_1.promises.readFile(node_path_1.default.join(dir, 'data/out.jsonl'), 'utf8')).trim().split('\n').map((line) => JSON.parse(line));
|
|
109
|
+
strict_1.default.equal(row.credit, '5%');
|
|
110
|
+
strict_1.default.equal(row.part, 2);
|
|
111
|
+
});
|
|
112
|
+
(0, node_test_1.it)('keeps the fields the model filled when it answers found=false', async () => {
|
|
113
|
+
const dir = await workspace({ 'data/text/form.txt': 'GL 4065 new endorsement text' });
|
|
114
|
+
const model = new test_1.MockLanguageModelV3({ doGenerate: async () => answer('{"purpose":"Adds a new exclusion","quote":null,"found":false}') });
|
|
115
|
+
await (0, extract_texts_1.extractFromTexts)({ paths: ['data/text/form.txt'], question: 'Summarize', fields: ['purpose'], outputPath: 'data/out.jsonl' }, { workspaceDir: dir, model, modelId: 'mock', recordUsage: () => undefined });
|
|
116
|
+
const [row] = (await node_fs_1.promises.readFile(node_path_1.default.join(dir, 'data/out.jsonl'), 'utf8')).trim().split('\n').map((line) => JSON.parse(line));
|
|
117
|
+
strict_1.default.equal(row.purpose, 'Adds a new exclusion');
|
|
118
|
+
});
|
|
119
|
+
});
|
|
120
|
+
//# sourceMappingURL=extract-texts.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extract-texts.test.js","sourceRoot":"","sources":["../../src/research/extract-texts.test.ts"],"names":[],"mappings":";;;;;AAAA,gEAAuC;AACvC,qCAAwC;AACxC,sDAAwB;AACxB,0DAA4B;AAC5B,yCAAwC;AACxC,kCAA6C;AAC7C,mDAAmG;AAEnG,SAAS,MAAM,CAAC,IAAY;IAC1B,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;QAC1C,YAAY,EAAE,EAAE,OAAO,EAAE,MAAe,EAAE,GAAG,EAAE,MAAM,EAAE;QACvD,KAAK,EAAE;YACL,WAAW,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE;YACtE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE;SACpD;QACD,QAAQ,EAAE,EAAE;KACb,CAAA;AACH,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAA6B;IACpD,MAAM,GAAG,GAAG,MAAM,kBAAE,CAAC,OAAO,CAAC,mBAAI,CAAC,IAAI,CAAC,iBAAE,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC,CAAA;IAChE,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACpD,MAAM,kBAAE,CAAC,KAAK,CAAC,mBAAI,CAAC,OAAO,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QACvE,MAAM,kBAAE,CAAC,SAAS,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAA;IACnD,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,IAAA,oBAAQ,EAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAA,cAAE,EAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,IAAI,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,IAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,qBAAqB,GAAG,CAAC,MAAM,CAAC,IAAK,CAAC,EAAE,CAAA;QAChH,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAA,yBAAS,EAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,GAAG,CAAC,CAAA;QAC5E,gBAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;QACxB,gBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAChC,gBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAA;QAC3C,gBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,gDAAgD,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,IAAA,cAAE,EAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,gBAAM,CAAC,SAAS,CAAC,IAAA,yBAAS,EAAC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;IACpF,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,IAAA,oBAAQ,EAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAA,cAAE,EAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,GAAG,GAAG,IAAA,4BAAY,EAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,WAAW,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,CAAC,CAAA;QAC5I,gBAAM,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;QACjC,gBAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,IAAA,oBAAQ,EAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAA,cAAE,EAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC;YAC1B,iBAAiB,EAAE,GAAG,SAAS,CAAC,MAAM,CAAC,KAAM,CAAC,uDAAuD;YACrG,iBAAiB,EAAE,wBAAwB;YAC3C,qBAAqB,EAAE,uFAAuF;SAC/G,CAAC,CAAA;QACF,MAAM,OAAO,GAAa,EAAE,CAAA;QAC5B,MAAM,KAAK,GAAG,IAAI,0BAAmB,CAAC;YACpC,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAA;gBAC5C,OAAO,MAAM,CAAC,wJAAwJ,CAAC,CAAA;YACzK,CAAC;SACF,CAAC,CAAA;QACF,MAAM,KAAK,GAAa,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE,qBAAqB;YAC/B,QAAQ,EAAE,gBAAgB;YAC1B,MAAM,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC;YAC9B,UAAU,EAAE,8BAA8B;YAC1C,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE;YAC/C,IAAI,EAAE,IAAI;SACX,CAAA;QACD,MAAM,IAAI,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAA0B,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE,CAAA;QAClI,MAAM,MAAM,GAAG,MAAM,IAAA,gCAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,CAAA;QAEpD,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAA;QACjC,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAA;QAC/C,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;QAC/B,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAC7B,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAC7B,gBAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,KAAM,EAAE,mCAAmC,CAAC,CAAA;QAC1E,gBAAM,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,GAAG,CAAC,CAAC,CAAA;QAC9B,MAAM,IAAI,GAAG,CAAC,MAAM,kBAAE,CAAC,QAAQ,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,8BAA8B,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC3I,gBAAM,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,WAAC,OAAA,CAAC,GAAG,CAAC,MAAM,EAAE,MAAA,GAAG,CAAC,IAAI,mCAAI,IAAI,CAAC,CAAA,EAAA,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,CAAC,CAAA;QAEzJ,MAAM,KAAK,GAAG,MAAM,IAAA,gCAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,CAAA;QACnD,gBAAM,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAA;QAChC,gBAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAA;QAClC,gBAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,IAAA,cAAE,EAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,EAAE,sBAAsB,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,sCAAsB,CAAC,0BAA0B,EAAE,CAAC,CAAA;QACxH,IAAI,KAAK,GAAG,CAAC,CAAA;QACb,MAAM,KAAK,GAAG,IAAI,0BAAmB,CAAC;YACpC,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5B,KAAK,IAAI,CAAC,CAAA;gBACV,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAA;gBAClE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,kDAAkD,CAAC,CAAC,CAAC,4CAA4C,CAAC,CAAA;YAC1H,CAAC;SACF,CAAC,CAAA;QACF,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,qBAAqB,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAA;QACtI,MAAM,IAAI,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE,CAAA;QAExF,MAAM,QAAQ,GAAG,MAAM,IAAA,gCAAgB,EAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,CAAC,CAAA;QAC3E,gBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAC/B,gBAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAEtB,MAAM,MAAM,GAAG,MAAM,IAAA,gCAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,CAAA;QACpD,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAC7B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,kBAAE,CAAC,QAAQ,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC9H,gBAAM,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QAC9B,gBAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;IAC3B,CAAC,CAAC,CAAA;IAEF,IAAA,cAAE,EAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,EAAE,oBAAoB,EAAE,8BAA8B,EAAE,CAAC,CAAA;QACrF,MAAM,KAAK,GAAG,IAAI,0BAAmB,CAAC,EAAE,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,+DAA+D,CAAC,EAAE,CAAC,CAAA;QAC1I,MAAM,IAAA,gCAAgB,EACpB,EAAE,KAAK,EAAE,CAAC,oBAAoB,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAC3G,EAAE,YAAY,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE,CAC5E,CAAA;QACD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,kBAAE,CAAC,QAAQ,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC9H,gBAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { type Tool } from 'ai';
|
|
2
|
+
import type { AgentModel } from '../agent/definition';
|
|
3
|
+
import type { SessionToolContext } from '../session/tool-context';
|
|
4
|
+
/** Bulk extraction on the cheaper model (extract-texts.ts): rows go to a JSONL file, the conversation gets counts. */
|
|
5
|
+
export declare function createExtractionTool(context: SessionToolContext, modelConfig: AgentModel): Tool;
|
|
6
|
+
//# sourceMappingURL=extraction-tool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extraction-tool.d.ts","sourceRoot":"","sources":["../../src/research/extraction-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,KAAK,IAAI,EAAE,MAAM,IAAI,CAAA;AAGpC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAA;AACrD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAGjE,sHAAsH;AACtH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,UAAU,GAAG,IAAI,CAkC/F"}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createExtractionTool = createExtractionTool;
|
|
4
|
+
const ai_1 = require("ai");
|
|
5
|
+
const zod_1 = require("zod");
|
|
6
|
+
const resolve_model_1 = require("../models/resolve-model");
|
|
7
|
+
const extract_texts_1 = require("./extract-texts");
|
|
8
|
+
/** Bulk extraction on the cheaper model (extract-texts.ts): rows go to a JSONL file, the conversation gets counts. */
|
|
9
|
+
function createExtractionTool(context, modelConfig) {
|
|
10
|
+
const { workspaceDir, abortSignal, recordUsage } = context;
|
|
11
|
+
return (0, ai_1.tool)({
|
|
12
|
+
description: `Ask the same question of many extracted-text files with a cheaper model, writing JSON rows to outputPath (JSONL); you get counts back, never the documents. Input: a manifest (fetch_attachments outputManifest or a saveHits file) or paths. Use windows.patterns (regexes) whenever you know what the passage looks like: only the text around matches is sent, and documents without a match cost nothing. Without windows each document is read in ${extract_texts_1.EXTRACTION_CHUNK_CHARS.toLocaleString()}-character chunks. Set many=true when one document can state the fact several times (one row per statement). allowed holds codebook values per field; rows outside it are flagged "invalid". Run dryRun first on large sets to see the call count and estimated tokens. Re-running with the same outputPath skips documents already done.`,
|
|
13
|
+
inputSchema: zod_1.z.object({
|
|
14
|
+
paths: zod_1.z.array(zod_1.z.string().min(1).max(500)).min(1).max(200).optional(),
|
|
15
|
+
manifest: zod_1.z.string().min(1).max(300).optional().describe('JSONL/JSON manifest whose rows carry textPath'),
|
|
16
|
+
question: zod_1.z.string().min(1).max(2000).describe('What to look for, precisely'),
|
|
17
|
+
fields: zod_1.z.array(zod_1.z.string().min(1).max(60)).min(1).max(15).describe('Field names each row must contain'),
|
|
18
|
+
outputPath: zod_1.z.string().min(1).max(300).describe('e.g. data/extracted/water-valve.jsonl'),
|
|
19
|
+
windows: zod_1.z
|
|
20
|
+
.object({
|
|
21
|
+
patterns: zod_1.z.array(zod_1.z.string().min(1).max(200)).min(1).max(20).describe('Case-insensitive regexes, e.g. "cancel", "non-?renew", "water (shut|leak)"'),
|
|
22
|
+
radius: zod_1.z.number().int().min(200).max(6000).optional().describe('Characters kept on each side of a match (default 1500)'),
|
|
23
|
+
maxWindows: zod_1.z.number().int().min(1).max(20).optional().describe('Passages per document (default 6)'),
|
|
24
|
+
})
|
|
25
|
+
.optional(),
|
|
26
|
+
many: zod_1.z.boolean().optional(),
|
|
27
|
+
allowed: zod_1.z.record(zod_1.z.string(), zod_1.z.array(zod_1.z.string().max(100)).max(30)).optional().describe('Codebook: allowed values per field'),
|
|
28
|
+
concurrency: zod_1.z.number().int().min(1).max(8).optional(),
|
|
29
|
+
dryRun: zod_1.z.boolean().optional(),
|
|
30
|
+
}),
|
|
31
|
+
execute: async (input, { toolCallId }) => (0, extract_texts_1.extractFromTexts)(input, {
|
|
32
|
+
workspaceDir,
|
|
33
|
+
model: (0, resolve_model_1.resolveLanguageModel)(modelConfig.provider, modelConfig.id, modelConfig.attribution),
|
|
34
|
+
modelId: modelConfig.id,
|
|
35
|
+
signal: abortSignal,
|
|
36
|
+
recordUsage: (usage) => recordUsage(usage, modelConfig.id),
|
|
37
|
+
onProgress: (done, total) => context.reportProgress([{ key: 'extract', label: 'Extracting facts', status: done < total ? 'active' : 'done', detail: `${done}/${total}` }], toolCallId),
|
|
38
|
+
}),
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
//# sourceMappingURL=extraction-tool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extraction-tool.js","sourceRoot":"","sources":["../../src/research/extraction-tool.ts"],"names":[],"mappings":";;AAQA,oDAkCC;AA1CD,2BAAoC;AACpC,6BAAuB;AACvB,2DAA8D;AAG9D,mDAA0E;AAE1E,sHAAsH;AACtH,SAAgB,oBAAoB,CAAC,OAA2B,EAAE,WAAuB;IACvF,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,OAAO,CAAA;IAC1D,OAAO,IAAA,SAAI,EAAC;QACV,WAAW,EACT,0bAA0b,sCAAsB,CAAC,cAAc,EAAE,2UAA2U;QAC9yB,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;YACrE,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;YACzG,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,6BAA6B,CAAC;YAC7E,MAAM,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;YACvG,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,uCAAuC,CAAC;YACxF,OAAO,EAAE,OAAC;iBACP,MAAM,CAAC;gBACN,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,4EAA4E,CAAC;gBACnJ,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAK,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wDAAwD,CAAC;gBAC1H,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;aACrG,CAAC;iBACD,QAAQ,EAAE;YACb,IAAI,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;YAC5B,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;YAC7H,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;YACtD,MAAM,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;SAC/B,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CACvC,IAAA,gCAAgB,EAAC,KAAK,EAAE;YACtB,YAAY;YACZ,KAAK,EAAE,IAAA,oCAAoB,EAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,EAAE,WAAW,CAAC,WAAW,CAAC;YAC1F,OAAO,EAAE,WAAW,CAAC,EAAE;YACvB,MAAM,EAAE,WAAW;YACnB,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;YAC1D,UAAU,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAC1B,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,kBAAkB,EAAE,MAAM,EAAE,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC,EAAE,UAAU,CAAC;SAC5J,CAAC;KACL,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extraction-tool.test.d.ts","sourceRoot":"","sources":["../../src/research/extraction-tool.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,102 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
19
|
+
var ownKeys = function(o) {
|
|
20
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
21
|
+
var ar = [];
|
|
22
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
23
|
+
return ar;
|
|
24
|
+
};
|
|
25
|
+
return ownKeys(o);
|
|
26
|
+
};
|
|
27
|
+
return function (mod) {
|
|
28
|
+
if (mod && mod.__esModule) return mod;
|
|
29
|
+
var result = {};
|
|
30
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
31
|
+
__setModuleDefault(result, mod);
|
|
32
|
+
return result;
|
|
33
|
+
};
|
|
34
|
+
})();
|
|
35
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
36
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
37
|
+
};
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
const strict_1 = __importDefault(require("node:assert/strict"));
|
|
40
|
+
const node_fs_1 = require("node:fs");
|
|
41
|
+
const node_os_1 = __importDefault(require("node:os"));
|
|
42
|
+
const node_path_1 = __importDefault(require("node:path"));
|
|
43
|
+
const node_test_1 = require("node:test");
|
|
44
|
+
const test_1 = require("ai/test");
|
|
45
|
+
const modelProvider = __importStar(require("../models/resolve-model"));
|
|
46
|
+
const attribution_1 = require("../models/attribution");
|
|
47
|
+
const extraction_tool_1 = require("./extraction-tool");
|
|
48
|
+
for (const cancelAfterFirst of [false, true]) {
|
|
49
|
+
(0, node_test_1.it)(`bulk extraction ${cancelAfterFirst ? 'stops after cancellation' : 'records usage and continues past missing files'}`, async (t) => {
|
|
50
|
+
const dir = await node_fs_1.promises.mkdtemp(node_path_1.default.join(node_os_1.default.tmpdir(), 'agent-extraction-'));
|
|
51
|
+
t.after(() => node_fs_1.promises.rm(dir, { recursive: true, force: true }));
|
|
52
|
+
await node_fs_1.promises.writeFile(node_path_1.default.join(dir, 'first.txt'), 'A water valve is required.');
|
|
53
|
+
await node_fs_1.promises.writeFile(node_path_1.default.join(dir, 'second.txt'), 'Another filing.');
|
|
54
|
+
const model = new test_1.MockLanguageModelV3({
|
|
55
|
+
doGenerate: async () => ({
|
|
56
|
+
content: [{ type: 'text', text: '{"found":true,"quote":"A water valve is required."}' }],
|
|
57
|
+
finishReason: { unified: 'stop', raw: 'stop' },
|
|
58
|
+
usage: {
|
|
59
|
+
inputTokens: { total: 10, noCache: 10, cacheRead: 0, cacheWrite: 0 },
|
|
60
|
+
outputTokens: { total: 5, text: 5, reasoning: 0 },
|
|
61
|
+
},
|
|
62
|
+
warnings: [],
|
|
63
|
+
}),
|
|
64
|
+
});
|
|
65
|
+
t.mock.method(modelProvider, 'resolveLanguageModel', (_provider, modelId, app) => {
|
|
66
|
+
strict_1.default.equal(modelId, 'extraction-model');
|
|
67
|
+
strict_1.default.equal(app, attribution_1.OPENROUTER_APPS.DEEP_RESEARCH);
|
|
68
|
+
return model;
|
|
69
|
+
});
|
|
70
|
+
const turn = { messageId: 'extract-turn', abort: new AbortController(), askedUser: false };
|
|
71
|
+
const events = [];
|
|
72
|
+
let usedTokens = 0;
|
|
73
|
+
const context = {
|
|
74
|
+
workspaceDir: dir,
|
|
75
|
+
abortSignal: turn.abort.signal,
|
|
76
|
+
reportProgress: (steps, toolCallId) => events.push({ type: 'progress', messageId: turn.messageId, toolCallId, steps }),
|
|
77
|
+
requestClarification() { },
|
|
78
|
+
publishFiles() { },
|
|
79
|
+
recordUsage(usage, modelId) {
|
|
80
|
+
usedTokens += usage.totalTokens;
|
|
81
|
+
strict_1.default.equal(modelId, 'extraction-model');
|
|
82
|
+
if (cancelAfterFirst)
|
|
83
|
+
turn.abort.abort();
|
|
84
|
+
},
|
|
85
|
+
};
|
|
86
|
+
const result = await (0, extraction_tool_1.createExtractionTool)(context, { id: 'extraction-model', provider: {}, attribution: attribution_1.OPENROUTER_APPS.DEEP_RESEARCH }).execute({
|
|
87
|
+
paths: ['missing.txt', 'first.txt', 'second.txt'], question: 'Is a valve required?', fields: ['found'], outputPath: 'results/rows.jsonl',
|
|
88
|
+
// One document at a time, so the row order and the stop after cancellation are exact.
|
|
89
|
+
concurrency: 1,
|
|
90
|
+
}, { toolCallId: 'extract', messages: [] });
|
|
91
|
+
const expectedRows = cancelAfterFirst ? 1 : 2;
|
|
92
|
+
strict_1.default.equal(result.rows, expectedRows);
|
|
93
|
+
strict_1.default.deepEqual(result.failures.map((failure) => failure.path), ['missing.txt']);
|
|
94
|
+
strict_1.default.equal(model.doGenerateCalls.length, expectedRows);
|
|
95
|
+
strict_1.default.equal(usedTokens, 15 * expectedRows);
|
|
96
|
+
const rows = (await node_fs_1.promises.readFile(node_path_1.default.join(dir, 'results/rows.jsonl'), 'utf8')).trim().split('\n').map((line) => JSON.parse(line));
|
|
97
|
+
strict_1.default.equal(rows.length, expectedRows);
|
|
98
|
+
strict_1.default.deepEqual(rows[0], { source: 'first.txt', found: true, quote: 'A water valve is required.' });
|
|
99
|
+
strict_1.default.ok(events.every((event) => event.type === 'progress' && event.messageId === 'extract-turn' && event.toolCallId === 'extract'));
|
|
100
|
+
});
|
|
101
|
+
}
|
|
102
|
+
//# sourceMappingURL=extraction-tool.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extraction-tool.test.js","sourceRoot":"","sources":["../../src/research/extraction-tool.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gEAAuC;AACvC,qCAAwC;AACxC,sDAAwB;AACxB,0DAA4B;AAC5B,yCAA8B;AAC9B,kCAA6C;AAC7C,uEAAwD;AACxD,uDAAuD;AAGvD,uDAAwD;AAExD,KAAK,MAAM,gBAAgB,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;IAC7C,IAAA,cAAE,EAAC,mBAAmB,gBAAgB,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACpI,MAAM,GAAG,GAAG,MAAM,kBAAE,CAAC,OAAO,CAAC,mBAAI,CAAC,IAAI,CAAC,iBAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAA;QACzE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,kBAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC3D,MAAM,kBAAE,CAAC,SAAS,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,EAAE,4BAA4B,CAAC,CAAA;QAC7E,MAAM,kBAAE,CAAC,SAAS,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,EAAE,iBAAiB,CAAC,CAAA;QACnE,MAAM,KAAK,GAAG,IAAI,0BAAmB,CAAC;YACpC,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qDAAqD,EAAE,CAAC;gBACxF,YAAY,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE;gBAC9C,KAAK,EAAE;oBACL,WAAW,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;oBACpE,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;iBAClD;gBACD,QAAQ,EAAE,EAAE;aACb,CAAC;SACH,CAAC,CAAA;QACF,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,sBAAsB,EAAE,CAAC,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE;YAC/E,gBAAM,CAAC,KAAK,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAA;YACzC,gBAAM,CAAC,KAAK,CAAC,GAAG,EAAE,6BAAe,CAAC,aAAa,CAAC,CAAA;YAChD,OAAO,KAAK,CAAA;QACd,CAAC,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,IAAI,eAAe,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QAC1F,MAAM,MAAM,GAAoB,EAAE,CAAA;QAClC,IAAI,UAAU,GAAG,CAAC,CAAA;QAClB,MAAM,OAAO,GAAuB;YAClC,YAAY,EAAE,GAAG;YACjB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YAC9B,cAAc,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;YACtH,oBAAoB,KAAI,CAAC;YACzB,YAAY,KAAI,CAAC;YACjB,WAAW,CAAC,KAAK,EAAE,OAAO;gBACxB,UAAU,IAAI,KAAK,CAAC,WAAW,CAAA;gBAC/B,gBAAM,CAAC,KAAK,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAA;gBACzC,IAAI,gBAAgB;oBAAE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAA;YAC1C,CAAC;SACF,CAAA;QACD,MAAM,MAAM,GAAG,MAAM,IAAA,sCAAoB,EAAC,OAAO,EAAE,EAAE,EAAE,EAAE,kBAAkB,EAAE,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,6BAAe,CAAC,aAAa,EAAE,CAAC,CAAC,OAAQ,CAAC;YAChJ,KAAK,EAAE,CAAC,aAAa,EAAE,WAAW,EAAE,YAAY,CAAC,EAAE,QAAQ,EAAE,sBAAsB,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,oBAAoB;YACxI,sFAAsF;YACtF,WAAW,EAAE,CAAC;SACf,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAwD,CAAA;QAClG,MAAM,YAAY,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QAC7C,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,CAAA;QACvC,gBAAM,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAA;QACjF,gBAAM,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;QACxD,gBAAM,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,GAAG,YAAY,CAAC,CAAA;QAC3C,MAAM,IAAI,GAAG,CAAC,MAAM,kBAAE,CAAC,QAAQ,CAAC,mBAAI,CAAC,IAAI,CAAC,GAAG,EAAE,oBAAoB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QACjI,gBAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;QACvC,gBAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE,CAAC,CAAA;QACpG,gBAAM,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,SAAS,KAAK,cAAc,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAA;IACvI,CAAC,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"attachment-tools.d.ts","sourceRoot":"","sources":["../../../src/research/filingramp/attachment-tools.ts"],"names":[],"mappings":"AAIA,OAAO,EAAQ,KAAK,IAAI,EAAE,MAAM,IAAI,CAAA;AAGpC,OAAO,EAA+D,KAAK,sBAAsB,EAAE,MAAM,UAAU,CAAA;AAyBnH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,sBAAsB,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAyJ3F"}
|
|
@@ -0,0 +1,198 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.createAttachmentTools = createAttachmentTools;
|
|
7
|
+
const node_child_process_1 = require("node:child_process");
|
|
8
|
+
const node_fs_1 = require("node:fs");
|
|
9
|
+
const node_path_1 = __importDefault(require("node:path"));
|
|
10
|
+
const node_util_1 = require("node:util");
|
|
11
|
+
const ai_1 = require("ai");
|
|
12
|
+
const zod_1 = require("zod");
|
|
13
|
+
const filing_detail_tool_1 = require("./filing-detail-tool");
|
|
14
|
+
const shared_1 = require("./shared");
|
|
15
|
+
const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
|
|
16
|
+
const MAX_INLINE_TEXT_CHARS = 4000;
|
|
17
|
+
const MAX_FETCH_ATTACHMENTS = 2000;
|
|
18
|
+
const FETCH_CONCURRENCY = 4;
|
|
19
|
+
/** PDF → text with whatever the image has (mupdf first, poppler second). Empty text = scanned or protected. */
|
|
20
|
+
async function extractPdfText(file, output) {
|
|
21
|
+
const attempts = [
|
|
22
|
+
['mutool', ['draw', '-F', 'txt', '-o', output, file]],
|
|
23
|
+
['pdftotext', ['-layout', file, output]],
|
|
24
|
+
];
|
|
25
|
+
for (const [bin, args] of attempts) {
|
|
26
|
+
try {
|
|
27
|
+
await execFileAsync(bin, args, { timeout: 120000 });
|
|
28
|
+
const stat = await node_fs_1.promises.stat(output);
|
|
29
|
+
return stat.size;
|
|
30
|
+
}
|
|
31
|
+
catch (_a) {
|
|
32
|
+
/* try the next extractor */
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
return 0;
|
|
36
|
+
}
|
|
37
|
+
function createAttachmentTools(options) {
|
|
38
|
+
const { client, workspaceDir, onProgress } = options;
|
|
39
|
+
const resolveFilingId = (0, filing_detail_tool_1.createFilingIdResolver)(options);
|
|
40
|
+
/** One attachment to data/attachments/, PDF text to data/text/. With `reuse`, a file already on disk is not fetched again. */
|
|
41
|
+
const saveAttachment = async (input) => {
|
|
42
|
+
var _a, _b, _c;
|
|
43
|
+
const folder = (0, shared_1.safeName)((_a = input.trackingNumber) !== null && _a !== void 0 ? _a : 'misc');
|
|
44
|
+
const fileName = (0, shared_1.safeName)((_c = (_b = input.name) !== null && _b !== void 0 ? _b : input.key.split('/').pop()) !== null && _c !== void 0 ? _c : 'attachment');
|
|
45
|
+
const filePath = `data/attachments/${folder}/${fileName}`;
|
|
46
|
+
const absoluteFile = node_path_1.default.resolve(workspaceDir, filePath);
|
|
47
|
+
const isPdfName = /\.pdf$/i.test(fileName);
|
|
48
|
+
const textPathFor = () => `data/text/${folder}/${fileName.replace(/\.pdf$/i, '')}.txt`;
|
|
49
|
+
if (input.reuse && (await (0, shared_1.exists)(absoluteFile))) {
|
|
50
|
+
const bytes = (await node_fs_1.promises.stat(absoluteFile)).size;
|
|
51
|
+
const textFile = node_path_1.default.resolve(workspaceDir, textPathFor());
|
|
52
|
+
const textChars = (await (0, shared_1.exists)(textFile)) ? (await node_fs_1.promises.stat(textFile)).size : 0;
|
|
53
|
+
return { path: filePath, bytes, isPdf: isPdfName, textPath: textChars > 0 ? textPathFor() : undefined, textChars, reused: true };
|
|
54
|
+
}
|
|
55
|
+
const bytes = await client.downloadAttachment(input.key);
|
|
56
|
+
const written = await (0, shared_1.writeWorkspaceFile)(workspaceDir, filePath, bytes);
|
|
57
|
+
const isPdf = isPdfName || bytes.subarray(0, 4).toString() === '%PDF';
|
|
58
|
+
let textPath;
|
|
59
|
+
let textChars = 0;
|
|
60
|
+
if (isPdf && input.extractText !== false) {
|
|
61
|
+
textPath = textPathFor();
|
|
62
|
+
const absoluteText = node_path_1.default.resolve(workspaceDir, textPath);
|
|
63
|
+
await node_fs_1.promises.mkdir(node_path_1.default.dirname(absoluteText), { recursive: true });
|
|
64
|
+
textChars = await extractPdfText(absoluteFile, absoluteText);
|
|
65
|
+
if (textChars === 0)
|
|
66
|
+
textPath = undefined;
|
|
67
|
+
}
|
|
68
|
+
return { path: written.path, bytes: written.bytes, isPdf, textPath, textChars, reused: false };
|
|
69
|
+
};
|
|
70
|
+
const downloadAttachment = (0, ai_1.tool)({
|
|
71
|
+
description: 'Download ONE attachment by key into data/attachments/<trackingNumber>/<name> and, for PDFs, extract its text to data/text/<trackingNumber>/<name>.txt. For more than a few documents use fetch_attachments instead. Returns paths and sizes, never the content.',
|
|
72
|
+
inputSchema: zod_1.z.object({
|
|
73
|
+
key: zod_1.z.string().min(1).max(2000),
|
|
74
|
+
trackingNumber: zod_1.z.string().max(100).optional(),
|
|
75
|
+
name: zod_1.z.string().max(300).optional(),
|
|
76
|
+
extractText: zod_1.z.boolean().optional().describe('default true for PDFs'),
|
|
77
|
+
}),
|
|
78
|
+
execute: async (input) => {
|
|
79
|
+
const saved = await saveAttachment(input);
|
|
80
|
+
return {
|
|
81
|
+
...saved,
|
|
82
|
+
note: saved.isPdf && saved.textChars === 0 ? 'No extractable text (scanned or protected PDF) — record it as unreadable in findings.' : undefined,
|
|
83
|
+
};
|
|
84
|
+
},
|
|
85
|
+
});
|
|
86
|
+
const fetchAttachments = (0, ai_1.tool)({
|
|
87
|
+
description: 'Download many attachments in one call, with PDF text extracted to data/text/<trackingNumber>/. Give either a manifest (JSONL rows with `key`, optional `trackingNumber` and `name` — get_filing writes one, and search hits saved with saveHits work too) or filingIds (their attachments are listed for you). Files already on disk are kept, so a repeated call resumes. Writes one row per attachment (path, textPath, textChars, readable, error) to outputManifest and returns only counts. This is the only way to read more than a handful of documents: never call download_attachment in a loop.',
|
|
88
|
+
inputSchema: zod_1.z.object({
|
|
89
|
+
manifest: zod_1.z.string().min(1).max(300).optional().describe('Workspace JSONL/JSON manifest of attachments'),
|
|
90
|
+
filingIds: zod_1.z.array(zod_1.z.string().min(1).max(100)).min(1).max(50).optional().describe('Filing _ids or tracking numbers whose attachments to fetch'),
|
|
91
|
+
sections: zod_1.z.array(zod_1.z.enum(['forms', 'rates', 'supportings', 'objections', 'dispositions'])).optional().describe('Only attachments in these filing sections'),
|
|
92
|
+
nameMatch: zod_1.z.string().max(200).optional().describe('Case-insensitive regex on the attachment name, e.g. "\\\\.pdf$" or "rule|manual"'),
|
|
93
|
+
limit: zod_1.z.number().int().min(1).max(MAX_FETCH_ATTACHMENTS).optional(),
|
|
94
|
+
outputManifest: zod_1.z.string().max(300).optional().describe('Default data/manifests/fetched-<n>.jsonl'),
|
|
95
|
+
}),
|
|
96
|
+
execute: async (input, { toolCallId }) => {
|
|
97
|
+
var _a, _b, _c;
|
|
98
|
+
if (!input.manifest === !input.filingIds)
|
|
99
|
+
throw new Error('Pass exactly one of manifest or filingIds.');
|
|
100
|
+
let rows;
|
|
101
|
+
if (input.manifest) {
|
|
102
|
+
rows = await (0, shared_1.readManifest)(workspaceDir, input.manifest);
|
|
103
|
+
}
|
|
104
|
+
else {
|
|
105
|
+
rows = [];
|
|
106
|
+
for (const id of (_a = input.filingIds) !== null && _a !== void 0 ? _a : []) {
|
|
107
|
+
const filingId = await resolveFilingId(id);
|
|
108
|
+
const filing = await client.request(`/v1/filings/${encodeURIComponent(filingId)}`);
|
|
109
|
+
for (const attachment of (0, filing_detail_tool_1.collectAttachments)(filing))
|
|
110
|
+
rows.push({ filingId, trackingNumber: filing.trackingNumber, ...attachment });
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
const pattern = input.nameMatch ? new RegExp(input.nameMatch, 'i') : undefined;
|
|
114
|
+
const selected = rows
|
|
115
|
+
.map((row) => {
|
|
116
|
+
var _a, _b, _c, _d, _e;
|
|
117
|
+
return ({
|
|
118
|
+
...row,
|
|
119
|
+
key: (_a = row.key) !== null && _a !== void 0 ? _a : row.attachmentKey,
|
|
120
|
+
name: (_b = row.name) !== null && _b !== void 0 ? _b : (typeof ((_c = row.key) !== null && _c !== void 0 ? _c : row.attachmentKey) === 'string' ? decodeURIComponent((_e = String((_d = row.key) !== null && _d !== void 0 ? _d : row.attachmentKey).split('/').pop()) !== null && _e !== void 0 ? _e : '') : undefined),
|
|
121
|
+
});
|
|
122
|
+
})
|
|
123
|
+
.filter((row) => typeof row.key === 'string' && row.key.length > 0)
|
|
124
|
+
.filter((row) => !input.sections || input.sections.includes(row.section))
|
|
125
|
+
.filter((row) => { var _a; return !pattern || pattern.test(String((_a = row.name) !== null && _a !== void 0 ? _a : '')); });
|
|
126
|
+
// The same document can be listed twice (two hits on one attachment, a manifest merged from searches).
|
|
127
|
+
const unique = [...new Map(selected.map((row) => [String(row.key), row])).values()];
|
|
128
|
+
const limit = (_b = input.limit) !== null && _b !== void 0 ? _b : MAX_FETCH_ATTACHMENTS;
|
|
129
|
+
const batch = unique.slice(0, limit);
|
|
130
|
+
let done = 0;
|
|
131
|
+
onProgress === null || onProgress === void 0 ? void 0 : onProgress(toolCallId, 0, batch.length, 'Downloading attachments');
|
|
132
|
+
const results = await (0, shared_1.mapPool)(batch, FETCH_CONCURRENCY, async (row) => {
|
|
133
|
+
try {
|
|
134
|
+
const saved = await saveAttachment({
|
|
135
|
+
key: String(row.key),
|
|
136
|
+
trackingNumber: row.trackingNumber === undefined ? undefined : String(row.trackingNumber),
|
|
137
|
+
name: row.name === undefined ? undefined : String(row.name),
|
|
138
|
+
reuse: true,
|
|
139
|
+
});
|
|
140
|
+
return { ...row, path: saved.path, bytes: saved.bytes, textPath: saved.textPath, textChars: saved.textChars, readable: !saved.isPdf || saved.textChars > 0, reused: saved.reused };
|
|
141
|
+
}
|
|
142
|
+
catch (error) {
|
|
143
|
+
return { ...row, error: error.message };
|
|
144
|
+
}
|
|
145
|
+
finally {
|
|
146
|
+
done += 1;
|
|
147
|
+
if (done % 10 === 0 || done === batch.length)
|
|
148
|
+
onProgress === null || onProgress === void 0 ? void 0 : onProgress(toolCallId, done, batch.length, 'Downloading attachments');
|
|
149
|
+
}
|
|
150
|
+
});
|
|
151
|
+
const outputPath = (_c = input.outputManifest) !== null && _c !== void 0 ? _c : `data/manifests/fetched-${Date.now()}.jsonl`;
|
|
152
|
+
const written = await (0, shared_1.writeWorkspaceFile)(workspaceDir, outputPath, results.map((row) => JSON.stringify(row)).join('\n') + (results.length ? '\n' : ''));
|
|
153
|
+
const failed = results.filter((row) => 'error' in row);
|
|
154
|
+
const unreadable = results.filter((row) => !('error' in row) && row.readable === false);
|
|
155
|
+
return {
|
|
156
|
+
outputManifest: written.path,
|
|
157
|
+
requested: unique.length,
|
|
158
|
+
fetched: results.length - failed.length,
|
|
159
|
+
reused: results.filter((row) => row.reused).length,
|
|
160
|
+
failed: failed.length,
|
|
161
|
+
unreadable: unreadable.length,
|
|
162
|
+
notFetched: unique.length - batch.length,
|
|
163
|
+
duplicatesDropped: selected.length - unique.length,
|
|
164
|
+
textChars: results.reduce((sum, row) => sum + (typeof row.textChars === 'number' ? row.textChars : 0), 0),
|
|
165
|
+
failures: failed.slice(0, 5).map((row) => ({ name: row.name, error: row.error })),
|
|
166
|
+
next: 'Analyze the texts with extract_from_texts (manifest = outputManifest) or run_command; do not read whole documents into the chat.',
|
|
167
|
+
};
|
|
168
|
+
},
|
|
169
|
+
});
|
|
170
|
+
const readText = (0, ai_1.tool)({
|
|
171
|
+
description: 'Read a slice of an extracted text file in the workspace (data/text/…). Prefer run_command with rg/grep to locate the passage first, then read only the lines around it. Hard cap per call keeps the conversation small.',
|
|
172
|
+
inputSchema: zod_1.z.object({
|
|
173
|
+
path: zod_1.z.string().min(1).max(500),
|
|
174
|
+
startLine: zod_1.z.number().int().min(1).optional(),
|
|
175
|
+
endLine: zod_1.z.number().int().min(1).optional(),
|
|
176
|
+
}),
|
|
177
|
+
execute: async (input) => {
|
|
178
|
+
var _a, _b;
|
|
179
|
+
const target = node_path_1.default.resolve(workspaceDir, input.path);
|
|
180
|
+
if (!target.startsWith(node_path_1.default.resolve(workspaceDir) + node_path_1.default.sep))
|
|
181
|
+
throw new Error('path escapes the workspace');
|
|
182
|
+
const content = await node_fs_1.promises.readFile(target, 'utf8');
|
|
183
|
+
const lines = content.split('\n');
|
|
184
|
+
const start = Math.max(1, (_a = input.startLine) !== null && _a !== void 0 ? _a : 1);
|
|
185
|
+
const end = Math.min(lines.length, (_b = input.endLine) !== null && _b !== void 0 ? _b : start + 80);
|
|
186
|
+
const slice = lines.slice(start - 1, end).join('\n');
|
|
187
|
+
return {
|
|
188
|
+
path: input.path,
|
|
189
|
+
totalLines: lines.length,
|
|
190
|
+
startLine: start,
|
|
191
|
+
endLine: end,
|
|
192
|
+
text: slice.length > MAX_INLINE_TEXT_CHARS ? `${slice.slice(0, MAX_INLINE_TEXT_CHARS)}…[truncated — narrow the line range]` : slice,
|
|
193
|
+
};
|
|
194
|
+
},
|
|
195
|
+
});
|
|
196
|
+
return { download_attachment: downloadAttachment, fetch_attachments: fetchAttachments, read_text: readText };
|
|
197
|
+
}
|
|
198
|
+
//# sourceMappingURL=attachment-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"attachment-tools.js","sourceRoot":"","sources":["../../../src/research/filingramp/attachment-tools.ts"],"names":[],"mappings":";;;;;AAgCA,sDAyJC;AAzLD,2DAA6C;AAC7C,qCAAwC;AACxC,0DAA4B;AAC5B,yCAAqC;AACrC,2BAAoC;AACpC,6BAAuB;AACvB,6DAAiF;AACjF,qCAAmH;AAEnH,MAAM,aAAa,GAAG,IAAA,qBAAS,EAAC,6BAAQ,CAAC,CAAA;AACzC,MAAM,qBAAqB,GAAG,IAAK,CAAA;AACnC,MAAM,qBAAqB,GAAG,IAAK,CAAA;AACnC,MAAM,iBAAiB,GAAG,CAAC,CAAA;AAE3B,+GAA+G;AAC/G,KAAK,UAAU,cAAc,CAAC,IAAY,EAAE,MAAc;IACxD,MAAM,QAAQ,GAA8B;QAC1C,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACrD,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;KACzC,CAAA;IACD,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,QAAQ,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,aAAa,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,MAAO,EAAE,CAAC,CAAA;YACpD,MAAM,IAAI,GAAG,MAAM,kBAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAClC,OAAO,IAAI,CAAC,IAAI,CAAA;QAClB,CAAC;QAAC,WAAM,CAAC;YACP,4BAA4B;QAC9B,CAAC;IACH,CAAC;IACD,OAAO,CAAC,CAAA;AACV,CAAC;AAED,SAAgB,qBAAqB,CAAC,OAA+B;IACnE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,OAAO,CAAA;IACpD,MAAM,eAAe,GAAG,IAAA,2CAAsB,EAAC,OAAO,CAAC,CAAA;IAEvD,8HAA8H;IAC9H,MAAM,cAAc,GAAG,KAAK,EAAE,KAAsG,EAAE,EAAE;;QACtI,MAAM,MAAM,GAAG,IAAA,iBAAQ,EAAC,MAAA,KAAK,CAAC,cAAc,mCAAI,MAAM,CAAC,CAAA;QACvD,MAAM,QAAQ,GAAG,IAAA,iBAAQ,EAAC,MAAA,MAAA,KAAK,CAAC,IAAI,mCAAI,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,mCAAI,YAAY,CAAC,CAAA;QACnF,MAAM,QAAQ,GAAG,oBAAoB,MAAM,IAAI,QAAQ,EAAE,CAAA;QACzD,MAAM,YAAY,GAAG,mBAAI,CAAC,OAAO,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAA;QACzD,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC1C,MAAM,WAAW,GAAG,GAAG,EAAE,CAAC,aAAa,MAAM,IAAI,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,MAAM,CAAA;QACtF,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC,MAAM,IAAA,eAAM,EAAC,YAAY,CAAC,CAAC,EAAE,CAAC;YAChD,MAAM,KAAK,GAAG,CAAC,MAAM,kBAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAA;YAChD,MAAM,QAAQ,GAAG,mBAAI,CAAC,OAAO,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,CAAA;YAC1D,MAAM,SAAS,GAAG,CAAC,MAAM,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,kBAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;YAC/E,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QAClI,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QACxD,MAAM,OAAO,GAAG,MAAM,IAAA,2BAAkB,EAAC,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAA;QACvE,MAAM,KAAK,GAAG,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,MAAM,CAAA;QACrE,IAAI,QAA4B,CAAA;QAChC,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,IAAI,KAAK,IAAI,KAAK,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YACzC,QAAQ,GAAG,WAAW,EAAE,CAAA;YACxB,MAAM,YAAY,GAAG,mBAAI,CAAC,OAAO,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAA;YACzD,MAAM,kBAAE,CAAC,KAAK,CAAC,mBAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/D,SAAS,GAAG,MAAM,cAAc,CAAC,YAAY,EAAE,YAAY,CAAC,CAAA;YAC5D,IAAI,SAAS,KAAK,CAAC;gBAAE,QAAQ,GAAG,SAAS,CAAA;QAC3C,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,CAAA;IAChG,CAAC,CAAA;IAED,MAAM,kBAAkB,GAAG,IAAA,SAAI,EAAC;QAC9B,WAAW,EACT,iQAAiQ;QACnQ,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,GAAG,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;YAChC,cAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;YAC9C,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;YACpC,WAAW,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;SACtE,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,CAAA;YACzC,OAAO;gBACL,GAAG,KAAK;gBACR,IAAI,EAAE,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,uFAAuF,CAAC,CAAC,CAAC,SAAS;aACjJ,CAAA;QACH,CAAC;KACF,CAAC,CAAA;IAEF,MAAM,gBAAgB,GAAG,IAAA,SAAI,EAAC;QAC5B,WAAW,EACT,2kBAA2kB;QAC7kB,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;YACxG,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;YAC/I,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;YAC3J,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kFAAkF,CAAC;YACtI,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,QAAQ,EAAE;YACpE,cAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;SACpG,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;;YACvC,IAAI,CAAC,KAAK,CAAC,QAAQ,KAAK,CAAC,KAAK,CAAC,SAAS;gBAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAA;YACvG,IAAI,IAA+B,CAAA;YACnC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACnB,IAAI,GAAG,MAAM,IAAA,qBAAY,EAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAA;YACzD,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG,EAAE,CAAA;gBACT,KAAK,MAAM,EAAE,IAAI,MAAA,KAAK,CAAC,SAAS,mCAAI,EAAE,EAAE,CAAC;oBACvC,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,EAAE,CAAC,CAAA;oBAC1C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAA0B,eAAe,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAA;oBAC3G,KAAK,MAAM,UAAU,IAAI,IAAA,uCAAkB,EAAC,MAAM,CAAC;wBAAE,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,GAAG,UAAU,EAAE,CAAC,CAAA;gBACpI,CAAC;YACH,CAAC;YACD,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAC9E,MAAM,QAAQ,GAAG,IAAI;iBAClB,GAAG,CAAC,CAAC,GAAG,EAA2B,EAAE;;gBAAC,OAAA,CAAC;oBACtC,GAAG,GAAG;oBACN,GAAG,EAAE,MAAA,GAAG,CAAC,GAAG,mCAAI,GAAG,CAAC,aAAa;oBACjC,IAAI,EAAE,MAAA,GAAG,CAAC,IAAI,mCAAI,CAAC,OAAO,CAAC,MAAA,GAAG,CAAC,GAAG,mCAAI,GAAG,CAAC,aAAa,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAA,MAAM,CAAC,MAAA,GAAG,CAAC,GAAG,mCAAI,GAAG,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,mCAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;iBACrK,CAAC,CAAA;aAAA,CAAC;iBACF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,CAAC,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;iBAClE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAgB,CAAC,CAAC;iBACjF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,WAAC,OAAA,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,MAAA,GAAG,CAAC,IAAI,mCAAI,EAAE,CAAC,CAAC,CAAA,EAAA,CAAC,CAAA;YACpE,uGAAuG;YACvG,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAA;YACnF,MAAM,KAAK,GAAG,MAAA,KAAK,CAAC,KAAK,mCAAI,qBAAqB,CAAA;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;YACpC,IAAI,IAAI,GAAG,CAAC,CAAA;YACZ,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAG,UAAU,EAAE,CAAC,EAAE,KAAK,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAA;YACpE,MAAM,OAAO,GAAG,MAAM,IAAA,gBAAO,EAAC,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,GAAG,EAAoC,EAAE;gBACtG,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC;wBACjC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;wBACpB,cAAc,EAAE,GAAG,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC;wBACzF,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;wBAC3D,KAAK,EAAE,IAAI;qBACZ,CAAC,CAAA;oBACF,OAAO,EAAE,GAAG,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAA;gBACpL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,EAAE,GAAG,GAAG,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAA;gBACpD,CAAC;wBAAS,CAAC;oBACT,IAAI,IAAI,CAAC,CAAA;oBACT,IAAI,IAAI,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,KAAK,KAAK,CAAC,MAAM;wBAAE,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAG,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAA;gBACvH,CAAC;YACH,CAAC,CAAC,CAAA;YACF,MAAM,UAAU,GAAG,MAAA,KAAK,CAAC,cAAc,mCAAI,0BAA0B,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAA;YACvF,MAAM,OAAO,GAAG,MAAM,IAAA,2BAAkB,EAAC,YAAY,EAAE,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;YACvJ,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,IAAI,GAAG,CAAC,CAAA;YACtD,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,GAAG,CAAC,IAAI,GAAG,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAA;YACvF,OAAO;gBACL,cAAc,EAAE,OAAO,CAAC,IAAI;gBAC5B,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM;gBACvC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM;gBAClD,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,UAAU,EAAE,UAAU,CAAC,MAAM;gBAC7B,UAAU,EAAE,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM;gBACxC,iBAAiB,EAAE,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM;gBAClD,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzG,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjF,IAAI,EAAE,kIAAkI;aACzI,CAAA;QACH,CAAC;KACF,CAAC,CAAA;IAEF,MAAM,QAAQ,GAAG,IAAA,SAAI,EAAC;QACpB,WAAW,EACT,yNAAyN;QAC3N,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;YAChC,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;YAC7C,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;SAC5C,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;;YACvB,MAAM,MAAM,GAAG,mBAAI,CAAC,OAAO,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,CAAC,CAAA;YACrD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,mBAAI,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,mBAAI,CAAC,GAAG,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAA;YAC5G,MAAM,OAAO,GAAG,MAAM,kBAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAA,KAAK,CAAC,SAAS,mCAAI,CAAC,CAAC,CAAA;YAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAA,KAAK,CAAC,OAAO,mCAAI,KAAK,GAAG,EAAE,CAAC,CAAA;YAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACpD,OAAO;gBACL,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,UAAU,EAAE,KAAK,CAAC,MAAM;gBACxB,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE,GAAG;gBACZ,IAAI,EAAE,KAAK,CAAC,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,qBAAqB,CAAC,sCAAsC,CAAC,CAAC,CAAC,KAAK;aACpI,CAAA;QACH,CAAC;KACF,CAAC,CAAA;IACF,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAA;AAC9G,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-search-tool.d.ts","sourceRoot":"","sources":["../../../src/research/filingramp/document-search-tool.ts"],"names":[],"mappings":"AAGA,OAAO,EAAQ,KAAK,IAAI,EAAE,MAAM,IAAI,CAAA;AAKpC,OAAO,EAA6G,KAAK,sBAAsB,EAAE,MAAM,UAAU,CAAA;AASjK,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,sBAAsB,GAAG,IAAI,CA0N9E"}
|