@mupt-ai/dari-gateway 0.2.0 → 0.2.1

package/src/providers/provider_attempt_runner.ts

@@ -1,4 +1,4 @@
-import { subscriptionUsageLimitFromFailure } from "./subscription_limits";
+import { subscriptionUsageLimitFromFailure } from "./subscription_limits.js";
 
 export type ProviderAttemptFailure =
   | {

package/src/providers/provider_call.ts

@@ -8,46 +8,46 @@ import {
   type RoutingDecision,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import { executionProviderForModel, keySourceForModel } from "../models/execution_provider";
-import { HttpError } from "../http/errors";
-import type { RequestTags } from "../platform/platform";
-import type { InteractionRequest, ProviderIdentity } from "../interaction/types";
+import { executionProviderForModel, keySourceForModel } from "../models/execution_provider.js";
+import { HttpError } from "../http/errors.js";
+import type { RequestTags } from "../platform/platform.js";
+import type { InteractionRequest, ProviderIdentity } from "../interaction/types.js";
 import {
   CustomerDisconnectedError,
   createProviderAttemptAbort,
   type ProviderAttemptAbort,
-} from "../runtime/customer_disconnect";
+} from "../runtime/customer_disconnect.js";
 import {
   isProviderFailureMessage,
   providerFailureFromCaught,
   providerFailureFromEvent,
-} from "../runtime/provider_stream_outcome";
-import { providerAuthForSelectedModel } from "../strategies/routing";
-import { scrubSecretValues } from "../telemetry/scrub";
+} from "../runtime/provider_stream_outcome.js";
+import { providerAuthForSelectedModel } from "../strategies/routing.js";
+import { scrubSecretValues } from "../telemetry/scrub.js";
 import {
   usageContextFor,
   type RoutingUsageContext,
-} from "../telemetry/usage";
+} from "../telemetry/usage.js";
 import type {
   ProviderAuth,
   ProviderKeySource,
   ResolvedRouter,
   ResolvedRouterModelConfig,
-} from "../types";
-import { awsRoleCredentials } from "./aws_assume_role";
+} from "../types.js";
+import { awsRoleCredentials } from "./aws_assume_role.js";
 import {
   attachProviderResponseCapture,
   createProviderResponseCapture,
   type ProviderFetch,
   type ProviderResponseCapture,
-} from "./response_capture";
+} from "./response_capture.js";
 import {
   completeInteractionModel,
   streamInteractionModel,
   type ProviderEndpoint,
   type ProviderEventStream,
-} from "./pi";
-import { promptTokenEstimateFor, type PromptTokenEstimate } from "./cancel_estimate";
+} from "./pi.js";
+import { promptTokenEstimateFor, type PromptTokenEstimate } from "./cancel_estimate.js";
 
 type ProviderCallConfig = {
   model: string;

package/src/providers/provider_reported_cost.ts

@@ -1,8 +1,8 @@
 import { isRecord } from "@mupt-ai/dari-router/policy-engine";
 
-import { gatewayFor } from "../models/gateways";
-import { providerFailureFromCaught } from "../runtime/provider_stream_outcome";
-import type { ProviderFetch } from "./response_capture";
+import { gatewayFor } from "../models/gateways.js";
+import { providerFailureFromCaught } from "../runtime/provider_stream_outcome.js";
+import type { ProviderFetch } from "./response_capture.js";
 
 // Gateways bill per call and say what they billed: OpenRouter as `usage.cost`
 // on the final chunk, Vercel AI Gateway as `providerMetadata.gateway.cost`.

package/src/providers/reasoning.ts

@@ -1,6 +1,6 @@
 import type { ReasoningEffort } from "@mupt-ai/dari-router/policy-engine";
-import { HttpError } from "../http/errors";
-import type { InteractionRequest } from "../interaction/types";
+import { HttpError } from "../http/errors.js";
+import type { InteractionRequest } from "../interaction/types.js";
 
 type ReasoningRequest = Pick<InteractionRequest, "reasoning">;
 

package/src/providers/stream_finalization.ts

@@ -3,32 +3,32 @@ import type {
   RoutingDecision,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import type { InteractionRequest, ProviderIdentity } from "../interaction/types";
+import type { InteractionRequest, ProviderIdentity } from "../interaction/types.js";
 import {
   captureOperationalError,
   type OperationalErrorCode,
-} from "../observability/errors";
-import type { ProviderStreamOutcome } from "../runtime/provider_stream_outcome";
-import type { RoutedProtocol } from "../runtime/routed_protocol";
+} from "../observability/errors.js";
+import type { ProviderStreamOutcome } from "../runtime/provider_stream_outcome.js";
+import type { RoutedProtocol } from "../runtime/routed_protocol.js";
 import {
   reportedExecution,
   type FrozenProviderTerminal,
   type ProviderAttemptRecord,
   type ReportableProviderOutcome,
   type RoutedRuntimeReporter,
-} from "../runtime/runtime_reporter";
-import type { ProviderKeySource } from "../types";
+} from "../runtime/runtime_reporter.js";
+import type { ProviderKeySource } from "../types.js";
 import {
   reportRoutingUsage,
   type RoutingUsageContext,
-} from "../telemetry/usage";
-import { customerCanceledProviderMessage } from "../runtime/customer_disconnect";
+} from "../telemetry/usage.js";
+import { customerCanceledProviderMessage } from "../runtime/customer_disconnect.js";
 import {
   estimatedCancelUsage,
   partialInteractionOutput,
   type DeliveredOutput,
   type PromptTokenEstimate,
-} from "./cancel_estimate";
+} from "./cancel_estimate.js";
 
 // Structurally a subset of ProviderExecution. The narrowing is deliberate: it
 // keeps apiKey and auth out of the finalizer.

