Skip to main content

batuta/
lib.rs

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