build: update app deps except carbon

2026-04-25 16:12:13 +00:00 · 2026-03-09 06:09:22 +00:00
parent 3c4377651e
commit 9bde7ef39f
23 changed files with 495 additions and 169 deletions
--- a/src/agents/anthropic-payload-log.test.ts
+++ b/src/agents/anthropic-payload-log.test.ts
@@ -28,8 +28,8 @@ describe("createAnthropicPayloadLogger", () => {
        },
      ],
    };
-    const streamFn: StreamFn = ((_, __, options) => {
-      options?.onPayload?.(payload);
+    const streamFn: StreamFn = ((model, __, options) => {
+      options?.onPayload?.(payload, model);
      return {} as never;
    }) as StreamFn;

--- a/src/agents/anthropic-payload-log.ts
+++ b/src/agents/anthropic-payload-log.ts
@@ -136,7 +136,7 @@ export function createAnthropicPayloadLogger(params: {
      if (!isAnthropicModel(model)) {
        return streamFn(model, context, options);
      }
-      const nextOnPayload = (payload: unknown) => {
+      const nextOnPayload = (payload: unknown, payloadModel: Parameters<StreamFn>[0]) => {
        const redactedPayload = redactImageDataForDiagnostics(payload);
        record({
          ...base,
@@ -145,7 +145,7 @@ export function createAnthropicPayloadLogger(params: {
          payload: redactedPayload,
          payloadDigest: digest(redactedPayload),
        });
-        options?.onPayload?.(payload);
+        return options?.onPayload?.(payload, payloadModel);
      };
      return streamFn(model, context, {
        ...options,
--- a/src/agents/auth-profiles/oauth.ts
+++ b/src/agents/auth-profiles/oauth.ts
@@ -1,9 +1,5 @@
-import {
-  getOAuthApiKey,
-  getOAuthProviders,
-  type OAuthCredentials,
-  type OAuthProvider,
-} from "@mariozechner/pi-ai";
+import type { OAuthCredentials, OAuthProvider } from "@mariozechner/pi-ai/oauth";
+import { getOAuthApiKey, getOAuthProviders } from "@mariozechner/pi-ai/oauth";
 import { loadConfig, type OpenClawConfig } from "../../config/config.js";
 import { coerceSecretRef } from "../../config/types.secrets.js";
 import { withFileLock } from "../../infra/file-lock.js";
--- a/src/agents/openai-ws-stream.ts
+++ b/src/agents/openai-ws-stream.ts
@@ -604,7 +604,7 @@ export function createOpenAIWebSocketStreamFn(
        ...(prevResponseId ? { previous_response_id: prevResponseId } : {}),
        ...extraParams,
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);

      try {
        session.manager.send(payload as Parameters<OpenAIWebSocketManager["send"]>[0]);
--- a/src/agents/pi-embedded-runner-extraparams.live.test.ts
+++ b/src/agents/pi-embedded-runner-extraparams.live.test.ts
@@ -101,7 +101,7 @@ describeGeminiLive("pi embedded extra params (gemini live)", () => {
    oneByOneRedPngBase64: string;
    includeImage?: boolean;
    prompt: string;
-    onPayload?: (payload: Record<string, unknown>) => void;
+    onPayload?: (payload: Record<string, unknown>, model: Model<"google-generative-ai">) => void;
  }): Promise<{ sawDone: boolean; stopReason?: string; errorMessage?: string }> {
    const userContent: Array<
      { type: "text"; text: string } | { type: "image"; mimeType: string; data: string }
@@ -129,8 +129,11 @@ describeGeminiLive("pi embedded extra params (gemini live)", () => {
        apiKey: params.apiKey,
        reasoning: "high",
        maxTokens: 64,
-        onPayload: (payload) => {
-          params.onPayload?.(payload as Record<string, unknown>);
+        onPayload: (payload, streamModel) => {
+          params.onPayload?.(
+            payload as Record<string, unknown>,
+            streamModel as Model<"google-generative-ai">,
+          );
        },
      },
    );
--- a/src/agents/pi-embedded-runner-extraparams.test.ts
+++ b/src/agents/pi-embedded-runner-extraparams.test.ts
@@ -207,8 +207,8 @@ describe("applyExtraParamsToAgent", () => {
    payload?: Record<string, unknown>;
  }) {
    const payload = params.payload ?? { store: false };
-    const baseStreamFn: StreamFn = (_model, _context, options) => {
-      options?.onPayload?.(payload);
+    const baseStreamFn: StreamFn = (model, _context, options) => {
+      options?.onPayload?.(payload, model);
      return {} as ReturnType<StreamFn>;
    };
    const agent = { streamFn: baseStreamFn };
@@ -232,8 +232,8 @@ describe("applyExtraParamsToAgent", () => {
    payload?: Record<string, unknown>;
  }) {
    const payload = params.payload ?? {};
-    const baseStreamFn: StreamFn = (_model, _context, options) => {
-      options?.onPayload?.(payload);
+    const baseStreamFn: StreamFn = (model, _context, options) => {
+      options?.onPayload?.(payload, model);
      return {} as ReturnType<StreamFn>;
    };
    const agent = { streamFn: baseStreamFn };
@@ -276,7 +276,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { model: "deepseek/deepseek-r1" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -308,7 +308,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = {};
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -332,7 +332,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { reasoning_effort: "high" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -357,7 +357,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { reasoning: { max_tokens: 256 } };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -381,7 +381,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { reasoning_effort: "medium" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -588,7 +588,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { thinking: "off" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -619,7 +619,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { thinking: "off" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -650,7 +650,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = {};
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -674,7 +674,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { tool_choice: "required" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -699,7 +699,7 @@ describe("applyExtraParamsToAgent", () => {
    const payloads: Record<string, unknown>[] = [];
    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = {};
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -757,7 +757,7 @@ describe("applyExtraParamsToAgent", () => {
        ],
        tool_choice: { type: "tool", name: "read" },
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -820,7 +820,7 @@ describe("applyExtraParamsToAgent", () => {
        ],
        tool_choice: input,
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -853,7 +853,7 @@ describe("applyExtraParamsToAgent", () => {
          },
        ],
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -892,7 +892,7 @@ describe("applyExtraParamsToAgent", () => {
          },
        ],
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -956,7 +956,7 @@ describe("applyExtraParamsToAgent", () => {
          },
        },
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
@@ -1003,7 +1003,7 @@ describe("applyExtraParamsToAgent", () => {
          },
        },
      };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      payloads.push(payload);
      return {} as ReturnType<StreamFn>;
    };
--- a/src/agents/pi-embedded-runner/anthropic-stream-wrappers.ts
+++ b/src/agents/pi-embedded-runner/anthropic-stream-wrappers.ts
@@ -277,7 +277,7 @@ export function createAnthropicToolPayloadCompatibilityWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (
          payload &&
          typeof payload === "object" &&
@@ -298,7 +298,7 @@ export function createAnthropicToolPayloadCompatibilityWrapper(
            );
          }
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
--- a/src/agents/pi-embedded-runner/extra-params.kilocode.test.ts
+++ b/src/agents/pi-embedded-runner/extra-params.kilocode.test.ts
@@ -19,7 +19,7 @@ function applyAndCapture(params: {

  const baseStreamFn: StreamFn = (_model, _context, options) => {
    captured.headers = options?.headers;
-    options?.onPayload?.({});
+    options?.onPayload?.({}, model);
    return createAssistantMessageEventStream();
  };
  const agent = { streamFn: baseStreamFn };
@@ -97,7 +97,7 @@ describe("extra-params: Kilocode kilo/auto reasoning", () => {

    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { reasoning_effort: "high" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      capturedPayload = payload;
      return createAssistantMessageEventStream();
    };
@@ -125,7 +125,7 @@ describe("extra-params: Kilocode kilo/auto reasoning", () => {

    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = {};
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      capturedPayload = payload;
      return createAssistantMessageEventStream();
    };
@@ -158,7 +158,7 @@ describe("extra-params: Kilocode kilo/auto reasoning", () => {

    const baseStreamFn: StreamFn = (_model, _context, options) => {
      const payload: Record<string, unknown> = { reasoning_effort: "high" };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, model);
      capturedPayload = payload;
      return createAssistantMessageEventStream();
    };
--- a/src/agents/pi-embedded-runner/extra-params.openrouter-cache-control.test.ts
+++ b/src/agents/pi-embedded-runner/extra-params.openrouter-cache-control.test.ts
@@ -13,7 +13,7 @@ type StreamPayload = {

 function runOpenRouterPayload(payload: StreamPayload, modelId: string) {
  const baseStreamFn: StreamFn = (_model, _context, options) => {
-    options?.onPayload?.(payload);
+    options?.onPayload?.(payload, model);
    return createAssistantMessageEventStream();
  };
  const agent = { streamFn: baseStreamFn };
--- a/src/agents/pi-embedded-runner/extra-params.ts
+++ b/src/agents/pi-embedded-runner/extra-params.ts
@@ -222,7 +222,7 @@ function createGoogleThinkingPayloadWrapper(
    const onPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (model.api === "google-generative-ai") {
          sanitizeGoogleThinkingPayload({
            payload,
@@ -230,7 +230,7 @@ function createGoogleThinkingPayloadWrapper(
            thinkingLevel,
          });
        }
-        onPayload?.(payload);
+        return onPayload?.(payload, payloadModel);
      },
    });
  };
@@ -258,12 +258,12 @@ function createZaiToolStreamWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          // Inject tool_stream: true for Z.AI API
          (payload as Record<string, unknown>).tool_stream = true;
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
@@ -306,11 +306,11 @@ function createParallelToolCallsWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          (payload as Record<string, unknown>).parallel_tool_calls = enabled;
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
--- a/src/agents/pi-embedded-runner/extra-params.zai-tool-stream.test.ts
+++ b/src/agents/pi-embedded-runner/extra-params.zai-tool-stream.test.ts
@@ -21,8 +21,8 @@ type ToolStreamCase = {

 function runToolStreamCase(params: ToolStreamCase) {
  const payload: Record<string, unknown> = { model: params.model.id, messages: [] };
-  const baseStreamFn: StreamFn = (_model, _context, options) => {
-    options?.onPayload?.(payload);
+  const baseStreamFn: StreamFn = (model, _context, options) => {
+    options?.onPayload?.(payload, model);
    return {} as ReturnType<StreamFn>;
  };
  const agent = { streamFn: baseStreamFn };
--- a/src/agents/pi-embedded-runner/moonshot-stream-wrappers.ts
+++ b/src/agents/pi-embedded-runner/moonshot-stream-wrappers.ts
@@ -53,14 +53,14 @@ export function createSiliconFlowThinkingWrapper(baseStreamFn: StreamFn | undefi
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          const payloadObj = payload as Record<string, unknown>;
          if (payloadObj.thinking === "off") {
            payloadObj.thinking = null;
          }
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
@@ -89,7 +89,7 @@ export function createMoonshotThinkingWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          const payloadObj = payload as Record<string, unknown>;
          let effectiveThinkingType = normalizeMoonshotThinkingType(payloadObj.thinking);
@@ -106,7 +106,7 @@ export function createMoonshotThinkingWrapper(
            payloadObj.tool_choice = "auto";
          }
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
--- a/src/agents/pi-embedded-runner/openai-stream-wrappers.ts
+++ b/src/agents/pi-embedded-runner/openai-stream-wrappers.ts
@@ -187,7 +187,7 @@ export function createOpenAIResponsesContextManagementWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          applyOpenAIResponsesPayloadOverrides({
            payloadObj: payload as Record<string, unknown>,
@@ -197,7 +197,7 @@ export function createOpenAIResponsesContextManagementWrapper(
            compactThreshold,
          });
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
@@ -219,14 +219,14 @@ export function createOpenAIServiceTierWrapper(
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        if (payload && typeof payload === "object") {
          const payloadObj = payload as Record<string, unknown>;
          if (payloadObj.service_tier === undefined) {
            payloadObj.service_tier = serviceTier;
          }
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
--- a/src/agents/pi-embedded-runner/proxy-stream-wrappers.ts
+++ b/src/agents/pi-embedded-runner/proxy-stream-wrappers.ts
@@ -73,7 +73,7 @@ export function createOpenRouterSystemCacheWrapper(baseStreamFn: StreamFn | unde
    const originalOnPayload = options?.onPayload;
    return underlying(model, context, {
      ...options,
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        const messages = (payload as Record<string, unknown>)?.messages;
        if (Array.isArray(messages)) {
          for (const msg of messages as Array<{ role?: string; content?: unknown }>) {
@@ -92,7 +92,7 @@ export function createOpenRouterSystemCacheWrapper(baseStreamFn: StreamFn | unde
            }
          }
        }
-        originalOnPayload?.(payload);
+        return originalOnPayload?.(payload, payloadModel);
      },
    });
  };
@@ -111,9 +111,9 @@ export function createOpenRouterWrapper(
        ...OPENROUTER_APP_HEADERS,
        ...options?.headers,
      },
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        normalizeProxyReasoningPayload(payload, thinkingLevel);
-        onPayload?.(payload);
+        return onPayload?.(payload, payloadModel);
      },
    });
  };
@@ -136,9 +136,9 @@ export function createKilocodeWrapper(
        ...options?.headers,
        ...resolveKilocodeAppHeaders(),
      },
-      onPayload: (payload) => {
+      onPayload: (payload, payloadModel) => {
        normalizeProxyReasoningPayload(payload, thinkingLevel);
-        onPayload?.(payload);
+        return onPayload?.(payload, payloadModel);
      },
    });
  };
--- a/src/agents/pi-embedded-runner/run/attempt.test.ts
+++ b/src/agents/pi-embedded-runner/run/attempt.test.ts
@@ -520,7 +520,7 @@ describe("wrapOllamaCompatNumCtx", () => {
    let payloadSeen: Record<string, unknown> | undefined;
    const baseFn = vi.fn((_model, _context, options) => {
      const payload: Record<string, unknown> = { options: { temperature: 0.1 } };
-      options?.onPayload?.(payload);
+      options?.onPayload?.(payload, _model);
      payloadSeen = payload;
      return {} as never;
    });
--- a/src/agents/pi-embedded-runner/run/attempt.ts
+++ b/src/agents/pi-embedded-runner/run/attempt.ts
@@ -227,17 +227,16 @@ export function wrapOllamaCompatNumCtx(baseFn: StreamFn | undefined, numCtx: num
  return (model, context, options) =>
    streamFn(model, context, {
      ...options,
-      onPayload: (payload: unknown) => {
+      onPayload: (payload: unknown, payloadModel) => {
        if (!payload || typeof payload !== "object") {
-          options?.onPayload?.(payload);
-          return;
+          return options?.onPayload?.(payload, payloadModel);
        }
        const payloadRecord = payload as Record<string, unknown>;
        if (!payloadRecord.options || typeof payloadRecord.options !== "object") {
          payloadRecord.options = {};
        }
        (payloadRecord.options as Record<string, unknown>).num_ctx = numCtx;
-        options?.onPayload?.(payload);
+        return options?.onPayload?.(payload, payloadModel);
      },
    });
 }
--- a/src/commands/openai-codex-oauth.ts
+++ b/src/commands/openai-codex-oauth.ts
@@ -1,5 +1,5 @@
-import type { OAuthCredentials } from "@mariozechner/pi-ai";
-import { loginOpenAICodex } from "@mariozechner/pi-ai";
+import type { OAuthCredentials } from "@mariozechner/pi-ai/oauth";
+import { loginOpenAICodex } from "@mariozechner/pi-ai/oauth";
 import type { RuntimeEnv } from "../runtime.js";
 import type { WizardPrompter } from "../wizard/prompts.js";
 import { createVpsAwareOAuthHandlers } from "./oauth-flow.js";
@@ -53,7 +53,7 @@ export async function loginOpenAICodexOAuth(params: {
    const creds = await loginOpenAICodex({
      onAuth: baseOnAuth,
      onPrompt,
-      onProgress: (msg) => spin.update(msg),
+      onProgress: (msg: string) => spin.update(msg),
    });
    spin.stop("OpenAI OAuth complete");
    return creds ?? null;
--- a/src/providers/google-shared.ensures-function-call-comes-after-user-turn.test.ts
+++ b/src/providers/google-shared.ensures-function-call-comes-after-user-turn.test.ts
@@ -1,6 +1,6 @@
-import { convertMessages } from "@mariozechner/pi-ai/dist/providers/google-shared.js";
-import type { Context } from "@mariozechner/pi-ai/dist/types.js";
+import type { Context } from "@mariozechner/pi-ai";
 import { describe, expect, it } from "vitest";
+import { convertMessages } from "../../node_modules/@mariozechner/pi-ai/dist/providers/google-shared.js";
 import {
  asRecord,
  expectConvertedRoles,
--- a/src/providers/google-shared.preserves-parameters-type-is-missing.test.ts
+++ b/src/providers/google-shared.preserves-parameters-type-is-missing.test.ts
@@ -1,6 +1,9 @@
-import { convertMessages, convertTools } from "@mariozechner/pi-ai/dist/providers/google-shared.js";
-import type { Context, Tool } from "@mariozechner/pi-ai/dist/types.js";
+import type { Context, Tool } from "@mariozechner/pi-ai";
 import { describe, expect, it } from "vitest";
+import {
+  convertMessages,
+  convertTools,
+} from "../../node_modules/@mariozechner/pi-ai/dist/providers/google-shared.js";
 import {
  asRecord,
  expectConvertedRoles,
--- a/src/providers/google-shared.test-helpers.ts
+++ b/src/providers/google-shared.test-helpers.ts
@@ -1,4 +1,4 @@
-import type { Model } from "@mariozechner/pi-ai/dist/types.js";
+import type { Model } from "@mariozechner/pi-ai";
 import { expect } from "vitest";
 import { makeZeroUsageSnapshot } from "../agents/usage.js";