package/src/providers/subscription_limits.ts

@@ -1,8 +1,8 @@
 import { isRecord } from "@mupt-ai/dari-router/policy-engine";
 
-import { providerFailureMessage } from "../runtime/provider_stream_outcome";
-import type { ProviderKeySource } from "../types";
-import type { CapturedProviderResponse } from "./response_capture";
+import { providerFailureMessage } from "../runtime/provider_stream_outcome.js";
+import type { ProviderKeySource } from "../types.js";
+import type { CapturedProviderResponse } from "./response_capture.js";
 
 export type SubscriptionOAuthProvider = "anthropic_claude_code" | "openai_codex";
 

package/src/runtime/claude_code/auxiliary_routing.ts

@@ -4,11 +4,11 @@ import {
   type RoutingDecision,
 } from "@mupt-ai/dari-router/policy-engine";
 
-import type { RoutedDecision } from "../../strategies/route_decision";
-import { isBackgroundRequest } from "../../interaction/types";
-import { HttpError } from "../../http/errors";
-import { assertCreditAvailableBeforeRouting } from "../../telemetry/usage";
-import type { PreparedRoutedCompletion } from "../prepare_routed_completion";
+import type { RoutedDecision } from "../../strategies/route_decision.js";
+import { isBackgroundRequest } from "../../interaction/types.js";
+import { HttpError } from "../../http/errors.js";
+import { assertCreditAvailableBeforeRouting } from "../../telemetry/usage.js";
+import type { PreparedRoutedCompletion } from "../prepare_routed_completion.js";
 import { leaseFallbackDecision } from "../routing_lease.js";
 
 // Explicit background classes bypass the selector and every lease path.

package/src/runtime/customer_disconnect.ts

@@ -1,4 +1,4 @@
-import { HttpError } from "../http/errors";
+import { HttpError } from "../http/errors.js";
 
 const CUSTOMER_DISCONNECTED_MESSAGE = "Client disconnected.";
 

package/src/runtime/explicit_model.ts

@@ -1,9 +1,9 @@
 import type { RoutingCandidate, RoutingDecision } from "@mupt-ai/dari-router/policy-engine";
 
