pi-harness-runtime 0.5.0 → 0.6.3-beta.13

package/packages/model-registry/src/index.ts

@@ -0,0 +1,57 @@
+/**
+ * Model Registry Package (RFC-0053)
+ *
+ * A centralized registry for model metadata with pricing.
+ *
+ * @example
+ * ```typescript
+ * import { createModelRegistry } from "@pi-harness/model-registry";
+ *
+ * const registry = createModelRegistry();
+ *
+ * // Get a model
+ * const model = registry.get("anthropic", "claude-sonnet-4");
+ *
+ * // Find by alias
+ * const byAlias = registry.getByAlias("sonnet");
+ *
+ * // Find active models with capabilities
+ * const models = registry.find({
+ *   capabilities: ["code_generation"],
+ *   status: "active",
+ * });
+ * ```
+ */
+
+// Types
+export type {
+	Currency,
+	Pricing,
+	ModelStatus,
+	ModelInfo,
+	ModelFilters,
+	ModelRegistry,
+	ModelRegistryEvent,
+} from "./types.js";
+
+// Registry
+export { InMemoryModelRegistry, createModelRegistry } from "./registry.js";
+
+// Defaults
+export { DEFAULT_MODELS } from "./defaults.js";
+
+// Query utilities
+export {
+	findModels,
+	getCheapestModel,
+	getLargestContextModel,
+	filterByCostEfficiency,
+} from "./query.js";
+
+// Cost utilities
+export {
+	calculateCost,
+	getCostPer1M,
+	compareCost,
+	calculateSavings,
+} from "./cost.js";

package/packages/model-registry/src/query.ts

@@ -0,0 +1,85 @@
+/**
+ * Model Query Functions (RFC-0053)
+ */
+
+import type { ModelFilters, ModelInfo } from "./types.js";
+import { getCostPer1M } from "./cost.js";
+
+/**
+ * Find models matching filters
+ */
+export function findModels(
+	models: ModelInfo[],
+	filters: ModelFilters,
+): ModelInfo[] {
+	let result = [...models];
+
+	// Filter by provider
+	if (filters.providerId) {
+		result = result.filter((m) => m.providerId === filters.providerId);
+	}
+
+	// Filter by context window
+	if (filters.minContextWindow !== undefined) {
+		result = result.filter((m) => m.contextWindow >= filters.minContextWindow!);
+	}
+
+	// Filter by cost
+	if (filters.maxCostPer1M !== undefined) {
+		result = result.filter(
+			(m) => getCostPer1M(m.pricing) <= filters.maxCostPer1M!,
+		);
+	}
+
+	// Filter by capabilities
+	if (filters.capabilities?.length) {
+		result = result.filter((m) =>
+			filters.capabilities!.every((c) => m.capabilities.includes(c)),
+		);
+	}
+
+	// Filter by status
+	if (filters.status) {
+		result = result.filter((m) => m.status === filters.status);
+	}
+
+	// Sort by cost (ascending)
+	result.sort((a, b) => getCostPer1M(a.pricing) - getCostPer1M(b.pricing));
+
+	return result;
+}
+
+/**
+ * Get the cheapest model
+ */
+export function getCheapestModel(models: ModelInfo[]): ModelInfo | undefined {
+	if (models.length === 0) return undefined;
+	return models.reduce((cheapest, current) =>
+		getCostPer1M(current.pricing) < getCostPer1M(cheapest.pricing)
+			? current
+			: cheapest,
+	);
+}
+
+/**
+ * Get the model with largest context window
+ */
+export function getLargestContextModel(
+	models: ModelInfo[],
+): ModelInfo | undefined {
+	if (models.length === 0) return undefined;
+	return models.reduce((largest, current) =>
+		current.contextWindow > largest.contextWindow ? current : largest,
+	);
+}
+
+/**
+ * Filter by cost efficiency (cost per context window)
+ */
+export function filterByCostEfficiency(models: ModelInfo[]): ModelInfo[] {
+	return [...models].sort((a, b) => {
+		const efficiencyA = getCostPer1M(a.pricing) / a.contextWindow;
+		const efficiencyB = getCostPer1M(b.pricing) / b.contextWindow;
+		return efficiencyA - efficiencyB;
+	});
+}

package/packages/model-registry/src/registry.ts

