@mupt-ai/dari-gateway 0.1.0 → 0.2.1

package/src/runtime/routed_completion.ts

@@ -1,16 +1,16 @@
 import { randomUUID } from "node:crypto";
 
-import { isBackgroundRequest, type InteractionRequest } from "../interaction/types";
-import { explicitModelId } from "../protocol/input/router_model";
-import { captureRequestError } from "../observability/errors";
-import type { SelectorBackend } from "../strategies/routing";
-import type { ClientHarness } from "../types";
-import { assertAutoroutingAllowed, assertCreditAvailableBeforeRouting, routerSelectsModels } from "../telemetry/usage";
-import type { HarnessConversationIdentity } from "./harness_identity";
-import { CustomerDisconnectedError } from "./customer_disconnect";
-import { prepareRoutedCompletion } from "./prepare_routed_completion";
-import type { RoutedProtocol } from "./routed_protocol";
-import { runPreparedRoutedCompletion } from "./routed_runtime";
+import { isBackgroundRequest, type InteractionRequest } from "../interaction/types.js";
+import { explicitModelId } from "../protocol/input/router_model.js";
+import { captureRequestError } from "../observability/errors.js";
+import type { SelectorBackend } from "../strategies/routing.js";
+import type { ClientHarness } from "../types.js";
+import { assertAutoroutingAllowed, assertCreditAvailableBeforeRouting, routerSelectsModels } from "../telemetry/usage.js";
+import type { HarnessConversationIdentity } from "./harness_identity/index.js";
+import { CustomerDisconnectedError } from "./customer_disconnect.js";
+import { prepareRoutedCompletion } from "./prepare_routed_completion.js";
+import type { RoutedProtocol } from "./routed_protocol.js";
+import { runPreparedRoutedCompletion } from "./routed_runtime.js";
 
 export async function handleRoutedCompletion(args: {
   request: Request;
@@ -37,6 +37,7 @@ export async function handleRoutedCompletion(args: {
     startedMs,
     requestId,
     clientRequestId,
+    tagsHeader: args.request.headers.get("x-dari-tags"),
     clientHarness: args.clientHarness,
     harnessConversationIdentity: args.harnessConversationIdentity,
   }).catch((error: unknown) => ({ error }));
@@ -56,6 +57,7 @@ export async function handleRoutedCompletion(args: {
     // Only auto-selection needs the plan: an explicitly requested model and
     // background work are served without the selector.
     const selects =
+      prepared.credentials.strict !== true &&
       explicitModelId(prepared.interaction.model) === null &&
       !isBackgroundRequest(prepared.interaction) &&
       routerSelectsModels(prepared.credentials);

package/src/runtime/routed_protocol.ts

@@ -1,6 +1,6 @@
 import type { RoutingDecision } from "@mupt-ai/dari-router/policy-engine";
-import type { InteractionOutput, InteractionRequest, ProviderIdentity } from "../interaction/types";
-import type { ProviderStreamOutcome } from "./provider_stream_outcome";
+import type { InteractionOutput, InteractionRequest, ProviderIdentity } from "../interaction/types.js";
+import type { ProviderStreamOutcome } from "./provider_stream_outcome.js";
 
 export type RoutedCompletionResponseArgs = {
   id: string;

package/src/runtime/routed_runtime.ts

@@ -8,8 +8,8 @@ import {
   type RoutingDecision,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import { buildRoutingContext } from "../context/routing_context";
-import type { InteractionRequest } from "../interaction/types";
+import { buildRoutingContext } from "../context/routing_context.js";
+import type { InteractionRequest } from "../interaction/types.js";
 import {
   completeNonStreaming,
   executeNonStreaming,
@@ -17,7 +17,7 @@ import {
   prepareStreamingExecution,
   type CompletedNonStreamingExecution,
   type ProviderChoice,
-} from "../providers/execution";
+} from "../providers/execution.js";
 import {
   prepareProviderExecution,
   startProviderCompletion,
@@ -25,18 +25,18 @@ import {
   type ProviderExecution,
   type StartedCompletion,
   type StartedStream,
-} from "../providers/provider_call";
-import type { SelectorBackend } from "../strategies/routing";
-import { routeDecision, type RoutedDecision } from "../strategies/route_decision";
-import { executionProviderForModel } from "../models/execution_provider";
-import { trackBackgroundTelemetry } from "../telemetry/background";
-import { estimateBenchmarkReplay } from "../telemetry/benchmark_replay";
-import { buildCompletedPrefixHit } from "../telemetry/prefix_state";
-import { scrubSecrets } from "../telemetry/scrub";
-import type { ResolvedRouter } from "../types";
-import type { PreparedRoutedCompletion } from "./prepare_routed_completion";
+} from "../providers/provider_call.js";
+import type { SelectorBackend } from "../strategies/routing.js";
+import { routeDecision, type RoutedDecision } from "../strategies/route_decision.js";
+import { executionProviderForModel } from "../models/execution_provider.js";
+import { trackBackgroundTelemetry } from "../telemetry/background.js";
+import { estimateBenchmarkReplay } from "../telemetry/benchmark_replay.js";
+import { buildCompletedPrefixHit } from "../telemetry/prefix_state.js";
+import { scrubSecrets } from "../telemetry/scrub.js";
+import type { ResolvedRouter } from "../types.js";
+import type { PreparedRoutedCompletion } from "./prepare_routed_completion.js";
 import { auxiliaryDecision } from "./claude_code/auxiliary_routing.js";
-import { explicitModelDecision } from "./explicit_model.js";
+import { explicitModelDecision, strictModelDecision } from "./explicit_model.js";
 import {
   clearPendingLease,
   committedLeaseDecision,
@@ -47,12 +47,12 @@ import {
   prefetchNextLease,
   selectorNegotiatesLeases,
 } from "./routing_lease.js";
-import type { RoutedProtocol } from "./routed_protocol";
+import type { RoutedProtocol } from "./routed_protocol.js";
 import {
   createExperimentRoutedRuntimeReporter,
   type RoutedRuntimeReporter,
-} from "./runtime_reporter";
-import { HttpError } from "../http/errors";
+} from "./runtime_reporter.js";
+import { HttpError } from "../http/errors.js";
 
 type SpeculativeExecution = {
   decision: RoutingDecision;
@@ -113,12 +113,12 @@ export type {
   SelectorAdapterArgs,
   SelectorBackend,
   SelectorRun,
-} from "../strategies/routing";
+} from "../strategies/routing.js";
 export type {
   ExperimentRoutedRuntimeReporter,
   RoutedRuntimeReporter,
-} from "./runtime_reporter";
-export { createExperimentRoutedRuntimeReporter } from "./runtime_reporter";
+} from "./runtime_reporter.js";
+export { createExperimentRoutedRuntimeReporter } from "./runtime_reporter.js";
 
 export async function runPreparedRoutedCompletion(args: {
   prepared: PreparedRoutedCompletion;
@@ -238,6 +238,7 @@ async function prepareExplicitRoutedCompletion(
     protocol: inProcessProtocol,
     credentials: args.credentials,
     billable: args.billable ?? false,
+    tags: {},
     harnessConversationIdentity: undefined,
     routing,
     reporter: args.reporter ?? createExperimentRoutedRuntimeReporter(),
@@ -252,6 +253,21 @@ async function prepareRuntimeExecution(args: {
 }): Promise<RuntimeExecutionPlan> {
   const { prepared } = args;
 
+  // A strict router never selects, leases, or falls back: the pinned pair or
+  // an error.
+  if (prepared.credentials.strict === true) {
+    const pinned = strictModelDecision(prepared);
+    prepared.reporter.attachServingDecision(pinned.decision);
+    prepared.reporter.attachDecisionDetails({
+      decision_source: "strict_model",
+      explicit_model: {
+        model: pinned.decision.selectedModel,
+        reasoning_effort: pinned.decision.reasoningEffort,
+      },
+    });
+    return selectedExecutionPlan(prepared, pinned);
+  }
+
   // A requested `dari/model/<id>` is served as chosen, before any lease.
   const explicit = args.requireSelectorDecision ? null : explicitModelDecision(prepared);
   if (explicit) {
@@ -621,6 +637,7 @@ function providerExecution(
     conversationId: prepared.routing.conversationId,
     costEstimates: prepared.routing.reportingCostEstimates,
     promptCacheKey: prepared.interaction.promptCacheKey,
+    tags: prepared.tags,
   });
 }
 
@@ -692,6 +709,7 @@ const inProcessProtocol: RoutedProtocol = {
       selected_model: args.routing.selectedModel,
       reasoning_effort: args.routing.reasoningEffort,
       reason: args.routing.reason,
+      request_id: args.id,
       ...(args.conversationId ? { conversation_id: args.conversationId } : {}),
     },
   }),

package/src/runtime/routing_lease.ts

@@ -6,16 +6,16 @@ import {
   type RoutingDecision,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import { deriveFallbackDecision } from "../strategies/fallback_decision";
-import { executionProviderForModel } from "../models/execution_provider";
-import type { SelectorBackend } from "../strategies/routing";
+import { deriveFallbackDecision } from "../strategies/fallback_decision.js";
+import { executionProviderForModel } from "../models/execution_provider.js";
+import type { SelectorBackend } from "../strategies/routing.js";
 import {
   routeDecision,
   type RoutedDecision,
-} from "../strategies/route_decision";
-import { reportPendingLease } from "../telemetry/prefix_state";
-import { scrubSecrets } from "../telemetry/scrub";
-import type { PreparedRoutedCompletion } from "./prepare_routed_completion";
+} from "../strategies/route_decision.js";
+import { reportPendingLease } from "../telemetry/prefix_state.js";
+import { scrubSecrets } from "../telemetry/scrub.js";
+import type { PreparedRoutedCompletion } from "./prepare_routed_completion.js";
 
 export type RoutingLeaseDecision = {
   decision: RoutingDecision;

package/src/runtime/runtime_reporter.ts

@@ -1,7 +1,7 @@
 import type { RoutingDecision } from "@mupt-ai/dari-router/policy-engine";
-import type { InteractionOutput } from "../interaction/types";
-import type { SelectorRun } from "../strategies/routing";
-import type { ProviderKeySource } from "../types";
+import type { InteractionOutput } from "../interaction/types.js";
+import type { SelectorRun } from "../strategies/routing.js";
+import type { ProviderKeySource } from "../types.js";
 
 export type ReportedProviderExecution = {
   provider: string;

package/src/self_host/main.ts

@@ -1,20 +1,20 @@
 #!/usr/bin/env bun
 import { chmodSync, existsSync } from "node:fs";
 import { Database } from "bun:sqlite";
-import { getEnvApiKey } from "@mupt-ai/pi-ai/compat";
+import { builtinModels } from "@mupt-ai/pi-ai/providers/all";
 
-import { bearerToken } from "../http/auth";
-import { errorResponse, HttpError } from "../http/errors";
-import { createRequestHandler } from "../http/handler";
-import { installShutdownHandlers, startServer } from "../http/serve";
-import { createAutoRoutingSelectorBackend } from "../strategies/auto_routing_selector";
-import { createPrioritySelectorBackend } from "../strategies/priority_selector";
-import type { SelectorBackend } from "../strategies/routing";
-import { loadGatewayConfig } from "./config";
-import { createSelfHostedPlatform } from "./platform";
-import { gatewayKeyIndex } from "./resolve";
-import { PrefixStore } from "./prefix_store";
-import { handleSubscriptionRequest, SubscriptionStore } from "./subscriptions";
+import { bearerToken } from "../http/auth.js";
+import { errorResponse, HttpError } from "../http/errors.js";
+import { createRequestHandler } from "../http/handler.js";
+import { installShutdownHandlers, startServer } from "../http/serve.js";
+import { createAutoRoutingSelectorBackend } from "../strategies/auto_routing_selector.js";
+import { createPrioritySelectorBackend } from "../strategies/priority_selector.js";
+import type { SelectorBackend } from "../strategies/routing.js";
+import { loadGatewayConfig } from "./config.js";
+import { createSelfHostedPlatform } from "./platform.js";
+import { gatewayKeyIndex } from "./resolve.js";
+import { PrefixStore } from "./prefix_store.js";
+import { handleSubscriptionRequest, SubscriptionStore } from "./subscriptions/index.js";
 
 const SUBSCRIPTION_ROUTES = "/v1/organizations/current/credentials";
 
@@ -45,13 +45,18 @@ export async function startSelfHostedGateway(env: Record<string, string> = proce
   db.exec("PRAGMA journal_mode = WAL");
   const subscriptions = new SubscriptionStore(db);
   const prefixes = new PrefixStore(db);
+  // Pi's provider auth reads each provider's standard key variable, such as
+  // OPENAI_API_KEY, from this environment only.
+  const piModels = builtinModels({
+    authContext: { env: async (name) => env[name], fileExists: async () => false },
+  });
 
   const platform = createSelfHostedPlatform({
     config,
     gatewayKeys,
     subscriptions,
     prefixes,
-    providerApiKey: (provider) => getEnvApiKey(provider, env),
+    providerApiKey: async (provider) => (await piModels.getAuth(provider))?.auth.apiKey,
     env,
     fetch,
   });

package/src/self_host/platform.ts

@@ -1,8 +1,8 @@
-import { errorCode } from "../http/errors";
-import type { GatewayPlatform } from "../platform/platform";
-import type { PrefixStore } from "./prefix_store";
-import { resolveSelfHostedRouter, type SelfHostedResolverDeps } from "./resolve";
-import type { SubscriptionStore } from "./subscriptions";
+import { errorCode } from "../http/errors.js";
+import type { GatewayPlatform } from "../platform/platform.js";
+import type { PrefixStore } from "./prefix_store.js";
+import { resolveSelfHostedRouter, type SelfHostedResolverDeps } from "./resolve.js";
+import type { SubscriptionStore } from "./subscriptions/index.js";
 
 /**
  * The self-hosted platform: routers from the config file, subscriptions and

package/src/self_host/prefix_store.ts

@@ -14,8 +14,8 @@
 import type { Database } from "bun:sqlite";
 import type { ReasoningEffort, RouterPrefixHit } from "@mupt-ai/dari-router/policy-engine";
 
-import type { PendingLeaseWrite, PrefixStateWrite } from "../platform/platform";
-import { ROUTING_PREFIX_HASHES_MAX_COUNT } from "../platform/routing_contract";
+import type { PendingLeaseWrite, PrefixStateWrite } from "../platform/platform.js";
+import { ROUTING_PREFIX_HASHES_MAX_COUNT } from "../platform/routing_contract.js";
 
 // Resolve returns entries up to this age so a continuation after a pause still
 // recovers its conversation id and previous decision; the runtime itself

package/src/self_host/resolve.ts

@@ -7,22 +7,22 @@ import {
   type RouterModelPrice,
   type RouterPrefixHit,
 } from "@mupt-ai/dari-router/policy-engine";
-import { HttpError } from "../http/errors";
-import type { ResolveRouterRequest } from "../platform/platform";
-import { registeredModelPrice } from "../providers/pi";
+import { HttpError } from "../http/errors.js";
+import type { ResolveRouterRequest } from "../platform/platform.js";
+import { registeredModelPrice } from "../providers/pi.js";
 import type {
   ProviderKeySource,
   ResolvedRouter,
   ResolvedRouterModelConfig,
-} from "../types";
-import type { GatewayConfig, RouterConfig } from "./config";
+} from "../types.js";
+import type { GatewayConfig, RouterConfig } from "./config.js";
 import {
   freshAccessToken,
   type OAuthFetch,
   type SubscriptionCredential,
   type SubscriptionProvider,
   type SubscriptionStore,
-} from "./subscriptions";
+} from "./subscriptions/index.js";
 
 // A self-hosted gateway is one tenant; these stand in for the hosted
 // organization and the user a personal subscription belongs to.
@@ -41,7 +41,7 @@ export type SelfHostedResolverDeps = {
     conversationId?: string;
   }) => RouterPrefixHit[];
   // The provider's API key from its standard environment variable.
-  providerApiKey: (provider: string) => string | undefined;
+  providerApiKey: (provider: string) => Promise<string | undefined>;
   env: Record<string, string | undefined>;
   fetch: OAuthFetch;
 };
@@ -92,7 +92,7 @@ export async function resolveSelfHostedRouter(
     if (subscription !== null && deps.subscriptions.list(subscription).length > 0) {
       exhausted.add(subscription);
     }
-    const apiKey = providerApiKey(deps, provider);
+    const apiKey = await providerApiKey(deps, provider);
     credentials.set(provider, apiKey ? { key: apiKey, source: "user" } : null);
   }
 
@@ -137,7 +137,7 @@ export async function resolveSelfHostedRouter(
     }
   }
   for (const model of models) {
-    const price = await registeredModelPrice(model.id);
+    const price = registeredModelPrice(model.id);
     if (price !== null) resolved.model_prices![model.id] = price;
     const provider = providerForModel(model.id);
     const endpoint = deps.config.providers[provider];
@@ -158,9 +158,9 @@ export async function resolveSelfHostedRouter(
 
 // Configured endpoints read their own key variable, or send a placeholder
 // when they take none; known providers use their standard variable.
-function providerApiKey(deps: SelfHostedResolverDeps, provider: string): string | undefined {
+async function providerApiKey(deps: SelfHostedResolverDeps, provider: string): Promise<string | undefined> {
   const endpoint = deps.config.providers[provider];
-  if (endpoint === undefined) return deps.providerApiKey(provider)?.trim() || undefined;
+  if (endpoint === undefined) return (await deps.providerApiKey(provider))?.trim() || undefined;
   if (endpoint.apiKeyEnv === undefined) return "no-key";
   return deps.env[endpoint.apiKeyEnv]?.trim() || undefined;
 }

package/src/self_host/subscriptions/access.ts

@@ -1,5 +1,5 @@
-import { refreshOAuth, type OAuthFetch } from "./oauth";
-import type { SubscriptionCredential, SubscriptionStore } from "./store";
+import { refreshOAuth, type OAuthFetch } from "./oauth.js";
+import type { SubscriptionCredential, SubscriptionStore } from "./store.js";
 
 const REFRESH_MARGIN_MS = 5 * 60_000;
 

package/src/self_host/subscriptions/http.ts

@@ -12,8 +12,8 @@ import {
   startOAuth,
   type OAuthFetch,
   type SubscriptionProvider,
-} from "./oauth";
-import { isHeld, type SubscriptionCredential, type SubscriptionStore } from "./store";
+} from "./oauth.js";
+import { isHeld, type SubscriptionCredential, type SubscriptionStore } from "./store.js";
 
 export type SubscriptionHttpDeps = {
   store: SubscriptionStore;

package/src/self_host/subscriptions/index.ts

@@ -18,8 +18,8 @@
  * `store.markUnavailable` holds that subscription until its reset and says
  * whether another one can serve the retry.
  */
-export { freshAccessToken } from "./access";
-export { handleSubscriptionRequest, type SubscriptionHttpDeps } from "./http";
+export { freshAccessToken } from "./access.js";
+export { handleSubscriptionRequest, type SubscriptionHttpDeps } from "./http.js";
 export {
   CODEX_APPROVAL_PENDING,
   SUBSCRIPTION_PROVIDERS,
@@ -34,10 +34,10 @@ export {
   type OAuthOptions,
   type OAuthTokens,
   type SubscriptionProvider,
-} from "./oauth";
+} from "./oauth.js";
 export {
   SubscriptionStore,
   isHeld,
   type PendingOAuthSession,
   type SubscriptionCredential,
-} from "./store";
+} from "./store.js";

package/src/self_host/subscriptions/store.ts

@@ -13,7 +13,7 @@
 import type { Database } from "bun:sqlite";
 import { randomBytes, randomUUID } from "node:crypto";
 
-import type { OAuthTokens, SubscriptionProvider } from "./oauth";
+import type { OAuthTokens, SubscriptionProvider } from "./oauth.js";
 
 export type SubscriptionCredential = {
   id: string;

package/src/strategies/auto_routing_selector.ts

@@ -1,6 +1,6 @@
 import { parseSelectorDecision } from "@mupt-ai/dari-router/policy-engine";
-import { HttpError } from "../http/errors";
-import type { SelectorBackend } from "./routing";
+import { HttpError } from "../http/errors.js";
+import type { SelectorBackend } from "./routing.js";
 
 export const DARI_AUTO_ROUTING_ENDPOINT = "https://routing.dari.dev/v1/auto-router";
 // Same selector identity and input budget as `createAutoRouter` in

package/src/strategies/eval_relevance.ts

@@ -8,9 +8,9 @@ import {
   type RouterEval,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import { publicCatalogSelectorEvals } from "../context/eval_catalog";
-import { scrubSecrets } from "../telemetry/scrub";
-import type { ResolvedRouter } from "../types";
+import { publicCatalogSelectorEvals } from "../context/eval_catalog.js";
+import { scrubSecrets } from "../telemetry/scrub.js";
+import type { ResolvedRouter } from "../types.js";
 
 export type EvalRelevanceDetails =
   | {

package/src/strategies/priority_selector.ts

@@ -1,4 +1,4 @@
-import type { SelectorBackend } from "./routing";
+import type { SelectorBackend } from "./routing.js";
 
 const PRIORITY_SELECTOR_MODEL = "dari-gateway/priority";
 // Priority selection reads no messages, so the sized selector input only

package/src/strategies/route_decision.ts

@@ -5,13 +5,13 @@ import type {
   StrategyCandidateResolution,
   SelectorLeaseHistoryEntry,
 } from "@mupt-ai/dari-router/policy-engine";
-import { publicCatalogSelectorEvals } from "../context/eval_catalog";
-import { HttpError } from "../http/errors";
-import { relevantEvals } from "./eval_relevance";
-import { executionProviderForModel } from "../models/execution_provider";
-import { selectModel, type SelectorBackend, type SelectorRun } from "./routing";
-import type { SelectorImage } from "./selector_projection";
-import type { ResolvedRouter } from "../types";
+import { publicCatalogSelectorEvals } from "../context/eval_catalog.js";
+import { HttpError } from "../http/errors.js";
+import { relevantEvals } from "./eval_relevance.js";
+import { executionProviderForModel } from "../models/execution_provider.js";
+import { selectModel, type SelectorBackend, type SelectorRun } from "./routing.js";
+import type { SelectorImage } from "./selector_projection.js";
+import type { ResolvedRouter } from "../types.js";
 
 export type RoutedDecision = {
   decision: RoutingDecision;

package/src/strategies/routing.ts

@@ -21,21 +21,21 @@ import {
   type SelectorInput,
   type SelectorLeaseHistoryEntry,
 } from "@mupt-ai/dari-router/policy-engine";
-import { HttpError } from "../http/errors";
-import { executionProviderForModel } from "../models/execution_provider";
-import { providerLabel } from "../models/provider_defaults";
-import { internalChatCompletionsToInteraction } from "../protocol/input/chat_completions";
-import { gatewayPlatform, type ResolveRouterRequest } from "../platform/platform";
-import { isBedrockApi } from "../providers/model_capabilities";
-import { completeInteractionModel, modelContextWindow } from "../providers/pi";
+import { HttpError } from "../http/errors.js";
+import { executionProviderForModel } from "../models/execution_provider.js";
+import { providerLabel } from "../models/provider_defaults.js";
+import { internalChatCompletionsToInteraction } from "../protocol/input/chat_completions.js";
+import { gatewayPlatform, type ResolveRouterRequest } from "../platform/platform.js";
+import { isBedrockApi } from "../providers/model_capabilities.js";
+import { completeInteractionModel, modelContextWindow } from "../providers/pi.js";
 import type {
   AwsAssumeRoleProviderAuth,
   AwsSigV4ProviderAuth,
   ProviderAuth,
   ResolvedRouter,
-} from "../types";
-import { deriveFallbackDecision } from "./fallback_decision";
-import type { SelectorImage } from "./selector_projection";
+} from "../types.js";
+import { deriveFallbackDecision } from "./fallback_decision.js";
+import type { SelectorImage } from "./selector_projection.js";
 
 export function resolveRouter(args: ResolveRouterRequest): Promise<ResolvedRouter> {
   return gatewayPlatform().resolveRouter(args);
@@ -188,7 +188,7 @@ export async function selectModel(
 }
 
 async function selectorContextWindowChars(selectorModel: string): Promise<number> {
-  const contextWindow = await modelContextWindow(selectorModel);
+  const contextWindow = modelContextWindow(selectorModel);
   if (contextWindow === null) {
     throw new HttpError(
       500,

package/src/strategies/selector_projection.ts

@@ -1,12 +1,12 @@
 import type { ChatMessage } from "@mupt-ai/dari-router/policy-engine";
 
-import { toolWireName } from "../interaction/tool_identity";
+import { toolWireName } from "../interaction/tool_identity.js";
 import type {
   InteractionContent,
   InteractionMessage,
   InteractionRequest,
   InteractionToolCall,
-} from "../interaction/types";
+} from "../interaction/types.js";
 
 export const IMAGE_OMITTED = "<image omitted>";
 // Matches every numbered placeholder in a rendered selector prompt.

package/src/telemetry/billing_finalization.ts

@@ -1,6 +1,6 @@
-import type { InteractionUsage } from "../interaction/types";
-import type { ProviderKeySource } from "../types";
-import { finiteNumber, isRecord, tokenCount } from "./ingest_values";
+import type { InteractionUsage } from "../interaction/types.js";
+import type { ProviderKeySource } from "../types.js";
+import { finiteNumber, isRecord, tokenCount } from "./ingest_values.js";
 
 const BILLING_EVENT_SCHEMA_VERSION = 1;
 

package/src/telemetry/model_cost.ts

@@ -1,4 +1,4 @@
-import type { RoutingBillingProviderAttemptFacts } from "./billing_finalization";
+import type { RoutingBillingProviderAttemptFacts } from "./billing_finalization.js";
 
 /** Model reference cost only, including retries; unknown/estimated usage is never free. */
 export function modelCost(attempts: readonly RoutingBillingProviderAttemptFacts[]): number | null {

package/src/telemetry/prefix_state.ts

@@ -6,15 +6,15 @@ import {
 import {
   extendInteractionChain,
   extendInteractionLooseChain,
-} from "../context/interaction_fingerprint";
+} from "../context/interaction_fingerprint.js";
 import type {
   InteractionOutput,
   InteractionRequest,
   InteractionUsage,
   ProviderIdentity,
-} from "../interaction/types";
-import { gatewayPlatform, type PrefixStateWrite } from "../platform/platform";
-import { tokenCount } from "./ingest_values";
+} from "../interaction/types.js";
+import { gatewayPlatform, type PrefixStateWrite } from "../platform/platform.js";
+import { tokenCount } from "./ingest_values.js";
 
 export type CompletedPrefixHit = Omit<RouterPrefixHit, "updated_at"> & {
   lease_depth: number;