github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-10 01:33:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(anthropic): preserve pi-ai default betas when injecting anthropic-beta header (openclaw#19789) thanks @minupla

Browse Source 
Verified:
- pnpm build
- pnpm check
- pnpm test:macmini

Co-authored-by: minupla <42547246+minupla@users.noreply.github.com>
Co-authored-by: Tak Hoffman <781889+Takhoffman@users.noreply.github.com>
This commit is contained in:
Dale Babiy
2026-02-19 22:23:00 -05:00
committed by GitHub
parent 38b4fb5d55
commit 10dab4f2c7
3 changed files with 89 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/agents/pi-embedded-runner-extraparams.e2e.test.ts
View File

@@ -144,15 +144,65 @@ describe("applyExtraParamsToAgent", () => {
} as Model<"anthropic-messages">;
const context: Context = { messages: [] };
void agent.streamFn?.(model, context, { headers: { "X-Custom": "1" } });
// Simulate pi-agent-core passing apiKey in options (API key, not OAuth token)
void agent.streamFn?.(model, context, {
apiKey: "sk-ant-api03-test",
headers: { "X-Custom": "1" },
});
expect(calls).toHaveLength(1);
expect(calls[0]?.headers).toEqual({
"X-Custom": "1",
"anthropic-beta": "context-1m-2025-08-07",
// Includes pi-ai default betas (preserved to avoid overwrite) + context1m
"anthropic-beta":
"fine-grained-tool-streaming-2025-05-14,interleaved-thinking-2025-05-14,context-1m-2025-08-07",
});
});
it("preserves oauth-2025-04-20 beta when context1m is enabled with an OAuth token", () => {
const calls: Array<SimpleStreamOptions | undefined> = [];
const baseStreamFn: StreamFn = (_model, _context, options) => {
calls.push(options);
return {} as ReturnType<StreamFn>;
};
const agent = { streamFn: baseStreamFn };
const cfg = {
agents: {
defaults: {
models: {
"anthropic/claude-sonnet-4-6": {
params: {
context1m: true,
},
},
},
},
},
};
applyExtraParamsToAgent(agent, cfg, "anthropic", "claude-sonnet-4-6");
const model = {
api: "anthropic-messages",
provider: "anthropic",
id: "claude-sonnet-4-6",
} as Model<"anthropic-messages">;
const context: Context = { messages: [] };
// Simulate pi-agent-core passing an OAuth token (sk-ant-oat-*) as apiKey
void agent.streamFn?.(model, context, {
apiKey: "sk-ant-oat01-test-oauth-token",
headers: { "X-Custom": "1" },
});
expect(calls).toHaveLength(1);
const betaHeader = calls[0]?.headers?.["anthropic-beta"] as string;
// Must include the OAuth-required betas so they aren't stripped by pi-ai's mergeHeaders
expect(betaHeader).toContain("oauth-2025-04-20");
expect(betaHeader).toContain("claude-code-20250219");
expect(betaHeader).toContain("context-1m-2025-08-07");
});
it("merges existing anthropic-beta headers with configured betas", () => {
const { calls, agent } = createOptionsCaptureAgent();
const cfg = buildAnthropicModelConfig("anthropic/claude-sonnet-4-5", {
@@ -170,12 +220,14 @@ describe("applyExtraParamsToAgent", () => {
const context: Context = { messages: [] };
void agent.streamFn?.(model, context, {
apiKey: "sk-ant-api03-test",
headers: { "anthropic-beta": "prompt-caching-2024-07-31" },
});
expect(calls).toHaveLength(1);
expect(calls[0]?.headers).toEqual({
"anthropic-beta": "prompt-caching-2024-07-31,files-api-2025-04-14,context-1m-2025-08-07",
"anthropic-beta":
"prompt-caching-2024-07-31,fine-grained-tool-streaming-2025-05-14,interleaved-thinking-2025-05-14,files-api-2025-04-14,context-1m-2025-08-07",
});
});