mcli-framework 7.9.5__py3-none-any.whl → 7.9.7__py3-none-any.whl

mcli/ml/database/__init__.py

@@ -0,0 +1,40 @@
+"""Database models and utilities"""
+
+from .models import (
+    Alert,
+    BacktestResult,
+    Base,
+    DataVersion,
+    Experiment,
+    FeatureSet,
+    Model,
+    Politician,
+    Portfolio,
+    Prediction,
+    StockData,
+    Trade,
+    User,
+)
+from .session import AsyncSessionLocal, SessionLocal, async_engine, engine, get_async_db, get_db
+
+__all__ = [
+    "Base",
+    "User",
+    "Trade",
+    "Politician",
+    "StockData",
+    "Prediction",
+    "Portfolio",
+    "Alert",
+    "BacktestResult",
+    "Experiment",
+    "Model",
+    "FeatureSet",
+    "DataVersion",
+    "get_db",
+    "get_async_db",
+    "SessionLocal",
+    "AsyncSessionLocal",
+    "engine",
+    "async_engine",
+]

mcli/ml/experimentation/__init__.py

@@ -0,0 +1,29 @@
+"""ML Experimentation and A/B Testing Framework"""
+
+from .ab_testing import (
+    ABTestingFramework,
+    ExperimentConfig,
+    ExperimentResult,
+    ExperimentStatus,
+    Metric,
+    MetricsCollector,
+    StatisticalAnalyzer,
+    TrafficSplitter,
+    UserAssignment,
+    Variant,
+    VariantType,
+)
+
+__all__ = [
+    "ABTestingFramework",
+    "ExperimentConfig",
+    "Variant",
+    "VariantType",
+    "Metric",
+    "ExperimentStatus",
+    "ExperimentResult",
+    "UserAssignment",
+    "TrafficSplitter",
+    "MetricsCollector",
+    "StatisticalAnalyzer",
+]

mcli/ml/features/__init__.py

@@ -0,0 +1,39 @@
+"""Feature Engineering Module for Stock Recommendation Models"""
+
+from .ensemble_features import (
+    DynamicFeatureSelector,
+    EnsembleFeatureBuilder,
+    FeatureInteractionEngine,
+)
+from .political_features import (
+    CongressionalTrackingFeatures,
+    PolicyImpactFeatures,
+    PoliticalInfluenceFeatures,
+)
+from .recommendation_engine import (
+    RecommendationConfig,
+    RecommendationResult,
+    StockRecommendationEngine,
+)
+from .stock_features import (
+    CrossAssetFeatures,
+    MarketRegimeFeatures,
+    StockRecommendationFeatures,
+    TechnicalIndicatorFeatures,
+)
+
+__all__ = [
+    "StockRecommendationFeatures",
+    "TechnicalIndicatorFeatures",
+    "MarketRegimeFeatures",
+    "CrossAssetFeatures",
+    "PoliticalInfluenceFeatures",
+    "CongressionalTrackingFeatures",
+    "PolicyImpactFeatures",
+    "EnsembleFeatureBuilder",
+    "FeatureInteractionEngine",
+    "DynamicFeatureSelector",
+    "StockRecommendationEngine",
+    "RecommendationConfig",
+    "RecommendationResult",
+]

mcli/ml/mlops/__init__.py

@@ -0,0 +1,19 @@
+"""MLOps components for ML pipeline management"""
+
+from .experiment_tracker import ExperimentRun, ExperimentTracker, MLflowConfig, ModelRegistry
+from .model_serving import ModelEndpoint, ModelServer, PredictionService
+from .pipeline_orchestrator import MLPipeline, PipelineConfig, PipelineExecutor, PipelineStep
+
+__all__ = [
+    "ExperimentTracker",
+    "ModelRegistry",
+    "MLflowConfig",
+    "ExperimentRun",
+    "ModelServer",
+    "PredictionService",
+    "ModelEndpoint",
+    "MLPipeline",
+    "PipelineStep",
+    "PipelineConfig",
+    "PipelineExecutor",
+]

mcli/ml/models/__init__.py

