Skip to main content

batuta/
lib.rs

1#![allow(unused_imports)]
2#![allow(dead_code)]
3// Production code: warn on unwrap (enforced here, allowed at workspace level for tests)
4#![cfg_attr(not(test), warn(clippy::unwrap_used))]
5#![cfg_attr(
6    test,
7    allow(
8        clippy::unwrap_used,
9        clippy::panic,
10        clippy::float_cmp,
11        clippy::float_cmp_const,
12        clippy::field_reassign_with_default,
13        clippy::unwrap_in_result,
14        clippy::single_char_pattern,
15        clippy::needless_borrows_for_generic_args,
16        clippy::manual_range_contains,
17        clippy::assertions_on_constants,
18    )
19)]
20
21#[macro_use]
22#[allow(unused_macros)]
23mod generated_contracts;
24#[macro_use]
25#[allow(unused_macros)]
26mod batuta_contracts;
27
28// ANSI colors module (DEP-REDUCE: replaces colored crate)
29pub mod ansi_colors;
30pub mod timing;
31
32// Library exports for Batuta orchestration framework
33pub mod analyzer;
34pub mod audit;
35pub mod backend;
36pub mod config;
37pub mod experiment;
38pub mod numpy_converter;
39pub mod oracle;
40pub mod parf;
41pub mod pipeline;
42pub mod pipeline_analysis;
43pub mod plugin;
44pub mod pytorch_converter;
45pub mod recipes;
46pub mod report;
47pub mod sklearn_converter;
48pub mod tools;
49pub mod types;
50
51// WASM-specific API (only compiled for wasm32 target)
52#[cfg(feature = "wasm")]
53pub mod wasm;
54
55// Model Serving Ecosystem (native-only)
56#[cfg(feature = "native")]
57pub mod serve;
58
59// Stack Dependency Orchestration (native-only)
60#[cfg(feature = "native")]
61pub mod stack;
62
63// Data Platforms Integration
64pub mod data;
65
66// Visualization Frameworks Integration
67pub mod viz;
68
69// Content Creation Tooling
70pub mod content;
71
72// TUI Graph Visualization (native-only, requires ratatui)
73#[cfg(feature = "native")]
74pub mod tui;
75
76// Popperian Falsification Checklist (native-only)
77#[cfg(feature = "native")]
78pub mod falsification;
79
80// Automated defect detection and localization (native-only)
81#[cfg(feature = "native")]
82pub mod bug_hunter;
83
84// Playbook Pipeline Orchestration (native-only)
85#[cfg(feature = "native")]
86pub mod playbook;
87
88// Stack Compliance Engine (native-only)
89#[cfg(feature = "native")]
90pub mod comply;
91
92// HuggingFace Hub Integration (native-only)
93#[cfg(feature = "native")]
94pub mod hf;
95
96// MCP (Model Context Protocol) Server
97#[cfg(feature = "native")]
98pub mod mcp;
99
100// Autonomous Agent Runtime (perceive-reason-act loop)
101#[cfg(feature = "agents")]
102pub mod agent;
103
104// Re-export key types for convenience
105pub use backend::{Backend, BackendSelector, OpComplexity};
106pub use numpy_converter::{NumPyConverter, NumPyOp};
107pub use parf::{
108    CodePattern, DeadCode, DependencyKind, FileDependency, ParfAnalyzer, SymbolKind,
109    SymbolReference,
110};
111pub use pipeline::{
112    AnalysisStage, BuildStage, OptimizationStage, PipelineStage, TranspilationPipeline,
113    TranspilationStage, ValidationStage, ValidationStrategy,
114};
115pub use plugin::{PluginMetadata, PluginRegistry, PluginStage, TranspilerPlugin};
116pub use pytorch_converter::{PyTorchConverter, PyTorchOperation, RealizarOperation};
117pub use report::{MigrationReport, ReportFormat};
118pub use sklearn_converter::{AprenderAlgorithm, SklearnAlgorithm, SklearnConverter};
119pub use types::{Language, PhaseStatus, ProjectAnalysis, WorkflowPhase, WorkflowState};
120
121// Audit Trail exports (entrenar-style provenance)
122pub use audit::{
123    new_audit_collector, record_failure, record_success, ChainVerification, ContextSnapshot,
124    HashChainEntry, PipelineAuditCollector, PipelinePath, PipelineTrace, StageTimer,
125};
126
127// Experiment Tracking exports (Entrenar v1.8.0 integration)
128pub use experiment::{
129    AppleChip, ArtifactSignature, ArtifactType, CitationMetadata, CitationType, ComputeDevice,
130    ComputeIntensity, CostMetrics, CostPerformanceBenchmark, CostPerformancePoint, CpuArchitecture,
131    CreditRole, EnergyMetrics, ExperimentError, ExperimentRun, ExperimentStorage, GpuVendor,
132    InMemoryExperimentStorage, ModelParadigm, OfflineRegistryConfig, Orcid, PlatformEfficiency,
133    PreRegistration, ResearchArtifact, ResearchContributor, RunStatus, SignatureAlgorithm,
134    SovereignArtifact, SovereignDistribution, TpuVersion,
135};
136
137// Orchestration Recipes exports
138pub use recipes::{
139    CiCdBenchmarkRecipe, CostPerformanceBenchmarkRecipe, ExperimentTrackingConfig,
140    ExperimentTrackingRecipe, RecipeResult, ResearchArtifactRecipe, SovereignDeploymentConfig,
141    SovereignDeploymentRecipe,
142};
143
144// Oracle Mode exports
145pub use oracle::{
146    Backend as OracleBackend, Capability, CapabilityCategory, ComponentRecommendation,
147    ComputeRecommendation, DataSize, DistributionRecommendation, HardwareSpec, IntegrationPattern,
148    KnowledgeGraph, OpComplexity as OracleOpComplexity, OptimizationTarget, OracleQuery,
149    OracleResponse, ProblemDomain, QueryConstraints, QueryEngine, QueryPreferences, Recommender,
150    StackComponent, StackLayer,
151};
152
153// Model Serving Ecosystem exports (native-only)
154#[cfg(feature = "native")]
155pub use serve::{
156    BackendSelector as ServeBackendSelector, ChatMessage, ChatTemplateEngine, CircuitBreakerConfig,
157    ContextManager, ContextWindow, CostCircuitBreaker, FailoverConfig, FailoverManager,
158    PrivacyTier, RejectReason, Role, RouterConfig, RoutingDecision, ServingBackend,
159    SpilloverRouter, StreamingContext, TemplateFormat, TokenEstimator, TruncationStrategy,
160};
161
162// Stack Dependency Orchestration exports (native-only)
163#[cfg(feature = "native")]
164pub use stack::{
165    // Diagnostics types
166    render_dashboard,
167    AndonStatus,
168    Anomaly,
169    AnomalyCategory,
170    ComponentMetrics,
171    ComponentNode,
172    // Quality types
173    ComponentQuality,
174    ErrorForecaster,
175    ForecastMetrics,
176    GraphMetrics,
177    HealthStatus,
178    HealthSummary,
179    HeroImageResult,
180    ImageFormat,
181    IsolationForest,
182    QualityChecker,
183    QualityGrade,
184    QualityIssue,
185    QualitySummary,
186    Score,
187    StackDiagnostics,
188    StackLayer as QualityStackLayer,
189    StackQualityReport,
190};
191
192// Popperian Falsification Checklist exports (native-only)
193#[cfg(feature = "native")]
194pub use falsification::{
195    evaluate_critical_only, evaluate_project, CheckItem, CheckStatus, ChecklistResult, Evidence,
196    EvidenceType, Severity, TpsGrade,
197};
198
199// Stack Compliance Engine exports (native-only)
200#[cfg(feature = "native")]
201pub use comply::{ComplyConfig, ComplyReport, ComplyReportFormat, StackComplyEngine};