@astragenie/astramemory-local 0.7.2
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 +341 -0
- package/README.md +419 -0
- package/dist/backup/retention.d.ts +15 -0
- package/dist/backup/retention.js +62 -0
- package/dist/backup/retention.js.map +1 -0
- package/dist/backup/snapshot.d.ts +21 -0
- package/dist/backup/snapshot.js +55 -0
- package/dist/backup/snapshot.js.map +1 -0
- package/dist/backup/verify.d.ts +23 -0
- package/dist/backup/verify.js +77 -0
- package/dist/backup/verify.js.map +1 -0
- package/dist/budget/tracker.d.ts +58 -0
- package/dist/budget/tracker.js +102 -0
- package/dist/budget/tracker.js.map +1 -0
- package/dist/capture/codex.d.ts +63 -0
- package/dist/capture/codex.js +0 -0
- package/dist/capture/codex.js.map +1 -0
- package/dist/cli/backup.d.ts +1 -0
- package/dist/cli/backup.js +112 -0
- package/dist/cli/backup.js.map +1 -0
- package/dist/cli/budget.d.ts +7 -0
- package/dist/cli/budget.js +44 -0
- package/dist/cli/budget.js.map +1 -0
- package/dist/cli/capture.d.ts +10 -0
- package/dist/cli/capture.js +113 -0
- package/dist/cli/capture.js.map +1 -0
- package/dist/cli/consolidate.d.ts +16 -0
- package/dist/cli/consolidate.js +146 -0
- package/dist/cli/consolidate.js.map +1 -0
- package/dist/cli/doctor.d.ts +1 -0
- package/dist/cli/doctor.js +54 -0
- package/dist/cli/doctor.js.map +1 -0
- package/dist/cli/entity-backfill.d.ts +10 -0
- package/dist/cli/entity-backfill.js +46 -0
- package/dist/cli/entity-backfill.js.map +1 -0
- package/dist/cli/hook-install.d.ts +45 -0
- package/dist/cli/hook-install.js +77 -0
- package/dist/cli/hook-install.js.map +1 -0
- package/dist/cli/index.d.ts +2 -0
- package/dist/cli/index.js +312 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/init.d.ts +16 -0
- package/dist/cli/init.js +431 -0
- package/dist/cli/init.js.map +1 -0
- package/dist/cli/mcp-stdio.d.ts +18 -0
- package/dist/cli/mcp-stdio.js +67 -0
- package/dist/cli/mcp-stdio.js.map +1 -0
- package/dist/cli/memory.d.ts +15 -0
- package/dist/cli/memory.js +52 -0
- package/dist/cli/memory.js.map +1 -0
- package/dist/cli/open-runtime-db.d.ts +15 -0
- package/dist/cli/open-runtime-db.js +37 -0
- package/dist/cli/open-runtime-db.js.map +1 -0
- package/dist/cli/pair.d.ts +29 -0
- package/dist/cli/pair.js +64 -0
- package/dist/cli/pair.js.map +1 -0
- package/dist/cli/providers.d.ts +10 -0
- package/dist/cli/providers.js +97 -0
- package/dist/cli/providers.js.map +1 -0
- package/dist/cli/queue-purge.d.ts +5 -0
- package/dist/cli/queue-purge.js +92 -0
- package/dist/cli/queue-purge.js.map +1 -0
- package/dist/cli/queue.d.ts +29 -0
- package/dist/cli/queue.js +73 -0
- package/dist/cli/queue.js.map +1 -0
- package/dist/cli/rebuild.d.ts +15 -0
- package/dist/cli/rebuild.js +70 -0
- package/dist/cli/rebuild.js.map +1 -0
- package/dist/cli/reembed-dim.d.ts +21 -0
- package/dist/cli/reembed-dim.js +199 -0
- package/dist/cli/reembed-dim.js.map +1 -0
- package/dist/cli/reinstall.d.ts +1 -0
- package/dist/cli/reinstall.js +205 -0
- package/dist/cli/reinstall.js.map +1 -0
- package/dist/cli/restore.d.ts +1 -0
- package/dist/cli/restore.js +167 -0
- package/dist/cli/restore.js.map +1 -0
- package/dist/cli/retag.d.ts +14 -0
- package/dist/cli/retag.js +62 -0
- package/dist/cli/retag.js.map +1 -0
- package/dist/cli/search.d.ts +66 -0
- package/dist/cli/search.js +174 -0
- package/dist/cli/search.js.map +1 -0
- package/dist/cli/serve.d.ts +9 -0
- package/dist/cli/serve.js +364 -0
- package/dist/cli/serve.js.map +1 -0
- package/dist/cli/service.d.ts +1 -0
- package/dist/cli/service.js +121 -0
- package/dist/cli/service.js.map +1 -0
- package/dist/cli/sync.d.ts +15 -0
- package/dist/cli/sync.js +61 -0
- package/dist/cli/sync.js.map +1 -0
- package/dist/cli/token.d.ts +24 -0
- package/dist/cli/token.js +77 -0
- package/dist/cli/token.js.map +1 -0
- package/dist/cli/wait-health.d.ts +4 -0
- package/dist/cli/wait-health.js +23 -0
- package/dist/cli/wait-health.js.map +1 -0
- package/dist/config/config.d.ts +127 -0
- package/dist/config/config.js +38 -0
- package/dist/config/config.js.map +1 -0
- package/dist/config/datadir.d.ts +30 -0
- package/dist/config/datadir.js +65 -0
- package/dist/config/datadir.js.map +1 -0
- package/dist/config/loader.d.ts +23 -0
- package/dist/config/loader.js +102 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/migrate-dirs.d.ts +36 -0
- package/dist/config/migrate-dirs.js +132 -0
- package/dist/config/migrate-dirs.js.map +1 -0
- package/dist/config/persist-envs.d.ts +23 -0
- package/dist/config/persist-envs.js +119 -0
- package/dist/config/persist-envs.js.map +1 -0
- package/dist/config/resolve-runtime.d.ts +19 -0
- package/dist/config/resolve-runtime.js +53 -0
- package/dist/config/resolve-runtime.js.map +1 -0
- package/dist/config/secrets.d.ts +28 -0
- package/dist/config/secrets.js +38 -0
- package/dist/config/secrets.js.map +1 -0
- package/dist/config/sync-settings.d.ts +16 -0
- package/dist/config/sync-settings.js +34 -0
- package/dist/config/sync-settings.js.map +1 -0
- package/dist/config/writer.d.ts +19 -0
- package/dist/config/writer.js +121 -0
- package/dist/config/writer.js.map +1 -0
- package/dist/consolidate/consolidate.d.ts +80 -0
- package/dist/consolidate/consolidate.js +0 -0
- package/dist/consolidate/consolidate.js.map +1 -0
- package/dist/consolidate/proposals.d.ts +35 -0
- package/dist/consolidate/proposals.js +66 -0
- package/dist/consolidate/proposals.js.map +1 -0
- package/dist/contracts/atom-wire.d.ts +48 -0
- package/dist/contracts/atom-wire.js +55 -0
- package/dist/contracts/atom-wire.js.map +1 -0
- package/dist/contracts/embed.d.ts +41 -0
- package/dist/contracts/embed.js +20 -0
- package/dist/contracts/embed.js.map +1 -0
- package/dist/contracts/index.d.ts +5 -0
- package/dist/contracts/index.js +6 -0
- package/dist/contracts/index.js.map +1 -0
- package/dist/contracts/job.d.ts +113 -0
- package/dist/contracts/job.js +32 -0
- package/dist/contracts/job.js.map +1 -0
- package/dist/contracts/llm.d.ts +30 -0
- package/dist/contracts/llm.js +2 -0
- package/dist/contracts/llm.js.map +1 -0
- package/dist/contracts/memory.d.ts +47 -0
- package/dist/contracts/memory.js +5 -0
- package/dist/contracts/memory.js.map +1 -0
- package/dist/contracts/vector.d.ts +29 -0
- package/dist/contracts/vector.js +2 -0
- package/dist/contracts/vector.js.map +1 -0
- package/dist/distill/flatten-turns.d.ts +1 -0
- package/dist/distill/flatten-turns.js +50 -0
- package/dist/distill/flatten-turns.js.map +1 -0
- package/dist/distill/pipeline.d.ts +45 -0
- package/dist/distill/pipeline.js +113 -0
- package/dist/distill/pipeline.js.map +1 -0
- package/dist/distill/prompts/extract.d.ts +122 -0
- package/dist/distill/prompts/extract.js +67 -0
- package/dist/distill/prompts/extract.js.map +1 -0
- package/dist/distill/stages/01-cleanup.d.ts +9 -0
- package/dist/distill/stages/01-cleanup.js +67 -0
- package/dist/distill/stages/01-cleanup.js.map +1 -0
- package/dist/distill/stages/02-normalize.d.ts +9 -0
- package/dist/distill/stages/02-normalize.js +76 -0
- package/dist/distill/stages/02-normalize.js.map +1 -0
- package/dist/distill/stages/03-chunk.d.ts +22 -0
- package/dist/distill/stages/03-chunk.js +138 -0
- package/dist/distill/stages/03-chunk.js.map +1 -0
- package/dist/distill/stages/04-compact.d.ts +28 -0
- package/dist/distill/stages/04-compact.js +69 -0
- package/dist/distill/stages/04-compact.js.map +1 -0
- package/dist/distill/stages/05-extract.d.ts +35 -0
- package/dist/distill/stages/05-extract.js +101 -0
- package/dist/distill/stages/05-extract.js.map +1 -0
- package/dist/distill/stages/06-reduce.d.ts +16 -0
- package/dist/distill/stages/06-reduce.js +30 -0
- package/dist/distill/stages/06-reduce.js.map +1 -0
- package/dist/distill/stages/07-memory-normalize.d.ts +27 -0
- package/dist/distill/stages/07-memory-normalize.js +65 -0
- package/dist/distill/stages/07-memory-normalize.js.map +1 -0
- package/dist/distill/stages/08-embed-index.d.ts +31 -0
- package/dist/distill/stages/08-embed-index.js +82 -0
- package/dist/distill/stages/08-embed-index.js.map +1 -0
- package/dist/doctor/checks.d.ts +77 -0
- package/dist/doctor/checks.js +626 -0
- package/dist/doctor/checks.js.map +1 -0
- package/dist/doctor/hardening-checks.d.ts +9 -0
- package/dist/doctor/hardening-checks.js +182 -0
- package/dist/doctor/hardening-checks.js.map +1 -0
- package/dist/doctor/probes/embed-probe.d.ts +19 -0
- package/dist/doctor/probes/embed-probe.js +47 -0
- package/dist/doctor/probes/embed-probe.js.map +1 -0
- package/dist/doctor/probes/llm-chat-probe.d.ts +11 -0
- package/dist/doctor/probes/llm-chat-probe.js +41 -0
- package/dist/doctor/probes/llm-chat-probe.js.map +1 -0
- package/dist/doctor/probes/plugin-coexistence.d.ts +14 -0
- package/dist/doctor/probes/plugin-coexistence.js +60 -0
- package/dist/doctor/probes/plugin-coexistence.js.map +1 -0
- package/dist/doctor/runner.d.ts +17 -0
- package/dist/doctor/runner.js +53 -0
- package/dist/doctor/runner.js.map +1 -0
- package/dist/doctor/types.d.ts +12 -0
- package/dist/doctor/types.js +2 -0
- package/dist/doctor/types.js.map +1 -0
- package/dist/entity/backfill.d.ts +30 -0
- package/dist/entity/backfill.js +55 -0
- package/dist/entity/backfill.js.map +1 -0
- package/dist/entity/extract-entities.d.ts +27 -0
- package/dist/entity/extract-entities.js +86 -0
- package/dist/entity/extract-entities.js.map +1 -0
- package/dist/entity/normalize.d.ts +17 -0
- package/dist/entity/normalize.js +20 -0
- package/dist/entity/normalize.js.map +1 -0
- package/dist/eval/harness.d.ts +96 -0
- package/dist/eval/harness.js +119 -0
- package/dist/eval/harness.js.map +1 -0
- package/dist/eval/metrics.d.ts +23 -0
- package/dist/eval/metrics.js +44 -0
- package/dist/eval/metrics.js.map +1 -0
- package/dist/log/correlation.d.ts +24 -0
- package/dist/log/correlation.js +33 -0
- package/dist/log/correlation.js.map +1 -0
- package/dist/log/logger.d.ts +38 -0
- package/dist/log/logger.js +129 -0
- package/dist/log/logger.js.map +1 -0
- package/dist/log/scrub.d.ts +33 -0
- package/dist/log/scrub.js +91 -0
- package/dist/log/scrub.js.map +1 -0
- package/dist/mcp/server.d.ts +36 -0
- package/dist/mcp/server.js +553 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/memory-tool/adapter.d.ts +73 -0
- package/dist/memory-tool/adapter.js +269 -0
- package/dist/memory-tool/adapter.js.map +1 -0
- package/dist/pipeline/errors.d.ts +21 -0
- package/dist/pipeline/errors.js +34 -0
- package/dist/pipeline/errors.js.map +1 -0
- package/dist/pipeline/failure-classifier.d.ts +13 -0
- package/dist/pipeline/failure-classifier.js +72 -0
- package/dist/pipeline/failure-classifier.js.map +1 -0
- package/dist/pipeline/handler-ctx-ext.d.ts +23 -0
- package/dist/pipeline/handler-ctx-ext.js +19 -0
- package/dist/pipeline/handler-ctx-ext.js.map +1 -0
- package/dist/pipeline/handler.d.ts +20 -0
- package/dist/pipeline/handler.js +2 -0
- package/dist/pipeline/handler.js.map +1 -0
- package/dist/pipeline/handlers/cleanup.d.ts +14 -0
- package/dist/pipeline/handlers/cleanup.js +47 -0
- package/dist/pipeline/handlers/cleanup.js.map +1 -0
- package/dist/pipeline/handlers/consolidate.d.ts +8 -0
- package/dist/pipeline/handlers/consolidate.js +23 -0
- package/dist/pipeline/handlers/consolidate.js.map +1 -0
- package/dist/pipeline/handlers/distill-events.d.ts +15 -0
- package/dist/pipeline/handlers/distill-events.js +134 -0
- package/dist/pipeline/handlers/distill-events.js.map +1 -0
- package/dist/pipeline/handlers/distill.d.ts +17 -0
- package/dist/pipeline/handlers/distill.js +110 -0
- package/dist/pipeline/handlers/distill.js.map +1 -0
- package/dist/pipeline/handlers/reembed.d.ts +10 -0
- package/dist/pipeline/handlers/reembed.js +34 -0
- package/dist/pipeline/handlers/reembed.js.map +1 -0
- package/dist/pipeline/job-repo.d.ts +86 -0
- package/dist/pipeline/job-repo.js +168 -0
- package/dist/pipeline/job-repo.js.map +1 -0
- package/dist/pipeline/mock-providers.d.ts +49 -0
- package/dist/pipeline/mock-providers.js +175 -0
- package/dist/pipeline/mock-providers.js.map +1 -0
- package/dist/pipeline/registry.d.ts +15 -0
- package/dist/pipeline/registry.js +20 -0
- package/dist/pipeline/registry.js.map +1 -0
- package/dist/pipeline/worker.d.ts +41 -0
- package/dist/pipeline/worker.js +167 -0
- package/dist/pipeline/worker.js.map +1 -0
- package/dist/providers/embed/azure-openai.d.ts +25 -0
- package/dist/providers/embed/azure-openai.js +138 -0
- package/dist/providers/embed/azure-openai.js.map +1 -0
- package/dist/providers/embed/ollama.d.ts +17 -0
- package/dist/providers/embed/ollama.js +106 -0
- package/dist/providers/embed/ollama.js.map +1 -0
- package/dist/providers/index.d.ts +19 -0
- package/dist/providers/index.js +72 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/llm/azure-openai.d.ts +20 -0
- package/dist/providers/llm/azure-openai.js +135 -0
- package/dist/providers/llm/azure-openai.js.map +1 -0
- package/dist/providers/llm/ollama.d.ts +13 -0
- package/dist/providers/llm/ollama.js +113 -0
- package/dist/providers/llm/ollama.js.map +1 -0
- package/dist/providers/llm/pricing.d.ts +21 -0
- package/dist/providers/llm/pricing.js +22 -0
- package/dist/providers/llm/pricing.js.map +1 -0
- package/dist/recall/pack.d.ts +32 -0
- package/dist/recall/pack.js +90 -0
- package/dist/recall/pack.js.map +1 -0
- package/dist/recall/policy.d.ts +39 -0
- package/dist/recall/policy.js +96 -0
- package/dist/recall/policy.js.map +1 -0
- package/dist/redact/detectors.d.ts +20 -0
- package/dist/redact/detectors.js +85 -0
- package/dist/redact/detectors.js.map +1 -0
- package/dist/redact/entropy.d.ts +24 -0
- package/dist/redact/entropy.js +77 -0
- package/dist/redact/entropy.js.map +1 -0
- package/dist/redact/index.d.ts +47 -0
- package/dist/redact/index.js +165 -0
- package/dist/redact/index.js.map +1 -0
- package/dist/search/fuse.d.ts +108 -0
- package/dist/search/fuse.js +135 -0
- package/dist/search/fuse.js.map +1 -0
- package/dist/search/query.d.ts +28 -0
- package/dist/search/query.js +70 -0
- package/dist/search/query.js.map +1 -0
- package/dist/search/search.d.ts +164 -0
- package/dist/search/search.js +310 -0
- package/dist/search/search.js.map +1 -0
- package/dist/server/app.d.ts +17 -0
- package/dist/server/app.js +133 -0
- package/dist/server/app.js.map +1 -0
- package/dist/server/health-state.d.ts +29 -0
- package/dist/server/health-state.js +28 -0
- package/dist/server/health-state.js.map +1 -0
- package/dist/server/lib/network.d.ts +12 -0
- package/dist/server/lib/network.js +16 -0
- package/dist/server/lib/network.js.map +1 -0
- package/dist/server/lib/score-contract.d.ts +36 -0
- package/dist/server/lib/score-contract.js +54 -0
- package/dist/server/lib/score-contract.js.map +1 -0
- package/dist/server/lib/stable-stringify.d.ts +10 -0
- package/dist/server/lib/stable-stringify.js +27 -0
- package/dist/server/lib/stable-stringify.js.map +1 -0
- package/dist/server/lib/wire-meta.d.ts +7 -0
- package/dist/server/lib/wire-meta.js +29 -0
- package/dist/server/lib/wire-meta.js.map +1 -0
- package/dist/server/queries/dashboard.d.ts +142 -0
- package/dist/server/queries/dashboard.js +166 -0
- package/dist/server/queries/dashboard.js.map +1 -0
- package/dist/server/routes/consolidation.d.ts +14 -0
- package/dist/server/routes/consolidation.js +67 -0
- package/dist/server/routes/consolidation.js.map +1 -0
- package/dist/server/routes/dashboard-api-html.d.ts +15 -0
- package/dist/server/routes/dashboard-api-html.js +144 -0
- package/dist/server/routes/dashboard-api-html.js.map +1 -0
- package/dist/server/routes/dashboard-consolidation-html.d.ts +26 -0
- package/dist/server/routes/dashboard-consolidation-html.js +202 -0
- package/dist/server/routes/dashboard-consolidation-html.js.map +1 -0
- package/dist/server/routes/dashboard-html.d.ts +15 -0
- package/dist/server/routes/dashboard-html.js +365 -0
- package/dist/server/routes/dashboard-html.js.map +1 -0
- package/dist/server/routes/dashboard-jobs-html.d.ts +18 -0
- package/dist/server/routes/dashboard-jobs-html.js +186 -0
- package/dist/server/routes/dashboard-jobs-html.js.map +1 -0
- package/dist/server/routes/dashboard-search-html.d.ts +18 -0
- package/dist/server/routes/dashboard-search-html.js +189 -0
- package/dist/server/routes/dashboard-search-html.js.map +1 -0
- package/dist/server/routes/dashboard.d.ts +19 -0
- package/dist/server/routes/dashboard.js +68 -0
- package/dist/server/routes/dashboard.js.map +1 -0
- package/dist/server/routes/digest.d.ts +9 -0
- package/dist/server/routes/digest.js +37 -0
- package/dist/server/routes/digest.js.map +1 -0
- package/dist/server/routes/entities.d.ts +12 -0
- package/dist/server/routes/entities.js +46 -0
- package/dist/server/routes/entities.js.map +1 -0
- package/dist/server/routes/health.d.ts +14 -0
- package/dist/server/routes/health.js +100 -0
- package/dist/server/routes/health.js.map +1 -0
- package/dist/server/routes/ingest.d.ts +209 -0
- package/dist/server/routes/ingest.js +454 -0
- package/dist/server/routes/ingest.js.map +1 -0
- package/dist/server/routes/lifecycle.d.ts +21 -0
- package/dist/server/routes/lifecycle.js +132 -0
- package/dist/server/routes/lifecycle.js.map +1 -0
- package/dist/server/routes/mcp.d.ts +15 -0
- package/dist/server/routes/mcp.js +36 -0
- package/dist/server/routes/mcp.js.map +1 -0
- package/dist/server/routes/memory-tool.d.ts +14 -0
- package/dist/server/routes/memory-tool.js +28 -0
- package/dist/server/routes/memory-tool.js.map +1 -0
- package/dist/server/routes/memory.d.ts +7 -0
- package/dist/server/routes/memory.js +19 -0
- package/dist/server/routes/memory.js.map +1 -0
- package/dist/server/routes/recall.d.ts +15 -0
- package/dist/server/routes/recall.js +74 -0
- package/dist/server/routes/recall.js.map +1 -0
- package/dist/server/routes/search.d.ts +12 -0
- package/dist/server/routes/search.js +203 -0
- package/dist/server/routes/search.js.map +1 -0
- package/dist/server/routes/version.d.ts +2 -0
- package/dist/server/routes/version.js +11 -0
- package/dist/server/routes/version.js.map +1 -0
- package/dist/server/routes/why.d.ts +9 -0
- package/dist/server/routes/why.js +38 -0
- package/dist/server/routes/why.js.map +1 -0
- package/dist/service/index.d.ts +10 -0
- package/dist/service/index.js +25 -0
- package/dist/service/index.js.map +1 -0
- package/dist/service/install-flow.d.ts +18 -0
- package/dist/service/install-flow.js +47 -0
- package/dist/service/install-flow.js.map +1 -0
- package/dist/service/instance-lock.d.ts +26 -0
- package/dist/service/instance-lock.js +150 -0
- package/dist/service/instance-lock.js.map +1 -0
- package/dist/service/launchd.d.ts +11 -0
- package/dist/service/launchd.js +196 -0
- package/dist/service/launchd.js.map +1 -0
- package/dist/service/schtasks.d.ts +31 -0
- package/dist/service/schtasks.js +274 -0
- package/dist/service/schtasks.js.map +1 -0
- package/dist/service/shim.d.ts +21 -0
- package/dist/service/shim.js +80 -0
- package/dist/service/shim.js.map +1 -0
- package/dist/service/systemd.d.ts +11 -0
- package/dist/service/systemd.js +150 -0
- package/dist/service/systemd.js.map +1 -0
- package/dist/service/task-xml.d.ts +36 -0
- package/dist/service/task-xml.js +91 -0
- package/dist/service/task-xml.js.map +1 -0
- package/dist/service/types.d.ts +47 -0
- package/dist/service/types.js +2 -0
- package/dist/service/types.js.map +1 -0
- package/dist/storage/archival.d.ts +29 -0
- package/dist/storage/archival.js +47 -0
- package/dist/storage/archival.js.map +1 -0
- package/dist/storage/bearer-keystore.d.ts +34 -0
- package/dist/storage/bearer-keystore.js +75 -0
- package/dist/storage/bearer-keystore.js.map +1 -0
- package/dist/storage/db.d.ts +37 -0
- package/dist/storage/db.js +92 -0
- package/dist/storage/db.js.map +1 -0
- package/dist/storage/entities.d.ts +71 -0
- package/dist/storage/entities.js +141 -0
- package/dist/storage/entities.js.map +1 -0
- package/dist/storage/ingest-idempotency.d.ts +26 -0
- package/dist/storage/ingest-idempotency.js +29 -0
- package/dist/storage/ingest-idempotency.js.map +1 -0
- package/dist/storage/keystore.d.ts +64 -0
- package/dist/storage/keystore.js +194 -0
- package/dist/storage/keystore.js.map +1 -0
- package/dist/storage/memories.d.ts +51 -0
- package/dist/storage/memories.js +67 -0
- package/dist/storage/memories.js.map +1 -0
- package/dist/storage/memory-events.d.ts +145 -0
- package/dist/storage/memory-events.js +287 -0
- package/dist/storage/memory-events.js.map +1 -0
- package/dist/storage/migrate-encrypt.d.ts +16 -0
- package/dist/storage/migrate-encrypt.js +121 -0
- package/dist/storage/migrate-encrypt.js.map +1 -0
- package/dist/storage/migrate.d.ts +27 -0
- package/dist/storage/migrate.js +105 -0
- package/dist/storage/migrate.js.map +1 -0
- package/dist/storage/redaction-log.d.ts +18 -0
- package/dist/storage/redaction-log.js +27 -0
- package/dist/storage/redaction-log.js.map +1 -0
- package/dist/storage/usefulness.d.ts +115 -0
- package/dist/storage/usefulness.js +203 -0
- package/dist/storage/usefulness.js.map +1 -0
- package/dist/sync/conflict-resolve.d.ts +26 -0
- package/dist/sync/conflict-resolve.js +139 -0
- package/dist/sync/conflict-resolve.js.map +1 -0
- package/dist/sync/puller.d.ts +115 -0
- package/dist/sync/puller.js +173 -0
- package/dist/sync/puller.js.map +1 -0
- package/dist/sync/shipper.d.ts +112 -0
- package/dist/sync/shipper.js +189 -0
- package/dist/sync/shipper.js.map +1 -0
- package/dist/tag-hygiene/backfill.d.ts +50 -0
- package/dist/tag-hygiene/backfill.js +117 -0
- package/dist/tag-hygiene/backfill.js.map +1 -0
- package/dist/tag-hygiene/derive-repo.d.ts +9 -0
- package/dist/tag-hygiene/derive-repo.js +19 -0
- package/dist/tag-hygiene/derive-repo.js.map +1 -0
- package/dist/tag-hygiene/tier2-infer.d.ts +28 -0
- package/dist/tag-hygiene/tier2-infer.js +72 -0
- package/dist/tag-hygiene/tier2-infer.js.map +1 -0
- package/dist/vector/sqlite-vec.d.ts +16 -0
- package/dist/vector/sqlite-vec.js +49 -0
- package/dist/vector/sqlite-vec.js.map +1 -0
- package/migrations/001-init.sql +117 -0
- package/migrations/002-wire-v1.sql +16 -0
- package/migrations/003-expand-memory-types.sql +81 -0
- package/migrations/004-provenance.sql +4 -0
- package/migrations/005-security.sql +12 -0
- package/migrations/006-atom-v3.sql +28 -0
- package/migrations/007-memory-events.sql +30 -0
- package/migrations/008-consolidation.sql +31 -0
- package/migrations/009-tag-hygiene.sql +13 -0
- package/migrations/010-sync-pull.sql +53 -0
- package/migrations/011-embed-dim-migration.sql +28 -0
- package/migrations/012-entities.sql +36 -0
- package/migrations/013-archival.sql +50 -0
- package/package.json +50 -0
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
import { DEFAULT_EMBED_DIM } from '../../contracts/embed.js';
|
|
2
|
+
import { TransientError, DeterministicError } from '../../pipeline/errors.js';
|
|
3
|
+
import { childLogger } from '../../log/logger.js';
|
|
4
|
+
const EMBED_TIMEOUT_MS = 15_000;
|
|
5
|
+
const HEALTH_TIMEOUT_MS = 10_000;
|
|
6
|
+
async function fetchWithRetry(url, init, timeoutMs) {
|
|
7
|
+
let lastError;
|
|
8
|
+
for (let i = 0; i < 2; i++) {
|
|
9
|
+
try {
|
|
10
|
+
const res = await fetch(url, { ...init, signal: AbortSignal.timeout(timeoutMs) });
|
|
11
|
+
if (res.status >= 400 && res.status < 500)
|
|
12
|
+
return res;
|
|
13
|
+
if (res.status >= 500 && i === 0) {
|
|
14
|
+
lastError = new Error(`HTTP ${res.status} from Ollama embed`);
|
|
15
|
+
continue;
|
|
16
|
+
}
|
|
17
|
+
return res;
|
|
18
|
+
}
|
|
19
|
+
catch (err) {
|
|
20
|
+
if (i === 0) {
|
|
21
|
+
lastError = err;
|
|
22
|
+
continue;
|
|
23
|
+
}
|
|
24
|
+
throw err;
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
throw lastError;
|
|
28
|
+
}
|
|
29
|
+
export class OllamaEmbedProvider {
|
|
30
|
+
name = 'ollama';
|
|
31
|
+
model;
|
|
32
|
+
dim;
|
|
33
|
+
baseUrl;
|
|
34
|
+
constructor(config = {}) {
|
|
35
|
+
this.baseUrl = (config.baseUrl ?? 'http://127.0.0.1:11434').replace(/\/$/, '');
|
|
36
|
+
this.model = config.model ?? 'mxbai-embed-large';
|
|
37
|
+
this.dim = config.dim ?? DEFAULT_EMBED_DIM;
|
|
38
|
+
}
|
|
39
|
+
async embed(texts) {
|
|
40
|
+
const log = childLogger({ provider: this.name, model: this.model });
|
|
41
|
+
const results = [];
|
|
42
|
+
for (const text of texts) {
|
|
43
|
+
const t0 = Date.now();
|
|
44
|
+
const res = await fetchWithRetry(`${this.baseUrl}/api/embeddings`, {
|
|
45
|
+
method: 'POST',
|
|
46
|
+
headers: { 'content-type': 'application/json' },
|
|
47
|
+
body: JSON.stringify({ model: this.model, prompt: text }),
|
|
48
|
+
}, EMBED_TIMEOUT_MS);
|
|
49
|
+
if (!res.ok) {
|
|
50
|
+
const raw = await res.text().catch(() => '');
|
|
51
|
+
const snippet = raw.slice(0, 200);
|
|
52
|
+
const latency_ms = Date.now() - t0;
|
|
53
|
+
log.warn({ latency_ms, status: res.status, error_kind: res.status >= 500 || res.status === 429 ? 'TransientError' : 'DeterministicError' }, 'ollama embed failed');
|
|
54
|
+
if (res.status >= 500) {
|
|
55
|
+
throw new TransientError(`Ollama embed failed: HTTP ${res.status} — ${snippet}`);
|
|
56
|
+
}
|
|
57
|
+
if (res.status === 429) {
|
|
58
|
+
throw new TransientError(`Ollama embed rate-limited (429): ${snippet}`);
|
|
59
|
+
}
|
|
60
|
+
throw new DeterministicError(`Ollama embed failed: HTTP ${res.status} — ${snippet}`);
|
|
61
|
+
}
|
|
62
|
+
const data = (await res.json());
|
|
63
|
+
if (!data.embedding) {
|
|
64
|
+
throw new Error(`Ollama embed response missing 'embedding' field`);
|
|
65
|
+
}
|
|
66
|
+
if (data.embedding.length !== this.dim) {
|
|
67
|
+
throw new Error(`Ollama embed dimension mismatch: expected ${this.dim}, got ${data.embedding.length} (model: ${this.model})`);
|
|
68
|
+
}
|
|
69
|
+
const latency_ms = Date.now() - t0;
|
|
70
|
+
log.info({ latency_ms, tokens_in: 0, tokens_out: this.dim }, 'ollama embed ok');
|
|
71
|
+
results.push(new Float32Array(data.embedding));
|
|
72
|
+
}
|
|
73
|
+
return results;
|
|
74
|
+
}
|
|
75
|
+
async health() {
|
|
76
|
+
try {
|
|
77
|
+
const t0 = Date.now();
|
|
78
|
+
const res = await fetch(`${this.baseUrl}/api/tags`, {
|
|
79
|
+
signal: AbortSignal.timeout(HEALTH_TIMEOUT_MS),
|
|
80
|
+
});
|
|
81
|
+
const latency_ms = Date.now() - t0;
|
|
82
|
+
if (!res.ok) {
|
|
83
|
+
return { ok: false, model: this.model, dim: this.dim, error: `HTTP ${res.status}` };
|
|
84
|
+
}
|
|
85
|
+
const data = (await res.json());
|
|
86
|
+
const models = data.models ?? [];
|
|
87
|
+
const found = models.some((m) => m.name === this.model || m.name.startsWith(`${this.model}:`));
|
|
88
|
+
void latency_ms; // latency not part of EmbedHealth interface
|
|
89
|
+
return {
|
|
90
|
+
ok: found,
|
|
91
|
+
model: this.model,
|
|
92
|
+
dim: this.dim,
|
|
93
|
+
...(found ? {} : { error: `model ${this.model} not found in ollama list` }),
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
catch (err) {
|
|
97
|
+
return {
|
|
98
|
+
ok: false,
|
|
99
|
+
model: this.model,
|
|
100
|
+
dim: this.dim,
|
|
101
|
+
error: err instanceof Error ? err.message : String(err),
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
//# sourceMappingURL=ollama.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../../src/providers/embed/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAChC,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAmBjC,KAAK,UAAU,cAAc,CAC3B,GAAW,EACX,IAAiB,EACjB,SAAiB;IAEjB,IAAI,SAAkB,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,GAAG,GAAG;gBAAE,OAAO,GAAG,CAAC;YACtD,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,SAAS,GAAG,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,MAAM,oBAAoB,CAAC,CAAC;gBAC9D,SAAS;YACX,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,SAAS,GAAG,GAAG,CAAC;gBAChB,SAAS;YACX,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IACD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,OAAO,mBAAmB;IACrB,IAAI,GAAG,QAAiB,CAAC;IACzB,KAAK,CAAS;IACd,GAAG,CAAS;IACJ,OAAO,CAAS;IAEjC,YAAY,SAA4B,EAAE;QACxC,IAAI,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/E,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,mBAAmB,CAAC;QACjD,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,iBAAiB,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,MAAM,GAAG,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,MAAM,GAAG,GAAG,MAAM,cAAc,CAC9B,GAAG,IAAI,CAAC,OAAO,iBAAiB,EAChC;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aAC1D,EACD,gBAAgB,CACjB,CAAC;YAEF,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC7C,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAClC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;gBACnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,oBAAoB,EAAE,EAAE,qBAAqB,CAAC,CAAC;gBACnK,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;oBACtB,MAAM,IAAI,cAAc,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;gBACnF,CAAC;gBACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACvB,MAAM,IAAI,cAAc,CAAC,oCAAoC,OAAO,EAAE,CAAC,CAAC;gBAC1E,CAAC;gBACD,MAAM,IAAI,kBAAkB,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;YACvF,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAwB,CAAC;YACvD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YAED,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,GAAG,EAAE,CAAC;gBACvC,MAAM,IAAI,KAAK,CACb,6CAA6C,IAAI,CAAC,GAAG,SAAS,IAAI,CAAC,SAAS,CAAC,MAAM,YAAY,IAAI,CAAC,KAAK,GAAG,CAC7G,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,iBAAiB,CAAC,CAAC;YAChF,OAAO,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBAClD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC;aAC/C,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YAEnC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YACtF,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CACvB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CACpE,CAAC;YACF,KAAK,UAAU,CAAC,CAAC,4CAA4C;YAC7D,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,GAAG,EAAE,IAAI,CAAC,GAAG;gBACb,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,2BAA2B,EAAE,CAAC;aAC5E,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,GAAG,EAAE,IAAI,CAAC,GAAG;gBACb,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { LLMProvider } from '../contracts/llm.js';
|
|
2
|
+
import type { EmbedProvider } from '../contracts/embed.js';
|
|
3
|
+
import type { Config } from '../config/config.js';
|
|
4
|
+
export interface ProviderSet {
|
|
5
|
+
llm: {
|
|
6
|
+
compaction: LLMProvider;
|
|
7
|
+
extraction: LLMProvider;
|
|
8
|
+
};
|
|
9
|
+
embed: EmbedProvider;
|
|
10
|
+
}
|
|
11
|
+
/**
|
|
12
|
+
* Factory: given config, return {llm: {compaction, extraction}, embed} instances.
|
|
13
|
+
*
|
|
14
|
+
* Provider selection driven by:
|
|
15
|
+
* config.llm.compaction.provider + config.llm.compaction.model
|
|
16
|
+
* config.llm.extraction.provider + config.llm.extraction.model
|
|
17
|
+
* config.embedding.provider + config.embedding.model
|
|
18
|
+
*/
|
|
19
|
+
export declare function getProviders(config: Config): ProviderSet;
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
import { OllamaLLMProvider } from './llm/ollama.js';
|
|
2
|
+
import { AzureOpenAILLMProvider } from './llm/azure-openai.js';
|
|
3
|
+
import { OllamaEmbedProvider } from './embed/ollama.js';
|
|
4
|
+
import { AzureOpenAIEmbedProvider } from './embed/azure-openai.js';
|
|
5
|
+
/**
|
|
6
|
+
* Resolve Azure credentials from environment if not present in config.
|
|
7
|
+
* AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_DEPLOYMENT are
|
|
8
|
+
* the conventional env vars for Azure OpenAI.
|
|
9
|
+
*/
|
|
10
|
+
function resolveAzureApiKey() {
|
|
11
|
+
const key = process.env.AZURE_OPENAI_API_KEY ?? '';
|
|
12
|
+
if (!key) {
|
|
13
|
+
throw new Error('Azure OpenAI provider selected but AZURE_OPENAI_API_KEY env var is not set');
|
|
14
|
+
}
|
|
15
|
+
return key;
|
|
16
|
+
}
|
|
17
|
+
function buildLLMProvider(providerName, model, config) {
|
|
18
|
+
if (providerName === 'ollama') {
|
|
19
|
+
return new OllamaLLMProvider({ baseUrl: config.ollama.baseUrl, model });
|
|
20
|
+
}
|
|
21
|
+
const endpoint = config.azure.endpoint ?? process.env.AZURE_OPENAI_ENDPOINT;
|
|
22
|
+
if (!endpoint) {
|
|
23
|
+
throw new Error('Azure OpenAI LLM provider selected but no endpoint configured. ' +
|
|
24
|
+
'Set config.azure.endpoint or AZURE_OPENAI_ENDPOINT env var.');
|
|
25
|
+
}
|
|
26
|
+
const deployment = config.azure.deployment ?? model;
|
|
27
|
+
return new AzureOpenAILLMProvider({
|
|
28
|
+
endpoint,
|
|
29
|
+
deployment,
|
|
30
|
+
apiKey: resolveAzureApiKey(),
|
|
31
|
+
apiVersion: config.azure.apiVersion,
|
|
32
|
+
model,
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
function buildEmbedProvider(config) {
|
|
36
|
+
const { provider, model, dim } = config.embedding;
|
|
37
|
+
if (provider === 'ollama') {
|
|
38
|
+
return new OllamaEmbedProvider({ baseUrl: config.ollama.baseUrl, model, dim });
|
|
39
|
+
}
|
|
40
|
+
const endpoint = config.azure.endpoint ?? process.env.AZURE_OPENAI_ENDPOINT;
|
|
41
|
+
if (!endpoint) {
|
|
42
|
+
throw new Error('Azure OpenAI embed provider selected but no endpoint configured. ' +
|
|
43
|
+
'Set config.azure.endpoint or AZURE_OPENAI_ENDPOINT env var.');
|
|
44
|
+
}
|
|
45
|
+
const deployment = config.azure.deployment ?? model;
|
|
46
|
+
return new AzureOpenAIEmbedProvider({
|
|
47
|
+
endpoint,
|
|
48
|
+
deployment,
|
|
49
|
+
apiKey: resolveAzureApiKey(),
|
|
50
|
+
apiVersion: config.azure.apiVersion,
|
|
51
|
+
model,
|
|
52
|
+
dim,
|
|
53
|
+
});
|
|
54
|
+
}
|
|
55
|
+
/**
|
|
56
|
+
* Factory: given config, return {llm: {compaction, extraction}, embed} instances.
|
|
57
|
+
*
|
|
58
|
+
* Provider selection driven by:
|
|
59
|
+
* config.llm.compaction.provider + config.llm.compaction.model
|
|
60
|
+
* config.llm.extraction.provider + config.llm.extraction.model
|
|
61
|
+
* config.embedding.provider + config.embedding.model
|
|
62
|
+
*/
|
|
63
|
+
export function getProviders(config) {
|
|
64
|
+
return {
|
|
65
|
+
llm: {
|
|
66
|
+
compaction: buildLLMProvider(config.llm.compaction.provider, config.llm.compaction.model, config),
|
|
67
|
+
extraction: buildLLMProvider(config.llm.extraction.provider, config.llm.extraction.model, config),
|
|
68
|
+
},
|
|
69
|
+
embed: buildEmbedProvider(config),
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AAUnE;;;;GAIG;AACH,SAAS,kBAAkB;IACzB,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,EAAE,CAAC;IACnD,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CACb,4EAA4E,CAC7E,CAAC;IACJ,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CACvB,YAAuC,EACvC,KAAa,EACb,MAAc;IAEd,IAAI,YAAY,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,IAAI,iBAAiB,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAC5E,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,iEAAiE;YAC/D,6DAA6D,CAChE,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC;IACpD,OAAO,IAAI,sBAAsB,CAAC;QAChC,QAAQ;QACR,UAAU;QACV,MAAM,EAAE,kBAAkB,EAAE;QAC5B,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU;QACnC,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAc;IACxC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC;IAElD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,IAAI,mBAAmB,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAC5E,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,mEAAmE;YACjE,6DAA6D,CAChE,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC;IACpD,OAAO,IAAI,wBAAwB,CAAC;QAClC,QAAQ;QACR,UAAU;QACV,MAAM,EAAE,kBAAkB,EAAE;QAC5B,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU;QACnC,KAAK;QACL,GAAG;KACJ,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,YAAY,CAAC,MAAc;IACzC,OAAO;QACL,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB,CAC1B,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,EAC9B,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,EAC3B,MAAM,CACP;YACD,UAAU,EAAE,gBAAgB,CAC1B,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,EAC9B,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,EAC3B,MAAM,CACP;SACF;QACD,KAAK,EAAE,kBAAkB,CAAC,MAAM,CAAC;KAClC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { ChatMsg, ChatOpts, ChatResult, LLMHealth, LLMProvider } from '../../contracts/llm.js';
|
|
2
|
+
export interface AzureOpenAILLMConfig {
|
|
3
|
+
endpoint: string;
|
|
4
|
+
deployment: string;
|
|
5
|
+
apiKey: string;
|
|
6
|
+
apiVersion?: string;
|
|
7
|
+
model?: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class AzureOpenAILLMProvider implements LLMProvider {
|
|
10
|
+
readonly name: "azure-openai";
|
|
11
|
+
readonly model: string;
|
|
12
|
+
private readonly endpoint;
|
|
13
|
+
private readonly deployment;
|
|
14
|
+
private readonly apiKey;
|
|
15
|
+
private readonly apiVersion;
|
|
16
|
+
constructor(config: AzureOpenAILLMConfig);
|
|
17
|
+
private get chatUrl();
|
|
18
|
+
chat(messages: ChatMsg[], opts?: ChatOpts): Promise<ChatResult>;
|
|
19
|
+
health(): Promise<LLMHealth>;
|
|
20
|
+
}
|
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
import { computeCostUsd } from './pricing.js';
|
|
2
|
+
import { TransientError, DeterministicError } from '../../pipeline/errors.js';
|
|
3
|
+
import { childLogger } from '../../log/logger.js';
|
|
4
|
+
const CHAT_TIMEOUT_MS = 60_000;
|
|
5
|
+
const HEALTH_TIMEOUT_MS = 15_000;
|
|
6
|
+
async function fetchWithRetry(url, init, timeoutMs) {
|
|
7
|
+
let lastError;
|
|
8
|
+
for (let i = 0; i < 2; i++) {
|
|
9
|
+
try {
|
|
10
|
+
const res = await fetch(url, { ...init, signal: AbortSignal.timeout(timeoutMs) });
|
|
11
|
+
// No retry on 4xx
|
|
12
|
+
if (res.status >= 400 && res.status < 500)
|
|
13
|
+
return res;
|
|
14
|
+
// Retry on 5xx (first attempt only)
|
|
15
|
+
if (res.status >= 500 && i === 0) {
|
|
16
|
+
lastError = new Error(`HTTP ${res.status} from Azure`);
|
|
17
|
+
continue;
|
|
18
|
+
}
|
|
19
|
+
return res;
|
|
20
|
+
}
|
|
21
|
+
catch (err) {
|
|
22
|
+
if (i === 0) {
|
|
23
|
+
lastError = err;
|
|
24
|
+
continue;
|
|
25
|
+
}
|
|
26
|
+
throw err;
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
throw lastError;
|
|
30
|
+
}
|
|
31
|
+
export class AzureOpenAILLMProvider {
|
|
32
|
+
name = 'azure-openai';
|
|
33
|
+
model;
|
|
34
|
+
endpoint;
|
|
35
|
+
deployment;
|
|
36
|
+
apiKey;
|
|
37
|
+
apiVersion;
|
|
38
|
+
constructor(config) {
|
|
39
|
+
this.endpoint = config.endpoint.replace(/\/$/, '');
|
|
40
|
+
this.deployment = config.deployment;
|
|
41
|
+
this.apiKey = config.apiKey;
|
|
42
|
+
this.apiVersion = config.apiVersion ?? '2024-10-21';
|
|
43
|
+
// model name used for pricing lookup; defaults to deployment name
|
|
44
|
+
this.model = config.model ?? config.deployment;
|
|
45
|
+
}
|
|
46
|
+
get chatUrl() {
|
|
47
|
+
return `${this.endpoint}/openai/deployments/${this.deployment}/chat/completions?api-version=${this.apiVersion}`;
|
|
48
|
+
}
|
|
49
|
+
async chat(messages, opts) {
|
|
50
|
+
const log = childLogger({ provider: this.name, model: this.model });
|
|
51
|
+
const t0 = Date.now();
|
|
52
|
+
const useJson = opts?.json ?? true;
|
|
53
|
+
const body = {
|
|
54
|
+
messages,
|
|
55
|
+
...(useJson ? { response_format: { type: 'json_object' } } : {}),
|
|
56
|
+
...(opts?.temperature !== undefined ? { temperature: opts.temperature } : {}),
|
|
57
|
+
...(opts?.maxTokens !== undefined ? { max_tokens: opts.maxTokens } : {}),
|
|
58
|
+
};
|
|
59
|
+
const res = await fetchWithRetry(this.chatUrl, {
|
|
60
|
+
method: 'POST',
|
|
61
|
+
headers: {
|
|
62
|
+
'content-type': 'application/json',
|
|
63
|
+
'api-key': this.apiKey,
|
|
64
|
+
},
|
|
65
|
+
body: JSON.stringify(body),
|
|
66
|
+
}, CHAT_TIMEOUT_MS);
|
|
67
|
+
if (!res.ok) {
|
|
68
|
+
const raw = await res.text().catch(() => '');
|
|
69
|
+
let detail = raw;
|
|
70
|
+
try {
|
|
71
|
+
const parsed = JSON.parse(raw);
|
|
72
|
+
detail = parsed.error?.message ?? raw;
|
|
73
|
+
}
|
|
74
|
+
catch {
|
|
75
|
+
// leave as raw text
|
|
76
|
+
}
|
|
77
|
+
const snippet = detail.slice(0, 200);
|
|
78
|
+
const latency_ms = Date.now() - t0;
|
|
79
|
+
log.warn({ latency_ms, status: res.status, error_kind: res.status >= 500 || res.status === 429 ? 'TransientError' : 'DeterministicError' }, 'azure-openai chat failed');
|
|
80
|
+
if (res.status >= 500) {
|
|
81
|
+
throw new TransientError(`Azure OpenAI chat failed: HTTP ${res.status} — ${snippet}`);
|
|
82
|
+
}
|
|
83
|
+
if (res.status === 429) {
|
|
84
|
+
throw new TransientError(`Azure OpenAI chat rate-limited (429): ${snippet}`);
|
|
85
|
+
}
|
|
86
|
+
throw new DeterministicError(`Azure OpenAI chat failed: HTTP ${res.status} — ${snippet}`);
|
|
87
|
+
}
|
|
88
|
+
const data = (await res.json());
|
|
89
|
+
const text = data.choices?.[0]?.message?.content ?? '';
|
|
90
|
+
const promptTokens = data.usage?.prompt_tokens ?? 0;
|
|
91
|
+
const completionTokens = data.usage?.completion_tokens ?? 0;
|
|
92
|
+
const usd = computeCostUsd(this.model, promptTokens, completionTokens);
|
|
93
|
+
const latency_ms = Date.now() - t0;
|
|
94
|
+
log.info({ latency_ms, tokens_in: promptTokens, tokens_out: completionTokens, usd }, 'azure-openai chat ok');
|
|
95
|
+
return {
|
|
96
|
+
text,
|
|
97
|
+
usage: { in: promptTokens, out: completionTokens, usd },
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
async health() {
|
|
101
|
+
const t0 = Date.now();
|
|
102
|
+
// Ping with a minimal 1-token completion
|
|
103
|
+
const body = {
|
|
104
|
+
messages: [{ role: 'user', content: 'ping' }],
|
|
105
|
+
max_tokens: 1,
|
|
106
|
+
};
|
|
107
|
+
try {
|
|
108
|
+
const res = await fetch(this.chatUrl, {
|
|
109
|
+
method: 'POST',
|
|
110
|
+
headers: {
|
|
111
|
+
'content-type': 'application/json',
|
|
112
|
+
'api-key': this.apiKey,
|
|
113
|
+
},
|
|
114
|
+
body: JSON.stringify(body),
|
|
115
|
+
signal: AbortSignal.timeout(HEALTH_TIMEOUT_MS),
|
|
116
|
+
});
|
|
117
|
+
const latency_ms = Date.now() - t0;
|
|
118
|
+
return {
|
|
119
|
+
ok: res.ok,
|
|
120
|
+
model: this.model,
|
|
121
|
+
latency_ms,
|
|
122
|
+
...(res.ok ? {} : { error: `HTTP ${res.status}` }),
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
catch (err) {
|
|
126
|
+
return {
|
|
127
|
+
ok: false,
|
|
128
|
+
model: this.model,
|
|
129
|
+
latency_ms: Date.now() - t0,
|
|
130
|
+
error: err instanceof Error ? err.message : String(err),
|
|
131
|
+
};
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
//# sourceMappingURL=azure-openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"azure-openai.js","sourceRoot":"","sources":["../../../src/providers/llm/azure-openai.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,MAAM,eAAe,GAAG,MAAM,CAAC;AAC/B,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAmBjC,KAAK,UAAU,cAAc,CAC3B,GAAW,EACX,IAAiB,EACjB,SAAiB;IAEjB,IAAI,SAAkB,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClF,kBAAkB;YAClB,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,GAAG,GAAG;gBAAE,OAAO,GAAG,CAAC;YACtD,oCAAoC;YACpC,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,SAAS,GAAG,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,MAAM,aAAa,CAAC,CAAC;gBACvD,SAAS;YACX,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,SAAS,GAAG,GAAG,CAAC;gBAChB,SAAS;YACX,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IACD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,OAAO,sBAAsB;IACxB,IAAI,GAAG,cAAuB,CAAC;IAC/B,KAAK,CAAS;IACN,QAAQ,CAAS;IACjB,UAAU,CAAS;IACnB,MAAM,CAAS;IACf,UAAU,CAAS;IAEpC,YAAY,MAA4B;QACtC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,YAAY,CAAC;QACpD,kEAAkE;QAClE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC;IACjD,CAAC;IAED,IAAY,OAAO;QACjB,OAAO,GAAG,IAAI,CAAC,QAAQ,uBAAuB,IAAI,CAAC,UAAU,iCAAiC,IAAI,CAAC,UAAU,EAAE,CAAC;IAClH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAmB,EAAE,IAAe;QAC7C,MAAM,GAAG,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,MAAM,OAAO,GAAG,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC;QACnC,MAAM,IAAI,GAAG;YACX,QAAQ;YACR,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChE,GAAG,CAAC,IAAI,EAAE,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7E,GAAG,CAAC,IAAI,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzE,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAC9B,IAAI,CAAC,OAAO,EACZ;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,SAAS,EAAE,IAAI,CAAC,MAAM;aACvB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,EACD,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,MAAM,GAAG,GAAG,CAAC;YACjB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAsB,CAAC;gBACpD,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,GAAG,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,oBAAoB;YACtB,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACrC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,oBAAoB,EAAE,EAAE,0BAA0B,CAAC,CAAC;YACxK,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gBACtB,MAAM,IAAI,cAAc,CAAC,kCAAkC,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;YACxF,CAAC;YACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,MAAM,IAAI,cAAc,CAAC,yCAAyC,OAAO,EAAE,CAAC,CAAC;YAC/E,CAAC;YACD,MAAM,IAAI,kBAAkB,CAAC,kCAAkC,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;QAC5F,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAsB,CAAC;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACvD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QACpD,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;QAC5D,MAAM,GAAG,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;QACvE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;QAEnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,GAAG,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAE7G,OAAO;YACL,IAAI;YACJ,KAAK,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE;SACxD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,yCAAyC;QACzC,MAAM,IAAI,GAAG;YACX,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,UAAU,EAAE,CAAC;SACd,CAAC;QACF,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE;gBACpC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,IAAI,CAAC,MAAM;iBACvB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAC1B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC;aAC/C,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACnC,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU;gBACV,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;aACnD,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;gBAC3B,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ChatMsg, ChatOpts, ChatResult, LLMHealth, LLMProvider } from '../../contracts/llm.js';
|
|
2
|
+
export interface OllamaLLMConfig {
|
|
3
|
+
baseUrl?: string;
|
|
4
|
+
model?: string;
|
|
5
|
+
}
|
|
6
|
+
export declare class OllamaLLMProvider implements LLMProvider {
|
|
7
|
+
readonly name: "ollama";
|
|
8
|
+
readonly model: string;
|
|
9
|
+
private readonly baseUrl;
|
|
10
|
+
constructor(config?: OllamaLLMConfig);
|
|
11
|
+
chat(messages: ChatMsg[], opts?: ChatOpts): Promise<ChatResult>;
|
|
12
|
+
health(): Promise<LLMHealth>;
|
|
13
|
+
}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
import { TransientError, DeterministicError } from '../../pipeline/errors.js';
|
|
2
|
+
import { childLogger } from '../../log/logger.js';
|
|
3
|
+
const CHAT_TIMEOUT_MS = 60_000;
|
|
4
|
+
const HEALTH_TIMEOUT_MS = 10_000;
|
|
5
|
+
async function fetchWithRetry(url, init, timeoutMs) {
|
|
6
|
+
const attempt = async () => {
|
|
7
|
+
return fetch(url, { ...init, signal: AbortSignal.timeout(timeoutMs) });
|
|
8
|
+
};
|
|
9
|
+
let lastError;
|
|
10
|
+
for (let i = 0; i < 2; i++) {
|
|
11
|
+
try {
|
|
12
|
+
const res = await attempt();
|
|
13
|
+
// No retry on 4xx
|
|
14
|
+
if (res.status >= 400 && res.status < 500)
|
|
15
|
+
return res;
|
|
16
|
+
// Retry on 5xx (only if first attempt)
|
|
17
|
+
if (res.status >= 500 && i === 0) {
|
|
18
|
+
lastError = new Error(`HTTP ${res.status} from ${url}`);
|
|
19
|
+
continue;
|
|
20
|
+
}
|
|
21
|
+
return res;
|
|
22
|
+
}
|
|
23
|
+
catch (err) {
|
|
24
|
+
// Network error or timeout — retry once
|
|
25
|
+
if (i === 0) {
|
|
26
|
+
lastError = err;
|
|
27
|
+
continue;
|
|
28
|
+
}
|
|
29
|
+
throw err;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
throw lastError;
|
|
33
|
+
}
|
|
34
|
+
export class OllamaLLMProvider {
|
|
35
|
+
name = 'ollama';
|
|
36
|
+
model;
|
|
37
|
+
baseUrl;
|
|
38
|
+
constructor(config = {}) {
|
|
39
|
+
this.baseUrl = (config.baseUrl ?? 'http://127.0.0.1:11434').replace(/\/$/, '');
|
|
40
|
+
this.model = config.model ?? 'qwen2.5-coder:7b';
|
|
41
|
+
}
|
|
42
|
+
async chat(messages, opts) {
|
|
43
|
+
const log = childLogger({ provider: this.name, model: this.model });
|
|
44
|
+
const t0 = Date.now();
|
|
45
|
+
const useJson = opts?.json ?? true;
|
|
46
|
+
const body = {
|
|
47
|
+
model: this.model,
|
|
48
|
+
messages,
|
|
49
|
+
stream: false,
|
|
50
|
+
...(useJson ? { format: 'json' } : {}),
|
|
51
|
+
...(opts?.temperature !== undefined ? { options: { temperature: opts.temperature } } : {}),
|
|
52
|
+
...(opts?.maxTokens !== undefined ? { options: { num_predict: opts.maxTokens } } : {}),
|
|
53
|
+
};
|
|
54
|
+
const res = await fetchWithRetry(`${this.baseUrl}/api/chat`, {
|
|
55
|
+
method: 'POST',
|
|
56
|
+
headers: { 'content-type': 'application/json' },
|
|
57
|
+
body: JSON.stringify(body),
|
|
58
|
+
}, CHAT_TIMEOUT_MS);
|
|
59
|
+
if (!res.ok) {
|
|
60
|
+
const rawBody = await res.text().catch(() => '');
|
|
61
|
+
const snippet = rawBody.slice(0, 200);
|
|
62
|
+
const latency_ms = Date.now() - t0;
|
|
63
|
+
log.warn({ latency_ms, status: res.status, error_kind: res.status >= 500 ? 'TransientError' : 'DeterministicError' }, 'ollama chat failed');
|
|
64
|
+
if (res.status >= 500) {
|
|
65
|
+
throw new TransientError(`Ollama chat failed: HTTP ${res.status} — ${snippet}`);
|
|
66
|
+
}
|
|
67
|
+
if (res.status === 429) {
|
|
68
|
+
throw new TransientError(`Ollama chat rate-limited (429): ${snippet}`);
|
|
69
|
+
}
|
|
70
|
+
throw new DeterministicError(`Ollama chat failed: HTTP ${res.status} — ${snippet}`);
|
|
71
|
+
}
|
|
72
|
+
const data = (await res.json());
|
|
73
|
+
const text = data.message?.content ?? '';
|
|
74
|
+
const inTokens = data.prompt_eval_count ?? 0;
|
|
75
|
+
const outTokens = data.eval_count ?? 0;
|
|
76
|
+
const latency_ms = Date.now() - t0;
|
|
77
|
+
log.info({ latency_ms, tokens_in: inTokens, tokens_out: outTokens, usd: 0 }, 'ollama chat ok');
|
|
78
|
+
return {
|
|
79
|
+
text,
|
|
80
|
+
usage: { in: inTokens, out: outTokens, usd: 0 },
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
async health() {
|
|
84
|
+
const t0 = Date.now();
|
|
85
|
+
try {
|
|
86
|
+
const res = await fetch(`${this.baseUrl}/api/tags`, {
|
|
87
|
+
signal: AbortSignal.timeout(HEALTH_TIMEOUT_MS),
|
|
88
|
+
});
|
|
89
|
+
const latency_ms = Date.now() - t0;
|
|
90
|
+
if (!res.ok) {
|
|
91
|
+
return { ok: false, model: this.model, latency_ms, error: `HTTP ${res.status}` };
|
|
92
|
+
}
|
|
93
|
+
const data = (await res.json());
|
|
94
|
+
const models = data.models ?? [];
|
|
95
|
+
const found = models.some((m) => m.name === this.model || m.name.startsWith(`${this.model}:`));
|
|
96
|
+
return {
|
|
97
|
+
ok: found,
|
|
98
|
+
model: this.model,
|
|
99
|
+
latency_ms,
|
|
100
|
+
...(found ? {} : { error: `model ${this.model} not found in ollama list` }),
|
|
101
|
+
};
|
|
102
|
+
}
|
|
103
|
+
catch (err) {
|
|
104
|
+
return {
|
|
105
|
+
ok: false,
|
|
106
|
+
model: this.model,
|
|
107
|
+
latency_ms: Date.now() - t0,
|
|
108
|
+
error: err instanceof Error ? err.message : String(err),
|
|
109
|
+
};
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=ollama.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../../src/providers/llm/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,MAAM,eAAe,GAAG,MAAM,CAAC;AAC/B,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAiBjC,KAAK,UAAU,cAAc,CAC3B,GAAW,EACX,IAAiB,EACjB,SAAiB;IAEjB,MAAM,OAAO,GAAG,KAAK,IAAuB,EAAE;QAC5C,OAAO,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACzE,CAAC,CAAC;IAEF,IAAI,SAAkB,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,OAAO,EAAE,CAAC;YAC5B,kBAAkB;YAClB,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,GAAG,GAAG;gBAAE,OAAO,GAAG,CAAC;YACtD,uCAAuC;YACvC,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,SAAS,GAAG,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,MAAM,SAAS,GAAG,EAAE,CAAC,CAAC;gBACxD,SAAS;YACX,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,wCAAwC;YACxC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,SAAS,GAAG,GAAG,CAAC;gBAChB,SAAS;YACX,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IACD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,QAAiB,CAAC;IACzB,KAAK,CAAS;IACN,OAAO,CAAS;IAEjC,YAAY,SAA0B,EAAE;QACtC,IAAI,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/E,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAmB,EAAE,IAAe;QAC7C,MAAM,GAAG,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,MAAM,OAAO,GAAG,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC;QACnC,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ;YACR,MAAM,EAAE,KAAK;YACb,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtC,GAAG,CAAC,IAAI,EAAE,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1F,GAAG,CAAC,IAAI,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvF,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAC9B,GAAG,IAAI,CAAC,OAAO,WAAW,EAC1B;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,EACD,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACtC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,oBAAoB,EAAE,EAAE,oBAAoB,CAAC,CAAC;YAC5I,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gBACtB,MAAM,IAAI,cAAc,CAAC,4BAA4B,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;YAClF,CAAC;YACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,MAAM,IAAI,cAAc,CAAC,mCAAmC,OAAO,EAAE,CAAC,CAAC;YACzE,CAAC;YACD,MAAM,IAAI,kBAAkB,CAAC,4BAA4B,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,IAAI,CAAC,CAAC;QAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;QAEnC,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,gBAAgB,CAAC,CAAC;QAE/F,OAAO;YACL,IAAI;YACJ,KAAK,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,EAAE;SAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBAClD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC;aAC/C,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACnC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YACnF,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CACvB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CACpE,CAAC;YACF,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU;gBACV,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,2BAA2B,EAAE,CAAC;aAC5E,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;gBAC3B,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Azure OpenAI pricing table — placeholder rates.
|
|
3
|
+
*
|
|
4
|
+
* UPDATE THESE with actual prices from:
|
|
5
|
+
* https://azure.microsoft.com/en-us/pricing/details/cognitive-services/openai-service/
|
|
6
|
+
*
|
|
7
|
+
* Rates are in USD per 1,000,000 tokens (per-1M).
|
|
8
|
+
* Last checked: 2026-06-27 (prices change — verify before production use).
|
|
9
|
+
*/
|
|
10
|
+
export interface ModelPricing {
|
|
11
|
+
/** USD per 1M input (prompt) tokens */
|
|
12
|
+
in_per_1m_usd: number;
|
|
13
|
+
/** USD per 1M output (completion) tokens */
|
|
14
|
+
out_per_1m_usd: number;
|
|
15
|
+
}
|
|
16
|
+
export declare const PRICING: Record<string, ModelPricing>;
|
|
17
|
+
/**
|
|
18
|
+
* Compute USD cost from token counts.
|
|
19
|
+
* Returns 0 if the model is not in the pricing table.
|
|
20
|
+
*/
|
|
21
|
+
export declare function computeCostUsd(model: string, promptTokens: number, completionTokens: number): number;
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
export const PRICING = {
|
|
2
|
+
// PLACEHOLDER: verify at https://azure.microsoft.com/pricing/details/cognitive-services/openai-service/
|
|
3
|
+
'gpt-4.1': { in_per_1m_usd: 2.50, out_per_1m_usd: 10.00 },
|
|
4
|
+
// PLACEHOLDER: verify pricing
|
|
5
|
+
'gpt-4.1-mini': { in_per_1m_usd: 0.15, out_per_1m_usd: 0.60 },
|
|
6
|
+
// PLACEHOLDER: verify pricing
|
|
7
|
+
'gpt-4o': { in_per_1m_usd: 2.50, out_per_1m_usd: 10.00 },
|
|
8
|
+
// PLACEHOLDER: verify pricing
|
|
9
|
+
'gpt-4o-mini': { in_per_1m_usd: 0.15, out_per_1m_usd: 0.60 },
|
|
10
|
+
};
|
|
11
|
+
/**
|
|
12
|
+
* Compute USD cost from token counts.
|
|
13
|
+
* Returns 0 if the model is not in the pricing table.
|
|
14
|
+
*/
|
|
15
|
+
export function computeCostUsd(model, promptTokens, completionTokens) {
|
|
16
|
+
const entry = PRICING[model];
|
|
17
|
+
if (!entry)
|
|
18
|
+
return 0;
|
|
19
|
+
return ((promptTokens / 1_000_000) * entry.in_per_1m_usd +
|
|
20
|
+
(completionTokens / 1_000_000) * entry.out_per_1m_usd);
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=pricing.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pricing.js","sourceRoot":"","sources":["../../../src/providers/llm/pricing.ts"],"names":[],"mappings":"AAgBA,MAAM,CAAC,MAAM,OAAO,GAAiC;IACnD,wGAAwG;IACxG,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE;IACzD,8BAA8B;IAC9B,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IAC7D,8BAA8B;IAC9B,QAAQ,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE;IACxD,8BAA8B;IAC9B,aAAa,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;CAC7D,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,KAAa,EACb,YAAoB,EACpB,gBAAwB;IAExB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAC7B,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,CAAC;IACrB,OAAO,CACL,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,aAAa;QAChD,CAAC,gBAAgB,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,cAAc,CACtD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Memory-pack selection (KF-B) — the "injection judgment" v1.
|
|
3
|
+
* Heuristic, no ML: score = typeWeight · recency · importance,
|
|
4
|
+
* take top-N under a token budget, render grouped Markdown.
|
|
5
|
+
* Recency half-life 30 days mirrors search freshness decay.
|
|
6
|
+
*/
|
|
7
|
+
import type { DB } from '../storage/db.js';
|
|
8
|
+
export interface PackOptions {
|
|
9
|
+
repo: string;
|
|
10
|
+
project?: string | null;
|
|
11
|
+
branch?: string | null;
|
|
12
|
+
budgetTokens?: number;
|
|
13
|
+
typeWeights?: Record<string, number>;
|
|
14
|
+
now?: number;
|
|
15
|
+
}
|
|
16
|
+
export interface PackMemory {
|
|
17
|
+
id: string;
|
|
18
|
+
type: string;
|
|
19
|
+
text: string;
|
|
20
|
+
score: number;
|
|
21
|
+
}
|
|
22
|
+
export declare const DEFAULT_TYPE_WEIGHTS: Record<string, number>;
|
|
23
|
+
export declare const DEFAULT_BUDGET_TOKENS = 1500;
|
|
24
|
+
/**
|
|
25
|
+
* Rough token estimate: ~4 chars per token. Real tokenizers vary ±20% —
|
|
26
|
+
* callers needing exact budgets should leave ~10% headroom. The default
|
|
27
|
+
* 1500-token budget is deliberately conservative for this reason.
|
|
28
|
+
*/
|
|
29
|
+
export declare function estimateTokens(text: string): number;
|
|
30
|
+
export declare function selectPack(db: DB, opts: PackOptions): PackMemory[];
|
|
31
|
+
/** Render the pack as compact Markdown grouped by type, memory ids inline. */
|
|
32
|
+
export declare function renderPack(memories: PackMemory[]): string;
|