@@ -0,0 +1,90 @@
+"""ML Models for Stock Recommendation System"""
+
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+import torch
+
+from .base_models import BaseStockModel, ModelMetrics, ValidationResult
+from .ensemble_models import (
+    AttentionStockPredictor,
+    CNNFeatureExtractor,
+    DeepEnsembleModel,
+    EnsembleConfig,
+    EnsembleTrainer,
+    LSTMStockPredictor,
+    ModelConfig,
+    TransformerStockModel,
+)
+from .recommendation_models import (
+    RecommendationConfig,
+    RecommendationTrainer,
+    StockRecommendationModel,
+)
+
+# Model registry
+_loaded_models: Dict[str, Any] = {}
+
+
+async def load_production_models():
+    """Load production models into memory"""
+    from mcli.ml.config import settings
+    from mcli.ml.logging import get_logger
+
+    logger = get_logger(__name__)
+    model_dir = settings.model.model_dir
+
+    if not model_dir.exists():
+        model_dir.mkdir(parents=True, exist_ok=True)
+        return
+
+    for model_path in model_dir.glob("*.pt"):
+        try:
+            model_id = model_path.stem
+            model = torch.load(model_path, map_location=settings.model.device)
+            _loaded_models[model_id] = model
+            logger.info(f"Loaded model: {model_id}")
+        except Exception as e:
+            logger.error(f"Failed to load model {model_path}: {e}")
+
+
+async def get_model_by_id(model_id: str):
+    """Get loaded model by ID"""
+    from mcli.ml.config import settings
+
+    if model_id not in _loaded_models:
+        # Try to load from disk
+        model_path = settings.model.model_dir / f"{model_id}.pt"
+        if model_path.exists():
+            _loaded_models[model_id] = torch.load(model_path, map_location=settings.model.device)
+
+    return _loaded_models.get(model_id)
+
+
+def initialize_models():
+    """Initialize models on startup"""
+    from mcli.ml.logging import get_logger
+
+    logger = get_logger(__name__)
+    logger.info("Initializing ML models...")
+
+
+__all__ = [
+    "DeepEnsembleModel",
+    "AttentionStockPredictor",
+    "TransformerStockModel",
+    "LSTMStockPredictor",
+    "CNNFeatureExtractor",
+    "EnsembleTrainer",
+    "ModelConfig",
+    "EnsembleConfig",
+    "BaseStockModel",
+    "ModelMetrics",
+    "ValidationResult",
+    "StockRecommendationModel",
+    "RecommendationTrainer",
+    "RecommendationConfig",
+    "load_production_models",
+    "get_model_by_id",
+    "initialize_models",
+]

mcli/ml/models/ensemble_models.py

@@ -9,6 +9,7 @@ import pandas as pd
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+
 from mcli.ml.models.base_models import BaseStockModel, ModelMetrics, ValidationResult
 
 logger = logging.getLogger(__name__)

mcli/ml/models/recommendation_models.py

@@ -10,6 +10,7 @@ import pandas as pd
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+
 from mcli.ml.models.base_models import BaseStockModel, ModelMetrics, ValidationResult
 from mcli.ml.models.ensemble_models import DeepEnsembleModel, EnsembleConfig, ModelConfig
 

mcli/ml/monitoring/__init__.py

@@ -0,0 +1,25 @@
+"""ML Model Monitoring and Drift Detection"""
+
+from .drift_detection import (
+    AlertSeverity,
+    ConceptDriftDetector,
+    DataProfile,
+    DriftAlert,
+    DriftType,
+    ModelMetrics,
+    ModelMonitor,
+    OutlierDetector,
+    StatisticalDriftDetector,
+)
+
+__all__ = [
+    "ModelMonitor",
+    "StatisticalDriftDetector",
+    "ConceptDriftDetector",
+    "OutlierDetector",
+    "DriftAlert",
+    "DriftType",
+    "AlertSeverity",
+    "ModelMetrics",
+    "DataProfile",
+]

mcli/ml/optimization/__init__.py

@@ -0,0 +1,27 @@
+"""Advanced Portfolio Optimization"""
+
+from .portfolio_optimizer import (
+    AdvancedPortfolioOptimizer,
+    BaseOptimizer,
+    BlackLittermanOptimizer,
+    CVaROptimizer,
+    KellyCriterionOptimizer,
+    MeanVarianceOptimizer,
+    OptimizationConstraints,
+    OptimizationObjective,
+    PortfolioAllocation,
+    RiskParityOptimizer,
+)
+
+__all__ = [
+    "AdvancedPortfolioOptimizer",
+    "OptimizationObjective",
+    "OptimizationConstraints",
+    "PortfolioAllocation",
+    "MeanVarianceOptimizer",
+    "RiskParityOptimizer",
+    "BlackLittermanOptimizer",
+    "CVaROptimizer",
+    "KellyCriterionOptimizer",
+    "BaseOptimizer",
+]

