feat(llm): attribute aiml-provider traffic with the AIMLAPI partner id - #5
Merged
Merged
Conversation
The aiml LiteLLM provider (VERIFIED_AIML_MODELS) sent no partner attribution, unlike the CLI's OAuth onboarding (openhands_cli/aimlapi/ config.py), so AIMLAPI had no way to attribute this traffic to OpenHands. Add _aiml_headers(), gated on the parsed provider being "aiml" (unlike _openrouter_headers(), which is unconditional), mirroring the existing OpenRouter HTTP-Referer/X-Title pattern in apply_extra_headers(). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
REST API breakage checks (OpenAPI) — ✅ PASSEDResult: ✅ PASSED |
Python API breakage checks — ✅ PASSEDResult: ✅ PASSED |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Reopened as #4 was auto-closed when its stacked base branch (sdk-upstream-sync-2026-08-21, merged as #3) got deleted on merge. Rebased onto main.
The
aimlLiteLLM provider (VERIFIED_AIML_MODELS) currently sends no partner attribution, unlike the CLI fork's OAuth onboarding (openhands_cli/aimlapi/config.py, which setsX-AIMLAPI-Partner-ID/X-AIMLAPI-Source) — so AIMLAPI has no way to attribute SDK/Canvas traffic to OpenHands short of sniffing User-Agent.Adds
LLM._aiml_headers(), mirroring the existing_openrouter_headers()→apply_extra_headers()pattern, but gated on the parsed LiteLLM provider actually beingaiml(OpenRouter's headers are sent unconditionally today; an unrecognizedX-AIMLAPI-Partner-IDheader on every other provider's request would just be dead weight, so this one only fires foraiml/*models).Same partner id as the CLI (
part_uDVajKg3xPLrOdNdQetOtoGA, OpenHands' provisioned partner — valid on both staging and prod).Tests:
tests/sdk/llm/test_llm.py(direct_aiml_headers()gating) +tests/sdk/llm/test_chat_options.py(header actually lands inextra_headersviaselect_chat_options, and does not leak to a non-aiml model). Fulltests/sdksuite: 5958 passed, 0 failed.