Skip to content

refactor(ai): convert chat_sdk.ai module to package for chat/ai subpath #116

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
patrick-chinchill merged 3 commits into from
May 30, 2026
Merged

refactor(ai): convert chat_sdk.ai module to package for chat/ai subpath #116

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ad864d1
refactor(ai): convert chat_sdk.ai module to package for chat/ai subpa…
claude May 29, 2026
a5ff616
fix(scripts): update AI test-fidelity mapping after module move (review)
patrick-chinchill May 30, 2026
85a8420
fix(ai): unbreak strict fidelity check + re-export TEXT_MIME_PREFIXES
claude May 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion scripts/verify_test_fidelity.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
"packages/chat/src/markdown.test.ts": "tests/test_markdown_faithful.py",
"packages/chat/src/streaming-markdown.test.ts": "tests/test_streaming_markdown.py",
"packages/chat/src/serialization.test.ts": "tests/test_serialization.py",
"packages/chat/src/ai.test.ts": "tests/test_ai.py",
"packages/chat/src/ai.test.ts": "tests/test_ai_messages.py",
"packages/chat/src/from-full-stream.test.ts": "tests/test_from_full_stream.py",
}

Expand Down
37 changes: 37 additions & 0 deletions src/chat_sdk/ai/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
"""AI SDK integration for the chat SDK.

Python port of the ``chat/ai`` subpath. Mirrors the upstream structure where
``ai.ts`` was split into ``ai/messages.ts`` (and, in later PRs, ``ai/tools.ts``)
to make room for tool factories.

Re-exports everything the former ``chat_sdk.ai`` module exposed so existing
imports such as ``from chat_sdk.ai import to_ai_messages`` keep working.
"""

from __future__ import annotations

from chat_sdk.ai.messages import (
TEXT_MIME_PREFIXES,
AiAssistantMessage,
AiFilePart,
AiImagePart,
AiMessage,
AiMessagePart,
AiTextPart,
AiUserMessage,
ToAiMessagesOptions,
to_ai_messages,
)
Comment on lines +13 to +24

@chatgpt-codex-connector chatgpt-codex-connector Bot May 30, 2026

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P2 Badge Re-export the existing MIME prefix constant

For consumers that imported the non-underscored TEXT_MIME_PREFIXES from the old chat_sdk.ai module (for example to inspect or reuse the text-file attachment policy), this package shim now raises ImportError because it only imports the message types and to_ai_messages. Since this refactor is intended to preserve the old chat_sdk.ai import surface, please re-export the constant from chat_sdk.ai.messages as well.

Useful? React with 👍 / 👎.


__all__ = [
"TEXT_MIME_PREFIXES",
"AiAssistantMessage",
"AiFilePart",
"AiImagePart",
"AiMessage",
"AiMessagePart",
"AiTextPart",
"AiUserMessage",
"ToAiMessagesOptions",
"to_ai_messages",
]
2 changes: 1 addition & 1 deletion src/chat_sdk/ai.py → src/chat_sdk/ai/messages.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Convert chat Messages to AI SDK format.

Python port of ai.ts.
Python port of ``ai/messages.ts``.
"""

from __future__ import annotations
Expand Down
0 tests/test_ai.py → tests/test_ai_messages.py
View file
Open in desktop
File renamed without changes.
Loading