-import { HttpError } from "../http/errors";
-import { explicitModelId } from "../protocol/input/router_model";
-import type { RoutedDecision } from "../strategies/route_decision";
-import type { PreparedRoutedCompletion } from "./prepare_routed_completion";
+import { HttpError } from "../http/errors.js";
+import { explicitModelId } from "../protocol/input/router_model.js";
+import type { RoutedDecision } from "../strategies/route_decision.js";
+import type { PreparedRoutedCompletion } from "./prepare_routed_completion.js";
 
 /* A request for `dari/model/<id>` is served by exactly that enabled model,
    with no selector, lease, or fallback: the user chose it, like picking a

package/src/runtime/harness_identity/conversation.ts

@@ -1,6 +1,6 @@
 import { createHash } from "node:crypto";
 
-import type { SourceProtocol } from "../../interaction/types";
+import type { SourceProtocol } from "../../interaction/types.js";
 
 export type HarnessConversationIdentity =
   | {

package/src/runtime/harness_identity/extract.ts

@@ -1,12 +1,12 @@
 import { isRecord } from "@mupt-ai/dari-router/policy-engine";
 
-import { HttpError } from "../../http/errors";
+import { HttpError } from "../../http/errors.js";
 import type {
   AgentRequestClass,
   AgentRequestContext,
   InteractionRequestPurpose,
-} from "../../interaction/types";
-import type { HarnessConversationIdentity } from "./conversation";
+} from "../../interaction/types.js";
+import type { HarnessConversationIdentity } from "./conversation.js";
 
 const MAX_ID_CHARS = 200;
 

package/src/runtime/harness_identity/index.ts

@@ -3,7 +3,7 @@ export {
   scopedHarnessSessionId,
   type HarnessConversationIdentity,
   type HarnessConversationScope,
-} from "./conversation";
+} from "./conversation.js";
 export {
   chatCompletionsConversationIdentity,
   claudeCodeConversationIdentity,
@@ -11,4 +11,4 @@ export {
   codexRequestPurpose,
   codexRequestContext,
   openCodeRequestContext,
-} from "./extract";
+} from "./extract.js";

package/src/runtime/prepare_routed_completion.ts

@@ -1,27 +1,27 @@
 import type { CandidatePruningAudit } from "@mupt-ai/dari-router/policy-engine";
 
-import { interactionConversationPrefix } from "../context/interaction_fingerprint";
+import { interactionConversationPrefix } from "../context/interaction_fingerprint.js";
 import {
   buildRoutingContext,
   derivePrefixLookupHashes,
   type ImageCapabilityPrunedModel,
   type RoutingContext,
-} from "../context/routing_context";
-import { isBackgroundRequest, type InteractionRequest } from "../interaction/types";
-import { executionProviderForModel } from "../models/execution_provider";
-import { gatewayPlatform, type RequestTags } from "../platform/platform";
-import { resolveRouter } from "../strategies/routing";
-import { RequestReporter, type ReportedCandidatePruning } from "../telemetry/report";
-import { RoutingBillingFinalization } from "../telemetry/billing_finalization";
-import type { ClientHarness, ProviderCredential, ResolvedRouter } from "../types";
-import { requestTags } from "./request_tags";
-import type { RoutedRuntimeReporter } from "./runtime_reporter";
-import type { RoutedProtocol } from "./routed_protocol";
+} from "../context/routing_context.js";
+import { isBackgroundRequest, type InteractionRequest } from "../interaction/types.js";
+import { executionProviderForModel } from "../models/execution_provider.js";
+import { gatewayPlatform, type RequestTags } from "../platform/platform.js";
+import { resolveRouter } from "../strategies/routing.js";
+import { RequestReporter, type ReportedCandidatePruning } from "../telemetry/report.js";
+import { RoutingBillingFinalization } from "../telemetry/billing_finalization.js";
+import type { ClientHarness, ProviderCredential, ResolvedRouter } from "../types.js";
+import { requestTags } from "./request_tags.js";
+import type { RoutedRuntimeReporter } from "./runtime_reporter.js";
+import type { RoutedProtocol } from "./routed_protocol.js";
 import {
   scopedHarnessConversationId,
   scopedHarnessSessionId,
   type HarnessConversationIdentity,
-} from "./harness_identity";
+} from "./harness_identity/index.js";
 
 export type PreparedRoutedCompletion = {
   requestId: string;

package/src/runtime/provider_stream_lifecycle.ts

@@ -1,4 +1,4 @@
-import type { ProviderStreamOutcome } from "./provider_stream_outcome";
+import type { ProviderStreamOutcome } from "./provider_stream_outcome.js";
 
 type LifecycleArgs = {
   partialResponse: () => Record<string, unknown>;

package/src/runtime/provider_stream_outcome.ts

@@ -1,6 +1,6 @@
 import { isRecord } from "@mupt-ai/dari-router/policy-engine";
-import type { InteractionOutput } from "../interaction/types";
-import { scrubSecrets } from "../telemetry/scrub";
+import type { InteractionOutput } from "../interaction/types.js";
+import { scrubSecrets } from "../telemetry/scrub.js";
 
 export type ProviderStreamOutcome =
   | {

package/src/runtime/request_tags.ts

@@ -1,5 +1,5 @@
-import { HttpError } from "../http/errors";
-import type { RequestTags } from "../platform/platform";
+import { HttpError } from "../http/errors.js";
+import type { RequestTags } from "../platform/platform.js";
 
 const TAGS_MAX_CHARS = 2048;
 const TAGS_MAX_COUNT = 32;

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;

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,16 +25,16 @@ 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, strictModelDecision } from "./explicit_model.js";
 import {
@@ -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;
@@ -709,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

@@ -3,18 +3,18 @@ import { chmodSync, existsSync } from "node:fs";
 import { Database } from "bun:sqlite";
 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";
 

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.

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