mcli/ml/optimization/optimize.py

@@ -18,13 +18,15 @@ def cli():
 @click.option("--end-date", required=True, help="End date (YYYY-MM-DD)")
 @click.option("--risk-free-rate", default=0.02, help="Risk-free rate")
 @click.option("--output", help="Output file for results")
-def optimize_portfolio(symbols: str, start_date: str, end_date: str, risk_free_rate: float, output: str):
+def optimize_portfolio(
+    symbols: str, start_date: str, end_date: str, risk_free_rate: float, output: str
+):
     """Optimize portfolio allocation for given symbols."""
     symbol_list = [s.strip() for s in symbols.split(",")]
     info(f"Optimizing portfolio for: {', '.join(symbol_list)}")
     info(f"Period: {start_date} to {end_date}")
     info(f"Risk-free rate: {risk_free_rate:.2%}")
-    
+
     # TODO: Implement actual optimization
     error("Portfolio optimization not yet implemented")
 
@@ -37,7 +39,7 @@ def efficient_frontier(symbols: str, points: int):
     symbol_list = [s.strip() for s in symbols.split(",")]
     info(f"Generating efficient frontier for: {', '.join(symbol_list)}")
     info(f"Points: {points}")
-    
+
     # TODO: Implement efficient frontier generation
     error("Efficient frontier generation not yet implemented")
 
@@ -48,4 +50,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

mcli/ml/predictions/__init__.py

@@ -0,0 +1,5 @@
+"""ML Predictions Module"""
+
+from .prediction_engine import PoliticianTradingPredictor
+
+__all__ = ["PoliticianTradingPredictor"]

mcli/ml/preprocessing/__init__.py

@@ -0,0 +1,24 @@
+"""ML Data Preprocessing Module"""
+
+from .data_cleaners import MissingValueHandler, OutlierDetector, TradingDataCleaner
+from .feature_extractors import (
+    MarketFeatureExtractor,
+    PoliticianFeatureExtractor,
+    SentimentFeatureExtractor,
+    TemporalFeatureExtractor,
+)
+from .ml_pipeline import MLDataPipeline, MLDataPipelineConfig
+from .politician_trading_preprocessor import PoliticianTradingPreprocessor
+
+__all__ = [
+    "PoliticianTradingPreprocessor",
+    "PoliticianFeatureExtractor",
+    "MarketFeatureExtractor",
+    "TemporalFeatureExtractor",
+    "SentimentFeatureExtractor",
+    "TradingDataCleaner",
+    "OutlierDetector",
+    "MissingValueHandler",
+    "MLDataPipeline",
+    "MLDataPipelineConfig",
+]

mcli/ml/scripts/__init__.py

@@ -0,0 +1 @@
+"""ML scripts module."""

mcli/ml/serving/__init__.py

@@ -0,0 +1 @@
+"""Model serving module for MCLI ML system."""

mcli/ml/serving/serve.py

@@ -20,7 +20,7 @@ def start_server(model: str, port: int, host: str):
     """Start the model serving server."""
     info(f"Starting model server for: {model}")
     info(f"Serving on {host}:{port}")
-    
+
     # TODO: Implement actual model serving
     error("Model serving functionality not yet implemented")
 
@@ -47,4 +47,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

mcli/ml/trading/__init__.py

@@ -0,0 +1,63 @@
+"""Trading module for portfolio management and trade execution"""
+
+from mcli.ml.trading.alpaca_client import (
+    AlpacaTradingClient,
+    create_trading_client,
+    get_alpaca_config_from_env,
+)
+from mcli.ml.trading.models import (  # Enums; Database models; Pydantic models
+    OrderCreate,
+    OrderResponse,
+    OrderSide,
+    OrderStatus,
+    OrderType,
+    Portfolio,
+    PortfolioCreate,
+    PortfolioPerformanceSnapshot,
+    PortfolioResponse,
+    PortfolioType,
+    Position,
+    PositionResponse,
+    PositionSide,
+    RiskLevel,
+    TradingAccount,
+    TradingAccountCreate,
+    TradingOrder,
+    TradingSignal,
+    TradingSignalResponse,
+)
+from mcli.ml.trading.paper_trading import PaperTradingEngine
+from mcli.ml.trading.risk_management import RiskManager
+from mcli.ml.trading.trading_service import TradingService
+
+__all__ = [
+    # Enums
+    "OrderStatus",
+    "OrderType",
+    "OrderSide",
+    "PositionSide",
+    "PortfolioType",
+    "RiskLevel",
+    # Database models
+    "TradingAccount",
+    "Portfolio",
+    "Position",
+    "TradingOrder",
+    "PortfolioPerformanceSnapshot",
+    "TradingSignal",
+    # Pydantic models
+    "TradingAccountCreate",
+    "PortfolioCreate",
+    "OrderCreate",
+    "PositionResponse",
+    "OrderResponse",
+    "PortfolioResponse",
+    "TradingSignalResponse",
+    # Services
+    "TradingService",
+    "AlpacaTradingClient",
+    "create_trading_client",
+    "get_alpaca_config_from_env",
+    "RiskManager",
+    "PaperTradingEngine",
+]

