[AINode] Append model management IT#16938
Merged
CRZbulabula merged 4 commits intomasterfrom Dec 23, 2025
Merged
Conversation
Codecov Report✅ All modified and coverable lines are covered by tests. Additional details and impacted files@@ Coverage Diff @@
## master #16938 +/- ##
============================================
+ Coverage 39.18% 39.20% +0.01%
Complexity 212 212
============================================
Files 5048 5050 +2
Lines 335053 335124 +71
Branches 42707 42720 +13
============================================
+ Hits 131303 131369 +66
- Misses 203750 203755 +5 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
Contributor
There was a problem hiding this comment.
Pull request overview
This PR adds integration tests for AINode model management and refines the model registration workflow, particularly for user-defined Transformers models.
Key Changes:
- Enhanced error handling for Transformers model registration with explicit exception propagation
- Refactored model storage to remove private field prefix (
_transformers_registered→transformers_registered) - Added comprehensive integration tests for user-defined model lifecycle (register, inference/forecast, drop)
Reviewed changes
Copilot reviewed 9 out of 9 changed files in this pull request and generated 3 comments.
Show a summary per file
| File | Description |
|---|---|
| ainode.thrift | Reordered TRegisterModelReq parameters (modelId before uri) for better API consistency |
| model_storage.py | Improved exception handling in model registration; refactored field naming; enhanced thread safety in registration flow |
| model_info.py | Removed private field prefix for transformers_registered; updated Chronos2 config reference |
| model_manager.py | Added generic exception handler for model registration; removed _refresh() call from show_models for performance |
| AINodeTestUtils.java | Added reusable data preparation methods for tree and table dialects |
| AINodeModelManageIT.java | Added comprehensive integration tests for user-defined model management in both dialects |
| AINodeForecastIT.java | Refactored to use shared data preparation utility; made forecastTableFunctionTest static for reuse |
| AINodeCallInferenceIT.java | Refactored to use shared data preparation utility; made callInferenceTest static for reuse |
| AINodeWrapper.java | Updated CACHE_BUILT_IN_MODEL_PATH to use more general directory path |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeModelManageIT.java
Outdated
Show resolved
Hide resolved
Contributor
|
LGTM! |
…deModelManageIT.java Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Contributor
|
LGTM! |
|
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
3 participants
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.
To verify the model management framework, we append corresponding correctness and error handling CI in this PR, employing the Chronos2 model as example.