@namzu/sdk 0.4.1 → 0.4.3
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/CHANGELOG.md +83 -0
- package/dist/__tests__/replay-public-surface.test.d.ts +2 -0
- package/dist/__tests__/replay-public-surface.test.d.ts.map +1 -0
- package/dist/__tests__/replay-public-surface.test.js +45 -0
- package/dist/__tests__/replay-public-surface.test.js.map +1 -0
- package/dist/advisory/context.test.d.ts +16 -0
- package/dist/advisory/context.test.d.ts.map +1 -0
- package/dist/advisory/context.test.js +92 -0
- package/dist/advisory/context.test.js.map +1 -0
- package/dist/advisory/evaluator.test.d.ts +34 -0
- package/dist/advisory/evaluator.test.d.ts.map +1 -0
- package/dist/advisory/evaluator.test.js +172 -0
- package/dist/advisory/evaluator.test.js.map +1 -0
- package/dist/advisory/executor.test.d.ts +35 -0
- package/dist/advisory/executor.test.d.ts.map +1 -0
- package/dist/advisory/executor.test.js +233 -0
- package/dist/advisory/executor.test.js.map +1 -0
- package/dist/advisory/registry.test.d.ts +16 -0
- package/dist/advisory/registry.test.d.ts.map +1 -0
- package/dist/advisory/registry.test.js +62 -0
- package/dist/advisory/registry.test.js.map +1 -0
- package/dist/agents/SupervisorAgent.d.ts.map +1 -1
- package/dist/agents/SupervisorAgent.js.map +1 -1
- package/dist/bridge/a2a/agent-card.test.d.ts +24 -0
- package/dist/bridge/a2a/agent-card.test.d.ts.map +1 -0
- package/dist/bridge/a2a/agent-card.test.js +118 -0
- package/dist/bridge/a2a/agent-card.test.js.map +1 -0
- package/dist/bridge/a2a/mapper.test.d.ts +29 -0
- package/dist/bridge/a2a/mapper.test.d.ts.map +1 -0
- package/dist/bridge/a2a/mapper.test.js +265 -0
- package/dist/bridge/a2a/mapper.test.js.map +1 -0
- package/dist/bridge/a2a/message.test.d.ts +20 -0
- package/dist/bridge/a2a/message.test.d.ts.map +1 -0
- package/dist/bridge/a2a/message.test.js +116 -0
- package/dist/bridge/a2a/message.test.js.map +1 -0
- package/dist/bridge/a2a/task.d.ts +2 -2
- package/dist/bridge/a2a/task.d.ts.map +1 -1
- package/dist/bridge/a2a/task.js.map +1 -1
- package/dist/bridge/a2a/task.test.d.ts +29 -0
- package/dist/bridge/a2a/task.test.d.ts.map +1 -0
- package/dist/bridge/a2a/task.test.js +198 -0
- package/dist/bridge/a2a/task.test.js.map +1 -0
- package/dist/bridge/mcp/connector/adapter.test.d.ts +27 -0
- package/dist/bridge/mcp/connector/adapter.test.d.ts.map +1 -0
- package/dist/bridge/mcp/connector/adapter.test.js +203 -0
- package/dist/bridge/mcp/connector/adapter.test.js.map +1 -0
- package/dist/bridge/sse/mapper.test.d.ts +27 -0
- package/dist/bridge/sse/mapper.test.d.ts.map +1 -0
- package/dist/bridge/sse/mapper.test.js +271 -0
- package/dist/bridge/sse/mapper.test.js.map +1 -0
- package/dist/bridge/tools/connector/adapter.test.d.ts +28 -0
- package/dist/bridge/tools/connector/adapter.test.d.ts.map +1 -0
- package/dist/bridge/tools/connector/adapter.test.js +182 -0
- package/dist/bridge/tools/connector/adapter.test.js.map +1 -0
- package/dist/bridge/tools/connector/definitions.test.d.ts +23 -0
- package/dist/bridge/tools/connector/definitions.test.d.ts.map +1 -0
- package/dist/bridge/tools/connector/definitions.test.js +158 -0
- package/dist/bridge/tools/connector/definitions.test.js.map +1 -0
- package/dist/bridge/tools/connector/router.test.d.ts +21 -0
- package/dist/bridge/tools/connector/router.test.d.ts.map +1 -0
- package/dist/bridge/tools/connector/router.test.js +139 -0
- package/dist/bridge/tools/connector/router.test.js.map +1 -0
- package/dist/bus/breaker.test.d.ts +41 -0
- package/dist/bus/breaker.test.d.ts.map +1 -0
- package/dist/bus/breaker.test.js +242 -0
- package/dist/bus/breaker.test.js.map +1 -0
- package/dist/bus/index.test.d.ts +25 -0
- package/dist/bus/index.test.d.ts.map +1 -0
- package/dist/bus/index.test.js +151 -0
- package/dist/bus/index.test.js.map +1 -0
- package/dist/bus/lock.test.d.ts +44 -0
- package/dist/bus/lock.test.d.ts.map +1 -0
- package/dist/bus/lock.test.js +226 -0
- package/dist/bus/lock.test.js.map +1 -0
- package/dist/bus/ownership.test.d.ts +26 -0
- package/dist/bus/ownership.test.d.ts.map +1 -0
- package/dist/bus/ownership.test.js +205 -0
- package/dist/bus/ownership.test.js.map +1 -0
- package/dist/config/runtime.d.ts +24 -25
- package/dist/config/runtime.d.ts.map +1 -1
- package/dist/config/runtime.js.map +1 -1
- package/dist/connector/BaseConnector.test.d.ts +21 -0
- package/dist/connector/BaseConnector.test.d.ts.map +1 -0
- package/dist/connector/BaseConnector.test.js +108 -0
- package/dist/connector/BaseConnector.test.js.map +1 -0
- package/dist/connector/builtins/http.test.d.ts +30 -0
- package/dist/connector/builtins/http.test.d.ts.map +1 -0
- package/dist/connector/builtins/http.test.js +232 -0
- package/dist/connector/builtins/http.test.js.map +1 -0
- package/dist/connector/builtins/webhook.test.d.ts +20 -0
- package/dist/connector/builtins/webhook.test.d.ts.map +1 -0
- package/dist/connector/builtins/webhook.test.js +113 -0
- package/dist/connector/builtins/webhook.test.js.map +1 -0
- package/dist/connector/execution/factory.test.d.ts +16 -0
- package/dist/connector/execution/factory.test.d.ts.map +1 -0
- package/dist/connector/execution/factory.test.js +64 -0
- package/dist/connector/execution/factory.test.js.map +1 -0
- package/dist/connector/execution/remote.test.d.ts +16 -0
- package/dist/connector/execution/remote.test.d.ts.map +1 -0
- package/dist/connector/execution/remote.test.js +53 -0
- package/dist/connector/execution/remote.test.js.map +1 -0
- package/dist/connector/mcp/adapter.test.d.ts +34 -0
- package/dist/connector/mcp/adapter.test.d.ts.map +1 -0
- package/dist/connector/mcp/adapter.test.js +199 -0
- package/dist/connector/mcp/adapter.test.js.map +1 -0
- package/dist/contracts/api.d.ts +2 -2
- package/dist/contracts/api.d.ts.map +1 -1
- package/dist/contracts/index.d.ts +1 -2
- package/dist/contracts/index.d.ts.map +1 -1
- package/dist/contracts/index.js.map +1 -1
- package/dist/contracts/schemas.d.ts +10 -10
- package/dist/index.d.ts +3 -103
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +9 -89
- package/dist/index.js.map +1 -1
- package/dist/manager/agent/__tests__/lifecycle.test.js.map +1 -1
- package/dist/manager/agent/lifecycle.d.ts +1 -1
- package/dist/manager/agent/lifecycle.d.ts.map +1 -1
- package/dist/manager/agent/lifecycle.js +1 -1
- package/dist/manager/agent/lifecycle.js.map +1 -1
- package/dist/manager/run/persistence.d.ts +3 -3
- package/dist/manager/run/persistence.d.ts.map +1 -1
- package/dist/manager/run/persistence.js.map +1 -1
- package/dist/manager/thread/__tests__/lifecycle.test.js.map +1 -1
- package/dist/manager/thread/lifecycle.d.ts +1 -1
- package/dist/manager/thread/lifecycle.d.ts.map +1 -1
- package/dist/plugin/loader.d.ts +0 -5
- package/dist/plugin/loader.d.ts.map +1 -1
- package/dist/plugin/loader.js +1 -13
- package/dist/plugin/loader.js.map +1 -1
- package/dist/public-runtime.d.ts +70 -0
- package/dist/public-runtime.d.ts.map +1 -0
- package/dist/public-runtime.js +108 -0
- package/dist/public-runtime.js.map +1 -0
- package/dist/public-tools.d.ts +19 -0
- package/dist/public-tools.d.ts.map +1 -0
- package/dist/public-tools.js +34 -0
- package/dist/public-tools.js.map +1 -0
- package/dist/public-types.d.ts +55 -0
- package/dist/public-types.d.ts.map +1 -0
- package/dist/public-types.js +14 -0
- package/dist/public-types.js.map +1 -0
- package/dist/rag/chunking.test.d.ts +20 -0
- package/dist/rag/chunking.test.d.ts.map +1 -0
- package/dist/rag/chunking.test.js +92 -0
- package/dist/rag/chunking.test.js.map +1 -0
- package/dist/rag/context-assembler.test.d.ts +19 -0
- package/dist/rag/context-assembler.test.d.ts.map +1 -0
- package/dist/rag/context-assembler.test.js +98 -0
- package/dist/rag/context-assembler.test.js.map +1 -0
- package/dist/rag/embedding.test.d.ts +19 -0
- package/dist/rag/embedding.test.d.ts.map +1 -0
- package/dist/rag/embedding.test.js +115 -0
- package/dist/rag/embedding.test.js.map +1 -0
- package/dist/rag/ingestion.test.d.ts +22 -0
- package/dist/rag/ingestion.test.d.ts.map +1 -0
- package/dist/rag/ingestion.test.js +99 -0
- package/dist/rag/ingestion.test.js.map +1 -0
- package/dist/rag/knowledge-base.test.d.ts +17 -0
- package/dist/rag/knowledge-base.test.d.ts.map +1 -0
- package/dist/rag/knowledge-base.test.js +77 -0
- package/dist/rag/knowledge-base.test.js.map +1 -0
- package/dist/rag/rag-tool.d.ts +1 -1
- package/dist/rag/rag-tool.test.d.ts +21 -0
- package/dist/rag/rag-tool.test.d.ts.map +1 -0
- package/dist/rag/rag-tool.test.js +149 -0
- package/dist/rag/rag-tool.test.js.map +1 -0
- package/dist/rag/retriever.test.d.ts +26 -0
- package/dist/rag/retriever.test.d.ts.map +1 -0
- package/dist/rag/retriever.test.js +180 -0
- package/dist/rag/retriever.test.js.map +1 -0
- package/dist/rag/vector-store.test.d.ts +38 -0
- package/dist/rag/vector-store.test.d.ts.map +1 -0
- package/dist/rag/vector-store.test.js +175 -0
- package/dist/rag/vector-store.test.js.map +1 -0
- package/dist/registry/ManagedRegistry.test.d.ts +21 -0
- package/dist/registry/ManagedRegistry.test.d.ts.map +1 -0
- package/dist/registry/ManagedRegistry.test.js +98 -0
- package/dist/registry/ManagedRegistry.test.js.map +1 -0
- package/dist/registry/Registry.test.d.ts +18 -0
- package/dist/registry/Registry.test.d.ts.map +1 -0
- package/dist/registry/Registry.test.js +79 -0
- package/dist/registry/Registry.test.js.map +1 -0
- package/dist/registry/agent/definitions.test.d.ts +15 -0
- package/dist/registry/agent/definitions.test.d.ts.map +1 -0
- package/dist/registry/agent/definitions.test.js +84 -0
- package/dist/registry/agent/definitions.test.js.map +1 -0
- package/dist/registry/connector/definitions.test.d.ts +13 -0
- package/dist/registry/connector/definitions.test.d.ts.map +1 -0
- package/dist/registry/connector/definitions.test.js +41 -0
- package/dist/registry/connector/definitions.test.js.map +1 -0
- package/dist/registry/connector/scoped.test.d.ts +21 -0
- package/dist/registry/connector/scoped.test.d.ts.map +1 -0
- package/dist/registry/connector/scoped.test.js +115 -0
- package/dist/registry/connector/scoped.test.js.map +1 -0
- package/dist/registry/plugin/index.test.d.ts +12 -0
- package/dist/registry/plugin/index.test.d.ts.map +1 -0
- package/dist/registry/plugin/index.test.js +69 -0
- package/dist/registry/plugin/index.test.js.map +1 -0
- package/dist/registry/tool/execute.test.d.ts +42 -0
- package/dist/registry/tool/execute.test.d.ts.map +1 -0
- package/dist/registry/tool/execute.test.js +281 -0
- package/dist/registry/tool/execute.test.js.map +1 -0
- package/dist/run/LimitChecker.d.ts +0 -1
- package/dist/run/LimitChecker.d.ts.map +1 -1
- package/dist/run/LimitChecker.js +0 -6
- package/dist/run/LimitChecker.js.map +1 -1
- package/dist/run/reporter.d.ts +2 -2
- package/dist/run/reporter.d.ts.map +1 -1
- package/dist/run/reporter.js.map +1 -1
- package/dist/runtime/query/__tests__/checkpoint.test.d.ts +2 -0
- package/dist/runtime/query/__tests__/checkpoint.test.d.ts.map +1 -0
- package/dist/runtime/query/__tests__/checkpoint.test.js +150 -0
- package/dist/runtime/query/__tests__/checkpoint.test.js.map +1 -0
- package/dist/runtime/query/checkpoint.d.ts +23 -0
- package/dist/runtime/query/checkpoint.d.ts.map +1 -1
- package/dist/runtime/query/checkpoint.js +49 -0
- package/dist/runtime/query/checkpoint.js.map +1 -1
- package/dist/runtime/query/context.d.ts +0 -2
- package/dist/runtime/query/context.d.ts.map +1 -1
- package/dist/runtime/query/context.js +0 -4
- package/dist/runtime/query/context.js.map +1 -1
- package/dist/runtime/query/index.d.ts +3 -3
- package/dist/runtime/query/index.d.ts.map +1 -1
- package/dist/runtime/query/iteration/phases/advisory.test.d.ts +42 -0
- package/dist/runtime/query/iteration/phases/advisory.test.d.ts.map +1 -0
- package/dist/runtime/query/iteration/phases/advisory.test.js +334 -0
- package/dist/runtime/query/iteration/phases/advisory.test.js.map +1 -0
- package/dist/runtime/query/replay/__tests__/mutate.test.d.ts +2 -0
- package/dist/runtime/query/replay/__tests__/mutate.test.d.ts.map +1 -0
- package/dist/runtime/query/replay/__tests__/mutate.test.js +123 -0
- package/dist/runtime/query/replay/__tests__/mutate.test.js.map +1 -0
- package/dist/runtime/query/replay/__tests__/prepare.test.d.ts +2 -0
- package/dist/runtime/query/replay/__tests__/prepare.test.d.ts.map +1 -0
- package/dist/runtime/query/replay/__tests__/prepare.test.js +171 -0
- package/dist/runtime/query/replay/__tests__/prepare.test.js.map +1 -0
- package/dist/runtime/query/replay/list.d.ts +24 -0
- package/dist/runtime/query/replay/list.d.ts.map +1 -0
- package/dist/runtime/query/replay/list.js +21 -0
- package/dist/runtime/query/replay/list.js.map +1 -0
- package/dist/runtime/query/replay/mutate.d.ts +17 -0
- package/dist/runtime/query/replay/mutate.d.ts.map +1 -0
- package/dist/runtime/query/replay/mutate.js +62 -0
- package/dist/runtime/query/replay/mutate.js.map +1 -0
- package/dist/runtime/query/replay/prepare.d.ts +56 -0
- package/dist/runtime/query/replay/prepare.d.ts.map +1 -0
- package/dist/runtime/query/replay/prepare.js +65 -0
- package/dist/runtime/query/replay/prepare.js.map +1 -0
- package/dist/runtime/query/result.d.ts +2 -2
- package/dist/runtime/query/result.d.ts.map +1 -1
- package/dist/session/__tests__/integration/_fixtures.d.ts +4 -4
- package/dist/session/__tests__/integration/_fixtures.d.ts.map +1 -1
- package/dist/session/__tests__/integration/_fixtures.js.map +1 -1
- package/dist/session/__tests__/integration/archive-gate.test.js.map +1 -1
- package/dist/session/__tests__/integration/capacity-caps.test.js.map +1 -1
- package/dist/session/__tests__/integration/e2e-spawn.test.js.map +1 -1
- package/dist/session/__tests__/integration/handoff-broadcast-e2e.test.js.map +1 -1
- package/dist/session/__tests__/integration/handoff-single-e2e.test.js.map +1 -1
- package/dist/session/__tests__/integration/prev-artifact-dag.test.js.map +1 -1
- package/dist/session/__tests__/integration/retention-archive.test.js.map +1 -1
- package/dist/session/__tests__/integration/spawn-rollback.test.js.map +1 -1
- package/dist/session/__tests__/integration/summary-materialization-e2e.test.js +1 -1
- package/dist/session/__tests__/integration/summary-materialization-e2e.test.js.map +1 -1
- package/dist/session/errors.d.ts +1 -1
- package/dist/session/errors.d.ts.map +1 -1
- package/dist/session/events/index.d.ts +3 -3
- package/dist/session/events/index.d.ts.map +1 -1
- package/dist/session/events/index.js +3 -2
- package/dist/session/events/index.js.map +1 -1
- package/dist/session/handoff/__tests__/broadcast.test.js.map +1 -1
- package/dist/session/handoff/__tests__/capacity.test.js.map +1 -1
- package/dist/session/handoff/__tests__/single.test.js.map +1 -1
- package/dist/session/handoff/assignment.d.ts +1 -1
- package/dist/session/handoff/assignment.d.ts.map +1 -1
- package/dist/session/handoff/broadcast.d.ts.map +1 -1
- package/dist/session/handoff/broadcast.js.map +1 -1
- package/dist/session/handoff/single.d.ts.map +1 -1
- package/dist/session/handoff/single.js.map +1 -1
- package/dist/session/index.d.ts +0 -1
- package/dist/session/index.d.ts.map +1 -1
- package/dist/session/index.js +7 -6
- package/dist/session/index.js.map +1 -1
- package/dist/session/intervention/__tests__/prev-artifact.test.js.map +1 -1
- package/dist/session/intervention/prev-artifact.d.ts +1 -1
- package/dist/session/intervention/prev-artifact.d.ts.map +1 -1
- package/dist/session/retention/__tests__/archive.test.js.map +1 -1
- package/dist/session/retention/__tests__/disk-backend.test.js.map +1 -1
- package/dist/session/retention/archive.d.ts +1 -1
- package/dist/session/retention/archive.d.ts.map +1 -1
- package/dist/session/retention/backend.d.ts +4 -4
- package/dist/session/retention/backend.d.ts.map +1 -1
- package/dist/session/retention/disk-backend.d.ts +1 -1
- package/dist/session/retention/disk-backend.d.ts.map +1 -1
- package/dist/session/retention/index.d.ts +3 -3
- package/dist/session/retention/index.d.ts.map +1 -1
- package/dist/session/retention/index.js +4 -2
- package/dist/session/retention/index.js.map +1 -1
- package/dist/session/status/__tests__/derive.test.d.ts +2 -0
- package/dist/session/status/__tests__/derive.test.d.ts.map +1 -0
- package/dist/session/{hierarchy/__tests__/session.test.js → status/__tests__/derive.test.js} +2 -2
- package/dist/session/status/__tests__/derive.test.js.map +1 -0
- package/dist/session/status/derive.d.ts +33 -0
- package/dist/session/status/derive.d.ts.map +1 -0
- package/dist/session/{hierarchy/session.js → status/derive.js} +6 -1
- package/dist/session/status/derive.js.map +1 -0
- package/dist/session/summary/__tests__/materialize.test.js +2 -1
- package/dist/session/summary/__tests__/materialize.test.js.map +1 -1
- package/dist/session/summary/errors.d.ts +43 -0
- package/dist/session/summary/errors.d.ts.map +1 -0
- package/dist/session/summary/errors.js +37 -0
- package/dist/session/summary/errors.js.map +1 -0
- package/dist/session/summary/index.d.ts +4 -3
- package/dist/session/summary/index.d.ts.map +1 -1
- package/dist/session/summary/index.js +5 -4
- package/dist/session/summary/index.js.map +1 -1
- package/dist/session/summary/materialize.d.ts +2 -2
- package/dist/session/summary/materialize.d.ts.map +1 -1
- package/dist/session/summary/materialize.js +2 -1
- package/dist/session/summary/materialize.js.map +1 -1
- package/dist/session/workspace/__tests__/git-worktree.test.js.map +1 -1
- package/dist/session/workspace/driver.d.ts +2 -2
- package/dist/session/workspace/driver.d.ts.map +1 -1
- package/dist/session/workspace/git-worktree.d.ts +1 -1
- package/dist/session/workspace/git-worktree.d.ts.map +1 -1
- package/dist/session/workspace/git-worktree.js.map +1 -1
- package/dist/session/workspace/index.d.ts +1 -1
- package/dist/session/workspace/index.d.ts.map +1 -1
- package/dist/session/workspace/index.js +3 -2
- package/dist/session/workspace/index.js.map +1 -1
- package/dist/store/run/disk.d.ts +4 -4
- package/dist/store/run/disk.d.ts.map +1 -1
- package/dist/store/run/disk.js.map +1 -1
- package/dist/store/session/__tests__/disk.test.js.map +1 -1
- package/dist/store/session/__tests__/memory.test.js.map +1 -1
- package/dist/store/session/disk.d.ts +6 -6
- package/dist/store/session/disk.d.ts.map +1 -1
- package/dist/store/session/disk.js +1 -1
- package/dist/store/session/disk.js.map +1 -1
- package/dist/store/session/index.d.ts +1 -1
- package/dist/store/session/index.d.ts.map +1 -1
- package/dist/store/session/index.js +2 -1
- package/dist/store/session/index.js.map +1 -1
- package/dist/store/session/linkage.d.ts +1 -1
- package/dist/store/session/linkage.d.ts.map +1 -1
- package/dist/store/session/memory.d.ts +5 -5
- package/dist/store/session/memory.d.ts.map +1 -1
- package/dist/store/session/memory.js +1 -1
- package/dist/store/session/memory.js.map +1 -1
- package/dist/store/thread/disk.d.ts +1 -1
- package/dist/store/thread/disk.d.ts.map +1 -1
- package/dist/store/thread/memory.d.ts +1 -1
- package/dist/store/thread/memory.d.ts.map +1 -1
- package/dist/telemetry/attributes.d.ts +0 -1
- package/dist/telemetry/attributes.d.ts.map +1 -1
- package/dist/telemetry/attributes.js +0 -3
- package/dist/telemetry/attributes.js.map +1 -1
- package/dist/telemetry/runtime-accessors.d.ts +1 -2
- package/dist/telemetry/runtime-accessors.d.ts.map +1 -1
- package/dist/telemetry/runtime-accessors.js +1 -4
- package/dist/telemetry/runtime-accessors.js.map +1 -1
- package/dist/test-setup.d.ts +22 -0
- package/dist/test-setup.d.ts.map +1 -0
- package/dist/test-setup.js +23 -0
- package/dist/test-setup.js.map +1 -0
- package/dist/tools/builtins/bash.d.ts +1 -1
- package/dist/tools/builtins/computer-use.d.ts +4 -4
- package/dist/tools/builtins/edit.d.ts +1 -1
- package/dist/tools/builtins/glob.d.ts +1 -1
- package/dist/tools/builtins/grep.d.ts +1 -1
- package/dist/tools/builtins/ls.d.ts +1 -1
- package/dist/tools/builtins/read-file.d.ts +1 -1
- package/dist/tools/builtins/search-tools.d.ts +1 -1
- package/dist/tools/builtins/write-file.d.ts +1 -1
- package/dist/types/agent/task.d.ts +2 -2
- package/dist/types/agent/task.d.ts.map +1 -1
- package/dist/{session/hierarchy/project.d.ts → types/project/entity.d.ts} +4 -4
- package/dist/types/project/entity.d.ts.map +1 -0
- package/dist/types/project/entity.js +2 -0
- package/dist/types/project/entity.js.map +1 -0
- package/dist/types/project/index.d.ts +2 -0
- package/dist/types/project/index.d.ts.map +1 -0
- package/dist/types/project/index.js +4 -0
- package/dist/types/project/index.js.map +1 -0
- package/dist/types/retention/archive-backend-ref.d.ts.map +1 -0
- package/dist/types/retention/archive-backend-ref.js.map +1 -0
- package/dist/types/retention/index.d.ts +4 -0
- package/dist/types/retention/index.d.ts.map +1 -0
- package/dist/types/retention/index.js +4 -0
- package/dist/types/retention/index.js.map +1 -0
- package/dist/types/retention/policy.d.ts.map +1 -0
- package/dist/types/retention/policy.js.map +1 -0
- package/dist/types/run/entity.d.ts +56 -0
- package/dist/types/run/entity.d.ts.map +1 -0
- package/dist/types/run/entity.js +2 -0
- package/dist/types/run/entity.js.map +1 -0
- package/dist/types/run/events.d.ts +2 -2
- package/dist/types/run/events.d.ts.map +1 -1
- package/dist/types/run/index.d.ts +7 -1
- package/dist/types/run/index.d.ts.map +1 -1
- package/dist/types/run/index.js +8 -9
- package/dist/types/run/index.js.map +1 -1
- package/dist/{session/hierarchy → types/run}/lineage.d.ts +1 -1
- package/dist/types/run/lineage.d.ts.map +1 -0
- package/dist/types/run/lineage.js.map +1 -0
- package/dist/types/run/replay.d.ts +53 -0
- package/dist/types/run/replay.d.ts.map +1 -0
- package/dist/types/run/replay.js +15 -0
- package/dist/types/run/replay.js.map +1 -0
- package/dist/types/run/schema-version.d.ts.map +1 -0
- package/dist/types/run/schema-version.js.map +1 -0
- package/dist/{session/events/types.d.ts → types/run/subsession-events.d.ts} +5 -5
- package/dist/types/run/subsession-events.d.ts.map +1 -0
- package/dist/types/run/subsession-events.js +2 -0
- package/dist/types/run/subsession-events.js.map +1 -0
- package/dist/{session/hierarchy → types/session}/actor.d.ts +1 -1
- package/dist/types/session/actor.d.ts.map +1 -0
- package/dist/{session/hierarchy → types/session}/actor.js.map +1 -1
- package/dist/types/session/entity.d.ts +51 -0
- package/dist/types/session/entity.d.ts.map +1 -0
- package/dist/types/session/entity.js +2 -0
- package/dist/types/session/entity.js.map +1 -0
- package/dist/types/session/index.d.ts +4 -0
- package/dist/types/session/index.d.ts.map +1 -1
- package/dist/{store → types}/session/messages.d.ts +2 -2
- package/dist/{store → types}/session/messages.d.ts.map +1 -1
- package/dist/types/session/messages.js.map +1 -0
- package/dist/types/session/store.d.ts +6 -6
- package/dist/types/session/store.d.ts.map +1 -1
- package/dist/{session/hierarchy → types/session}/sub-session.d.ts +9 -9
- package/dist/types/session/sub-session.d.ts.map +1 -0
- package/dist/types/session/sub-session.js.map +1 -0
- package/dist/{session → types}/summary/deliverable.d.ts +2 -2
- package/dist/types/summary/deliverable.d.ts.map +1 -0
- package/dist/types/summary/deliverable.js.map +1 -0
- package/dist/types/summary/index.d.ts +4 -0
- package/dist/types/summary/index.d.ts.map +1 -0
- package/dist/types/summary/index.js +4 -0
- package/dist/types/summary/index.js.map +1 -0
- package/dist/{session → types}/summary/ref.d.ts +9 -36
- package/dist/types/summary/ref.d.ts.map +1 -0
- package/dist/types/summary/ref.js +28 -0
- package/dist/types/summary/ref.js.map +1 -0
- package/dist/{session/hierarchy/tenant.d.ts → types/tenant/entity.d.ts} +2 -2
- package/dist/types/tenant/entity.d.ts.map +1 -0
- package/dist/types/tenant/entity.js +2 -0
- package/dist/types/tenant/entity.js.map +1 -0
- package/dist/types/tenant/index.d.ts +2 -0
- package/dist/types/tenant/index.d.ts.map +1 -0
- package/dist/types/tenant/index.js +4 -0
- package/dist/types/tenant/index.js.map +1 -0
- package/dist/{session/hierarchy/thread.d.ts → types/thread/entity.d.ts} +31 -7
- package/dist/types/thread/entity.d.ts.map +1 -0
- package/dist/types/thread/entity.js +2 -0
- package/dist/types/thread/entity.js.map +1 -0
- package/dist/types/thread/index.d.ts +1 -0
- package/dist/types/thread/index.d.ts.map +1 -1
- package/dist/types/thread/store.d.ts +1 -1
- package/dist/types/thread/store.d.ts.map +1 -1
- package/dist/types/workspace/index.d.ts +2 -0
- package/dist/types/workspace/index.d.ts.map +1 -0
- package/dist/types/workspace/index.js +7 -0
- package/dist/types/workspace/index.js.map +1 -0
- package/dist/{session → types}/workspace/ref.d.ts +1 -1
- package/dist/types/workspace/ref.d.ts.map +1 -0
- package/dist/types/workspace/ref.js.map +1 -0
- package/dist/utils/logger.d.ts +1 -1
- package/dist/utils/logger.d.ts.map +1 -1
- package/dist/utils/logger.js +5 -0
- package/dist/utils/logger.js.map +1 -1
- package/package.json +4 -1
- package/src/__tests__/replay-public-surface.test.ts +54 -0
- package/src/advisory/context.test.ts +109 -0
- package/src/advisory/evaluator.test.ts +192 -0
- package/src/advisory/executor.test.ts +272 -0
- package/src/advisory/registry.test.ts +75 -0
- package/src/agents/SupervisorAgent.ts +1 -1
- package/src/bridge/a2a/agent-card.test.ts +140 -0
- package/src/bridge/a2a/mapper.test.ts +293 -0
- package/src/bridge/a2a/message.test.ts +138 -0
- package/src/bridge/a2a/task.test.ts +235 -0
- package/src/bridge/a2a/task.ts +4 -4
- package/src/bridge/mcp/connector/adapter.test.ts +230 -0
- package/src/bridge/sse/mapper.test.ts +422 -0
- package/src/bridge/tools/connector/adapter.test.ts +224 -0
- package/src/bridge/tools/connector/definitions.test.ts +183 -0
- package/src/bridge/tools/connector/router.test.ts +159 -0
- package/src/bus/breaker.test.ts +274 -0
- package/src/bus/index.test.ts +183 -0
- package/src/bus/lock.test.ts +265 -0
- package/src/bus/ownership.test.ts +243 -0
- package/src/config/runtime.ts +0 -2
- package/src/connector/BaseConnector.test.ts +130 -0
- package/src/connector/builtins/http.test.ts +290 -0
- package/src/connector/builtins/webhook.test.ts +138 -0
- package/src/connector/execution/factory.test.ts +75 -0
- package/src/connector/execution/remote.test.ts +63 -0
- package/src/connector/mcp/adapter.test.ts +249 -0
- package/src/contracts/api.ts +2 -2
- package/src/contracts/index.ts +1 -3
- package/src/index.ts +11 -362
- package/src/manager/agent/__tests__/lifecycle.test.ts +2 -2
- package/src/manager/agent/lifecycle.ts +5 -5
- package/src/manager/run/persistence.ts +4 -4
- package/src/manager/thread/__tests__/lifecycle.test.ts +1 -1
- package/src/manager/thread/lifecycle.ts +2 -2
- package/src/plugin/loader.ts +0 -16
- package/src/public-runtime.ts +373 -0
- package/src/public-tools.ts +63 -0
- package/src/public-types.ts +156 -0
- package/src/rag/chunking.test.ts +107 -0
- package/src/rag/context-assembler.test.ts +114 -0
- package/src/rag/embedding.test.ts +130 -0
- package/src/rag/ingestion.test.ts +114 -0
- package/src/rag/knowledge-base.test.ts +106 -0
- package/src/rag/rag-tool.test.ts +167 -0
- package/src/rag/retriever.test.ts +210 -0
- package/src/rag/vector-store.test.ts +196 -0
- package/src/registry/ManagedRegistry.test.ts +118 -0
- package/src/registry/Registry.test.ts +91 -0
- package/src/registry/agent/definitions.test.ts +100 -0
- package/src/registry/connector/definitions.test.ts +51 -0
- package/src/registry/connector/scoped.test.ts +129 -0
- package/src/registry/plugin/index.test.ts +85 -0
- package/src/registry/tool/execute.test.ts +330 -0
- package/src/run/LimitChecker.ts +0 -9
- package/src/run/reporter.ts +3 -3
- package/src/runtime/query/__tests__/checkpoint.test.ts +169 -0
- package/src/runtime/query/checkpoint.ts +54 -0
- package/src/runtime/query/context.ts +0 -5
- package/src/runtime/query/index.ts +4 -4
- package/src/runtime/query/iteration/phases/advisory.test.ts +412 -0
- package/src/runtime/query/replay/__tests__/mutate.test.ts +134 -0
- package/src/runtime/query/replay/__tests__/prepare.test.ts +207 -0
- package/src/runtime/query/replay/list.ts +32 -0
- package/src/runtime/query/replay/mutate.ts +76 -0
- package/src/runtime/query/replay/prepare.ts +114 -0
- package/src/runtime/query/result.ts +2 -2
- package/src/session/__tests__/integration/_fixtures.ts +2 -2
- package/src/session/__tests__/integration/archive-gate.test.ts +1 -1
- package/src/session/__tests__/integration/capacity-caps.test.ts +1 -1
- package/src/session/__tests__/integration/e2e-spawn.test.ts +1 -1
- package/src/session/__tests__/integration/handoff-broadcast-e2e.test.ts +1 -1
- package/src/session/__tests__/integration/handoff-single-e2e.test.ts +1 -1
- package/src/session/__tests__/integration/prev-artifact-dag.test.ts +4 -1
- package/src/session/__tests__/integration/retention-archive.test.ts +1 -1
- package/src/session/__tests__/integration/spawn-rollback.test.ts +2 -2
- package/src/session/__tests__/integration/summary-materialization-e2e.test.ts +1 -1
- package/src/session/errors.ts +1 -1
- package/src/session/events/index.ts +5 -4
- package/src/session/handoff/__tests__/broadcast.test.ts +1 -1
- package/src/session/handoff/__tests__/capacity.test.ts +1 -1
- package/src/session/handoff/__tests__/single.test.ts +1 -1
- package/src/session/handoff/assignment.ts +1 -1
- package/src/session/handoff/broadcast.ts +2 -2
- package/src/session/handoff/single.ts +2 -2
- package/src/session/index.ts +7 -6
- package/src/session/intervention/__tests__/prev-artifact.test.ts +4 -1
- package/src/session/intervention/prev-artifact.ts +1 -1
- package/src/session/retention/__tests__/archive.test.ts +3 -3
- package/src/session/retention/__tests__/disk-backend.test.ts +4 -4
- package/src/session/retention/archive.ts +2 -2
- package/src/session/retention/backend.ts +4 -4
- package/src/session/retention/disk-backend.ts +4 -4
- package/src/session/retention/index.ts +6 -4
- package/src/session/{hierarchy/__tests__/session.test.ts → status/__tests__/derive.test.ts} +3 -2
- package/src/session/{hierarchy/session.ts → status/derive.ts} +9 -55
- package/src/session/summary/__tests__/materialize.test.ts +4 -7
- package/src/session/summary/errors.ts +51 -0
- package/src/session/summary/index.ts +7 -10
- package/src/session/summary/materialize.ts +4 -5
- package/src/session/workspace/__tests__/git-worktree.test.ts +1 -1
- package/src/session/workspace/driver.ts +2 -2
- package/src/session/workspace/git-worktree.ts +1 -1
- package/src/session/workspace/index.ts +7 -6
- package/src/store/run/disk.ts +4 -4
- package/src/store/session/__tests__/disk.test.ts +2 -2
- package/src/store/session/__tests__/memory.test.ts +2 -2
- package/src/store/session/disk.ts +12 -12
- package/src/store/session/index.ts +3 -2
- package/src/store/session/linkage.ts +1 -1
- package/src/store/session/memory.ts +6 -6
- package/src/store/thread/disk.ts +1 -1
- package/src/store/thread/memory.ts +1 -1
- package/src/telemetry/attributes.ts +0 -4
- package/src/telemetry/runtime-accessors.ts +1 -5
- package/src/test-setup.ts +24 -0
- package/src/types/agent/task.ts +2 -2
- package/src/{session/hierarchy/project.ts → types/project/entity.ts} +3 -3
- package/src/types/project/index.ts +4 -0
- package/src/types/retention/index.ts +6 -0
- package/src/types/run/entity.ts +63 -0
- package/src/types/run/events.ts +6 -6
- package/src/types/run/index.ts +16 -9
- package/src/{session/hierarchy → types/run}/lineage.ts +1 -1
- package/src/types/run/replay.ts +61 -0
- package/src/{session/events/types.ts → types/run/subsession-events.ts} +4 -4
- package/src/{session/hierarchy → types/session}/actor.ts +1 -1
- package/src/types/session/entity.ts +59 -0
- package/src/types/session/index.ts +15 -0
- package/src/{store → types}/session/messages.ts +2 -2
- package/src/types/session/store.ts +6 -6
- package/src/{session/hierarchy → types/session}/sub-session.ts +9 -9
- package/src/{session → types}/summary/deliverable.ts +2 -2
- package/src/types/summary/index.ts +18 -0
- package/src/{session → types}/summary/ref.ts +9 -44
- package/src/{session/hierarchy/tenant.ts → types/tenant/entity.ts} +1 -1
- package/src/types/tenant/index.ts +4 -0
- package/src/{session/hierarchy/thread.ts → types/thread/entity.ts} +30 -6
- package/src/types/thread/index.ts +1 -0
- package/src/types/thread/store.ts +1 -1
- package/src/types/workspace/index.ts +12 -0
- package/src/{session → types}/workspace/ref.ts +1 -1
- package/src/utils/logger.ts +6 -1
- package/dist/contracts/ids.d.ts +0 -2
- package/dist/contracts/ids.d.ts.map +0 -1
- package/dist/contracts/ids.js +0 -2
- package/dist/contracts/ids.js.map +0 -1
- package/dist/session/events/schema-version.d.ts.map +0 -1
- package/dist/session/events/schema-version.js.map +0 -1
- package/dist/session/events/types.d.ts.map +0 -1
- package/dist/session/events/types.js +0 -2
- package/dist/session/events/types.js.map +0 -1
- package/dist/session/hierarchy/__tests__/session.test.d.ts +0 -2
- package/dist/session/hierarchy/__tests__/session.test.d.ts.map +0 -1
- package/dist/session/hierarchy/__tests__/session.test.js.map +0 -1
- package/dist/session/hierarchy/actor.d.ts.map +0 -1
- package/dist/session/hierarchy/index.d.ts +0 -9
- package/dist/session/hierarchy/index.d.ts.map +0 -1
- package/dist/session/hierarchy/index.js +0 -4
- package/dist/session/hierarchy/index.js.map +0 -1
- package/dist/session/hierarchy/lineage.d.ts.map +0 -1
- package/dist/session/hierarchy/lineage.js.map +0 -1
- package/dist/session/hierarchy/project.d.ts.map +0 -1
- package/dist/session/hierarchy/project.js +0 -2
- package/dist/session/hierarchy/project.js.map +0 -1
- package/dist/session/hierarchy/session.d.ts +0 -71
- package/dist/session/hierarchy/session.d.ts.map +0 -1
- package/dist/session/hierarchy/session.js.map +0 -1
- package/dist/session/hierarchy/sub-session.d.ts.map +0 -1
- package/dist/session/hierarchy/sub-session.js.map +0 -1
- package/dist/session/hierarchy/tenant.d.ts.map +0 -1
- package/dist/session/hierarchy/tenant.js +0 -2
- package/dist/session/hierarchy/tenant.js.map +0 -1
- package/dist/session/hierarchy/thread.d.ts.map +0 -1
- package/dist/session/hierarchy/thread.js +0 -2
- package/dist/session/hierarchy/thread.js.map +0 -1
- package/dist/session/retention/archive-backend-ref.d.ts.map +0 -1
- package/dist/session/retention/archive-backend-ref.js.map +0 -1
- package/dist/session/retention/policy.d.ts.map +0 -1
- package/dist/session/retention/policy.js.map +0 -1
- package/dist/session/summary/deliverable.d.ts.map +0 -1
- package/dist/session/summary/deliverable.js.map +0 -1
- package/dist/session/summary/ref.d.ts.map +0 -1
- package/dist/session/summary/ref.js +0 -51
- package/dist/session/summary/ref.js.map +0 -1
- package/dist/session/workspace/ref.d.ts.map +0 -1
- package/dist/session/workspace/ref.js.map +0 -1
- package/dist/store/session/messages.js.map +0 -1
- package/dist/tools/builtins/__tests__/structuredOutput.example.d.ts +0 -140
- package/dist/tools/builtins/__tests__/structuredOutput.example.d.ts.map +0 -1
- package/dist/tools/builtins/__tests__/structuredOutput.example.js +0 -183
- package/dist/tools/builtins/__tests__/structuredOutput.example.js.map +0 -1
- package/dist/types/run/state.d.ts +0 -30
- package/dist/types/run/state.d.ts.map +0 -1
- package/dist/types/run/state.js +0 -2
- package/dist/types/run/state.js.map +0 -1
- package/src/contracts/ids.ts +0 -1
- package/src/session/hierarchy/index.ts +0 -18
- package/src/tools/builtins/__tests__/structuredOutput.example.ts +0 -221
- package/src/types/run/state.ts +0 -35
- /package/dist/{session → types}/retention/archive-backend-ref.d.ts +0 -0
- /package/dist/{session → types}/retention/archive-backend-ref.js +0 -0
- /package/dist/{session → types}/retention/policy.d.ts +0 -0
- /package/dist/{session → types}/retention/policy.js +0 -0
- /package/dist/{session/hierarchy → types/run}/lineage.js +0 -0
- /package/dist/{session/events → types/run}/schema-version.d.ts +0 -0
- /package/dist/{session/events → types/run}/schema-version.js +0 -0
- /package/dist/{session/hierarchy → types/session}/actor.js +0 -0
- /package/dist/{store → types}/session/messages.js +0 -0
- /package/dist/{session/hierarchy → types/session}/sub-session.js +0 -0
- /package/dist/{session → types}/summary/deliverable.js +0 -0
- /package/dist/{session → types}/workspace/ref.js +0 -0
- /package/src/{session → types}/retention/archive-backend-ref.ts +0 -0
- /package/src/{session → types}/retention/policy.ts +0 -0
- /package/src/{session/events → types/run}/schema-version.ts +0 -0
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `OpenRouterEmbeddingProvider`:
|
|
5
|
+
* - Defaults: dimensions = 1536; baseUrl = openrouter.ai/api/v1;
|
|
6
|
+
* batchSize = 64.
|
|
7
|
+
* - `embed(texts)` batches into `batchSize` slices and concatenates
|
|
8
|
+
* results in input order.
|
|
9
|
+
* - Each HTTP call posts `{model, input, dimensions}` to
|
|
10
|
+
* `${baseUrl}/embeddings` with the Bearer authorization header.
|
|
11
|
+
* - The API response is sorted by `index` ascending before extracting
|
|
12
|
+
* `.embedding`, so results match input order even if the server
|
|
13
|
+
* re-orders.
|
|
14
|
+
* - `embedQuery(query)` returns the first result from `embed([query])`;
|
|
15
|
+
* throws when the response is empty.
|
|
16
|
+
* - `!response.ok` → throws with `Embedding API error (<status>): <body>`.
|
|
17
|
+
*/
|
|
18
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
19
|
+
import { OpenRouterEmbeddingProvider } from './embedding.js';
|
|
20
|
+
describe('OpenRouterEmbeddingProvider', () => {
|
|
21
|
+
let fetchMock;
|
|
22
|
+
beforeEach(() => {
|
|
23
|
+
fetchMock = vi.fn();
|
|
24
|
+
global.fetch = fetchMock;
|
|
25
|
+
});
|
|
26
|
+
afterEach(() => {
|
|
27
|
+
vi.restoreAllMocks();
|
|
28
|
+
});
|
|
29
|
+
it('carries model / dimensions defaults + batchSize', () => {
|
|
30
|
+
const p = new OpenRouterEmbeddingProvider({ apiKey: 'k', model: 'm' });
|
|
31
|
+
expect(p.model).toBe('m');
|
|
32
|
+
expect(p.dimensions).toBe(1536);
|
|
33
|
+
});
|
|
34
|
+
it('honors overrides for dimensions + baseUrl + batchSize', async () => {
|
|
35
|
+
fetchMock.mockResolvedValue({
|
|
36
|
+
ok: true,
|
|
37
|
+
json: async () => ({ data: [{ index: 0, embedding: [1, 2, 3] }] }),
|
|
38
|
+
});
|
|
39
|
+
const p = new OpenRouterEmbeddingProvider({
|
|
40
|
+
apiKey: 'k',
|
|
41
|
+
model: 'm',
|
|
42
|
+
dimensions: 256,
|
|
43
|
+
baseUrl: 'https://custom.example/api',
|
|
44
|
+
});
|
|
45
|
+
await p.embed(['x']);
|
|
46
|
+
expect(fetchMock).toHaveBeenCalledWith('https://custom.example/api/embeddings', expect.objectContaining({
|
|
47
|
+
method: 'POST',
|
|
48
|
+
headers: expect.objectContaining({
|
|
49
|
+
Authorization: 'Bearer k',
|
|
50
|
+
'Content-Type': 'application/json',
|
|
51
|
+
}),
|
|
52
|
+
}));
|
|
53
|
+
const body = JSON.parse((fetchMock.mock.calls[0]?.[1]).body);
|
|
54
|
+
expect(body).toEqual({ model: 'm', input: ['x'], dimensions: 256 });
|
|
55
|
+
expect(p.dimensions).toBe(256);
|
|
56
|
+
});
|
|
57
|
+
it('batches into batchSize slices', async () => {
|
|
58
|
+
fetchMock.mockResolvedValue({
|
|
59
|
+
ok: true,
|
|
60
|
+
json: async () => ({
|
|
61
|
+
data: [
|
|
62
|
+
{ index: 0, embedding: [1] },
|
|
63
|
+
{ index: 1, embedding: [2] },
|
|
64
|
+
],
|
|
65
|
+
}),
|
|
66
|
+
});
|
|
67
|
+
const p = new OpenRouterEmbeddingProvider({
|
|
68
|
+
apiKey: 'k',
|
|
69
|
+
model: 'm',
|
|
70
|
+
batchSize: 2,
|
|
71
|
+
});
|
|
72
|
+
await p.embed(['a', 'b', 'c', 'd']);
|
|
73
|
+
expect(fetchMock).toHaveBeenCalledTimes(2);
|
|
74
|
+
});
|
|
75
|
+
it('sorts response by index before extracting embeddings', async () => {
|
|
76
|
+
fetchMock.mockResolvedValue({
|
|
77
|
+
ok: true,
|
|
78
|
+
json: async () => ({
|
|
79
|
+
data: [
|
|
80
|
+
{ index: 2, embedding: [3] },
|
|
81
|
+
{ index: 0, embedding: [1] },
|
|
82
|
+
{ index: 1, embedding: [2] },
|
|
83
|
+
],
|
|
84
|
+
}),
|
|
85
|
+
});
|
|
86
|
+
const p = new OpenRouterEmbeddingProvider({ apiKey: 'k', model: 'm' });
|
|
87
|
+
expect(await p.embed(['a', 'b', 'c'])).toEqual([[1], [2], [3]]);
|
|
88
|
+
});
|
|
89
|
+
it('embedQuery returns the first result', async () => {
|
|
90
|
+
fetchMock.mockResolvedValue({
|
|
91
|
+
ok: true,
|
|
92
|
+
json: async () => ({ data: [{ index: 0, embedding: [9, 9] }] }),
|
|
93
|
+
});
|
|
94
|
+
const p = new OpenRouterEmbeddingProvider({ apiKey: 'k', model: 'm' });
|
|
95
|
+
expect(await p.embedQuery('hi')).toEqual([9, 9]);
|
|
96
|
+
});
|
|
97
|
+
it('embedQuery throws when the response is empty', async () => {
|
|
98
|
+
fetchMock.mockResolvedValue({
|
|
99
|
+
ok: true,
|
|
100
|
+
json: async () => ({ data: [] }),
|
|
101
|
+
});
|
|
102
|
+
const p = new OpenRouterEmbeddingProvider({ apiKey: 'k', model: 'm' });
|
|
103
|
+
await expect(p.embedQuery('hi')).rejects.toThrow(/no results/);
|
|
104
|
+
});
|
|
105
|
+
it('throws on non-OK HTTP response', async () => {
|
|
106
|
+
fetchMock.mockResolvedValue({
|
|
107
|
+
ok: false,
|
|
108
|
+
status: 503,
|
|
109
|
+
text: async () => 'service unavailable',
|
|
110
|
+
});
|
|
111
|
+
const p = new OpenRouterEmbeddingProvider({ apiKey: 'k', model: 'm' });
|
|
112
|
+
await expect(p.embed(['hi'])).rejects.toThrow(/503.*service unavailable/);
|
|
113
|
+
});
|
|
114
|
+
});
|
|
115
|
+
//# sourceMappingURL=embedding.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.test.js","sourceRoot":"","sources":["../../src/rag/embedding.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExE,OAAO,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAA;AAE5D,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,IAAI,SAAmC,CAAA;IAEvC,UAAU,CAAC,GAAG,EAAE;QACf,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QACnB,MAAM,CAAC,KAAK,GAAG,SAAoC,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,eAAe,EAAE,CAAA;IACrB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QACtE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACzB,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACtE,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;SAClE,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC;YACzC,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;YACV,UAAU,EAAE,GAAG;YACf,OAAO,EAAE,4BAA4B;SACrC,CAAC,CAAA;QACF,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAA;QAEpB,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACrC,uCAAuC,EACvC,MAAM,CAAC,gBAAgB,CAAC;YACvB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAChC,aAAa,EAAE,UAAU;gBACzB,cAAc,EAAE,kBAAkB;aAClC,CAAC;SACF,CAAC,CACF,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAsB,CAAA,CAAC,IAAI,CAAC,CAAA;QAChF,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAA;QACnE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC9C,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE;oBACL,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;oBAC5B,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;iBAC5B;aACD,CAAC;SACF,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC;YACzC,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;YACV,SAAS,EAAE,CAAC;SACZ,CAAC,CAAA;QACF,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QACnC,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACrE,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE;oBACL,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;oBAC5B,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;oBAC5B,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;iBAC5B;aACD,CAAC;SACF,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QACtE,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAChE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;SAC/D,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QACtE,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC7D,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;SAChC,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QACtE,MAAM,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;IAC/D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC/C,SAAS,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,KAAK;YACT,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,qBAAqB;SACvC,CAAC,CAAA;QACF,MAAM,CAAC,GAAG,IAAI,2BAA2B,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QACtE,MAAM,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `DefaultIngestionPipeline.ingest(content, metadata, scope, kbId)`:
|
|
5
|
+
* - Generates a fresh `documentId` per call.
|
|
6
|
+
* - Chunks via `TextChunker.chunk(content, chunkingConfig)`.
|
|
7
|
+
* - Returns zero-chunk result when the chunker emits nothing.
|
|
8
|
+
* - Calls `embeddingProvider.embed([...chunkTexts])` exactly once
|
|
9
|
+
* with every chunk in order.
|
|
10
|
+
* - Each resulting `Chunk` carries:
|
|
11
|
+
* - a fresh `id`,
|
|
12
|
+
* - the generated `documentId`,
|
|
13
|
+
* - the passed `knowledgeBaseId`,
|
|
14
|
+
* - `scope.tenantId`,
|
|
15
|
+
* - the original `chunkContent` + `chunkIndex` metadata,
|
|
16
|
+
* - `tokenCount = Math.ceil(content.length / 4)`.
|
|
17
|
+
* - Calls `vectorStore.upsert(chunks)` exactly once.
|
|
18
|
+
* - Totals `tokenCount` across chunks and reports `durationMs`.
|
|
19
|
+
* - `remove(documentId)` delegates to `vectorStore.deleteByDocument`.
|
|
20
|
+
*/
|
|
21
|
+
export {};
|
|
22
|
+
//# sourceMappingURL=ingestion.test.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ingestion.test.d.ts","sourceRoot":"","sources":["../../src/rag/ingestion.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG"}
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `DefaultIngestionPipeline.ingest(content, metadata, scope, kbId)`:
|
|
5
|
+
* - Generates a fresh `documentId` per call.
|
|
6
|
+
* - Chunks via `TextChunker.chunk(content, chunkingConfig)`.
|
|
7
|
+
* - Returns zero-chunk result when the chunker emits nothing.
|
|
8
|
+
* - Calls `embeddingProvider.embed([...chunkTexts])` exactly once
|
|
9
|
+
* with every chunk in order.
|
|
10
|
+
* - Each resulting `Chunk` carries:
|
|
11
|
+
* - a fresh `id`,
|
|
12
|
+
* - the generated `documentId`,
|
|
13
|
+
* - the passed `knowledgeBaseId`,
|
|
14
|
+
* - `scope.tenantId`,
|
|
15
|
+
* - the original `chunkContent` + `chunkIndex` metadata,
|
|
16
|
+
* - `tokenCount = Math.ceil(content.length / 4)`.
|
|
17
|
+
* - Calls `vectorStore.upsert(chunks)` exactly once.
|
|
18
|
+
* - Totals `tokenCount` across chunks and reports `durationMs`.
|
|
19
|
+
* - `remove(documentId)` delegates to `vectorStore.deleteByDocument`.
|
|
20
|
+
*/
|
|
21
|
+
import { describe, expect, it, vi } from 'vitest';
|
|
22
|
+
import { z } from 'zod';
|
|
23
|
+
import { DefaultIngestionPipeline } from './ingestion.js';
|
|
24
|
+
const KB = 'kb_1';
|
|
25
|
+
const TENANT = 't_1';
|
|
26
|
+
const scope = { tenantId: TENANT };
|
|
27
|
+
function makeVectorStore() {
|
|
28
|
+
return {
|
|
29
|
+
upsert: vi.fn(),
|
|
30
|
+
search: vi.fn(),
|
|
31
|
+
delete: vi.fn(),
|
|
32
|
+
deleteByDocument: vi.fn(),
|
|
33
|
+
deleteByKnowledgeBase: vi.fn(),
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
function makeEmbedder() {
|
|
37
|
+
return {
|
|
38
|
+
id: 'mock',
|
|
39
|
+
model: 'x',
|
|
40
|
+
dimensions: 3,
|
|
41
|
+
embed: vi.fn(async (texts) => texts.map(() => [1, 0, 0])),
|
|
42
|
+
embedQuery: vi.fn(async () => [1, 0, 0]),
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
describe('DefaultIngestionPipeline — ingest', () => {
|
|
46
|
+
it('returns zero-chunk result when the chunker emits nothing', async () => {
|
|
47
|
+
const vs = makeVectorStore();
|
|
48
|
+
const emb = makeEmbedder();
|
|
49
|
+
// chunkSize 10, no content at all → chunker returns []
|
|
50
|
+
const pipeline = new DefaultIngestionPipeline(vs, emb, {
|
|
51
|
+
strategy: 'fixed',
|
|
52
|
+
chunkSize: 10,
|
|
53
|
+
chunkOverlap: 0,
|
|
54
|
+
});
|
|
55
|
+
const result = await pipeline.ingest(' ', {}, scope, KB);
|
|
56
|
+
expect(result.chunkCount).toBe(0);
|
|
57
|
+
expect(result.totalTokens).toBe(0);
|
|
58
|
+
expect(vs.upsert).not.toHaveBeenCalled();
|
|
59
|
+
});
|
|
60
|
+
it('chunks content, embeds every chunk once, upserts once', async () => {
|
|
61
|
+
const vs = makeVectorStore();
|
|
62
|
+
const emb = makeEmbedder();
|
|
63
|
+
const pipeline = new DefaultIngestionPipeline(vs, emb, {
|
|
64
|
+
strategy: 'fixed',
|
|
65
|
+
chunkSize: 10,
|
|
66
|
+
chunkOverlap: 0,
|
|
67
|
+
});
|
|
68
|
+
const result = await pipeline.ingest('a'.repeat(25), { source: 'repo' }, scope, KB);
|
|
69
|
+
expect(emb.embed).toHaveBeenCalledTimes(1);
|
|
70
|
+
expect(vs.upsert).toHaveBeenCalledTimes(1);
|
|
71
|
+
expect(result.chunkCount).toBeGreaterThan(1);
|
|
72
|
+
const upsertedChunks = vi.mocked(vs.upsert).mock.calls[0]?.[0] ?? [];
|
|
73
|
+
for (const c of upsertedChunks) {
|
|
74
|
+
expect(c.documentId).toBe(result.documentId);
|
|
75
|
+
expect(c.knowledgeBaseId).toBe(KB);
|
|
76
|
+
expect(c.tenantId).toBe(TENANT);
|
|
77
|
+
expect(c.metadata.source).toBe('repo');
|
|
78
|
+
expect(c.metadata.chunkIndex).toBeDefined();
|
|
79
|
+
expect(c.tokenCount).toBe(Math.ceil(c.content.length / 4));
|
|
80
|
+
}
|
|
81
|
+
});
|
|
82
|
+
it('generates a fresh documentId per ingest call', async () => {
|
|
83
|
+
const pipeline = new DefaultIngestionPipeline(makeVectorStore(), makeEmbedder());
|
|
84
|
+
const a = await pipeline.ingest('alpha', {}, scope, KB);
|
|
85
|
+
const b = await pipeline.ingest('beta', {}, scope, KB);
|
|
86
|
+
expect(a.documentId).not.toBe(b.documentId);
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
describe('DefaultIngestionPipeline — remove', () => {
|
|
90
|
+
it('delegates to vectorStore.deleteByDocument', async () => {
|
|
91
|
+
const vs = makeVectorStore();
|
|
92
|
+
const pipeline = new DefaultIngestionPipeline(vs, makeEmbedder());
|
|
93
|
+
await pipeline.remove('doc_9');
|
|
94
|
+
expect(vs.deleteByDocument).toHaveBeenCalledWith('doc_9');
|
|
95
|
+
});
|
|
96
|
+
});
|
|
97
|
+
// Avoid the unused zod import — used in type coverage indirectly.
|
|
98
|
+
void z;
|
|
99
|
+
//# sourceMappingURL=ingestion.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ingestion.test.js","sourceRoot":"","sources":["../../src/rag/ingestion.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACjD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAKvB,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAA;AAEzD,MAAM,EAAE,GAAG,MAAyB,CAAA;AACpC,MAAM,MAAM,GAAG,KAAiB,CAAA;AAChC,MAAM,KAAK,GAAgB,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;AAE/C,SAAS,eAAe;IACvB,OAAO;QACN,MAAM,EAAE,EAAE,CAAC,EAAE,EAAsC;QACnD,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;KACJ,CAAA;AAC5B,CAAC;AAED,SAAS,YAAY;IACpB,OAAO;QACN,EAAE,EAAE,MAAM;QACV,KAAK,EAAE,GAAG;QACV,UAAU,EAAE,CAAC;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAe,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnE,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;KACxC,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,GAAG,GAAG,YAAY,EAAE,CAAA;QAC1B,uDAAuD;QACvD,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAAC,EAAE,EAAE,GAAG,EAAE;YACtD,QAAQ,EAAE,OAAO;YACjB,SAAS,EAAE,EAAE;YACb,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAA;QAC1D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAClC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,GAAG,GAAG,YAAY,EAAE,CAAA;QAC1B,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAAC,EAAE,EAAE,GAAG,EAAE;YACtD,QAAQ,EAAE,OAAO;YACjB,SAAS,EAAE,EAAE;YACb,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAA;QAEnF,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC1C,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC1C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QAE5C,MAAM,cAAc,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QACpE,KAAK,MAAM,CAAC,IAAI,cAAc,EAAE,CAAC;YAChC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAA;YAC5C,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YAClC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAC/B,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACtC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAA;YAC3C,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;QAC3D,CAAC;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAAC,eAAe,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;QAChF,MAAM,CAAC,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAA;QACvD,MAAM,CAAC,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAA;QACtD,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAAC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;QACjE,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAqB,CAAC,CAAA;QAC5C,MAAM,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,kEAAkE;AAClE,KAAK,CAAC,CAAA"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `DefaultKnowledgeBase` is a composition layer: it builds an
|
|
5
|
+
* ingestion pipeline + a retriever from its config and delegates
|
|
6
|
+
* `ingest` / `query` / `remove` / `clear` to them.
|
|
7
|
+
* - `id` is set from `config.id` if provided, else generated once
|
|
8
|
+
* via `generateKnowledgeBaseId` — never regenerated later.
|
|
9
|
+
* - `ingest(content, metadata)` invokes the ingestion pipeline with
|
|
10
|
+
* the configured scope + kb id; returns whatever the pipeline
|
|
11
|
+
* returns.
|
|
12
|
+
* - `query(query)` invokes the retriever with scope + kb id.
|
|
13
|
+
* - `clear()` calls `vectorStore.deleteByKnowledgeBase(id, tenantId)`.
|
|
14
|
+
* - `remove(documentId)` calls `vectorStore.deleteByDocument`.
|
|
15
|
+
*/
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=knowledge-base.test.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"knowledge-base.test.d.ts","sourceRoot":"","sources":["../../src/rag/knowledge-base.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG"}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `DefaultKnowledgeBase` is a composition layer: it builds an
|
|
5
|
+
* ingestion pipeline + a retriever from its config and delegates
|
|
6
|
+
* `ingest` / `query` / `remove` / `clear` to them.
|
|
7
|
+
* - `id` is set from `config.id` if provided, else generated once
|
|
8
|
+
* via `generateKnowledgeBaseId` — never regenerated later.
|
|
9
|
+
* - `ingest(content, metadata)` invokes the ingestion pipeline with
|
|
10
|
+
* the configured scope + kb id; returns whatever the pipeline
|
|
11
|
+
* returns.
|
|
12
|
+
* - `query(query)` invokes the retriever with scope + kb id.
|
|
13
|
+
* - `clear()` calls `vectorStore.deleteByKnowledgeBase(id, tenantId)`.
|
|
14
|
+
* - `remove(documentId)` calls `vectorStore.deleteByDocument`.
|
|
15
|
+
*/
|
|
16
|
+
import { describe, expect, it, vi } from 'vitest';
|
|
17
|
+
import { DefaultKnowledgeBase } from './knowledge-base.js';
|
|
18
|
+
const TENANT = 't_1';
|
|
19
|
+
function makeVectorStore() {
|
|
20
|
+
return {
|
|
21
|
+
upsert: vi.fn(),
|
|
22
|
+
search: vi.fn(async () => []),
|
|
23
|
+
delete: vi.fn(),
|
|
24
|
+
deleteByDocument: vi.fn(),
|
|
25
|
+
deleteByKnowledgeBase: vi.fn(),
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
function makeEmbedder() {
|
|
29
|
+
return {
|
|
30
|
+
id: 'mock',
|
|
31
|
+
model: 'x',
|
|
32
|
+
dimensions: 3,
|
|
33
|
+
embed: vi.fn(async (texts) => texts.map(() => [1, 0, 0])),
|
|
34
|
+
embedQuery: vi.fn(async () => [1, 0, 0]),
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
describe('DefaultKnowledgeBase', () => {
|
|
38
|
+
it('uses the id provided in config when set', () => {
|
|
39
|
+
const kb = new DefaultKnowledgeBase({ id: 'kb_fixed', name: 'kb', tenantId: TENANT }, makeVectorStore(), makeEmbedder());
|
|
40
|
+
expect(kb.id).toBe('kb_fixed');
|
|
41
|
+
expect(kb.config.id).toBe('kb_fixed');
|
|
42
|
+
});
|
|
43
|
+
it('generates an id when none is provided', () => {
|
|
44
|
+
const kb = new DefaultKnowledgeBase({ name: 'kb', tenantId: TENANT }, makeVectorStore(), makeEmbedder());
|
|
45
|
+
expect(kb.id).toMatch(/^kb_/);
|
|
46
|
+
});
|
|
47
|
+
it('ingest delegates to the ingestion pipeline and carries metadata through', async () => {
|
|
48
|
+
const vs = makeVectorStore();
|
|
49
|
+
const kb = new DefaultKnowledgeBase({ id: 'kb_fixed', name: 'kb', tenantId: TENANT }, vs, makeEmbedder());
|
|
50
|
+
const result = await kb.ingest('hello world', { source: 'readme' });
|
|
51
|
+
expect(result.documentId).toMatch(/^doc_/);
|
|
52
|
+
expect(vs.upsert).toHaveBeenCalled();
|
|
53
|
+
const chunks = vi.mocked(vs.upsert).mock.calls[0]?.[0] ?? [];
|
|
54
|
+
expect(chunks[0]?.knowledgeBaseId).toBe('kb_fixed');
|
|
55
|
+
expect(chunks[0]?.tenantId).toBe(TENANT);
|
|
56
|
+
});
|
|
57
|
+
it('remove delegates to vectorStore.deleteByDocument', async () => {
|
|
58
|
+
const vs = makeVectorStore();
|
|
59
|
+
const kb = new DefaultKnowledgeBase({ name: 'kb', tenantId: TENANT }, vs, makeEmbedder());
|
|
60
|
+
await kb.remove('doc_1');
|
|
61
|
+
expect(vs.deleteByDocument).toHaveBeenCalledWith('doc_1');
|
|
62
|
+
});
|
|
63
|
+
it('clear delegates to vectorStore.deleteByKnowledgeBase with id + tenantId', async () => {
|
|
64
|
+
const vs = makeVectorStore();
|
|
65
|
+
const kb = new DefaultKnowledgeBase({ id: 'kb_fixed', name: 'kb', tenantId: TENANT }, vs, makeEmbedder());
|
|
66
|
+
await kb.clear();
|
|
67
|
+
expect(vs.deleteByKnowledgeBase).toHaveBeenCalledWith('kb_fixed', TENANT);
|
|
68
|
+
});
|
|
69
|
+
it('query delegates to retriever', async () => {
|
|
70
|
+
const vs = makeVectorStore();
|
|
71
|
+
const kb = new DefaultKnowledgeBase({ name: 'kb', tenantId: TENANT }, vs, makeEmbedder());
|
|
72
|
+
const out = await kb.query({ text: 'hi' });
|
|
73
|
+
expect(out.mode).toBeDefined();
|
|
74
|
+
expect(vs.search).toHaveBeenCalled();
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
//# sourceMappingURL=knowledge-base.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"knowledge-base.test.js","sourceRoot":"","sources":["../../src/rag/knowledge-base.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAKjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAA;AAE1D,MAAM,MAAM,GAAG,KAAiB,CAAA;AAEhC,SAAS,eAAe;IACvB,OAAO;QACN,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;QAC7B,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC9B,CAAA;AACF,CAAC;AAED,SAAS,YAAY;IACpB,OAAO;QACN,EAAE,EAAE,MAAM;QACV,KAAK,EAAE,GAAG;QACV,UAAU,EAAE,CAAC;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAe,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnE,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;KACxC,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QAClD,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAClC,EAAE,EAAE,EAAE,UAA6B,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EACnE,eAAe,EAAE,EACjB,YAAY,EAAE,CACd,CAAA;QACD,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC9B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAChD,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAClC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAChC,eAAe,EAAE,EACjB,YAAY,EAAE,CACd,CAAA;QACD,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IAC9B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAClC,EAAE,EAAE,EAAE,UAA6B,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EACnE,EAAE,EACF,YAAY,EAAE,CACd,CAAA;QACD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QACnE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QAC1C,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACpC,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;QACzF,MAAM,EAAE,CAAC,MAAM,CAAC,OAAqB,CAAC,CAAA;QACtC,MAAM,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAClC,EAAE,EAAE,EAAE,UAA6B,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EACnE,EAAE,EACF,YAAY,EAAE,CACd,CAAA;QACD,MAAM,EAAE,CAAC,KAAK,EAAE,CAAA;QAChB,MAAM,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,EAAE,GAAG,eAAe,EAAE,CAAA;QAC5B,MAAM,EAAE,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;QACzF,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;QAC1C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAA;QAC9B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAA;IACrC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
package/dist/rag/rag-tool.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { RAGToolConfig } from '../types/rag/index.js';
|
|
2
|
-
export declare function createRAGTool(config: RAGToolConfig): import("../
|
|
2
|
+
export declare function createRAGTool(config: RAGToolConfig): import("../public-types.js").ToolDefinition<{
|
|
3
3
|
query: string;
|
|
4
4
|
knowledge_base_id?: string | undefined;
|
|
5
5
|
top_k?: number | undefined;
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `createRAGTool(config)` returns a `defineTool`-wrapped
|
|
5
|
+
* `knowledge_search` tool with `category: 'analysis'`,
|
|
6
|
+
* `permissions: ['network_access']`, read-only.
|
|
7
|
+
* - Knowledge base selection:
|
|
8
|
+
* - If `input.knowledge_base_id` is set, looks up by it.
|
|
9
|
+
* - Else uses `config.defaultKnowledgeBaseId`.
|
|
10
|
+
* - Else picks the first value from
|
|
11
|
+
* `config.knowledgeBases.values().next()`.
|
|
12
|
+
* - Returns `{success: false, error: ...}` if no knowledge base is
|
|
13
|
+
* resolved.
|
|
14
|
+
* - When the query returns no chunks, returns the literal
|
|
15
|
+
* "No relevant information found..." message.
|
|
16
|
+
* - When chunks exist, assembles them via `assembleRAGContext` and
|
|
17
|
+
* returns `{success: true, output: content, data: {sources, mode, durationMs}}`.
|
|
18
|
+
* - `top_k` override precedence: input > config.topK > 5.
|
|
19
|
+
*/
|
|
20
|
+
export {};
|
|
21
|
+
//# sourceMappingURL=rag-tool.test.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-tool.test.d.ts","sourceRoot":"","sources":["../../src/rag/rag-tool.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG"}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `createRAGTool(config)` returns a `defineTool`-wrapped
|
|
5
|
+
* `knowledge_search` tool with `category: 'analysis'`,
|
|
6
|
+
* `permissions: ['network_access']`, read-only.
|
|
7
|
+
* - Knowledge base selection:
|
|
8
|
+
* - If `input.knowledge_base_id` is set, looks up by it.
|
|
9
|
+
* - Else uses `config.defaultKnowledgeBaseId`.
|
|
10
|
+
* - Else picks the first value from
|
|
11
|
+
* `config.knowledgeBases.values().next()`.
|
|
12
|
+
* - Returns `{success: false, error: ...}` if no knowledge base is
|
|
13
|
+
* resolved.
|
|
14
|
+
* - When the query returns no chunks, returns the literal
|
|
15
|
+
* "No relevant information found..." message.
|
|
16
|
+
* - When chunks exist, assembles them via `assembleRAGContext` and
|
|
17
|
+
* returns `{success: true, output: content, data: {sources, mode, durationMs}}`.
|
|
18
|
+
* - `top_k` override precedence: input > config.topK > 5.
|
|
19
|
+
*/
|
|
20
|
+
import { describe, expect, it, vi } from 'vitest';
|
|
21
|
+
import { createRAGTool } from './rag-tool.js';
|
|
22
|
+
const TENANT = 't_1';
|
|
23
|
+
const KB_A = 'kb_a';
|
|
24
|
+
const KB_B = 'kb_b';
|
|
25
|
+
function makeKB(retrievalResult) {
|
|
26
|
+
const kb = {
|
|
27
|
+
id: KB_A,
|
|
28
|
+
config: { id: KB_A, tenantId: TENANT },
|
|
29
|
+
ingest: vi.fn(),
|
|
30
|
+
remove: vi.fn(),
|
|
31
|
+
query: vi.fn(async () => retrievalResult),
|
|
32
|
+
clear: vi.fn(),
|
|
33
|
+
};
|
|
34
|
+
return kb;
|
|
35
|
+
}
|
|
36
|
+
function searchResult(content) {
|
|
37
|
+
return {
|
|
38
|
+
chunk: {
|
|
39
|
+
id: 'c_1',
|
|
40
|
+
documentId: 'd_1',
|
|
41
|
+
knowledgeBaseId: KB_A,
|
|
42
|
+
tenantId: TENANT,
|
|
43
|
+
content,
|
|
44
|
+
index: 0,
|
|
45
|
+
tokenCount: 0,
|
|
46
|
+
metadata: {},
|
|
47
|
+
createdAt: 0,
|
|
48
|
+
},
|
|
49
|
+
score: 0.9,
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
const ctx = {};
|
|
53
|
+
describe('createRAGTool', () => {
|
|
54
|
+
it('is named knowledge_search + read-only', () => {
|
|
55
|
+
const tool = createRAGTool({ knowledgeBases: new Map() });
|
|
56
|
+
expect(tool.name).toBe('knowledge_search');
|
|
57
|
+
expect(tool.isReadOnly?.({ query: 'x' })).toBe(true);
|
|
58
|
+
});
|
|
59
|
+
it('returns an error when no knowledge base is resolved', async () => {
|
|
60
|
+
const tool = createRAGTool({ knowledgeBases: new Map() });
|
|
61
|
+
const result = await tool.execute({ query: 'hi' }, ctx);
|
|
62
|
+
expect(result.success).toBe(false);
|
|
63
|
+
expect(result.error).toMatch(/Knowledge base not found/);
|
|
64
|
+
});
|
|
65
|
+
it('picks KB by explicit input.knowledge_base_id when set', async () => {
|
|
66
|
+
const kbA = makeKB({
|
|
67
|
+
chunks: [searchResult('from A')],
|
|
68
|
+
query: 'hi',
|
|
69
|
+
mode: 'vector',
|
|
70
|
+
durationMs: 1,
|
|
71
|
+
});
|
|
72
|
+
const kbB = makeKB({
|
|
73
|
+
chunks: [searchResult('from B')],
|
|
74
|
+
query: 'hi',
|
|
75
|
+
mode: 'vector',
|
|
76
|
+
durationMs: 1,
|
|
77
|
+
});
|
|
78
|
+
const tool = createRAGTool({
|
|
79
|
+
knowledgeBases: new Map([
|
|
80
|
+
[KB_A, kbA],
|
|
81
|
+
[KB_B, kbB],
|
|
82
|
+
]),
|
|
83
|
+
});
|
|
84
|
+
const out = await tool.execute({ query: 'hi', knowledge_base_id: KB_B }, ctx);
|
|
85
|
+
expect(out.success).toBe(true);
|
|
86
|
+
expect(kbB.query).toHaveBeenCalled();
|
|
87
|
+
expect(kbA.query).not.toHaveBeenCalled();
|
|
88
|
+
});
|
|
89
|
+
it('falls back to defaultKnowledgeBaseId when input.knowledge_base_id is absent', async () => {
|
|
90
|
+
const kbA = makeKB({
|
|
91
|
+
chunks: [searchResult('from A')],
|
|
92
|
+
query: 'hi',
|
|
93
|
+
mode: 'vector',
|
|
94
|
+
durationMs: 1,
|
|
95
|
+
});
|
|
96
|
+
const kbB = makeKB({
|
|
97
|
+
chunks: [],
|
|
98
|
+
query: 'hi',
|
|
99
|
+
mode: 'vector',
|
|
100
|
+
durationMs: 1,
|
|
101
|
+
});
|
|
102
|
+
const tool = createRAGTool({
|
|
103
|
+
knowledgeBases: new Map([
|
|
104
|
+
[KB_A, kbA],
|
|
105
|
+
[KB_B, kbB],
|
|
106
|
+
]),
|
|
107
|
+
defaultKnowledgeBaseId: KB_B,
|
|
108
|
+
});
|
|
109
|
+
await tool.execute({ query: 'hi' }, ctx);
|
|
110
|
+
expect(kbB.query).toHaveBeenCalled();
|
|
111
|
+
expect(kbA.query).not.toHaveBeenCalled();
|
|
112
|
+
});
|
|
113
|
+
it('returns "No relevant information..." when the query returns zero chunks', async () => {
|
|
114
|
+
const kb = makeKB({ chunks: [], query: 'hi', mode: 'vector', durationMs: 1 });
|
|
115
|
+
const tool = createRAGTool({ knowledgeBases: new Map([[KB_A, kb]]) });
|
|
116
|
+
const result = await tool.execute({ query: 'hi' }, ctx);
|
|
117
|
+
expect(result.success).toBe(true);
|
|
118
|
+
expect(result.output).toBe('No relevant information found in the knowledge base for this query.');
|
|
119
|
+
});
|
|
120
|
+
it('returns assembled content + metadata on hit', async () => {
|
|
121
|
+
const kb = makeKB({
|
|
122
|
+
chunks: [searchResult('answer text')],
|
|
123
|
+
query: 'hi',
|
|
124
|
+
mode: 'vector',
|
|
125
|
+
durationMs: 42,
|
|
126
|
+
});
|
|
127
|
+
const tool = createRAGTool({ knowledgeBases: new Map([[KB_A, kb]]) });
|
|
128
|
+
const result = await tool.execute({ query: 'hi' }, ctx);
|
|
129
|
+
expect(result.success).toBe(true);
|
|
130
|
+
expect(result.output).toContain('answer text');
|
|
131
|
+
expect(result.data?.mode).toBe('vector');
|
|
132
|
+
expect(result.data?.durationMs).toBe(42);
|
|
133
|
+
});
|
|
134
|
+
it('top_k override: input > config.topK > 5 default', async () => {
|
|
135
|
+
const kb = makeKB({
|
|
136
|
+
chunks: [searchResult('x')],
|
|
137
|
+
query: 'hi',
|
|
138
|
+
mode: 'vector',
|
|
139
|
+
durationMs: 1,
|
|
140
|
+
});
|
|
141
|
+
const tool = createRAGTool({
|
|
142
|
+
knowledgeBases: new Map([[KB_A, kb]]),
|
|
143
|
+
topK: 8,
|
|
144
|
+
});
|
|
145
|
+
await tool.execute({ query: 'hi', top_k: 3 }, ctx);
|
|
146
|
+
expect(kb.query).toHaveBeenCalledWith(expect.objectContaining({ config: { topK: 3 } }));
|
|
147
|
+
});
|
|
148
|
+
});
|
|
149
|
+
//# sourceMappingURL=rag-tool.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-tool.test.js","sourceRoot":"","sources":["../../src/rag/rag-tool.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAMjD,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAE7C,MAAM,MAAM,GAAG,KAAiB,CAAA;AAChC,MAAM,IAAI,GAAG,MAAyB,CAAA;AACtC,MAAM,IAAI,GAAG,MAAyB,CAAA;AAEtC,SAAS,MAAM,CAAC,eAAgC;IAC/C,MAAM,EAAE,GAAG;QACV,EAAE,EAAE,IAAI;QACR,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE;QACtC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,eAAe,CAAC;QACzC,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACc,CAAA;IAC7B,OAAO,EAAE,CAAA;AACV,CAAC;AAED,SAAS,YAAY,CAAC,OAAe;IACpC,OAAO;QACN,KAAK,EAAE;YACN,EAAE,EAAE,KAAgB;YACpB,UAAU,EAAE,KAAmB;YAC/B,eAAe,EAAE,IAAI;YACrB,QAAQ,EAAE,MAAM;YAChB,OAAO;YACP,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC;YACb,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,CAAC;SACZ;QACD,KAAK,EAAE,GAAG;KACV,CAAA;AACF,CAAC;AAED,MAAM,GAAG,GAAgB,EAAiB,CAAA;AAE1C,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAChD,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,cAAc,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC,CAAA;QACzD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAC1C,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,cAAc,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC,CAAA;QACzD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,CAAA;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,GAAG,GAAG,MAAM,CAAC;YAClB,MAAM,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,CAAC;SACb,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,MAAM,CAAC;YAClB,MAAM,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,CAAC;SACb,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,aAAa,CAAC;YAC1B,cAAc,EAAE,IAAI,GAAG,CAAC;gBACvB,CAAC,IAAI,EAAE,GAAG,CAAC;gBACX,CAAC,IAAI,EAAE,GAAG,CAAC;aACX,CAAC;SACF,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,CAAA;QAC7E,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC9B,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,GAAG,GAAG,MAAM,CAAC;YAClB,MAAM,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,CAAC;SACb,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,MAAM,CAAC;YAClB,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,CAAC;SACb,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,aAAa,CAAC;YAC1B,cAAc,EAAE,IAAI,GAAG,CAAC;gBACvB,CAAC,IAAI,EAAE,GAAG,CAAC;gBACX,CAAC,IAAI,EAAE,GAAG,CAAC;aACX,CAAC;YACF,sBAAsB,EAAE,IAAI;SAC5B,CAAC,CAAA;QACF,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,CAAA;QACxC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,GAAG,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAA;QAC7E,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,cAAc,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QACrE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,CAAA;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACzB,qEAAqE,CACrE,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,GAAG,MAAM,CAAC;YACjB,MAAM,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACrC,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACd,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,cAAc,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QACrE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,CAAA;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;QAC9C,MAAM,CAAE,MAAM,CAAC,IAA0B,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/D,MAAM,CAAE,MAAM,CAAC,IAAgC,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,GAAG,MAAM,CAAC;YACjB,MAAM,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAC3B,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,CAAC;SACb,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,aAAa,CAAC;YAC1B,cAAc,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;YACrC,IAAI,EAAE,CAAC;SACP,CAAC,CAAA;QACF,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;QAClD,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;IACxF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Current-code invariants asserted (2026-04-21, ses_006 Phase 4):
|
|
3
|
+
*
|
|
4
|
+
* - `DefaultRetriever.retrieve(query, scope, kbId)` dispatches by
|
|
5
|
+
* `effectiveConfig.mode` (query.config overrides instance config).
|
|
6
|
+
* - Query expansion: if `recentMessages` is present and non-empty,
|
|
7
|
+
* appends the last 3 messages joined with spaces after a
|
|
8
|
+
* `\n\nContext: ` marker. Otherwise passes `query.text` verbatim.
|
|
9
|
+
* - `vector` mode: embeds the (expanded) query once, forwards
|
|
10
|
+
* `{embedding, topK, tenantId, knowledgeBaseId, minScore}` to the
|
|
11
|
+
* store, returns store results unchanged.
|
|
12
|
+
* - `keyword` mode: fetches `topK * 2` results from vector search with
|
|
13
|
+
* `minScore: 0`, then rescores via BM25 against the query's
|
|
14
|
+
* tokenized form. Filters score > 0, sorts desc, slices to topK.
|
|
15
|
+
* Tokenization: lowercase, strip non-`\w\s`, split by whitespace,
|
|
16
|
+
* drop tokens length ≤ 1.
|
|
17
|
+
* - `hybrid` mode: runs vector + keyword in parallel, merges by
|
|
18
|
+
* `chunk.id`, weights vector by `alpha` and keyword by `1 - alpha`
|
|
19
|
+
* (default alpha = 0.7). Sorts desc, slices to topK.
|
|
20
|
+
* - Result shape: `{chunks, query, expandedQuery?, mode, durationMs}`.
|
|
21
|
+
* `expandedQuery` is omitted when equal to original `query.text`.
|
|
22
|
+
* - Unknown mode would hit the exhaustive throw (unreachable via
|
|
23
|
+
* types; not asserted).
|
|
24
|
+
*/
|
|
25
|
+
export {};
|
|
26
|
+
//# sourceMappingURL=retriever.test.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retriever.test.d.ts","sourceRoot":"","sources":["../../src/rag/retriever.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG"}
|