mcli/ml/training/__init__.py

@@ -0,0 +1,7 @@
+"""ML model training module"""
+
+from .train_model import PoliticianTradingNet, fetch_training_data
+from .train_model import main as train_model
+from .train_model import prepare_dataset
+
+__all__ = ["PoliticianTradingNet", "train_model", "fetch_training_data", "prepare_dataset"]

mcli/ml/training/train.py

@@ -19,12 +19,19 @@ def cli():
 @click.option("--batch-size", default=32, help="Batch size for training")
 @click.option("--learning-rate", default=0.001, help="Learning rate")
 @click.option("--output-dir", help="Directory to save trained model")
-def train_model(model_type: str, dataset: str, epochs: int, batch_size: int, learning_rate: float, output_dir: str):
+def train_model(
+    model_type: str,
+    dataset: str,
+    epochs: int,
+    batch_size: int,
+    learning_rate: float,
+    output_dir: str,
+):
     """Train a model with the specified parameters."""
     info(f"Training {model_type} model")
     info(f"Dataset: {dataset}")
     info(f"Epochs: {epochs}, Batch size: {batch_size}, Learning rate: {learning_rate}")
-    
+
     # TODO: Implement actual training logic
     error("Model training functionality not yet implemented")
 
@@ -36,7 +43,7 @@ def resume_training(checkpoint: str, epochs: int):
     """Resume training from a checkpoint."""
     info(f"Resuming training from: {checkpoint}")
     info(f"Additional epochs: {epochs}")
-    
+
     # TODO: Implement resume functionality
     error("Resume training not yet implemented")
 
@@ -46,14 +53,14 @@ def resume_training(checkpoint: str, epochs: int):
 def train_politician_trading(output_dir: str):
     """Train the politician trading prediction model."""
     info("Training politician trading prediction model...")
-    
+
     try:
         # Import the actual training function
         from mcli.ml.training.train_model import train_politician_trading_model
-        
+
         # Run the training
         metrics = train_politician_trading_model(output_dir)
-        
+
         if metrics:
             success(f"Training completed! Final loss: {metrics.get('final_loss', 'N/A')}")
         else:
@@ -70,4 +77,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

mcli/mygroup/__init__.py

@@ -0,0 +1,3 @@
+"""
+Mygroup commands for mcli.
+"""

mcli/public/__init__.py

@@ -0,0 +1 @@
+# logger.info("I am in mcli.public.__init__.py")

mcli/public/commands/__init__.py

@@ -0,0 +1,2 @@
+# User-generated commands directory
+# Commands created through the MCLI chat interface are stored here

mcli/self/__init__.py

@@ -0,0 +1,3 @@
+"""
+Self-management module for mcli.
+"""

mcli/workflow/daemon/__init__.py

@@ -0,0 +1,15 @@
+"""
+Daemon service for command management and execution.
+
+This module provides a background daemon service that can store, manage, and execute
+commands written in various programming languages (Python, Node.js, Lua, Shell).
+Commands are stored in a SQLite database with embeddings for similarity search and
+hierarchical grouping.
+
+The daemon CLI commands are now loaded from portable JSON files in ~/.mcli/commands/
+"""
+
+from .daemon import Command, CommandExecutor, DaemonService
+
+# Export main components
+__all__ = ["Command", "CommandExecutor", "DaemonService"]

mcli/workflow/dashboard/__init__.py

@@ -0,0 +1,5 @@
+"""Dashboard workflow module."""
+
+from .dashboard_cmd import dashboard
+
+__all__ = ["dashboard"]