fix(core): root re-export deprecated chat/ai type aliases

[patrick-chinchill]

Summary

Pre-existing parity gap from the 0.4.30 pre-ship upstream-parity audit (core-reexports group; not a regression).

Upstream packages/chat/src/index.ts:8-27 re-exports toAiMessages plus eight deprecated AI type aliases from the package root for backwards compatibility, each marked @deprecated and pointing at the canonical chat/ai subpath:

• AiAssistantMessage
• AiFilePart
• AiImagePart
• AiMessage
• AiMessagePart
• AiTextPart
• AiUserMessage
• ToAiMessagesOptions

Ours (src/chat_sdk/__init__.py) re-exported only to_ai_messages, so hasattr(chat_sdk, "AiMessage") == False. This adds the eight deprecated root re-exports from chat_sdk.ai (which stays the canonical home), matching upstream's deprecated-alias surface.

Changes

• src/chat_sdk/__init__.py — import the eight AI type aliases from chat_sdk.ai (alongside the existing to_ai_messages) and list them in __all__. The root re-export is the same object as the chat_sdk.ai canonical type, so identity/isinstance checks agree. No new optional-dep cost: chat_sdk.ai is already imported at the root for to_ai_messages, and the aliased types are plain TypedDict/dataclass/Union with no ai/zod peer-dep imports.
• tests/test_root_reexports.py — new test asserting each alias resolves at the root, is identical to its chat_sdk.ai home, and appears in chat_sdk.__all__; plus that to_ai_messages stays re-exported.

Verification

Gauntlet green from the worktree:

• ruff check — all checks passed
• ruff format --check — 258 files already formatted
• audit_test_quality.py — 0 hard failures
• pyrefly check — 0 errors
• pytest tests/ — 4777 passed, 3 skipped

The three alias-resolution tests fail on the pre-fix code (aliases absent from root + __all__) and pass after.

Summary by CodeRabbit

• Improvements
  • AI message types (AiMessage, AiUserMessage, AiAssistantMessage) and message components (AiTextPart, AiImagePart, AiFilePart) are now directly accessible from the package root for simpler imports.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Repository UI

Review profile: CHILL

Plan: Pro Plus

Run ID: 66c0f32d-235c-4bb0-9555-8aeae2ef0f8f

📥 Commits

Reviewing files that changed from the base of the PR and between 17aa41a and b43e8d9.

📒 Files selected for processing (2)

• src/chat_sdk/__init__.py
• tests/test_root_reexports.py

---

📝 Walkthrough

Walkthrough

chat_sdk/__init__.py is updated to import and re-export eight AI type aliases (AiAssistantMessage, AiFilePart, AiImagePart, AiMessage, AiMessagePart, AiTextPart, AiUserMessage, ToAiMessagesOptions) from chat_sdk.ai alongside the existing to_ai_messages, and all are added to __all__. A new test module verifies each alias resolves at the root, is the canonical object from chat_sdk.ai, and is listed in __all__.

Changes

AI Type Alias Root Re-exports

Layer / File(s)	Summary
Re-export imports and __all__ extension src/chat_sdk/__init__.py	Broadens the import from chat_sdk.ai to include eight AI message part/type aliases and ToAiMessagesOptions, and adds all of them to __all__.
Root re-export contract tests tests/test_root_reexports.py	Introduces _DEPRECATED_AI_TYPE_ALIASES and four test functions that assert attribute presence, object identity against chat_sdk.ai, __all__ membership, and continued availability of to_ai_messages.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~3 minutes

Poem

🐇 Hop hop, the aliases bloom,
Re-exported to lighten the gloom!
AiMessage here, AiFilePart there,
All in __all__ beyond compare.
The root package greets them with cheer~ 🌸

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title 'fix(core): root re-export deprecated chat/ai type aliases' is specific and accurately summarizes the main change: adding type alias re-exports to the package root for backwards compatibility parity with upstream.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing Touches

📝 Generate docstrings

• Create stacked PR
• Commit on current branch

---

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 and usage tips.}

[gemini-code-assist]

Code Review

This pull request re-exports several deprecated AI type aliases from 'chat_sdk.ai' at the root package level in 'src/chat_sdk/init.py' to maintain backwards compatibility and mirror the upstream TypeScript SDK. It also adds a new test suite in 'tests/test_root_reexports.py' to verify that these aliases resolve correctly, match their canonical objects, and are included in 'all'. There are no review comments, so I have no feedback to provide.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.