+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| openrouter.ts | +
+
+ |
+ 86.95% | +20/23 | +85% | +17/20 | +100% | +3/3 | +86.95% | +20/23 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 +66 +67 +68 +69 +70 +71 +72 +73 +74 +75 +76 +77 +78 +79 +80 +81 +82 +83 +84 +85 +86 +87 +88 +89 +90 +91 +92 +93 +94 +95 +96 +97 +98 +99 +100 +101 | + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +13x +9x + +4x +1x + +3x +3x + + + + +13x +4x + + +4x + + +4x +3x + +4x +2x + +4x + + + +3x + + + + +13x +13x +13x + + + + + + + + +13x + + + + + + + + + + + +3x + + + | import { OpenRouterCore } from '@openrouter/sdk/core';
+import { SDKHooks } from '@openrouter/sdk/hooks/hooks';
+import type {
+ AfterErrorHook,
+ AfterSuccessHook,
+ BeforeCreateRequestHook,
+ BeforeRequestHook,
+ SDKInitHook,
+} from '@openrouter/sdk/hooks/types';
+import type { SDKOptions } from '@openrouter/sdk/lib/config';
+import type { RequestOptions } from '@openrouter/sdk/lib/sdks';
+import type { $ZodObject, $ZodShape, infer as zodInfer } from 'zod/v4/core';
+
+import { callModel } from './inner-loop/call-model.js';
+import type { CallModelInput } from './lib/async-params.js';
+import type { ModelResult } from './lib/model-result.js';
+import type { Tool } from './lib/tool-types.js';
+
+export type { SDKOptions } from '@openrouter/sdk/lib/config';
+
+/** Any single hook interface supported by the underlying SDK. */
+export type Hook =
+ | SDKInitHook
+ | BeforeCreateRequestHook
+ | BeforeRequestHook
+ | AfterSuccessHook
+ | AfterErrorHook;
+
+/**
+ * SDK options extended with optional hooks for request/response interception.
+ * `hooks` accepts an `SDKHooks` instance, a single hook object, or an array of
+ * hook objects; the constructor normalizes shorthand input into an `SDKHooks`
+ * instance before passing it to the underlying SDK, which only recognizes
+ * `hooks` when it is an `SDKHooks` instance.
+ */
+export type OpenRouterOptions = SDKOptions & {
+ hooks?: SDKHooks | Hook | readonly Hook[];
+};
+
+function buildSDKHooks(input: SDKHooks | Hook | readonly Hook[] | undefined): SDKHooks | undefined {
+ if (input === undefined) {
+ return undefined;
+ }
+ if (input instanceof SDKHooks) {
+ return input;
+ }
+ const hooks = new SDKHooks();
+ const list: readonly Hook[] = Array.isArray(input)
+ ? input
+ : [
+ input,
+ ];
+ for (const hook of list) {
+ Iif ('sdkInit' in hook) {
+ hooks.registerSDKInitHook(hook);
+ }
+ Iif ('beforeCreateRequest' in hook) {
+ hooks.registerBeforeCreateRequestHook(hook);
+ }
+ if ('beforeRequest' in hook) {
+ hooks.registerBeforeRequestHook(hook);
+ }
+ if ('afterSuccess' in hook) {
+ hooks.registerAfterSuccessHook(hook);
+ }
+ Iif ('afterError' in hook) {
+ hooks.registerAfterErrorHook(hook);
+ }
+ }
+ return hooks;
+}
+
+export class OpenRouter extends OpenRouterCore {
+ constructor(options?: OpenRouterOptions) {
+ const { hooks: hooksInput, ...rest } = options ?? {};
+ const hooks = buildSDKHooks(hooksInput);
+ super(
+ hooks === undefined
+ ? rest
+ : Object.assign({}, rest, {
+ hooks,
+ }),
+ );
+ }
+
+ callModel = <
+ TTools extends readonly Tool[],
+ TSharedSchema extends $ZodObject<$ZodShape> | undefined = undefined,
+ TShared extends Record<string, unknown> = TSharedSchema extends $ZodObject<$ZodShape>
+ ? zodInfer<TSharedSchema>
+ : Record<string, never>,
+ >(
+ request: CallModelInput<TTools, TShared> & {
+ sharedContextSchema?: TSharedSchema;
+ },
+ options?: RequestOptions,
+ ): ModelResult<TTools, TShared> => {
+ return callModel(this, request, options);
+ };
+}
+ |