feat: Add MLflow integration utilities

- client: Connection management and helpers
- tracker: General experiment tracking
- inference_tracker: Async metrics for NATS handlers
- model_registry: Model registration with KServe metadata
- kfp_components: Kubeflow Pipeline components
- experiment_comparison: Run comparison tools
- cli: Command-line interface
This commit is contained in:
2026-02-01 20:43:13 -05:00
parent 944bd8bc06
commit 2df3f27af7
10 changed files with 3315 additions and 1 deletions

19
requirements.txt Normal file
View File

@@ -0,0 +1,19 @@
# MLflow Utils Module Requirements
# Core MLflow
mlflow>=2.10.0
# Database backends
psycopg2-binary>=2.9.0 # PostgreSQL (CNPG)
boto3>=1.34.0 # S3-compatible artifact storage (optional)
# For async tracking
aiohttp>=3.9.0
# YAML generation for KServe manifests
PyYAML>=6.0
# Already in chat-handler/voice-assistant requirements:
# httpx (for health checks)
# Used but typically installed with mlflow:
# numpy
# pandas