examples/query-router.mjs

Rama master · réplica de lectura del repositorio original.

#!/usr/bin/env node
// Example: QueryRouter — one-call grounded Q&A over a markdown corpus.
//
// Point the router at your docs directory, call route(), and get back a
// fully-attributed answer (text + sources + classification metadata).
//
// This script demonstrates:
//   - Initializing a corpus from local markdown directories
//   - The three-stage pipeline (classify -> retrieve -> generate)
//   - Lifecycle hooks for observability (onClassification, onRetrieval)
//   - Inspecting the result shape: answer, classification, sources,
//     tiersUsed, fallbacksUsed
//
// Usage:
//   export OPENAI_API_KEY="sk-..."   # Required (used for embeddings + LLM)
//   node examples/query-router.mjs
//
// Note: If OPENROUTER_API_KEY is also set, it may be auto-detected first.
// OpenRouter doesn't support embeddings, so unset it or ensure OPENAI_API_KEY is set.

import { QueryRouter } from '../dist/index.js';

async function main() {
  const router = new QueryRouter({
    knowledgeCorpus: ['./docs', './packages/agentos/docs'],
    availableTools: ['web_search', 'deep_research'],
    onClassification: (result) => {
      console.log(
        `[classification] tier=${result.tier} strategy=${result.strategy} confidence=${result.confidence.toFixed(2)}`
      );
    },
    onRetrieval: (result) => {
      console.log(
        `[retrieval] chunks=${result.chunks.length} durationMs=${result.durationMs}`
      );
    },
  });

  await router.init();
  console.log('\n=== corpus stats ===\n');
  console.log(router.getCorpusStats());

  const result = await router.route(
    'How does AgentOS memory retrieval work, and when does it fall back to keyword search?'
  );

  console.log('\n=== answer ===\n');
  console.log(result.answer);

  console.log('\n=== routing metadata ===\n');
  console.log('classified tier:', result.classification.tier);
  console.log('tiers used:', result.tiersUsed);
  console.log('fallbacks used:', result.fallbacksUsed);

  console.log('\n=== sources ===\n');
  for (const source of result.sources) {
    console.log(`- ${source.heading} (${source.path})`);
  }

  await router.close();
}

main().catch((error) => {
  console.error(error);
  process.exitCode = 1;
});