@@ -88,27 +88,10 @@ describe("createOpenAIThinkingLevelWrapper", () => {
8888});
89899090it("overrides existing reasoning.effort from upstream wrappers", () => {
91-const baseStreamFn: StreamFn = (model, _context, options) => {
92-const payload: Record<string, unknown> = {
93-model: model.id,
94-reasoning: { effort: "none" },
95-};
96-options?.onPayload?.(payload, model);
97-return createAssistantMessageEventStream();
98-};
99-100-const payloads: Array<Record<string, unknown>> = [];
101-const capture: StreamFn = (model, context, options) => {
102-return baseStreamFn(model, context, {
103- ...options,
104-onPayload: (payload, m) => {
105-options?.onPayload?.(payload, m);
106-payloads.push(structuredClone(payload as Record<string, unknown>));
107-},
108-});
109-};
110-111-const wrapped = createOpenAIThinkingLevelWrapper(capture, "medium");
91+const { baseStreamFn, payloads } = createPayloadCapture({
92+initialReasoning: { effort: "none" },
93+});
94+const wrapped = createOpenAIThinkingLevelWrapper(baseStreamFn, "medium");
11295void wrapped(codexModel, { messages: [] }, {});
1139611497expect(payloads[0]?.reasoning).toEqual({ effort: "medium" });
@@ -121,27 +104,10 @@ describe("createOpenAIThinkingLevelWrapper", () => {
121104});
122105123106it("preserves other reasoning properties when overriding effort", () => {
124-const baseStreamFn: StreamFn = (model, _context, options) => {
125-const payload: Record<string, unknown> = {
126-model: model.id,
127-reasoning: { effort: "none", summary: "auto" },
128-};
129-options?.onPayload?.(payload, model);
130-return createAssistantMessageEventStream();
131-};
132-133-const payloads: Array<Record<string, unknown>> = [];
134-const capture: StreamFn = (model, context, options) => {
135-return baseStreamFn(model, context, {
136- ...options,
137-onPayload: (payload, m) => {
138-options?.onPayload?.(payload, m);
139-payloads.push(structuredClone(payload as Record<string, unknown>));
140-},
141-});
142-};
143-144-const wrapped = createOpenAIThinkingLevelWrapper(capture, "high");
107+const { baseStreamFn, payloads } = createPayloadCapture({
108+initialReasoning: { effort: "none", summary: "auto" },
109+});
110+const wrapped = createOpenAIThinkingLevelWrapper(baseStreamFn, "high");
145111void wrapped(codexModel, { messages: [] }, {});
146112147113expect(payloads[0]?.reasoning).toEqual({ effort: "high", summary: "auto" });
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。