@@ -0,0 +1,188 @@
+/**
+ * Model Registry Implementation (RFC-0053)
+ */
+
+import type {
+	ModelFilters,
+	ModelInfo,
+	ModelRegistry,
+	ModelRegistryEvent,
+	ModelStatus,
+} from "./types.js";
+import { DEFAULT_MODELS } from "./defaults.js";
+import { findModels } from "./query.js";
+
+type EventHandler = (event: ModelRegistryEvent) => void;
+
+/**
+ * In-memory model registry with event emission
+ */
+export class InMemoryModelRegistry implements ModelRegistry {
+	private models: Map<string, ModelInfo> = new Map();
+	private aliasIndex: Map<string, ModelInfo> = new Map();
+	private eventHandlers: Set<EventHandler> = new Set();
+
+	constructor(loadDefaults = true) {
+		if (loadDefaults) {
+			this.loadDefaults();
+		}
+	}
+
+	/**
+	 * Load default models
+	 */
+	private loadDefaults(): void {
+		for (const model of DEFAULT_MODELS) {
+			this.register(model);
+		}
+	}
+
+	/**
+	 * Create a unique key for provider/model pair
+	 */
+	private makeKey(providerId: string, modelId: string): string {
+		return `${providerId}::${modelId}`;
+	}
+
+	/**
+	 * Register a model
+	 */
+	register(model: ModelInfo): void {
+		const key = this.makeKey(model.providerId, model.id);
+		this.models.set(key, model);
+
+		// Index by aliases
+		for (const alias of model.aliases) {
+			this.aliasIndex.set(alias.toLowerCase(), model);
+		}
+
+		this.emit({
+			type: "model.registered",
+			providerId: model.providerId,
+			modelId: model.id,
+		});
+	}
+
+	/**
+	 * Unregister a model
+	 */
+	unregister(providerId: string, modelId: string): void {
+		const key = this.makeKey(providerId, modelId);
+		const model = this.models.get(key);
+
+		if (model) {
+			// Remove from alias index
+			for (const alias of model.aliases) {
+				this.aliasIndex.delete(alias.toLowerCase());
+			}
+			this.models.delete(key);
+		}
+
+		this.emit({
+			type: "model.unregistered",
+			providerId,
+			modelId,
+		});
+	}
+
+	/**
+	 * Get a model by provider and ID
+	 */
+	get(providerId: string, modelId: string): ModelInfo | undefined {
+		const key = this.makeKey(providerId, modelId);
+		return this.models.get(key);
+	}
+
+	/**
+	 * Get a model by alias
+	 */
+	getByAlias(alias: string): ModelInfo | undefined {
+		return this.aliasIndex.get(alias.toLowerCase());
+	}
+
+	/**
+	 * List all models, optionally filtered by provider
+	 */
+	list(providerId?: string): ModelInfo[] {
+		const all = Array.from(this.models.values());
+		if (providerId) {
+			return all.filter((m) => m.providerId === providerId);
+		}
+		return all;
+	}
+
+	/**
+	 * List only active models
+	 */
+	listActive(): ModelInfo[] {
+		return Array.from(this.models.values()).filter(
+			(m) => m.status === "active",
+		);
+	}
+
+	/**
+	 * Find models matching filters
+	 */
+	find(filters: ModelFilters): ModelInfo[] {
+		const results = findModels(this.list(), filters);
+
+		this.emit({
+			type: "model.queried",
+			filters,
+			results: results.length,
+		});
+
+		return results;
+	}
+
+	/**
+	 * Update model status
+	 */
+	updateStatus(providerId: string, modelId: string, status: ModelStatus): void {
+		const key = this.makeKey(providerId, modelId);
+		const model = this.models.get(key);
+
+		if (model) {
+			model.status = status;
+			if (status === "deprecated") {
+				model.deprecatedAt = new Date().toISOString();
+			}
+			this.emit({
+				type: "model.status_changed",
+				providerId,
+				modelId,
+				status,
+			});
+		}
+	}
+
+	/**
+	 * Subscribe to events
+	 */
+	onEvent(handler: EventHandler): () => void {
+		this.eventHandlers.add(handler);
+		return () => this.eventHandlers.delete(handler);
+	}
+
+	/**
+	 * Emit an event
+	 */
+	private emit(event: ModelRegistryEvent): void {
+		for (const handler of this.eventHandlers) {
+			try {
+				handler(event);
+			} catch {
+				// Ignore handler errors
+			}
+		}
+	}
+}
+
+/**
+ * Create a new model registry
+ */
+export function createModelRegistry(
+	loadDefaults = true,
+): InMemoryModelRegistry {
+	return new InMemoryModelRegistry(loadDefaults);
+}

package/packages/model-registry/src/types.d.ts

