pi-smart-router 0.1.0

package/.pi/extensions/smart-router/index.ts

@@ -0,0 +1,115 @@
+/**
+ * pi-smart-router project-local extension.
+ *
+ * Discovers authenticated models from pi's model registry, maps them to a
+ * router fleet, registers the smart-router/auto provider, and wires middleware
+ * hooks for routing state. Stream delegation routes each request through the
+ * pipeline and forwards to the selected provider's built-in streaming API.
+ *
+ * Imports from ../../../src/** (not dist/) because the extension is loaded by pi
+ * from source at dev time and is excluded from the npm dist artifact.
+ */
+
+import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
+
+import { getSmartRouterArgumentCompletions } from './commands.js';
+import {
+  createExtensionDatasetRecorder,
+  createExtensionOutcomeRecorder,
+  exportDatasetToFile,
+  formatDatasetExportJsonl,
+  formatDatasetExportTimestamp,
+  getDatasetExportPath,
+  toDatasetExportRecord,
+} from './dataset-export.js';
+import {
+  createDispatchOptions,
+  discoverFleet,
+  formatLmuStatus,
+  initHydraMatcher,
+  bindSharedModelRegistry,
+  computeCurrentFleetScopeFingerprint,
+  computeFleetScopeFingerprint,
+  ensureFleetFresh,
+  rebuildFleet,
+} from './fleet-bootstrap.js';
+import {
+  formatHistoryMessage,
+  formatStatusMessage,
+  parseSmartRouterArgs,
+} from './command-formatters.js';
+import { formatPricingStalenessLine, refreshPricingCatalog } from './pricing-lifecycle.js';
+import {
+  buildRoutingRequest,
+  deriveTurnType,
+  extractPromptText,
+  mapContextMessages,
+} from './routing-context.js';
+import { capturePreRouteOutcomes, updateSessionRoutingSnapshot } from './routing-outcomes.js';
+import {
+  buildDelegationContext,
+  createStreamSimple,
+  getRoutingFeatureSidecar,
+  logRoutingDecision,
+  resolveDelegationOptions,
+} from './stream-delegation.js';
+import { createSmartRouterRuntime, wireSmartRouterExtension } from './extension-setup.js';
+import { getRouterStateDbPath } from './utils.js';
+
+export {
+  buildRoutingRequest,
+  buildDelegationContext,
+  createDispatchOptions,
+  createExtensionDatasetRecorder,
+  createExtensionOutcomeRecorder,
+  createSmartRouterRuntime,
+  createStreamSimple,
+  deriveTurnType,
+  discoverFleet,
+  bindSharedModelRegistry,
+  computeCurrentFleetScopeFingerprint,
+  computeFleetScopeFingerprint,
+  ensureFleetFresh,
+  rebuildFleet,
+  exportDatasetToFile,
+  extractPromptText,
+  formatDatasetExportJsonl,
+  formatDatasetExportTimestamp,
+  formatLmuStatus,
+  formatPricingStalenessLine,
+  formatHistoryMessage,
+  formatStatusMessage,
+  getDatasetExportPath,
+  getRouterStateDbPath,
+  getRoutingFeatureSidecar,
+  getSmartRouterArgumentCompletions,
+  mapContextMessages,
+  parseSmartRouterArgs,
+  refreshPricingCatalog,
+  resolveDelegationOptions,
+  logRoutingDecision,
+  toDatasetExportRecord,
+  capturePreRouteOutcomes,
+  updateSessionRoutingSnapshot,
+  initHydraMatcher,
+  wireSmartRouterExtension,
+};
+export { SMART_ROUTER_FULL_INVOCATIONS, SMART_ROUTER_USAGE } from './commands.js';
+export { routeAndDelegate } from './route-and-delegate.js';
+export {
+  formatGeminiThoughtSignatureErrorMessage,
+  isGeminiThoughtSignatureAssistantError,
+} from '../../../src/infrastructure/delegation/provider-error.js';
+export {
+  GEMINI_TOOL_HISTORY_EXCLUDED,
+  hasToolCallHistory,
+  hasToolCallHistoryFromContext,
+  isGoogleGeminiProfile,
+  resolveEffectiveFleet,
+} from '../../../src/domain/routing/tool-history-guard.js';
+
+export default async function smartRouterExtension(pi: ExtensionAPI): Promise<void> {
+  const cwd = process.cwd();
+  const { runtime, datasetNotify } = await createSmartRouterRuntime(cwd);
+  await wireSmartRouterExtension(pi, runtime, datasetNotify);
+}

