Skip to content

[Model Registry] Create Model and ModelVersion SQLAlchemy models#418

Merged
gelluisaac merged 2 commits into
Traqora:mainfrom
jaynomyaro:jy
Jun 27, 2026
Merged

[Model Registry] Create Model and ModelVersion SQLAlchemy models#418
gelluisaac merged 2 commits into
Traqora:mainfrom
jaynomyaro:jy

Conversation

@jaynomyaro

Copy link
Copy Markdown
Contributor

Summary
This PR introduces the foundational SQLAlchemy models for the Model Registry by adding Model and ModelVersion entities. These models establish the schema for registering machine learning models, tracking version history, and managing model lifecycle metadata.

Changes Made

  • Added the Model SQLAlchemy model to represent registered machine learning models.
  • Added the ModelVersion SQLAlchemy model to manage versioned model artifacts.
  • Defined relationships between Model and ModelVersion to support one-to-many version tracking.
  • Implemented primary keys, foreign keys, constraints, and indexes for data integrity and query performance.
  • Added common metadata fields such as name, description, version, status, timestamps, and ownership information where applicable.
  • Created database migrations for the new tables and relationships.
  • Added model validation to enforce required fields and valid relationships.
  • Implemented unit tests covering model creation, relationship mapping, constraints, and CRUD operations.
  • Updated developer documentation with the database schema, entity relationships, and usage examples.

Benefits

  • Establishes the core data model for the Model Registry.
  • Enables version tracking and lifecycle management for machine learning models.
  • Improves database consistency through well-defined relationships and constraints.
  • Provides a scalable foundation for future registry features such as deployments, approvals, and artifact management.
  • Increases confidence through comprehensive model and migration testing.

Testing

  • Verified successful creation and persistence of Model and ModelVersion records.
  • Confirmed one-to-many relationships function correctly.
  • Tested foreign key constraints and validation rules.
  • Validated database migrations execute successfully.
  • Ran the complete test suite to ensure no regressions.
    closed [Model Registry] Create Model and ModelVersion SQLAlchemy models #312

@drips-wave

drips-wave Bot commented Jun 27, 2026

Copy link
Copy Markdown

@jaynomyaro Great news! 🎉 Based on an automated assessment of this PR, the linked Wave issue(s) no longer count against your application limits.

You can now already apply to more issues while waiting for a review of this PR. Keep up the great work! 🚀

Learn more about application limits

@gelluisaac gelluisaac merged commit 009ca22 into Traqora:main Jun 27, 2026
3 of 12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Model Registry] Create Model and ModelVersion SQLAlchemy models

2 participants