perf: skip provider plugin loading in extra param tests ·...
steipete
·
2026-04-24
·
via Recent Commits to openclaw:main
| Original file line number | Diff line number | Diff line change |
|---|
@@ -10,6 +10,7 @@ import {
|
10 | 10 | beforeEach(() => { |
11 | 11 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
12 | 12 | prepareProviderExtraParams: ({ context }) => context.extraParams, |
| 13 | +resolveProviderExtraParamsForTransport: () => undefined, |
13 | 14 | wrapProviderStreamFn: (params) => { |
14 | 15 | if (params.provider === "moonshot") { |
15 | 16 | const thinkingType = resolveMoonshotThinkingType({ |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -11,6 +11,7 @@ import {
|
11 | 11 | beforeEach(() => { |
12 | 12 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
13 | 13 | prepareProviderExtraParams: ({ context }) => context.extraParams, |
| 14 | +resolveProviderExtraParamsForTransport: () => undefined, |
14 | 15 | wrapProviderStreamFn: (params) => { |
15 | 16 | if (params.provider !== "openrouter") { |
16 | 17 | return params.context.streamFn; |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -324,6 +324,7 @@ function installFullProviderRuntimeDepsForTest() {
|
324 | 324 | transport: "auto", |
325 | 325 | }; |
326 | 326 | }, |
| 327 | +resolveProviderExtraParamsForTransport: () => undefined, |
327 | 328 | wrapProviderStreamFn: (params) => { |
328 | 329 | if (params.provider === "openai") { |
329 | 330 | return createTestOpenAIProviderWrapper(params, true); |
@@ -398,6 +399,7 @@ function installFullProviderRuntimeDepsForTest() {
|
398 | 399 | function withMinimalProviderRuntimeDepsForTest<T>(run: () => T): T { |
399 | 400 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
400 | 401 | prepareProviderExtraParams: () => undefined, |
| 402 | +resolveProviderExtraParamsForTransport: () => undefined, |
401 | 403 | wrapProviderStreamFn: (params) => params.context.streamFn, |
402 | 404 | }); |
403 | 405 | try { |
|
| Original file line number | Diff line number | Diff line change |
|---|
|
1 | 1 | import type { StreamFn } from "@mariozechner/pi-agent-core"; |
2 | 2 | import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; |
| 3 | +import { createPiAiStreamSimpleMock } from "../../../test/helpers/agents/pi-ai-stream-simple-mock.js"; |
3 | 4 | import { isOpenRouterAnthropicModelRef } from "./anthropic-family-cache-semantics.js"; |
4 | 5 | import { __testing as extraParamsTesting, applyExtraParamsToAgent } from "./extra-params.js"; |
5 | 6 | import { resolveCacheRetention } from "./prompt-cache-retention.js"; |
@@ -36,9 +37,12 @@ vi.mock("./logger.js", () => ({
|
36 | 37 | }, |
37 | 38 | })); |
38 | 39 | |
| 40 | +vi.mock("@mariozechner/pi-ai", () => createPiAiStreamSimpleMock()); |
| 41 | + |
39 | 42 | beforeEach(() => { |
40 | 43 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
41 | 44 | prepareProviderExtraParams: () => undefined, |
| 45 | +resolveProviderExtraParamsForTransport: () => undefined, |
42 | 46 | wrapProviderStreamFn: () => undefined, |
43 | 47 | }); |
44 | 48 | }); |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -4,15 +4,12 @@ import { createPiAiStreamSimpleMock } from "../../../test/helpers/agents/pi-ai-s
|
4 | 4 | import { __testing as extraParamsTesting } from "./extra-params.js"; |
5 | 5 | import { runExtraParamsCase } from "./extra-params.test-support.js"; |
6 | 6 | |
7 | | -vi.mock("@mariozechner/pi-ai", async () => |
8 | | -createPiAiStreamSimpleMock(() => |
9 | | -vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai"), |
10 | | -), |
11 | | -); |
| 7 | +vi.mock("@mariozechner/pi-ai", () => createPiAiStreamSimpleMock()); |
12 | 8 | |
13 | 9 | beforeEach(() => { |
14 | 10 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
15 | 11 | prepareProviderExtraParams: (params) => params.context.extraParams, |
| 12 | +resolveProviderExtraParamsForTransport: () => undefined, |
16 | 13 | wrapProviderStreamFn: () => undefined, |
17 | 14 | }); |
18 | 15 | }); |
|
| Original file line number | Diff line number | Diff line change |
|---|
|
1 | 1 | import type { Model } from "@mariozechner/pi-ai"; |
2 | 2 | import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; |
| 3 | +import { createPiAiStreamSimpleMock } from "../../../test/helpers/agents/pi-ai-stream-simple-mock.js"; |
3 | 4 | import { __testing as extraParamsTesting } from "./extra-params.js"; |
4 | 5 | import { runExtraParamsCase } from "./extra-params.test-support.js"; |
5 | 6 | |
6 | | -vi.mock("@mariozechner/pi-ai", async () => { |
7 | | -const original = |
8 | | -await vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai"); |
9 | | -return { |
10 | | - ...original, |
11 | | -streamSimple: vi.fn(() => ({ |
12 | | -push: vi.fn(), |
13 | | -result: vi.fn(), |
14 | | -})), |
15 | | -}; |
16 | | -}); |
| 7 | +vi.mock("@mariozechner/pi-ai", () => createPiAiStreamSimpleMock()); |
17 | 8 | |
18 | 9 | beforeEach(() => { |
19 | 10 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
20 | 11 | prepareProviderExtraParams: ({ context }) => context.extraParams, |
| 12 | +resolveProviderExtraParamsForTransport: () => undefined, |
21 | 13 | wrapProviderStreamFn: ({ provider, context }) => { |
22 | 14 | if (provider !== "local-provider" || context.thinkingLevel !== "off") { |
23 | 15 | return context.streamFn; |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -56,6 +56,7 @@ export function runExtraParamsCase<
|
56 | 56 | if (params.mockProviderRuntime === true) { |
57 | 57 | extraParamsTesting.setProviderRuntimeDepsForTest({ |
58 | 58 | prepareProviderExtraParams: () => undefined, |
| 59 | +resolveProviderExtraParamsForTransport: () => undefined, |
59 | 60 | wrapProviderStreamFn: () => undefined, |
60 | 61 | }); |
61 | 62 | } |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -3,11 +3,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
3 | 3 | import { createPiAiStreamSimpleMock } from "../../../test/helpers/agents/pi-ai-stream-simple-mock.js"; |
4 | 4 | import type { OpenClawConfig } from "../../config/config.js"; |
5 | 5 | |
6 | | -vi.mock("@mariozechner/pi-ai", async () => |
7 | | -createPiAiStreamSimpleMock(() => |
8 | | -vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai"), |
9 | | -), |
10 | | -); |
| 6 | +vi.mock("@mariozechner/pi-ai", () => createPiAiStreamSimpleMock()); |
11 | 7 | |
12 | 8 | let runExtraParamsCase: typeof import("./extra-params.test-support.js").runExtraParamsCase; |
13 | 9 | let extraParamsTesting: typeof import("./extra-params.js").__testing; |
@@ -46,6 +42,7 @@ describe("extra-params: provider tool_stream support", () => {
|
46 | 42 | } |
47 | 43 | return extraParams; |
48 | 44 | }, |
| 45 | +resolveProviderExtraParamsForTransport: () => undefined, |
49 | 46 | wrapProviderStreamFn: (params) => { |
50 | 47 | const extraParams = params.context.extraParams ?? {}; |
51 | 48 | if (extraParams.tool_stream !== true) { |
|
| Original file line number | Diff line number | Diff line change |
|---|
@@ -2,12 +2,8 @@ import { vi } from "vitest";
|
2 | 2 | |
3 | 3 | type PiAiMockModule = Record<string, unknown>; |
4 | 4 | |
5 | | -export async function createPiAiStreamSimpleMock( |
6 | | -loadActual: () => Promise<PiAiMockModule>, |
7 | | -): Promise<PiAiMockModule> { |
8 | | -const original = await loadActual(); |
| 5 | +export function createPiAiStreamSimpleMock(): PiAiMockModule { |
9 | 6 | return { |
10 | | - ...original, |
11 | 7 | streamSimple: vi.fn(() => ({ |
12 | 8 | push: vi.fn(), |
13 | 9 | result: vi.fn(async () => undefined), |
|
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。