package/.pi/extensions/smart-router/package.json

@@ -0,0 +1,3 @@
+{
+  "type": "module"
+}

package/.pi/extensions/smart-router/pi-model-scope.ts

@@ -0,0 +1,55 @@
+import { existsSync } from 'node:fs';
+import { dirname, join } from 'node:path';
+import { fileURLToPath, pathToFileURL } from 'node:url';
+
+import type { ModelRegistry } from '@earendil-works/pi-coding-agent';
+import type { Api, Model } from '@earendil-works/pi-ai/compat';
+
+export interface ScopedModel {
+  model: Model<Api>;
+  thinkingLevel?: 'off' | 'minimal' | 'low' | 'medium' | 'high' | 'xhigh';
+}
+
+type ResolveModelScopeFn = (
+  patterns: string[],
+  modelRegistry: ModelRegistry,
+) => Promise<ScopedModel[]>;
+
+function findPiCodingAgentDir(): string {
+  const candidates: string[] = [];
+  const roots = [dirname(fileURLToPath(import.meta.url)), process.cwd()];
+
+  for (const start of roots) {
+    let dir = start;
+    for (let i = 0; i < 12; i++) {
+      candidates.push(join(dir, 'node_modules/@earendil-works/pi-coding-agent'));
+      const parent = dirname(dir);
+      if (parent === dir) {
+        break;
+      }
+      dir = parent;
+    }
+  }
+
+  for (const candidate of candidates) {
+    if (existsSync(join(candidate, 'dist/core/model-resolver.js'))) {
+      return candidate;
+    }
+  }
+
+  throw new Error('Unable to locate @earendil-works/pi-coding-agent for resolveModelScope');
+}
+
+const piPkgDir = findPiCodingAgentDir();
+const modelResolver = await import(
+  pathToFileURL(join(piPkgDir, 'dist/core/model-resolver.js')).href
+);
+const piResolveModelScope = modelResolver.resolveModelScope as ResolveModelScopeFn;
+
+/** Delegate to pi's resolveModelScope (not yet on the public package export surface). */
+export function resolveModelScope(
+  patterns: string[],
+  modelRegistry: ModelRegistry,
+): Promise<ScopedModel[]> {
+  return piResolveModelScope(patterns, modelRegistry);
+}

package/.pi/extensions/smart-router/pricing-lifecycle.ts

@@ -0,0 +1,43 @@
+import { DEFAULT_OPERATOR_CONFIG } from '../../../src/config/defaults.js';
+import type { PriceCatalog } from '../../../src/domain/types/index.js';
+import { fetchLitellmPriceCatalog } from '../../../src/infrastructure/pricing/litellm-fetch.js';
+import { checkStaleness } from '../../../src/infrastructure/pricing/pricing-monitor.js';
+import type { SmartRouterRuntime } from './types.js';
+
+export async function refreshPricingCatalog(
+  runtime: SmartRouterRuntime,
+  fetchFn?: typeof fetch,
+): Promise<{ modelCount: number; lastUpdated: string }> {
+  const existing = await runtime.store.getPriceCatalog();
+  const { catalog: fetched, model_count: modelCount } = await fetchLitellmPriceCatalog(
+    fetchFn ? { fetchFn } : {},
+  );
+
+  const catalog: PriceCatalog = {
+    ...fetched,
+    user_overrides: existing?.user_overrides ?? {},
+  };
+
+  await runtime.store.putPriceCatalog(catalog);
+  runtime.priceCatalog = catalog;
+
+  return { modelCount, lastUpdated: catalog.last_updated };
+}
+
+export function formatPricingStalenessLine(catalog: PriceCatalog | null): string | undefined {
+  const staleness = checkStaleness(
+    catalog,
+    DEFAULT_OPERATOR_CONFIG.pricing.staleness_days,
+  );
+  return staleness.warning;
+}
+
+export function notifyPricingStalenessIfNeeded(
+  runtime: SmartRouterRuntime,
+  notify: (message: string, level: 'info' | 'warning' | 'error') => void,
+): void {
+  const stalenessLine = formatPricingStalenessLine(runtime.priceCatalog);
+  if (stalenessLine) {
+    notify(stalenessLine, 'warning');
+  }
+}