fix(moonshot): bound normalized tool-schema expansion

[luvs01]

Summary

Combine #5464's Moonshot schema corrections with the request-wide expansion budget carried by 7c05f085d81eb9ae85fb7dedd9b1f6add72b575f and 3f161b8c3b21b3ee671d79168e2df90598314757.

• Re-normalize composed properties and infer appropriate scalar/object types without turning non-object allOf schemas into objects.
• Share the inline-byte allowance across the entire request's tool catalog, preserving the existing depth, node and reference-expansion bounds.
• Account for growth caused by normalization before retaining an expanded reference. If type inference pushes an otherwise admissible raw target over the remaining allowance, preserve the bare reference instead of retaining an oversized copy.
• Preserve all original contributions and follow-ups, document the contract in chat compatibility and ADR-0355, and add regressions covering the combined normalization/budget boundary.

This is an integration proposal. The original contributor PR #5464 remains open; any closure or change to its review state requires its author or maintainer to authorize it. Contributor attribution is preserved.

Verification

• Base: dev 7bd9985963fd344a2c512fa5cf3c0f57f6ba83bd. Head: 93831de7520df0ceaff43b9dd6bb4d251683bd90.
• On Bun 1.4.2, focused Moonshot, endpoint, xAI schema, Azure router schema and file-size checks: 50 passed / 190 assertions / 0 failures.
• The normalization-growth regression failed before the integration follow-up; both combined cases passed after it (8 assertions).
• bun run typecheck, bun run structure:check, bun run privacy:scan and base-to-head git diff --check passed.
• The source budget implementation and tests match their carried state before layering the complete upstream corrections. Original author and cherry-pick provenance are preserved, including the module-split migration.
• Existing bounds remain depth 64, 4,096 nodes, 512 reference expansions and 1 MiB shared inline bytes. No limit or source test timeout was raised.
• Full suite, exact-head cross-platform CI, live-provider acceptance and independent review remain pending. This PR stays draft; tests build synthetic requests and make no provider call.

Checklist

• Scope stays focused and avoids unrelated cleanup.
• Docs or release notes were updated when needed.
• Security-sensitive changes were reviewed for secrets, auth, and unsafe defaults.

[coderabbitai]

Important

Draft PR not reviewed

Draft PRs are not automatically reviewed by default.

• Trigger a manual review

To automatically review draft PRs, update your CodeRabbit configuration:

reviews:
  auto_review:
    drafts: true

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands.}

[github-actions]

✅ Deterministic PR hygiene checks passed.

[lidge-jun]

리뷰 · 우선순위 61 / 80

이 PR은 Moonshot/Kimi용 도구 스키마를 고치는 통합안입니다. 원래 #5464에서 하던 “형제 키워드가 붙은 $ref를 다시 정규화하고, properties/allOf/const/enum에 빠진 type을 채워 주기”와, 요청 하나에서 $ref를 펼칠 때 커지는 크기를 막는 1 MiB 바이트 예산을 한데 모았습니다. 예산은 도구마다 따로가 아니라 toolsToChatFormat 한 번당 하나를 나눠 씁니다. 펼치기 전에 원본 크기만큼 먼저 깎고, 타입 추론이나 안쪽 정규화로 커진 분량도 남은 예산에 맞으면 쓰고, 넘치면 예전처럼 맨 $ref만 남깁니다. 깊이 64·노드 4096·참조 펼침 512 한도는 그대로입니다. ADR-0355와 chat-compat에도 같은 계약을 적어 두었고, 회귀 테스트가 여러 개 붙었습니다. base는 선호하는 dev이고, draft입니다. 원작 #5464는 아직 열려 있고, 본문도 닫는 일은 원 작성자·메인테이너 허락이 필요하다고 적었습니다.

라인 - src/adapters/openai-chat/tool-schema.ts normalizeMoonshotSchemaNode — 원본 바이트를 예산에서 뺀 뒤 정규화하고, “커진 분량”이 남은 예산을 넘기면 { $ref }만 돌려줍니다. 이때 이미 깎인 원본 바이트·펼침 횟수·안쪽에서 쓴 예산을 되돌리지 않습니다. 결과는 맨 $ref인데 예산만 줄어, 같은 요청의 뒤 도구·형제 $ref가 더 일찍 막힐 수 있습니다. 실패 시 환불하거나, 커진 분량까지 맞춰 본 뒤에만 예산을 확정하는 편이 맞습니다.
라인 - serializedJsonBytesUpTo — 실제 JSON.stringify 전체가 아니라 괄호·쉼표·키를 대략 센 값입니다. 가드로 쓰기엔 충분해 보이지만, 캐시 한도를 MOONSHOT_MAX_INLINED_SCHEMA_BYTES로 자르면 아주 큰 대상의 “진짜 크기”를 작게 잡을 여지는 있습니다. 지금 테스트가 큰 맵을 막는지 한 번만 눈여겨보면 됩니다.
라인 - #5464 — 이 PR이 그 정규화·타입 추론을 이미 가져왔습니다. 둘 다 draft로 열려 있으면 나중에 같은 수정이 두 번 들어갈 수 있습니다.
라인 - draft / 검증 — 본문 기준 집중 테스트·typecheck·structure는 통과했다고 적혀 있고, 전체 스위트·호스티드 CI·보안 체크리스트는 아직입니다. 머지 전에 초록을 봐야 합니다.

메인테이너의 판단이 필요한 지점

성장 한도 초과 때 예산을 환불할지, 아니면 “맨 $ref로 떨어지는 순간에도 비용은 쓴 것으로 본다”를 의도된 정책으로 둘지 정해 주세요. 또 #5464를 이 PR이 대체한다고 보고 닫을지(원 작성자 확인 포함), draft를 언제 ready로 올릴지도 정하면 됩니다.

너의 추천

방향은 맞습니다. 요청 단위 공유 예산·성장분 과금·구성 property 재정규화·문서·회귀가 한줄로 이어집니다. 머지 전에 성장 실패 시 환불(또는 확정 전 측정)만 고치거나, 의도라면 주석·테스트로 “환불 없음”을 박아 두세요. CI가 초록이면 #5464는 중복으로 닫는 쪽을 추천합니다. types.ts/config.ts 쪼개기·preview deploy는 이번 범위와 무관합니다.

이 댓글은 grok-bot이 작성했습니다

[lidge-jun]

Carried into #5619 at 93831de with your authorship kept, plus a follow-up repair: a rejected reference expansion now returns the inline budget it consumed, and nested copies are charged only once. The provider-driven object typing from #5464 is kept and documented. Closing in favor of #5619. Thank you!