@yasserkhanorg/impact-gate 2.0.0
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/LICENSE +168 -0
- package/README.md +520 -0
- package/dist/adapters/cypress.d.ts +10 -0
- package/dist/adapters/cypress.d.ts.map +1 -0
- package/dist/adapters/cypress.js +86 -0
- package/dist/adapters/framework_adapter.d.ts +41 -0
- package/dist/adapters/framework_adapter.d.ts.map +1 -0
- package/dist/adapters/framework_adapter.js +152 -0
- package/dist/adapters/playwright.d.ts +10 -0
- package/dist/adapters/playwright.d.ts.map +1 -0
- package/dist/adapters/playwright.js +86 -0
- package/dist/adapters/pytest.d.ts +10 -0
- package/dist/adapters/pytest.d.ts.map +1 -0
- package/dist/adapters/pytest.js +96 -0
- package/dist/adapters/supertest.d.ts +12 -0
- package/dist/adapters/supertest.d.ts.map +1 -0
- package/dist/adapters/supertest.js +85 -0
- package/dist/agent/api_catalog.d.ts +11 -0
- package/dist/agent/api_catalog.d.ts.map +1 -0
- package/dist/agent/api_catalog.js +210 -0
- package/dist/agent/config.d.ts +193 -0
- package/dist/agent/config.d.ts.map +1 -0
- package/dist/agent/config.js +875 -0
- package/dist/agent/feedback.d.ts +91 -0
- package/dist/agent/feedback.d.ts.map +1 -0
- package/dist/agent/feedback.js +323 -0
- package/dist/agent/git.d.ts +19 -0
- package/dist/agent/git.d.ts.map +1 -0
- package/dist/agent/git.js +257 -0
- package/dist/agent/handoff.d.ts +22 -0
- package/dist/agent/handoff.d.ts.map +1 -0
- package/dist/agent/handoff.js +180 -0
- package/dist/agent/llm_agents_flow.d.ts +15 -0
- package/dist/agent/llm_agents_flow.d.ts.map +1 -0
- package/dist/agent/llm_agents_flow.js +434 -0
- package/dist/agent/native_flow.d.ts +6 -0
- package/dist/agent/native_flow.d.ts.map +1 -0
- package/dist/agent/native_flow.js +179 -0
- package/dist/agent/pipeline.d.ts +7 -0
- package/dist/agent/pipeline.d.ts.map +1 -0
- package/dist/agent/pipeline.js +260 -0
- package/dist/agent/pipeline_types.d.ts +54 -0
- package/dist/agent/pipeline_types.d.ts.map +1 -0
- package/dist/agent/pipeline_types.js +4 -0
- package/dist/agent/pipeline_utils.d.ts +12 -0
- package/dist/agent/pipeline_utils.d.ts.map +1 -0
- package/dist/agent/pipeline_utils.js +156 -0
- package/dist/agent/plan.d.ts +170 -0
- package/dist/agent/plan.d.ts.map +1 -0
- package/dist/agent/plan.js +86 -0
- package/dist/agent/playwright_report.d.ts +8 -0
- package/dist/agent/playwright_report.d.ts.map +1 -0
- package/dist/agent/playwright_report.js +126 -0
- package/dist/agent/process_runner.d.ts +10 -0
- package/dist/agent/process_runner.d.ts.map +1 -0
- package/dist/agent/process_runner.js +92 -0
- package/dist/agent/spec_generator.d.ts +5 -0
- package/dist/agent/spec_generator.d.ts.map +1 -0
- package/dist/agent/spec_generator.js +253 -0
- package/dist/agent/test_path.d.ts +2 -0
- package/dist/agent/test_path.d.ts.map +1 -0
- package/dist/agent/test_path.js +23 -0
- package/dist/agent/traceability_capture.d.ts +18 -0
- package/dist/agent/traceability_capture.d.ts.map +1 -0
- package/dist/agent/traceability_capture.js +313 -0
- package/dist/agent/traceability_ingest.d.ts +21 -0
- package/dist/agent/traceability_ingest.d.ts.map +1 -0
- package/dist/agent/traceability_ingest.js +237 -0
- package/dist/agent/types.d.ts +42 -0
- package/dist/agent/types.d.ts.map +1 -0
- package/dist/agent/types.js +4 -0
- package/dist/agent/utils.d.ts +13 -0
- package/dist/agent/utils.d.ts.map +1 -0
- package/dist/agent/utils.js +152 -0
- package/dist/agent/validation_runner.d.ts +5 -0
- package/dist/agent/validation_runner.d.ts.map +1 -0
- package/dist/agent/validation_runner.js +77 -0
- package/dist/agentic/fix_loop.d.ts +26 -0
- package/dist/agentic/fix_loop.d.ts.map +1 -0
- package/dist/agentic/fix_loop.js +96 -0
- package/dist/agentic/playwright_runner.d.ts +43 -0
- package/dist/agentic/playwright_runner.d.ts.map +1 -0
- package/dist/agentic/playwright_runner.js +165 -0
- package/dist/agentic/runner.d.ts +27 -0
- package/dist/agentic/runner.d.ts.map +1 -0
- package/dist/agentic/runner.js +210 -0
- package/dist/agentic/types.d.ts +62 -0
- package/dist/agentic/types.d.ts.map +1 -0
- package/dist/agentic/types.js +4 -0
- package/dist/agents/coverage-evaluator.d.ts +8 -0
- package/dist/agents/coverage-evaluator.d.ts.map +1 -0
- package/dist/agents/coverage-evaluator.js +41 -0
- package/dist/agents/cross-impact.d.ts +13 -0
- package/dist/agents/cross-impact.d.ts.map +1 -0
- package/dist/agents/cross-impact.js +140 -0
- package/dist/agents/executor.d.ts +8 -0
- package/dist/agents/executor.d.ts.map +1 -0
- package/dist/agents/executor.js +75 -0
- package/dist/agents/explorer.d.ts +12 -0
- package/dist/agents/explorer.d.ts.map +1 -0
- package/dist/agents/explorer.js +43 -0
- package/dist/agents/generator.d.ts +8 -0
- package/dist/agents/generator.d.ts.map +1 -0
- package/dist/agents/generator.js +77 -0
- package/dist/agents/healer.d.ts +8 -0
- package/dist/agents/healer.d.ts.map +1 -0
- package/dist/agents/healer.js +31 -0
- package/dist/agents/impact-analyst.d.ts +8 -0
- package/dist/agents/impact-analyst.d.ts.map +1 -0
- package/dist/agents/impact-analyst.js +38 -0
- package/dist/agents/regression-advisor.d.ts +8 -0
- package/dist/agents/regression-advisor.d.ts.map +1 -0
- package/dist/agents/regression-advisor.js +116 -0
- package/dist/agents/strategist.d.ts +9 -0
- package/dist/agents/strategist.d.ts.map +1 -0
- package/dist/agents/strategist.js +92 -0
- package/dist/agents/test-designer.d.ts +8 -0
- package/dist/agents/test-designer.d.ts.map +1 -0
- package/dist/agents/test-designer.js +111 -0
- package/dist/anthropic_provider.d.ts +65 -0
- package/dist/anthropic_provider.d.ts.map +1 -0
- package/dist/anthropic_provider.js +334 -0
- package/dist/api.d.ts +48 -0
- package/dist/api.d.ts.map +1 -0
- package/dist/api.js +151 -0
- package/dist/base_provider.d.ts +109 -0
- package/dist/base_provider.d.ts.map +1 -0
- package/dist/base_provider.js +203 -0
- package/dist/budget_ledger.d.ts +28 -0
- package/dist/budget_ledger.d.ts.map +1 -0
- package/dist/budget_ledger.js +62 -0
- package/dist/cache/cached_provider.d.ts +49 -0
- package/dist/cache/cached_provider.d.ts.map +1 -0
- package/dist/cache/cached_provider.js +91 -0
- package/dist/cache/response_cache.d.ts +79 -0
- package/dist/cache/response_cache.d.ts.map +1 -0
- package/dist/cache/response_cache.js +177 -0
- package/dist/cli/commands/analyze.d.ts +3 -0
- package/dist/cli/commands/analyze.d.ts.map +1 -0
- package/dist/cli/commands/analyze.js +77 -0
- package/dist/cli/commands/bootstrap.d.ts +3 -0
- package/dist/cli/commands/bootstrap.d.ts.map +1 -0
- package/dist/cli/commands/bootstrap.js +109 -0
- package/dist/cli/commands/cost_report.d.ts +3 -0
- package/dist/cli/commands/cost_report.d.ts.map +1 -0
- package/dist/cli/commands/cost_report.js +115 -0
- package/dist/cli/commands/crew.d.ts +3 -0
- package/dist/cli/commands/crew.d.ts.map +1 -0
- package/dist/cli/commands/crew.js +255 -0
- package/dist/cli/commands/feedback.d.ts +3 -0
- package/dist/cli/commands/feedback.d.ts.map +1 -0
- package/dist/cli/commands/feedback.js +39 -0
- package/dist/cli/commands/finalize.d.ts +3 -0
- package/dist/cli/commands/finalize.d.ts.map +1 -0
- package/dist/cli/commands/finalize.js +41 -0
- package/dist/cli/commands/gate.d.ts +3 -0
- package/dist/cli/commands/gate.d.ts.map +1 -0
- package/dist/cli/commands/gate.js +89 -0
- package/dist/cli/commands/generate.d.ts +4 -0
- package/dist/cli/commands/generate.d.ts.map +1 -0
- package/dist/cli/commands/generate.js +108 -0
- package/dist/cli/commands/heal.d.ts +3 -0
- package/dist/cli/commands/heal.d.ts.map +1 -0
- package/dist/cli/commands/heal.js +60 -0
- package/dist/cli/commands/impact.d.ts +4 -0
- package/dist/cli/commands/impact.d.ts.map +1 -0
- package/dist/cli/commands/impact.js +33 -0
- package/dist/cli/commands/init.d.ts +2 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +169 -0
- package/dist/cli/commands/llm_health.d.ts +2 -0
- package/dist/cli/commands/llm_health.d.ts.map +1 -0
- package/dist/cli/commands/llm_health.js +22 -0
- package/dist/cli/commands/plan.d.ts +4 -0
- package/dist/cli/commands/plan.d.ts.map +1 -0
- package/dist/cli/commands/plan.js +120 -0
- package/dist/cli/commands/plan_crew.d.ts +17 -0
- package/dist/cli/commands/plan_crew.d.ts.map +1 -0
- package/dist/cli/commands/plan_crew.js +316 -0
- package/dist/cli/commands/traceability.d.ts +4 -0
- package/dist/cli/commands/traceability.d.ts.map +1 -0
- package/dist/cli/commands/traceability.js +77 -0
- package/dist/cli/commands/train.d.ts +3 -0
- package/dist/cli/commands/train.d.ts.map +1 -0
- package/dist/cli/commands/train.js +391 -0
- package/dist/cli/defaults.d.ts +35 -0
- package/dist/cli/defaults.d.ts.map +1 -0
- package/dist/cli/defaults.js +172 -0
- package/dist/cli/errors.d.ts +27 -0
- package/dist/cli/errors.d.ts.map +1 -0
- package/dist/cli/errors.js +57 -0
- package/dist/cli/parse_args.d.ts +6 -0
- package/dist/cli/parse_args.d.ts.map +1 -0
- package/dist/cli/parse_args.js +257 -0
- package/dist/cli/types.d.ts +87 -0
- package/dist/cli/types.d.ts.map +1 -0
- package/dist/cli/types.js +4 -0
- package/dist/cli/usage.d.ts +2 -0
- package/dist/cli/usage.d.ts.map +1 -0
- package/dist/cli/usage.js +109 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +194 -0
- package/dist/crew/context.d.ts +55 -0
- package/dist/crew/context.d.ts.map +1 -0
- package/dist/crew/context.js +36 -0
- package/dist/crew/orchestrator.d.ts +50 -0
- package/dist/crew/orchestrator.d.ts.map +1 -0
- package/dist/crew/orchestrator.js +329 -0
- package/dist/crew/protocol.d.ts +46 -0
- package/dist/crew/protocol.d.ts.map +1 -0
- package/dist/crew/protocol.js +4 -0
- package/dist/crew/provider.d.ts +17 -0
- package/dist/crew/provider.d.ts.map +1 -0
- package/dist/crew/provider.js +36 -0
- package/dist/crew/sanitize.d.ts +3 -0
- package/dist/crew/sanitize.d.ts.map +1 -0
- package/dist/crew/sanitize.js +31 -0
- package/dist/crew/types.d.ts +52 -0
- package/dist/crew/types.d.ts.map +1 -0
- package/dist/crew/types.js +4 -0
- package/dist/crew/workflows.d.ts +52 -0
- package/dist/crew/workflows.d.ts.map +1 -0
- package/dist/crew/workflows.js +36 -0
- package/dist/custom_provider.d.ts +20 -0
- package/dist/custom_provider.d.ts.map +1 -0
- package/dist/custom_provider.js +277 -0
- package/dist/engine/ai_enrichment.d.ts +44 -0
- package/dist/engine/ai_enrichment.d.ts.map +1 -0
- package/dist/engine/ai_enrichment.js +267 -0
- package/dist/engine/diff_loader.d.ts +11 -0
- package/dist/engine/diff_loader.d.ts.map +1 -0
- package/dist/engine/diff_loader.js +63 -0
- package/dist/engine/impact_engine.d.ts +72 -0
- package/dist/engine/impact_engine.d.ts.map +1 -0
- package/dist/engine/impact_engine.js +298 -0
- package/dist/engine/plan_builder.d.ts +11 -0
- package/dist/engine/plan_builder.d.ts.map +1 -0
- package/dist/engine/plan_builder.js +599 -0
- package/dist/esm/adapters/cypress.js +49 -0
- package/dist/esm/adapters/framework_adapter.js +114 -0
- package/dist/esm/adapters/playwright.js +49 -0
- package/dist/esm/adapters/pytest.js +59 -0
- package/dist/esm/adapters/supertest.js +48 -0
- package/dist/esm/agent/api_catalog.js +199 -0
- package/dist/esm/agent/config.js +872 -0
- package/dist/esm/agent/feedback.js +317 -0
- package/dist/esm/agent/git.js +252 -0
- package/dist/esm/agent/handoff.js +177 -0
- package/dist/esm/agent/llm_agents_flow.js +421 -0
- package/dist/esm/agent/native_flow.js +175 -0
- package/dist/esm/agent/pipeline.js +256 -0
- package/dist/esm/agent/pipeline_types.js +3 -0
- package/dist/esm/agent/pipeline_utils.js +146 -0
- package/dist/esm/agent/plan.js +83 -0
- package/dist/esm/agent/playwright_report.js +123 -0
- package/dist/esm/agent/process_runner.js +83 -0
- package/dist/esm/agent/spec_generator.js +249 -0
- package/dist/esm/agent/test_path.js +20 -0
- package/dist/esm/agent/traceability_capture.js +310 -0
- package/dist/esm/agent/traceability_ingest.js +234 -0
- package/dist/esm/agent/types.js +3 -0
- package/dist/esm/agent/utils.js +138 -0
- package/dist/esm/agent/validation_runner.js +73 -0
- package/dist/esm/agentic/fix_loop.js +91 -0
- package/dist/esm/agentic/playwright_runner.js +161 -0
- package/dist/esm/agentic/runner.js +207 -0
- package/dist/esm/agentic/types.js +3 -0
- package/dist/esm/agents/coverage-evaluator.js +37 -0
- package/dist/esm/agents/cross-impact.js +136 -0
- package/dist/esm/agents/executor.js +71 -0
- package/dist/esm/agents/explorer.js +39 -0
- package/dist/esm/agents/generator.js +73 -0
- package/dist/esm/agents/healer.js +27 -0
- package/dist/esm/agents/impact-analyst.js +34 -0
- package/dist/esm/agents/regression-advisor.js +112 -0
- package/dist/esm/agents/strategist.js +88 -0
- package/dist/esm/agents/test-designer.js +107 -0
- package/dist/esm/anthropic_provider.js +326 -0
- package/dist/esm/api.js +143 -0
- package/dist/esm/base_provider.js +198 -0
- package/dist/esm/budget_ledger.js +58 -0
- package/dist/esm/cache/cached_provider.js +85 -0
- package/dist/esm/cache/response_cache.js +140 -0
- package/dist/esm/cli/commands/analyze.js +74 -0
- package/dist/esm/cli/commands/bootstrap.js +106 -0
- package/dist/esm/cli/commands/cost_report.js +112 -0
- package/dist/esm/cli/commands/crew.js +252 -0
- package/dist/esm/cli/commands/feedback.js +36 -0
- package/dist/esm/cli/commands/finalize.js +38 -0
- package/dist/esm/cli/commands/gate.js +86 -0
- package/dist/esm/cli/commands/generate.js +105 -0
- package/dist/esm/cli/commands/heal.js +57 -0
- package/dist/esm/cli/commands/impact.js +30 -0
- package/dist/esm/cli/commands/init.js +133 -0
- package/dist/esm/cli/commands/llm_health.js +19 -0
- package/dist/esm/cli/commands/plan.js +117 -0
- package/dist/esm/cli/commands/plan_crew.js +309 -0
- package/dist/esm/cli/commands/traceability.js +73 -0
- package/dist/esm/cli/commands/train.js +355 -0
- package/dist/esm/cli/defaults.js +165 -0
- package/dist/esm/cli/errors.js +52 -0
- package/dist/esm/cli/parse_args.js +251 -0
- package/dist/esm/cli/types.js +3 -0
- package/dist/esm/cli/usage.js +106 -0
- package/dist/esm/cli.js +192 -0
- package/dist/esm/crew/context.js +32 -0
- package/dist/esm/crew/orchestrator.js +325 -0
- package/dist/esm/crew/protocol.js +3 -0
- package/dist/esm/crew/provider.js +33 -0
- package/dist/esm/crew/sanitize.js +27 -0
- package/dist/esm/crew/types.js +3 -0
- package/dist/esm/crew/workflows.js +33 -0
- package/dist/esm/custom_provider.js +273 -0
- package/dist/esm/engine/ai_enrichment.js +264 -0
- package/dist/esm/engine/diff_loader.js +59 -0
- package/dist/esm/engine/impact_engine.js +291 -0
- package/dist/esm/engine/plan_builder.js +593 -0
- package/dist/esm/index.js +72 -0
- package/dist/esm/knowledge/api_surface.js +408 -0
- package/dist/esm/knowledge/cluster_utils.js +60 -0
- package/dist/esm/knowledge/context_loader.js +85 -0
- package/dist/esm/knowledge/failure_history.js +121 -0
- package/dist/esm/knowledge/kg_bridge.js +381 -0
- package/dist/esm/knowledge/kg_types.js +3 -0
- package/dist/esm/knowledge/route_families.js +393 -0
- package/dist/esm/knowledge/spec_index.js +122 -0
- package/dist/esm/logger.js +115 -0
- package/dist/esm/mcp-server.js +621 -0
- package/dist/esm/metrics/prometheus.js +149 -0
- package/dist/esm/model_router.js +59 -0
- package/dist/esm/ollama_provider.js +301 -0
- package/dist/esm/openai_provider.js +243 -0
- package/dist/esm/package.json +3 -0
- package/dist/esm/pipeline/orchestrator.js +228 -0
- package/dist/esm/pipeline/spec_verifier.js +75 -0
- package/dist/esm/pipeline/stage0_preprocess.js +102 -0
- package/dist/esm/pipeline/stage1_impact.js +140 -0
- package/dist/esm/pipeline/stage2_coverage.js +153 -0
- package/dist/esm/pipeline/stage3_generation.js +284 -0
- package/dist/esm/pipeline/stage4_heal.js +288 -0
- package/dist/esm/progress.js +112 -0
- package/dist/esm/prompts/coverage.js +57 -0
- package/dist/esm/prompts/cross-impact.js +53 -0
- package/dist/esm/prompts/generation.js +297 -0
- package/dist/esm/prompts/generation_profile.js +147 -0
- package/dist/esm/prompts/heal.js +91 -0
- package/dist/esm/prompts/impact.js +63 -0
- package/dist/esm/prompts/json_extract.js +36 -0
- package/dist/esm/prompts/strategist.js +61 -0
- package/dist/esm/prompts/test-designer.js +92 -0
- package/dist/esm/provider_factory.js +366 -0
- package/dist/esm/provider_interface.js +23 -0
- package/dist/esm/provider_utils.js +96 -0
- package/dist/esm/qa-agent/cli.js +205 -0
- package/dist/esm/qa-agent/orchestrator.js +120 -0
- package/dist/esm/qa-agent/phase1/runner.js +139 -0
- package/dist/esm/qa-agent/phase1/scope.js +126 -0
- package/dist/esm/qa-agent/phase2/agent_browser.js +95 -0
- package/dist/esm/qa-agent/phase2/agent_loop.js +351 -0
- package/dist/esm/qa-agent/phase2/exploration_state.js +97 -0
- package/dist/esm/qa-agent/phase2/tools.js +386 -0
- package/dist/esm/qa-agent/phase2/vision.js +75 -0
- package/dist/esm/qa-agent/phase3/feedback.js +34 -0
- package/dist/esm/qa-agent/phase3/reporter.js +145 -0
- package/dist/esm/qa-agent/phase3/spec_generator.js +62 -0
- package/dist/esm/qa-agent/phase3/verdict.js +66 -0
- package/dist/esm/qa-agent/safe_env.js +23 -0
- package/dist/esm/qa-agent/types.js +3 -0
- package/dist/esm/reporters/junit.js +86 -0
- package/dist/esm/reporters/reporter.js +3 -0
- package/dist/esm/reporters/sarif.js +132 -0
- package/dist/esm/resilience/circuit_breaker.js +78 -0
- package/dist/esm/resilience/retry.js +56 -0
- package/dist/esm/sanitize.js +66 -0
- package/dist/esm/training/enricher.js +345 -0
- package/dist/esm/training/kg_scanner.js +115 -0
- package/dist/esm/training/merger.js +204 -0
- package/dist/esm/training/scanner.js +923 -0
- package/dist/esm/training/types.js +6 -0
- package/dist/esm/training/validator.js +254 -0
- package/dist/esm/validation/guardrails.js +101 -0
- package/dist/esm/validation/output_schema.js +80 -0
- package/dist/esm/version.js +33 -0
- package/dist/index.d.ts +99 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +169 -0
- package/dist/knowledge/api_surface.d.ts +37 -0
- package/dist/knowledge/api_surface.d.ts.map +1 -0
- package/dist/knowledge/api_surface.js +418 -0
- package/dist/knowledge/cluster_utils.d.ts +28 -0
- package/dist/knowledge/cluster_utils.d.ts.map +1 -0
- package/dist/knowledge/cluster_utils.js +67 -0
- package/dist/knowledge/context_loader.d.ts +13 -0
- package/dist/knowledge/context_loader.d.ts.map +1 -0
- package/dist/knowledge/context_loader.js +90 -0
- package/dist/knowledge/failure_history.d.ts +39 -0
- package/dist/knowledge/failure_history.d.ts.map +1 -0
- package/dist/knowledge/failure_history.js +128 -0
- package/dist/knowledge/kg_bridge.d.ts +31 -0
- package/dist/knowledge/kg_bridge.d.ts.map +1 -0
- package/dist/knowledge/kg_bridge.js +388 -0
- package/dist/knowledge/kg_types.d.ts +75 -0
- package/dist/knowledge/kg_types.d.ts.map +1 -0
- package/dist/knowledge/kg_types.js +4 -0
- package/dist/knowledge/route_families.d.ts +98 -0
- package/dist/knowledge/route_families.d.ts.map +1 -0
- package/dist/knowledge/route_families.js +410 -0
- package/dist/knowledge/spec_index.d.ts +18 -0
- package/dist/knowledge/spec_index.d.ts.map +1 -0
- package/dist/knowledge/spec_index.js +128 -0
- package/dist/logger.d.ts +31 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +119 -0
- package/dist/mcp-server.d.ts +68 -0
- package/dist/mcp-server.d.ts.map +1 -0
- package/dist/mcp-server.js +629 -0
- package/dist/metrics/prometheus.d.ts +37 -0
- package/dist/metrics/prometheus.d.ts.map +1 -0
- package/dist/metrics/prometheus.js +153 -0
- package/dist/model_router.d.ts +28 -0
- package/dist/model_router.d.ts.map +1 -0
- package/dist/model_router.js +63 -0
- package/dist/ollama_provider.d.ts +65 -0
- package/dist/ollama_provider.d.ts.map +1 -0
- package/dist/ollama_provider.js +309 -0
- package/dist/openai_provider.d.ts +23 -0
- package/dist/openai_provider.d.ts.map +1 -0
- package/dist/openai_provider.js +251 -0
- package/dist/pipeline/orchestrator.d.ts +33 -0
- package/dist/pipeline/orchestrator.d.ts.map +1 -0
- package/dist/pipeline/orchestrator.js +231 -0
- package/dist/pipeline/spec_verifier.d.ts +20 -0
- package/dist/pipeline/spec_verifier.d.ts.map +1 -0
- package/dist/pipeline/spec_verifier.js +79 -0
- package/dist/pipeline/stage0_preprocess.d.ts +31 -0
- package/dist/pipeline/stage0_preprocess.d.ts.map +1 -0
- package/dist/pipeline/stage0_preprocess.js +105 -0
- package/dist/pipeline/stage1_impact.d.ts +19 -0
- package/dist/pipeline/stage1_impact.d.ts.map +1 -0
- package/dist/pipeline/stage1_impact.js +143 -0
- package/dist/pipeline/stage2_coverage.d.ts +19 -0
- package/dist/pipeline/stage2_coverage.d.ts.map +1 -0
- package/dist/pipeline/stage2_coverage.js +156 -0
- package/dist/pipeline/stage3_generation.d.ts +43 -0
- package/dist/pipeline/stage3_generation.d.ts.map +1 -0
- package/dist/pipeline/stage3_generation.js +287 -0
- package/dist/pipeline/stage4_heal.d.ts +62 -0
- package/dist/pipeline/stage4_heal.d.ts.map +1 -0
- package/dist/pipeline/stage4_heal.js +294 -0
- package/dist/progress.d.ts +22 -0
- package/dist/progress.d.ts.map +1 -0
- package/dist/progress.js +116 -0
- package/dist/prompts/coverage.d.ts +39 -0
- package/dist/prompts/coverage.d.ts.map +1 -0
- package/dist/prompts/coverage.js +61 -0
- package/dist/prompts/cross-impact.d.ts +23 -0
- package/dist/prompts/cross-impact.d.ts.map +1 -0
- package/dist/prompts/cross-impact.js +57 -0
- package/dist/prompts/generation.d.ts +25 -0
- package/dist/prompts/generation.d.ts.map +1 -0
- package/dist/prompts/generation.js +302 -0
- package/dist/prompts/generation_profile.d.ts +29 -0
- package/dist/prompts/generation_profile.d.ts.map +1 -0
- package/dist/prompts/generation_profile.js +151 -0
- package/dist/prompts/heal.d.ts +23 -0
- package/dist/prompts/heal.d.ts.map +1 -0
- package/dist/prompts/heal.js +95 -0
- package/dist/prompts/impact.d.ts +31 -0
- package/dist/prompts/impact.d.ts.map +1 -0
- package/dist/prompts/impact.js +67 -0
- package/dist/prompts/json_extract.d.ts +14 -0
- package/dist/prompts/json_extract.d.ts.map +1 -0
- package/dist/prompts/json_extract.js +39 -0
- package/dist/prompts/strategist.d.ts +25 -0
- package/dist/prompts/strategist.d.ts.map +1 -0
- package/dist/prompts/strategist.js +65 -0
- package/dist/prompts/test-designer.d.ts +35 -0
- package/dist/prompts/test-designer.d.ts.map +1 -0
- package/dist/prompts/test-designer.js +96 -0
- package/dist/provider_factory.d.ts +104 -0
- package/dist/provider_factory.d.ts.map +1 -0
- package/dist/provider_factory.js +371 -0
- package/dist/provider_interface.d.ts +365 -0
- package/dist/provider_interface.d.ts.map +1 -0
- package/dist/provider_interface.js +28 -0
- package/dist/provider_utils.d.ts +39 -0
- package/dist/provider_utils.d.ts.map +1 -0
- package/dist/provider_utils.js +103 -0
- package/dist/qa-agent/cli.d.ts +3 -0
- package/dist/qa-agent/cli.d.ts.map +1 -0
- package/dist/qa-agent/cli.js +207 -0
- package/dist/qa-agent/orchestrator.d.ts +3 -0
- package/dist/qa-agent/orchestrator.d.ts.map +1 -0
- package/dist/qa-agent/orchestrator.js +123 -0
- package/dist/qa-agent/phase1/runner.d.ts +3 -0
- package/dist/qa-agent/phase1/runner.d.ts.map +1 -0
- package/dist/qa-agent/phase1/runner.js +142 -0
- package/dist/qa-agent/phase1/scope.d.ts +6 -0
- package/dist/qa-agent/phase1/scope.d.ts.map +1 -0
- package/dist/qa-agent/phase1/scope.js +129 -0
- package/dist/qa-agent/phase2/agent_browser.d.ts +35 -0
- package/dist/qa-agent/phase2/agent_browser.d.ts.map +1 -0
- package/dist/qa-agent/phase2/agent_browser.js +99 -0
- package/dist/qa-agent/phase2/agent_loop.d.ts +3 -0
- package/dist/qa-agent/phase2/agent_loop.d.ts.map +1 -0
- package/dist/qa-agent/phase2/agent_loop.js +357 -0
- package/dist/qa-agent/phase2/exploration_state.d.ts +12 -0
- package/dist/qa-agent/phase2/exploration_state.d.ts.map +1 -0
- package/dist/qa-agent/phase2/exploration_state.js +109 -0
- package/dist/qa-agent/phase2/tools.d.ts +28 -0
- package/dist/qa-agent/phase2/tools.d.ts.map +1 -0
- package/dist/qa-agent/phase2/tools.js +390 -0
- package/dist/qa-agent/phase2/vision.d.ts +3 -0
- package/dist/qa-agent/phase2/vision.d.ts.map +1 -0
- package/dist/qa-agent/phase2/vision.js +78 -0
- package/dist/qa-agent/phase3/feedback.d.ts +3 -0
- package/dist/qa-agent/phase3/feedback.d.ts.map +1 -0
- package/dist/qa-agent/phase3/feedback.js +37 -0
- package/dist/qa-agent/phase3/reporter.d.ts +3 -0
- package/dist/qa-agent/phase3/reporter.d.ts.map +1 -0
- package/dist/qa-agent/phase3/reporter.js +148 -0
- package/dist/qa-agent/phase3/spec_generator.d.ts +3 -0
- package/dist/qa-agent/phase3/spec_generator.d.ts.map +1 -0
- package/dist/qa-agent/phase3/spec_generator.js +65 -0
- package/dist/qa-agent/phase3/verdict.d.ts +3 -0
- package/dist/qa-agent/phase3/verdict.d.ts.map +1 -0
- package/dist/qa-agent/phase3/verdict.js +69 -0
- package/dist/qa-agent/safe_env.d.ts +3 -0
- package/dist/qa-agent/safe_env.d.ts.map +1 -0
- package/dist/qa-agent/safe_env.js +26 -0
- package/dist/qa-agent/types.d.ts +130 -0
- package/dist/qa-agent/types.d.ts.map +1 -0
- package/dist/qa-agent/types.js +4 -0
- package/dist/reporters/junit.d.ts +6 -0
- package/dist/reporters/junit.d.ts.map +1 -0
- package/dist/reporters/junit.js +89 -0
- package/dist/reporters/reporter.d.ts +42 -0
- package/dist/reporters/reporter.d.ts.map +1 -0
- package/dist/reporters/reporter.js +4 -0
- package/dist/reporters/sarif.d.ts +7 -0
- package/dist/reporters/sarif.d.ts.map +1 -0
- package/dist/reporters/sarif.js +135 -0
- package/dist/resilience/circuit_breaker.d.ts +36 -0
- package/dist/resilience/circuit_breaker.d.ts.map +1 -0
- package/dist/resilience/circuit_breaker.js +82 -0
- package/dist/resilience/retry.d.ts +11 -0
- package/dist/resilience/retry.d.ts.map +1 -0
- package/dist/resilience/retry.js +59 -0
- package/dist/sanitize.d.ts +15 -0
- package/dist/sanitize.d.ts.map +1 -0
- package/dist/sanitize.js +71 -0
- package/dist/training/enricher.d.ts +17 -0
- package/dist/training/enricher.d.ts.map +1 -0
- package/dist/training/enricher.js +350 -0
- package/dist/training/kg_scanner.d.ts +13 -0
- package/dist/training/kg_scanner.d.ts.map +1 -0
- package/dist/training/kg_scanner.js +118 -0
- package/dist/training/merger.d.ts +15 -0
- package/dist/training/merger.d.ts.map +1 -0
- package/dist/training/merger.js +208 -0
- package/dist/training/scanner.d.ts +36 -0
- package/dist/training/scanner.d.ts.map +1 -0
- package/dist/training/scanner.js +932 -0
- package/dist/training/types.d.ts +117 -0
- package/dist/training/types.d.ts.map +1 -0
- package/dist/training/types.js +9 -0
- package/dist/training/validator.d.ts +21 -0
- package/dist/training/validator.d.ts.map +1 -0
- package/dist/training/validator.js +262 -0
- package/dist/validation/guardrails.d.ts +31 -0
- package/dist/validation/guardrails.d.ts.map +1 -0
- package/dist/validation/guardrails.js +112 -0
- package/dist/validation/output_schema.d.ts +67 -0
- package/dist/validation/output_schema.d.ts.map +1 -0
- package/dist/validation/output_schema.js +84 -0
- package/dist/version.d.ts +6 -0
- package/dist/version.d.ts.map +1 -0
- package/dist/version.js +36 -0
- package/package.json +126 -0
- package/schemas/flow-decision.schema.json +83 -0
- package/schemas/gap.schema.json +18 -0
- package/schemas/impact.schema.json +455 -0
- package/schemas/plan.schema.json +491 -0
- package/schemas/route-families.schema.json +137 -0
- package/schemas/subsystem-risk-map.schema.json +62 -0
- package/schemas/traceability-input.schema.json +122 -0
|
@@ -0,0 +1,153 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
3
|
+
// See LICENSE.txt for license information.
|
|
4
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
5
|
+
exports.PrometheusMetrics = void 0;
|
|
6
|
+
const DURATION_BUCKETS = [0.1, 0.5, 1, 2, 5, 10, 30, 60, 120, 300];
|
|
7
|
+
class PrometheusMetrics {
|
|
8
|
+
constructor() {
|
|
9
|
+
this.counters = [];
|
|
10
|
+
this.gauges = [];
|
|
11
|
+
this.histograms = [];
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* Record an LLM request.
|
|
15
|
+
*/
|
|
16
|
+
recordLLMRequest(provider, agent, durationMs, costUSD, tokens) {
|
|
17
|
+
this.incrementCounter('e2e_agents_llm_requests_total', 'Total LLM requests', { provider, agent });
|
|
18
|
+
this.incrementCounter('e2e_agents_llm_tokens_total', 'Total tokens consumed', { provider, agent }, tokens);
|
|
19
|
+
this.incrementCounter('e2e_agents_llm_cost_usd_total', 'Total LLM cost in USD', { provider, agent }, costUSD);
|
|
20
|
+
this.observeHistogram('e2e_agents_llm_request_duration_seconds', 'LLM request duration', { provider, agent }, durationMs / 1000);
|
|
21
|
+
}
|
|
22
|
+
/**
|
|
23
|
+
* Record a crew workflow run.
|
|
24
|
+
*/
|
|
25
|
+
recordCrewRun(workflow, families, durationMs, costUSD) {
|
|
26
|
+
this.incrementCounter('e2e_agents_crew_runs_total', 'Total crew workflow runs', { workflow });
|
|
27
|
+
this.incrementCounter('e2e_agents_crew_families_processed_total', 'Total families processed', { workflow }, families);
|
|
28
|
+
this.incrementCounter('e2e_agents_crew_cost_usd_total', 'Total crew cost in USD', { workflow }, costUSD);
|
|
29
|
+
this.observeHistogram('e2e_agents_crew_duration_seconds', 'Crew workflow duration', { workflow }, durationMs / 1000);
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* Record a budget check event.
|
|
33
|
+
*/
|
|
34
|
+
recordBudgetCheck(exceeded, currentUSD, limitUSD) {
|
|
35
|
+
this.incrementCounter('e2e_agents_budget_checks_total', 'Total budget checks', { exceeded: String(exceeded) });
|
|
36
|
+
this.setGauge('e2e_agents_budget_used_usd', 'Current budget usage in USD', {}, currentUSD);
|
|
37
|
+
this.setGauge('e2e_agents_budget_limit_usd', 'Budget limit in USD', {}, limitUSD);
|
|
38
|
+
}
|
|
39
|
+
/**
|
|
40
|
+
* Record a circuit breaker state change.
|
|
41
|
+
*/
|
|
42
|
+
recordCircuitBreakerState(state) {
|
|
43
|
+
this.setGauge('e2e_agents_circuit_breaker_state', 'Circuit breaker state (0=closed, 1=open, 2=half-open)', {}, state === 'closed' ? 0 : state === 'open' ? 1 : 2);
|
|
44
|
+
}
|
|
45
|
+
/**
|
|
46
|
+
* Record a cache hit or miss.
|
|
47
|
+
*/
|
|
48
|
+
recordCacheResult(hit, agent) {
|
|
49
|
+
this.incrementCounter('e2e_agents_cache_lookups_total', 'Total cache lookups', { result: hit ? 'hit' : 'miss', agent });
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* Export all metrics in Prometheus text exposition format.
|
|
53
|
+
*/
|
|
54
|
+
export() {
|
|
55
|
+
const lines = [];
|
|
56
|
+
const seenHelp = new Set();
|
|
57
|
+
// Export counters
|
|
58
|
+
for (const counter of this.counters) {
|
|
59
|
+
if (!seenHelp.has(counter.name)) {
|
|
60
|
+
lines.push(`# HELP ${counter.name} ${counter.help}`);
|
|
61
|
+
lines.push(`# TYPE ${counter.name} counter`);
|
|
62
|
+
seenHelp.add(counter.name);
|
|
63
|
+
}
|
|
64
|
+
const labelStr = formatLabels(counter.labels);
|
|
65
|
+
lines.push(`${counter.name}${labelStr} ${counter.value}`);
|
|
66
|
+
}
|
|
67
|
+
// Export gauges
|
|
68
|
+
for (const gauge of this.gauges) {
|
|
69
|
+
if (!seenHelp.has(gauge.name)) {
|
|
70
|
+
lines.push(`# HELP ${gauge.name} ${gauge.help}`);
|
|
71
|
+
lines.push(`# TYPE ${gauge.name} gauge`);
|
|
72
|
+
seenHelp.add(gauge.name);
|
|
73
|
+
}
|
|
74
|
+
const labelStr = formatLabels(gauge.labels);
|
|
75
|
+
lines.push(`${gauge.name}${labelStr} ${gauge.value}`);
|
|
76
|
+
}
|
|
77
|
+
// Export histograms
|
|
78
|
+
for (const hist of this.histograms) {
|
|
79
|
+
if (!seenHelp.has(hist.name)) {
|
|
80
|
+
lines.push(`# HELP ${hist.name} ${hist.help}`);
|
|
81
|
+
lines.push(`# TYPE ${hist.name} histogram`);
|
|
82
|
+
seenHelp.add(hist.name);
|
|
83
|
+
}
|
|
84
|
+
const labelStr = formatLabels(hist.labels);
|
|
85
|
+
let cumulative = 0;
|
|
86
|
+
for (const bucket of DURATION_BUCKETS) {
|
|
87
|
+
cumulative += hist.buckets.get(bucket) || 0;
|
|
88
|
+
lines.push(`${hist.name}_bucket${formatLabels({ ...hist.labels, le: String(bucket) })} ${cumulative}`);
|
|
89
|
+
}
|
|
90
|
+
lines.push(`${hist.name}_bucket${formatLabels({ ...hist.labels, le: '+Inf' })} ${hist.count}`);
|
|
91
|
+
lines.push(`${hist.name}_sum${labelStr} ${hist.sum}`);
|
|
92
|
+
lines.push(`${hist.name}_count${labelStr} ${hist.count}`);
|
|
93
|
+
}
|
|
94
|
+
return lines.join('\n') + '\n';
|
|
95
|
+
}
|
|
96
|
+
/**
|
|
97
|
+
* Reset all metrics to zero.
|
|
98
|
+
*/
|
|
99
|
+
reset() {
|
|
100
|
+
this.counters = [];
|
|
101
|
+
this.gauges = [];
|
|
102
|
+
this.histograms = [];
|
|
103
|
+
}
|
|
104
|
+
incrementCounter(name, help, labels, value = 1) {
|
|
105
|
+
const existing = this.counters.find((c) => c.name === name && labelsMatch(c.labels, labels));
|
|
106
|
+
if (existing) {
|
|
107
|
+
existing.value += value;
|
|
108
|
+
}
|
|
109
|
+
else {
|
|
110
|
+
this.counters.push({ name, help, labels, value });
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
setGauge(name, help, labels, value) {
|
|
114
|
+
const existing = this.gauges.find((c) => c.name === name && labelsMatch(c.labels, labels));
|
|
115
|
+
if (existing) {
|
|
116
|
+
existing.value = value;
|
|
117
|
+
}
|
|
118
|
+
else {
|
|
119
|
+
this.gauges.push({ name, help, labels, value });
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
observeHistogram(name, help, labels, value) {
|
|
123
|
+
let existing = this.histograms.find((h) => h.name === name && labelsMatch(h.labels, labels));
|
|
124
|
+
if (!existing) {
|
|
125
|
+
existing = { name, help, labels, sum: 0, count: 0, buckets: new Map() };
|
|
126
|
+
this.histograms.push(existing);
|
|
127
|
+
}
|
|
128
|
+
existing.sum += value;
|
|
129
|
+
existing.count++;
|
|
130
|
+
for (const bucket of DURATION_BUCKETS) {
|
|
131
|
+
if (value <= bucket) {
|
|
132
|
+
existing.buckets.set(bucket, (existing.buckets.get(bucket) || 0) + 1);
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
exports.PrometheusMetrics = PrometheusMetrics;
|
|
138
|
+
function escapeLabel(v) {
|
|
139
|
+
return v.replace(/\\/g, '\\\\').replace(/"/g, '\\"').replace(/\n/g, '\\n');
|
|
140
|
+
}
|
|
141
|
+
function formatLabels(labels) {
|
|
142
|
+
const entries = Object.entries(labels);
|
|
143
|
+
if (entries.length === 0)
|
|
144
|
+
return '';
|
|
145
|
+
return `{${entries.map(([k, v]) => `${k}="${escapeLabel(v)}"`).join(',')}}`;
|
|
146
|
+
}
|
|
147
|
+
function labelsMatch(a, b) {
|
|
148
|
+
const keysA = Object.keys(a);
|
|
149
|
+
const keysB = Object.keys(b);
|
|
150
|
+
if (keysA.length !== keysB.length)
|
|
151
|
+
return false;
|
|
152
|
+
return keysA.every((k) => a[k] === b[k]);
|
|
153
|
+
}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Model Router — routes agent tasks to appropriate models based on task complexity.
|
|
3
|
+
*
|
|
4
|
+
* Classification/extraction tasks → cheap model (Haiku)
|
|
5
|
+
* Generation/reasoning tasks → capable model (Sonnet)
|
|
6
|
+
*/
|
|
7
|
+
export type TaskComplexity = 'classification' | 'extraction' | 'generation' | 'reasoning';
|
|
8
|
+
export interface ModelRoutingConfig {
|
|
9
|
+
classification?: string;
|
|
10
|
+
extraction?: string;
|
|
11
|
+
generation?: string;
|
|
12
|
+
reasoning?: string;
|
|
13
|
+
}
|
|
14
|
+
export declare class ModelRouter {
|
|
15
|
+
private overrides;
|
|
16
|
+
private providerType;
|
|
17
|
+
constructor(providerType: string, overrides?: ModelRoutingConfig);
|
|
18
|
+
/**
|
|
19
|
+
* Get the recommended model for a given agent role.
|
|
20
|
+
* Returns undefined if no routing recommendation (use provider default).
|
|
21
|
+
*/
|
|
22
|
+
getModel(role: string): string | undefined;
|
|
23
|
+
/**
|
|
24
|
+
* Get the task complexity for an agent role.
|
|
25
|
+
*/
|
|
26
|
+
getComplexity(role: string): TaskComplexity;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=model_router.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model_router.d.ts","sourceRoot":"","sources":["../src/model_router.ts"],"names":[],"mappings":"AAGA;;;;;GAKG;AAEH,MAAM,MAAM,cAAc,GAAG,gBAAgB,GAAG,YAAY,GAAG,YAAY,GAAG,WAAW,CAAC;AAE1F,MAAM,WAAW,kBAAkB;IAC/B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AA8BD,qBAAa,WAAW;IACpB,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,YAAY,CAAS;gBAEjB,YAAY,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,kBAAkB;IAKhE;;;OAGG;IACH,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAgB1C;;OAEG;IACH,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;CAG9C"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
3
|
+
// See LICENSE.txt for license information.
|
|
4
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
5
|
+
exports.ModelRouter = void 0;
|
|
6
|
+
const AGENT_COMPLEXITY = {
|
|
7
|
+
'impact-analyst': 'classification',
|
|
8
|
+
'coverage-evaluator': 'classification',
|
|
9
|
+
'cross-impact': 'extraction',
|
|
10
|
+
'regression-advisor': 'extraction',
|
|
11
|
+
'strategist': 'classification',
|
|
12
|
+
'test-designer': 'generation',
|
|
13
|
+
'generator': 'generation',
|
|
14
|
+
'executor': 'generation',
|
|
15
|
+
'healer': 'reasoning',
|
|
16
|
+
'explorer': 'reasoning',
|
|
17
|
+
};
|
|
18
|
+
const DEFAULT_MODELS = {
|
|
19
|
+
anthropic: {
|
|
20
|
+
classification: 'claude-haiku-4-5-20251001',
|
|
21
|
+
extraction: 'claude-haiku-4-5-20251001',
|
|
22
|
+
generation: 'claude-sonnet-4-5-20250514',
|
|
23
|
+
reasoning: 'claude-sonnet-4-5-20250514',
|
|
24
|
+
},
|
|
25
|
+
openai: {
|
|
26
|
+
classification: 'gpt-4o-mini',
|
|
27
|
+
extraction: 'gpt-4o-mini',
|
|
28
|
+
generation: 'gpt-4o',
|
|
29
|
+
reasoning: 'gpt-4o',
|
|
30
|
+
},
|
|
31
|
+
};
|
|
32
|
+
class ModelRouter {
|
|
33
|
+
constructor(providerType, overrides) {
|
|
34
|
+
this.providerType = providerType;
|
|
35
|
+
this.overrides = overrides || {};
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* Get the recommended model for a given agent role.
|
|
39
|
+
* Returns undefined if no routing recommendation (use provider default).
|
|
40
|
+
*/
|
|
41
|
+
getModel(role) {
|
|
42
|
+
const complexity = AGENT_COMPLEXITY[role];
|
|
43
|
+
if (!complexity)
|
|
44
|
+
return undefined;
|
|
45
|
+
// Check user overrides first
|
|
46
|
+
const override = this.overrides[complexity];
|
|
47
|
+
if (override)
|
|
48
|
+
return override;
|
|
49
|
+
// Check provider defaults
|
|
50
|
+
const defaults = DEFAULT_MODELS[this.providerType];
|
|
51
|
+
if (defaults)
|
|
52
|
+
return defaults[complexity];
|
|
53
|
+
// No recommendation — use provider's default model
|
|
54
|
+
return undefined;
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* Get the task complexity for an agent role.
|
|
58
|
+
*/
|
|
59
|
+
getComplexity(role) {
|
|
60
|
+
return AGENT_COMPLEXITY[role] || 'generation';
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
exports.ModelRouter = ModelRouter;
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
import type { GenerateOptions, ImageInput, LLMResponse, OllamaConfig, ProviderCapabilities } from './provider_interface.js';
|
|
2
|
+
import { BaseProvider } from './base_provider.js';
|
|
3
|
+
/**
|
|
4
|
+
* Ollama Provider - Free, local LLM execution
|
|
5
|
+
*
|
|
6
|
+
* Features:
|
|
7
|
+
* - Zero cost (runs locally)
|
|
8
|
+
* - Full privacy (no data leaves your machine)
|
|
9
|
+
* - OpenAI-compatible API
|
|
10
|
+
* - Supports DeepSeek-R1, Llama 4, and other open models
|
|
11
|
+
*
|
|
12
|
+
* Limitations:
|
|
13
|
+
* - No vision support (most models)
|
|
14
|
+
* - Slower inference than cloud APIs (~2-5 sec vs <1 sec)
|
|
15
|
+
* - Requires local installation and model downloads
|
|
16
|
+
*
|
|
17
|
+
* Recommended models:
|
|
18
|
+
* - deepseek-r1:7b - Fast, good quality, low memory (4GB)
|
|
19
|
+
* - deepseek-r1:14b - Better quality, medium memory (8GB)
|
|
20
|
+
* - llama4:13b - High quality, medium memory (8GB)
|
|
21
|
+
* - deepseek-r1:7b-q4 - Quantized for speed, lower quality
|
|
22
|
+
*
|
|
23
|
+
* Setup:
|
|
24
|
+
* 1. Install Ollama: curl -fsSL https://ollama.com/install.sh | sh
|
|
25
|
+
* 2. Pull model: ollama pull deepseek-r1:7b
|
|
26
|
+
* 3. Start: ollama serve (runs on localhost:11434)
|
|
27
|
+
*/
|
|
28
|
+
export declare class OllamaProvider extends BaseProvider {
|
|
29
|
+
name: string;
|
|
30
|
+
private client;
|
|
31
|
+
private model;
|
|
32
|
+
capabilities: ProviderCapabilities;
|
|
33
|
+
constructor(config: OllamaConfig);
|
|
34
|
+
generateText(prompt: string, options?: GenerateOptions): Promise<LLMResponse>;
|
|
35
|
+
/**
|
|
36
|
+
* Ollama does not support vision by default
|
|
37
|
+
* This method throws an error to help users understand the limitation
|
|
38
|
+
*/
|
|
39
|
+
analyzeImage(images: ImageInput[], prompt: string, options?: GenerateOptions): Promise<LLMResponse>;
|
|
40
|
+
/**
|
|
41
|
+
* Stream text generation for real-time feedback
|
|
42
|
+
*/
|
|
43
|
+
streamText(prompt: string, options?: GenerateOptions): AsyncGenerator<string, void, unknown>;
|
|
44
|
+
/**
|
|
45
|
+
* Check if Ollama is running and accessible
|
|
46
|
+
*/
|
|
47
|
+
checkHealth(): Promise<{
|
|
48
|
+
healthy: boolean;
|
|
49
|
+
message: string;
|
|
50
|
+
}>;
|
|
51
|
+
/**
|
|
52
|
+
* List available models in Ollama
|
|
53
|
+
*/
|
|
54
|
+
listModels(): Promise<string[]>;
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* Helper function to check if Ollama is installed and suggest setup
|
|
58
|
+
*/
|
|
59
|
+
export declare function checkOllamaSetup(): Promise<{
|
|
60
|
+
installed: boolean;
|
|
61
|
+
running: boolean;
|
|
62
|
+
modelAvailable: boolean;
|
|
63
|
+
setupInstructions: string;
|
|
64
|
+
}>;
|
|
65
|
+
//# sourceMappingURL=ollama_provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama_provider.d.ts","sourceRoot":"","sources":["../src/ollama_provider.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACR,eAAe,EACf,UAAU,EACV,WAAW,EACX,YAAY,EACZ,oBAAoB,EACvB,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAC,YAAY,EAAC,MAAM,oBAAoB,CAAC;AAiEhD;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,qBAAa,cAAe,SAAQ,YAAY;IAC5C,IAAI,SAAY;IAChB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IAEtB,YAAY,EAAE,oBAAoB,CAShC;gBAEU,MAAM,EAAE,YAAY;IA8B1B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IAmEnF;;;OAGG;IAEG,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IAIzG;;OAEG;IACI,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC;IAqDnG;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAC,CAAC;IAgBjE;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAaxC;AAED;;GAEG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC;IAC9C,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,cAAc,EAAE,OAAO,CAAC;IACxB,iBAAiB,EAAE,MAAM,CAAC;CAC7B,CAAC,CAoCD"}
|
|
@@ -0,0 +1,309 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
3
|
+
// See LICENSE.txt for license information.
|
|
4
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
5
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
6
|
+
};
|
|
7
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
8
|
+
exports.OllamaProvider = void 0;
|
|
9
|
+
exports.checkOllamaSetup = checkOllamaSetup;
|
|
10
|
+
const openai_1 = __importDefault(require("openai"));
|
|
11
|
+
const provider_interface_js_1 = require("./provider_interface.js");
|
|
12
|
+
const provider_utils_js_1 = require("./provider_utils.js");
|
|
13
|
+
const base_provider_js_1 = require("./base_provider.js");
|
|
14
|
+
const logger_js_1 = require("./logger.js");
|
|
15
|
+
/**
|
|
16
|
+
* SECURITY: Validate Ollama base URL and enforce HTTPS for remote connections
|
|
17
|
+
*/
|
|
18
|
+
function normalizeOllamaBaseUrl(baseUrl) {
|
|
19
|
+
const raw = baseUrl || 'http://localhost:11434';
|
|
20
|
+
try {
|
|
21
|
+
const parsed = new URL(raw);
|
|
22
|
+
if (!parsed.pathname || parsed.pathname === '/') {
|
|
23
|
+
parsed.pathname = '/v1';
|
|
24
|
+
}
|
|
25
|
+
return parsed.toString().replace(/\/$/, '');
|
|
26
|
+
}
|
|
27
|
+
catch {
|
|
28
|
+
return 'http://localhost:11434/v1';
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
function validateOllamaUrl(baseUrl) {
|
|
32
|
+
const url = normalizeOllamaBaseUrl(baseUrl);
|
|
33
|
+
try {
|
|
34
|
+
const parsed = new URL(url);
|
|
35
|
+
// For non-localhost URLs, warn about HTTP risks
|
|
36
|
+
const isLocalhost = parsed.hostname === 'localhost' || parsed.hostname === '127.0.0.1' || parsed.hostname === '::1';
|
|
37
|
+
if (!isLocalhost && parsed.protocol === 'http:') {
|
|
38
|
+
logger_js_1.logger.warn('Ollama connection over plaintext HTTP to remote server. ' +
|
|
39
|
+
'Prompts and responses will be transmitted unencrypted. Consider using HTTPS proxy or local Ollama.');
|
|
40
|
+
}
|
|
41
|
+
return { valid: true, url };
|
|
42
|
+
}
|
|
43
|
+
catch {
|
|
44
|
+
return {
|
|
45
|
+
valid: false,
|
|
46
|
+
url: 'http://localhost:11434/v1',
|
|
47
|
+
warning: `Invalid Ollama URL: ${baseUrl}. Using default: http://localhost:11434/v1`,
|
|
48
|
+
};
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* SECURITY: Validate model name to prevent injection issues
|
|
53
|
+
*/
|
|
54
|
+
function validateModelName(model) {
|
|
55
|
+
// Allow alphanumeric, dash, colon, underscore
|
|
56
|
+
// Typical format: deepseek-r1:7b, llama4:13b, etc.
|
|
57
|
+
return /^[a-z0-9_:.\-]+$/i.test(model) && model.length < 256;
|
|
58
|
+
}
|
|
59
|
+
/**
|
|
60
|
+
* SECURITY: Validate timeout value
|
|
61
|
+
*/
|
|
62
|
+
function validateTimeout(timeout) {
|
|
63
|
+
if (!timeout)
|
|
64
|
+
return 60000;
|
|
65
|
+
if (timeout < 1000 || timeout > 600000) {
|
|
66
|
+
logger_js_1.logger.warn('Timeout out of valid range (1s-10m). Using 60 second default.');
|
|
67
|
+
return 60000;
|
|
68
|
+
}
|
|
69
|
+
return timeout;
|
|
70
|
+
}
|
|
71
|
+
/**
|
|
72
|
+
* Ollama Provider - Free, local LLM execution
|
|
73
|
+
*
|
|
74
|
+
* Features:
|
|
75
|
+
* - Zero cost (runs locally)
|
|
76
|
+
* - Full privacy (no data leaves your machine)
|
|
77
|
+
* - OpenAI-compatible API
|
|
78
|
+
* - Supports DeepSeek-R1, Llama 4, and other open models
|
|
79
|
+
*
|
|
80
|
+
* Limitations:
|
|
81
|
+
* - No vision support (most models)
|
|
82
|
+
* - Slower inference than cloud APIs (~2-5 sec vs <1 sec)
|
|
83
|
+
* - Requires local installation and model downloads
|
|
84
|
+
*
|
|
85
|
+
* Recommended models:
|
|
86
|
+
* - deepseek-r1:7b - Fast, good quality, low memory (4GB)
|
|
87
|
+
* - deepseek-r1:14b - Better quality, medium memory (8GB)
|
|
88
|
+
* - llama4:13b - High quality, medium memory (8GB)
|
|
89
|
+
* - deepseek-r1:7b-q4 - Quantized for speed, lower quality
|
|
90
|
+
*
|
|
91
|
+
* Setup:
|
|
92
|
+
* 1. Install Ollama: curl -fsSL https://ollama.com/install.sh | sh
|
|
93
|
+
* 2. Pull model: ollama pull deepseek-r1:7b
|
|
94
|
+
* 3. Start: ollama serve (runs on localhost:11434)
|
|
95
|
+
*/
|
|
96
|
+
class OllamaProvider extends base_provider_js_1.BaseProvider {
|
|
97
|
+
constructor(config) {
|
|
98
|
+
super();
|
|
99
|
+
this.name = 'ollama';
|
|
100
|
+
this.capabilities = {
|
|
101
|
+
vision: false, // Most Ollama models don't support vision
|
|
102
|
+
streaming: true,
|
|
103
|
+
maxTokens: 8000, // Varies by model
|
|
104
|
+
costPer1MInputTokens: 0, // Free!
|
|
105
|
+
costPer1MOutputTokens: 0, // Free!
|
|
106
|
+
supportsTools: true, // DeepSeek, Llama 4 support function calling
|
|
107
|
+
supportsPromptCaching: false,
|
|
108
|
+
typicalResponseTimeMs: 3000, // ~2-5 seconds on decent hardware
|
|
109
|
+
};
|
|
110
|
+
// SECURITY: Validate and sanitize URL
|
|
111
|
+
const urlValidation = validateOllamaUrl(config.baseUrl);
|
|
112
|
+
if (!urlValidation.valid && urlValidation.warning) {
|
|
113
|
+
logger_js_1.logger.warn(urlValidation.warning);
|
|
114
|
+
}
|
|
115
|
+
// SECURITY: Validate timeout
|
|
116
|
+
const timeout = validateTimeout(config.timeout);
|
|
117
|
+
// Ollama uses OpenAI-compatible API
|
|
118
|
+
this.client = new openai_1.default({
|
|
119
|
+
baseURL: urlValidation.url,
|
|
120
|
+
apiKey: 'ollama', // Ollama doesn't require real API key
|
|
121
|
+
timeout,
|
|
122
|
+
maxRetries: 0, // Don't retry to avoid hanging on connection issues
|
|
123
|
+
});
|
|
124
|
+
const model = config.model || 'deepseek-r1:7b';
|
|
125
|
+
// SECURITY: Validate model name format
|
|
126
|
+
if (!validateModelName(model)) {
|
|
127
|
+
throw new Error('Invalid model name format');
|
|
128
|
+
}
|
|
129
|
+
this.model = model;
|
|
130
|
+
}
|
|
131
|
+
async generateText(prompt, options) {
|
|
132
|
+
this.checkBudget();
|
|
133
|
+
const startTime = Date.now();
|
|
134
|
+
try {
|
|
135
|
+
// SECURITY: Validate prompt length
|
|
136
|
+
if (prompt.length > 10 * 1024 * 1024) {
|
|
137
|
+
throw new Error('Prompt exceeds maximum size (10MB)');
|
|
138
|
+
}
|
|
139
|
+
const messages = [];
|
|
140
|
+
// Add system message if provided
|
|
141
|
+
if (options?.systemPrompt) {
|
|
142
|
+
messages.push({
|
|
143
|
+
role: 'system',
|
|
144
|
+
content: options.systemPrompt,
|
|
145
|
+
});
|
|
146
|
+
}
|
|
147
|
+
// Add user prompt
|
|
148
|
+
messages.push({
|
|
149
|
+
role: 'user',
|
|
150
|
+
content: prompt,
|
|
151
|
+
});
|
|
152
|
+
const response = await (0, provider_utils_js_1.withTimeout)(this.client.chat.completions.create({
|
|
153
|
+
model: this.model,
|
|
154
|
+
messages,
|
|
155
|
+
max_tokens: options?.maxTokens,
|
|
156
|
+
temperature: options?.temperature,
|
|
157
|
+
top_p: options?.topP,
|
|
158
|
+
stop: options?.stopSequences,
|
|
159
|
+
}), options?.timeout, 'generateText');
|
|
160
|
+
const responseTime = Date.now() - startTime;
|
|
161
|
+
const text = response.choices[0]?.message?.content || '';
|
|
162
|
+
const usage = {
|
|
163
|
+
inputTokens: response.usage?.prompt_tokens || 0,
|
|
164
|
+
outputTokens: response.usage?.completion_tokens || 0,
|
|
165
|
+
totalTokens: response.usage?.total_tokens || 0,
|
|
166
|
+
};
|
|
167
|
+
// Update stats
|
|
168
|
+
this.updateStats(usage, responseTime, 0); // Cost is always 0 for Ollama
|
|
169
|
+
return {
|
|
170
|
+
text,
|
|
171
|
+
usage,
|
|
172
|
+
cost: 0, // Free!
|
|
173
|
+
metadata: {
|
|
174
|
+
model: this.model,
|
|
175
|
+
responseTimeMs: responseTime,
|
|
176
|
+
finishReason: response.choices[0]?.finish_reason,
|
|
177
|
+
},
|
|
178
|
+
};
|
|
179
|
+
}
|
|
180
|
+
catch (error) {
|
|
181
|
+
this.stats.failedRequests++;
|
|
182
|
+
throw new provider_interface_js_1.LLMProviderError((0, provider_utils_js_1.sanitizeErrorMessage)(error, 'generateText'), this.name, undefined, error);
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
/**
|
|
186
|
+
* Ollama does not support vision by default
|
|
187
|
+
* This method throws an error to help users understand the limitation
|
|
188
|
+
*/
|
|
189
|
+
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
190
|
+
async analyzeImage(images, prompt, options) {
|
|
191
|
+
throw new provider_interface_js_1.UnsupportedCapabilityError(this.name, 'vision');
|
|
192
|
+
}
|
|
193
|
+
/**
|
|
194
|
+
* Stream text generation for real-time feedback
|
|
195
|
+
*/
|
|
196
|
+
async *streamText(prompt, options) {
|
|
197
|
+
try {
|
|
198
|
+
// SECURITY: Validate prompt length
|
|
199
|
+
if (prompt.length > 10 * 1024 * 1024) {
|
|
200
|
+
throw new Error('Prompt exceeds maximum size (10MB)');
|
|
201
|
+
}
|
|
202
|
+
const messages = [];
|
|
203
|
+
if (options?.systemPrompt) {
|
|
204
|
+
messages.push({
|
|
205
|
+
role: 'system',
|
|
206
|
+
content: options.systemPrompt,
|
|
207
|
+
});
|
|
208
|
+
}
|
|
209
|
+
messages.push({
|
|
210
|
+
role: 'user',
|
|
211
|
+
content: prompt,
|
|
212
|
+
});
|
|
213
|
+
const stream = await (0, provider_utils_js_1.withTimeout)(this.client.chat.completions.create({
|
|
214
|
+
model: this.model,
|
|
215
|
+
messages,
|
|
216
|
+
max_tokens: options?.maxTokens,
|
|
217
|
+
temperature: options?.temperature,
|
|
218
|
+
top_p: options?.topP,
|
|
219
|
+
stop: options?.stopSequences,
|
|
220
|
+
stream: true,
|
|
221
|
+
}), options?.timeout, 'streamText');
|
|
222
|
+
for await (const chunk of stream) {
|
|
223
|
+
const content = chunk.choices[0]?.delta?.content;
|
|
224
|
+
if (content) {
|
|
225
|
+
yield content;
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
// Note: Streaming doesn't provide detailed usage stats
|
|
229
|
+
// We increment request count but can't track exact tokens
|
|
230
|
+
this.stats.requestCount++;
|
|
231
|
+
this.stats.lastUpdated = new Date();
|
|
232
|
+
}
|
|
233
|
+
catch (error) {
|
|
234
|
+
this.stats.failedRequests++;
|
|
235
|
+
throw new provider_interface_js_1.LLMProviderError((0, provider_utils_js_1.sanitizeErrorMessage)(error, 'streamText'), this.name, undefined, error);
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
/**
|
|
239
|
+
* Check if Ollama is running and accessible
|
|
240
|
+
*/
|
|
241
|
+
async checkHealth() {
|
|
242
|
+
try {
|
|
243
|
+
// Try a simple request
|
|
244
|
+
await (0, provider_utils_js_1.withTimeout)(this.client.models.list(), 5000, 'health check');
|
|
245
|
+
return {
|
|
246
|
+
healthy: true,
|
|
247
|
+
message: `Ollama is running with model: ${this.model}`,
|
|
248
|
+
};
|
|
249
|
+
}
|
|
250
|
+
catch (error) {
|
|
251
|
+
return {
|
|
252
|
+
healthy: false,
|
|
253
|
+
message: `Ollama not accessible: ${(0, provider_utils_js_1.sanitizeErrorMessage)(error, 'health check')}`,
|
|
254
|
+
};
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
/**
|
|
258
|
+
* List available models in Ollama
|
|
259
|
+
*/
|
|
260
|
+
async listModels() {
|
|
261
|
+
try {
|
|
262
|
+
const response = await (0, provider_utils_js_1.withTimeout)(this.client.models.list(), 5000, 'listModels');
|
|
263
|
+
return response.data.map((model) => model.id);
|
|
264
|
+
}
|
|
265
|
+
catch (error) {
|
|
266
|
+
throw new provider_interface_js_1.LLMProviderError((0, provider_utils_js_1.sanitizeErrorMessage)(error, 'listModels'), this.name, undefined, error);
|
|
267
|
+
}
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
exports.OllamaProvider = OllamaProvider;
|
|
271
|
+
/**
|
|
272
|
+
* Helper function to check if Ollama is installed and suggest setup
|
|
273
|
+
*/
|
|
274
|
+
async function checkOllamaSetup() {
|
|
275
|
+
const provider = new OllamaProvider({});
|
|
276
|
+
try {
|
|
277
|
+
const health = await provider.checkHealth();
|
|
278
|
+
const models = await provider.listModels();
|
|
279
|
+
return {
|
|
280
|
+
installed: true,
|
|
281
|
+
running: health.healthy,
|
|
282
|
+
modelAvailable: models.length > 0,
|
|
283
|
+
setupInstructions: health.healthy ? 'Ollama is ready to use!' : 'Run: ollama serve',
|
|
284
|
+
};
|
|
285
|
+
}
|
|
286
|
+
catch {
|
|
287
|
+
return {
|
|
288
|
+
installed: false,
|
|
289
|
+
running: false,
|
|
290
|
+
modelAvailable: false,
|
|
291
|
+
setupInstructions: `
|
|
292
|
+
Ollama is not installed. To set up:
|
|
293
|
+
|
|
294
|
+
1. Install Ollama:
|
|
295
|
+
curl -fsSL https://ollama.com/install.sh | sh
|
|
296
|
+
|
|
297
|
+
2. Pull a model (choose one):
|
|
298
|
+
ollama pull deepseek-r1:7b # Recommended: Fast, 4GB RAM
|
|
299
|
+
ollama pull deepseek-r1:14b # Better quality, 8GB RAM
|
|
300
|
+
ollama pull llama4:13b # Alternative, 8GB RAM
|
|
301
|
+
|
|
302
|
+
3. Start Ollama:
|
|
303
|
+
ollama serve
|
|
304
|
+
|
|
305
|
+
For more info: https://ollama.com
|
|
306
|
+
`.trim(),
|
|
307
|
+
};
|
|
308
|
+
}
|
|
309
|
+
}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { GenerateOptions, ImageInput, LLMResponse, OpenAIConfig, ProviderCapabilities } from './provider_interface.js';
|
|
2
|
+
import { BaseProvider } from './base_provider.js';
|
|
3
|
+
export declare class OpenAIProvider extends BaseProvider {
|
|
4
|
+
name: string;
|
|
5
|
+
private client;
|
|
6
|
+
private model;
|
|
7
|
+
capabilities: ProviderCapabilities;
|
|
8
|
+
constructor(config: OpenAIConfig);
|
|
9
|
+
generateText(prompt: string, options?: GenerateOptions): Promise<LLMResponse>;
|
|
10
|
+
analyzeImage(images: ImageInput[], prompt: string, options?: GenerateOptions): Promise<LLMResponse>;
|
|
11
|
+
streamText(prompt: string, options?: GenerateOptions): AsyncGenerator<string, void, unknown>;
|
|
12
|
+
private extractUsage;
|
|
13
|
+
private extractStatusCode;
|
|
14
|
+
checkHealth(): Promise<{
|
|
15
|
+
healthy: boolean;
|
|
16
|
+
message: string;
|
|
17
|
+
}>;
|
|
18
|
+
}
|
|
19
|
+
export declare function checkOpenAISetup(apiKey: string): Promise<{
|
|
20
|
+
valid: boolean;
|
|
21
|
+
message: string;
|
|
22
|
+
}>;
|
|
23
|
+
//# sourceMappingURL=openai_provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai_provider.d.ts","sourceRoot":"","sources":["../src/openai_provider.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACR,eAAe,EACf,UAAU,EACV,WAAW,EACX,YAAY,EACZ,oBAAoB,EACvB,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAC,YAAY,EAAC,MAAM,oBAAoB,CAAC;AAehD,qBAAa,cAAe,SAAQ,YAAY;IAC5C,IAAI,SAAY;IAChB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IAEtB,YAAY,EAAE,oBAAoB,CAAC;gBAEvB,MAAM,EAAE,YAAY;IAyC1B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IA4D7E,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC;IAgGlG,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC;IA8CnG,OAAO,CAAC,YAAY;IAYpB,OAAO,CAAC,iBAAiB;IAWnB,WAAW,IAAI,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAC,CAAC;CAuBpE;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IAC5D,KAAK,EAAE,OAAO,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACnB,CAAC,CAsBD"}
|