@@ -0,0 +1,63 @@
+/**
+ * Model Registry Types (RFC-0053)
+ */
+export type Currency = "USD" | "THB";
+export interface Pricing {
+    currency: Currency;
+    inputPer1M: number;
+    outputPer1M: number;
+    batchInputPer1M?: number;
+    cacheReadPer1M?: number;
+    cacheWritePer1M?: number;
+}
+export type ModelStatus = "active" | "deprecated" | "disabled";
+export interface ModelInfo {
+    id: string;
+    providerId: string;
+    name: string;
+    version?: string;
+    contextWindow: number;
+    maxOutputTokens: number;
+    pricing: Pricing;
+    capabilities: string[];
+    aliases: string[];
+    status: ModelStatus;
+    releasedAt?: string;
+    deprecatedAt?: string;
+}
+export interface ModelFilters {
+    providerId?: string;
+    minContextWindow?: number;
+    maxCostPer1M?: number;
+    capabilities?: string[];
+    status?: ModelStatus;
+}
+export interface ModelRegistry {
+    register(model: ModelInfo): void;
+    unregister(providerId: string, modelId: string): void;
+    get(providerId: string, modelId: string): ModelInfo | undefined;
+    getByAlias(alias: string): ModelInfo | undefined;
+    list(providerId?: string): ModelInfo[];
+    listActive(): ModelInfo[];
+    find(filters: ModelFilters): ModelInfo[];
+    updateStatus(providerId: string, modelId: string, status: ModelStatus): void;
+}
+export type ModelRegistryEvent = {
+    type: "model.registered";
+    providerId: string;
+    modelId: string;
+} | {
+    type: "model.unregistered";
+    providerId: string;
+    modelId: string;
+} | {
+    type: "model.status_changed";
+    providerId: string;
+    modelId: string;
+    status: ModelStatus;
+} | {
+    type: "model.queried";
+    filters: ModelFilters;
+    results: number;
+};
+//# sourceMappingURL=types.d.ts.map

package/packages/model-registry/src/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,MAAM,MAAM,QAAQ,GAAG,KAAK,GAAG,KAAK,CAAC;AAErC,MAAM,WAAW,OAAO;IACvB,QAAQ,EAAE,QAAQ,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CACzB;AAID,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,YAAY,GAAG,UAAU,CAAC;AAE/D,MAAM,WAAW,SAAS;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB;AAID,MAAM,WAAW,YAAY;IAC5B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,MAAM,CAAC,EAAE,WAAW,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC7B,QAAQ,CAAC,KAAK,EAAE,SAAS,GAAG,IAAI,CAAC;IACjC,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACtD,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAAC;IAChE,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAAC;IACjD,IAAI,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,EAAE,CAAC;IACvC,UAAU,IAAI,SAAS,EAAE,CAAC;IAC1B,IAAI,CAAC,OAAO,EAAE,YAAY,GAAG,SAAS,EAAE,CAAC;IACzC,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,IAAI,CAAC;CAC7E;AAID,MAAM,MAAM,kBAAkB,GAC3B;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACjE;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACnE;IACA,IAAI,EAAE,sBAAsB,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,WAAW,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,OAAO,EAAE,YAAY,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC"}

package/packages/model-registry/src/types.js

@@ -0,0 +1,5 @@
+/**
+ * Model Registry Types (RFC-0053)
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/packages/model-registry/src/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/packages/model-registry/src/types.ts

@@ -0,0 +1,69 @@
+/**
+ * Model Registry Types (RFC-0053)
+ */
+
+// ─── Pricing Types ───────────────────────────────────────────────────
+
+export type Currency = "USD" | "THB";
+
+export interface Pricing {
+	currency: Currency;
+	inputPer1M: number; // cost per 1M input tokens
+	outputPer1M: number; // cost per 1M output tokens
+	batchInputPer1M?: number;
+	cacheReadPer1M?: number;
+	cacheWritePer1M?: number;
+}
+
+// ─── Model Info ───────────────────────────────────────────────────────
+
+export type ModelStatus = "active" | "deprecated" | "disabled";
+
+export interface ModelInfo {
+	id: string;
+	providerId: string;
+	name: string;
+	version?: string;
+	contextWindow: number;
+	maxOutputTokens: number;
+	pricing: Pricing;
+	capabilities: string[]; // references to Capability IDs
+	aliases: string[];
+	status: ModelStatus;
+	releasedAt?: string;
+	deprecatedAt?: string;
+}
+
+// ─── Registry Types ───────────────────────────────────────────────────
+
+export interface ModelFilters {
+	providerId?: string;
+	minContextWindow?: number;
+	maxCostPer1M?: number;
+	capabilities?: string[];
+	status?: ModelStatus;
+}
+
+export interface ModelRegistry {
+	register(model: ModelInfo): void;
+	unregister(providerId: string, modelId: string): void;
+	get(providerId: string, modelId: string): ModelInfo | undefined;
+	getByAlias(alias: string): ModelInfo | undefined;
+	list(providerId?: string): ModelInfo[];
+	listActive(): ModelInfo[];
+	find(filters: ModelFilters): ModelInfo[];
+	updateStatus(providerId: string, modelId: string, status: ModelStatus): void;
+}
+
+// ─── Event Types ─────────────────────────────────────────────────────
+
+export type ModelRegistryEvent =
+	| { type: "model.registered"; providerId: string; modelId: string }
+	| { type: "model.unregistered"; providerId: string; modelId: string }
+	| {
+			type: "model.status_changed";
+			providerId: string;
+			modelId: string;
+			status: ModelStatus;
+	  }
+	| { type: "model.queried"; filters: ModelFilters; results: number };