[PR #27709] refactor(api): add SQLAlchemy 2.x Mapped type hints to Message model #31840

Closed
opened 2026-02-21 20:50:14 -05:00 by yindo · 0 comments
Owner

Original Pull Request: https://github.com/langgenius/dify/pull/27709

State: closed
Merged: Yes


Important

  1. Make sure you have read our contribution guidelines
  2. Ensure there is an associated issue and you have been assigned to it
  3. Use the correct syntax to link this PR: Fixes #<issue number>.

Summary

This PR adds comprehensive Mapped type hints to all fields in the Message model following SQLAlchemy 2.x conventions for improved type safety and IDE support.

Changes:

  • Add Decimal import for numeric field typing
  • Apply Mapped[Type] annotations to 24 fields in the Message model
  • Correctly handle nullable vs non-nullable fields based on column definitions
  • Use appropriate types: str, Decimal, dict[str, Any], float, datetime

Fixes #27708

Screenshots

N/A - Backend refactoring only (no UI changes)

Checklist

  • This change requires a documentation update, included: Dify Document
  • I understand that this PR may be closed in case there was no previous discussion or issues. (This doesn't apply to typos!)
  • I've added a test for each change that was introduced, and I tried as much as possible to make a single atomic change.
  • I've updated the documentation accordingly.
  • I ran dev/reformat(backend) and cd web && npx lint-staged(frontend) to appease the lint gods
**Original Pull Request:** https://github.com/langgenius/dify/pull/27709 **State:** closed **Merged:** Yes --- > [!IMPORTANT] > > 1. Make sure you have read our [contribution guidelines](https://github.com/langgenius/dify/blob/main/CONTRIBUTING.md) > 1. Ensure there is an associated issue and you have been assigned to it > 1. Use the correct syntax to link this PR: `Fixes #<issue number>`. ## Summary This PR adds comprehensive `Mapped` type hints to all fields in the `Message` model following SQLAlchemy 2.x conventions for improved type safety and IDE support. **Changes:** - Add `Decimal` import for numeric field typing - Apply `Mapped[Type]` annotations to 24 fields in the `Message` model - Correctly handle nullable vs non-nullable fields based on column definitions - Use appropriate types: `str`, `Decimal`, `dict[str, Any]`, `float`, `datetime` Fixes #27708 ## Screenshots N/A - Backend refactoring only (no UI changes) ## Checklist - [ ] This change requires a documentation update, included: [Dify Document](https://github.com/langgenius/dify-docs) - [x] I understand that this PR may be closed in case there was no previous discussion or issues. (This doesn't apply to typos!) - [x] I've added a test for each change that was introduced, and I tried as much as possible to make a single atomic change. - [x] I've updated the documentation accordingly. - [x] I ran `dev/reformat`(backend) and `cd web && npx lint-staged`(frontend) to appease the lint gods
yindo added the pull-request label 2026-02-21 20:50:14 -05:00
yindo closed this issue 2026-02-21 20:50:14 -05:00
Sign in to join this conversation.
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: langgenius/dify#31840