mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-11 10:33:18 +08:00
重构主控编排与运行时预热链路,统一工作区提示词/专家调度协议并补齐 wiki 记忆注入与写回闭环。
同时收敛启动与运维脚本默认行为(含 wiki worker)、更新 Admin 可观测性与相关测试,降低首轮时延并提高运行稳定性。 Made-with: Cursor
This commit is contained in:
parent
4a23b715a2
commit
dbbe3add6a
14438 changed files with 2693620 additions and 2546 deletions
155
openclaw/extensions/fireworks/stream.test.ts
Normal file
155
openclaw/extensions/fireworks/stream.test.ts
Normal file
|
|
@ -0,0 +1,155 @@
|
|||
import type { StreamFn } from "@mariozechner/pi-agent-core";
|
||||
import type { Context, Model } from "@mariozechner/pi-ai";
|
||||
import { describe, expect, it } from "vitest";
|
||||
import {
|
||||
createFireworksKimiThinkingDisabledWrapper,
|
||||
wrapFireworksProviderStream,
|
||||
} from "./stream.js";
|
||||
|
||||
function capturePayload(params: {
|
||||
provider: string;
|
||||
api: string;
|
||||
modelId: string;
|
||||
initialPayload?: Record<string, unknown>;
|
||||
}): Record<string, unknown> {
|
||||
let captured: Record<string, unknown> = {};
|
||||
const baseStreamFn: StreamFn = (_model, _context, options) => {
|
||||
const payload = { ...params.initialPayload };
|
||||
options?.onPayload?.(payload, _model);
|
||||
captured = payload;
|
||||
return {} as ReturnType<StreamFn>;
|
||||
};
|
||||
|
||||
const wrapped = createFireworksKimiThinkingDisabledWrapper(baseStreamFn);
|
||||
void wrapped(
|
||||
{
|
||||
api: params.api,
|
||||
provider: params.provider,
|
||||
id: params.modelId,
|
||||
} as Model<"openai-completions">,
|
||||
{ messages: [] } as Context,
|
||||
{},
|
||||
);
|
||||
|
||||
return captured;
|
||||
}
|
||||
|
||||
describe("createFireworksKimiThinkingDisabledWrapper", () => {
|
||||
it("forces thinking disabled for Fireworks Kimi models", () => {
|
||||
expect(
|
||||
capturePayload({
|
||||
provider: "fireworks",
|
||||
api: "openai-completions",
|
||||
modelId: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
}),
|
||||
).toMatchObject({ thinking: { type: "disabled" } });
|
||||
});
|
||||
|
||||
it("forces thinking disabled for Fireworks Kimi k2.5 aliases", () => {
|
||||
expect(
|
||||
capturePayload({
|
||||
provider: "fireworks",
|
||||
api: "openai-completions",
|
||||
modelId: "accounts/fireworks/routers/kimi-k2.5-turbo",
|
||||
}),
|
||||
).toMatchObject({ thinking: { type: "disabled" } });
|
||||
});
|
||||
|
||||
it("strips reasoning fields when disabling Fireworks Kimi thinking", () => {
|
||||
const payload = capturePayload({
|
||||
provider: "fireworks",
|
||||
api: "openai-completions",
|
||||
modelId: "accounts/fireworks/models/kimi-k2p5",
|
||||
initialPayload: {
|
||||
reasoning_effort: "low",
|
||||
reasoning: { effort: "low" },
|
||||
reasoningEffort: "low",
|
||||
},
|
||||
});
|
||||
|
||||
expect(payload).toEqual({ thinking: { type: "disabled" } });
|
||||
});
|
||||
|
||||
it("passes sanitized payloads to caller onPayload hooks", () => {
|
||||
let callbackPayload: Record<string, unknown> = {};
|
||||
const baseStreamFn: StreamFn = (_model, _context, options) => {
|
||||
const payload = {
|
||||
reasoning_effort: "high",
|
||||
reasoning: { effort: "high" },
|
||||
};
|
||||
options?.onPayload?.(payload, _model);
|
||||
return {} as ReturnType<StreamFn>;
|
||||
};
|
||||
|
||||
const wrapped = createFireworksKimiThinkingDisabledWrapper(baseStreamFn);
|
||||
void wrapped(
|
||||
{
|
||||
api: "openai-completions",
|
||||
provider: "fireworks",
|
||||
id: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
} as Model<"openai-completions">,
|
||||
{ messages: [] } as Context,
|
||||
{
|
||||
onPayload: (payload) => {
|
||||
callbackPayload = payload as Record<string, unknown>;
|
||||
},
|
||||
},
|
||||
);
|
||||
|
||||
expect(callbackPayload).toEqual({ thinking: { type: "disabled" } });
|
||||
});
|
||||
|
||||
it("returns no provider wrapper for non-target Fireworks requests", () => {
|
||||
expect(
|
||||
wrapFireworksProviderStream({
|
||||
provider: "fireworks",
|
||||
modelId: "accounts/fireworks/models/qwen3.6-plus",
|
||||
model: {
|
||||
api: "openai-completions",
|
||||
provider: "fireworks",
|
||||
id: "accounts/fireworks/models/qwen3.6-plus",
|
||||
} as Model<"openai-completions">,
|
||||
streamFn: undefined,
|
||||
} as never),
|
||||
).toBeUndefined();
|
||||
|
||||
expect(
|
||||
wrapFireworksProviderStream({
|
||||
provider: "fireworks",
|
||||
modelId: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
model: {
|
||||
api: "openai-responses",
|
||||
provider: "fireworks",
|
||||
id: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
} as Model<"openai-responses">,
|
||||
streamFn: undefined,
|
||||
} as never),
|
||||
).toBeUndefined();
|
||||
|
||||
expect(
|
||||
wrapFireworksProviderStream({
|
||||
provider: "fireworks-ai",
|
||||
modelId: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
model: {
|
||||
api: "openai-completions",
|
||||
provider: "fireworks-ai",
|
||||
id: "accounts/fireworks/routers/kimi-k2p5-turbo",
|
||||
} as Model<"openai-completions">,
|
||||
streamFn: undefined,
|
||||
} as never),
|
||||
).toBeTypeOf("function");
|
||||
|
||||
expect(
|
||||
wrapFireworksProviderStream({
|
||||
provider: "openai",
|
||||
modelId: "gpt-5.4",
|
||||
model: {
|
||||
api: "openai-completions",
|
||||
provider: "openai",
|
||||
id: "gpt-5.4",
|
||||
} as Model<"openai-completions">,
|
||||
streamFn: undefined,
|
||||
} as never),
|
||||
).toBeUndefined();
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue