adaptive-memory-multi-model-router 2.14.59 → 2.15.0

package/src/server/router.ts

@@ -0,0 +1,157 @@
+/**
+ * A3M Router - HTTP Route Registry
+ *
+ * Pluggable route table pattern replacing monolithic if/else routing.
+ * Adding a new endpoint = 1 handler file + 1 registration line.
+ *
+ * Usage:
+ *   import { createRouter, registerRoute } from './router';
+ *   registerRoute('GET', /^\/health$/, handleHealth);
+ *   registerRoute('POST', /^\/v1\/chat\/completions$/, handleChatCompletions);
+ *   const router = createRouter();
+ *   server.use(router);
+ */
+
+import * as http from 'http';
+
+// ============================================================
+// TYPES
+// ============================================================
+
+export type HTTPMethod = 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH' | 'OPTIONS';
+
+export interface RouteContext {
+  requestId: string;
+  startTime: number;
+  method: HTTPMethod;
+  path: string;
+  query: Record<string, string>;
+}
+
+export type RouteHandler = (
+  req: http.IncomingMessage,
+  res: http.ServerResponse,
+  ctx: RouteContext
+) => Promise<void> | void;
+
+export interface Route {
+  method: HTTPMethod;
+  pattern: RegExp;
+  handler: RouteHandler;
+  name: string;
+}
+
+// ============================================================
+// ROUTE REGISTRY
+// ============================================================
+
+const routes: Route[] = [];
+
+/**
+ * Generate a short request ID.
+ */
+function generateRequestId(): string {
+  return 'req-' + Math.random().toString(36).substring(2, 10) + Date.now().toString(36);
+}
+
+/**
+ * Parse query string from URL.
+ */
+function parseQuery(url: string): Record<string, string> {
+  try {
+    const urlObj = new URL(url, 'http://localhost');
+    const query: Record<string, string> = {};
+    urlObj.searchParams.forEach((value, key) => {
+      query[key] = value;
+    });
+    return query;
+  } catch {
+    return {};
+  }
+}
+
+/**
+ * Parse HTTP method safely.
+ */
+function parseMethod(method: string | undefined): HTTPMethod {
+  const m = (method || 'GET').toUpperCase();
+  if (['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS'].includes(m)) {
+    return m as HTTPMethod;
+  }
+  return 'GET';
+}
+
+/**
+ * Register a route. Routes are matched in registration order.
+ */
+export function registerRoute(
+  method: HTTPMethod,
+  pattern: RegExp,
+  handler: RouteHandler,
+  name?: string
+): void {
+  routes.push({ method, pattern, handler, name: name || `${method} ${pattern}` });
+}
+
+/**
+ * Create the request handler function for the registered routes.
+ * Returns a function compatible with http.createServer's requestListener.
+ */
+export function createRequestHandler(): (req: http.IncomingMessage, res: http.ServerResponse) => Promise<void> {
+  return async function handleRequest(req: http.IncomingMessage, res: http.ServerResponse): Promise<void> {
+    const method = parseMethod(req.method);
+    const url = req.url || '/';
+    const ctx: RouteContext = {
+      requestId: generateRequestId(),
+      startTime: Date.now(),
+      method,
+      path: url.split('?')[0],
+      query: parseQuery(url),
+    };
+
+    // CORS preflight
+    if (method === 'OPTIONS') {
+      res.writeHead(204, {
+        'Access-Control-Allow-Origin': '*',
+        'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, PATCH, OPTIONS',
+        'Access-Control-Allow-Headers': 'Content-Type, Authorization, x-request-id',
+        'Access-Control-Max-Age': '86400',
+      });
+      res.end();
+      return;
+    }
+
+    // Find matching route
+    for (const route of routes) {
+      if (route.method !== method) continue;
+      const match = ctx.path.match(route.pattern);
+      if (match) {
+        try {
+          await route.handler(req, res, ctx);
+        } catch (err: any) {
+          console.error(`[a3m-router] Route ${route.name} error: ${err.message}`);
+          if (!res.headersSent) {
+            res.writeHead(500, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
+            res.end(JSON.stringify({ error: { message: err.message, type: 'server_error' } }));
+          }
+        }
+        return;
+      }
+    }
+
+    // No route matched — 404
+    if (!res.headersSent) {
+      res.writeHead(404, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
+      res.end(JSON.stringify({ error: { message: `Not found: ${method} ${url}`, type: 'not_found' } }));
+    }
+  };
+}
+
+/**
+ * Get all registered routes (for debugging and /routes endpoint).
+ */
+export function getRegisteredRoutes(): Route[] {
+  return [...routes];
+}
+
+export default { registerRoute, createRequestHandler, getRegisteredRoutes };

package/src/server/state.ts

@@ -0,0 +1,34 @@
+/**
+ * A3M Router - Shared Server State
+ *
+ * Singleton state shared across the proxy server and all handlers.
+ * Extracted from proxyServer.ts to enable modular handler architecture.
+ */
+
+import { CostTracker } from '../cost/costTracker';
+
+// ============================================================
+// SHARED STATE
+// ============================================================
+
+/** Singleton cost tracker instance */
+export const costTracker = new CostTracker();
+
+/** In-memory request log (last 1000 entries) */
+export interface RequestLog {
+  id: string;
+  model: string;
+  resolvedProvider: string;
+  resolvedModel: string;
+  latencyMs: number;
+  tokensIn: number;
+  tokensOut: number;
+  cost: number;
+  status: 'success' | 'error';
+  error?: string;
+  timestamp: number;
+}
+
+export const requestLogs: RequestLog[] = [];
+
+export default { costTracker, requestLogs };

package/src/skills/__tests__/skill_manager.test.ts

@@ -1,6 +1,6 @@
-"""
-Tests for SkillManager and TMLEnhancedAgent
-"""
+/**
+ * Tests for SkillManager and TMLEnhancedAgent
+ */
 
 import { describe, it, expect, beforeEach, jest } from '@jest/globals';
 import { SkillManager, Skill } from '../skill_manager';