Skip to main content

mockforge_intelligence/ai_studio/
mod.rs

1//! Pillars: [AI]
2//!
3//! MockForge AI Studio - Unified AI Copilot
4//!
5//! This module provides a unified interface for all AI-powered features in MockForge,
6//! including natural language mock generation, AI-guided debugging, persona generation,
7//! and artifact freezing for deterministic testing.
8//!
9//! # Features
10//!
11//! - **Natural Language Mock Generation**: Generate mocks from conversational descriptions
12//! - **AI-Guided Debugging**: Analyze test failures and suggest fixes
13//! - **Persona Generation**: Create and tweak personas using AI
14//! - **Artifact Freezing**: Convert AI outputs to deterministic YAML/JSON
15//! - **Cost & Budget Management**: Track tokens and enforce budgets
16//!
17//! # Example Usage
18//!
19//! ```rust,ignore
20//! use mockforge_core::ai_studio::{ChatOrchestrator, ChatRequest};
21//! use mockforge_core::intelligent_behavior::IntelligentBehaviorConfig;
22//!
23//! async fn example() -> mockforge_core::Result<()> {
24//!     let config = IntelligentBehaviorConfig::default();
25//!     let orchestrator = ChatOrchestrator::new(config);
26//!
27//!     // Process a natural language command
28//!     let request = ChatRequest {
29//!         message: "Create a user API with CRUD operations".to_string(),
30//!         context: None,
31//!     };
32//!     let response = orchestrator.process(&request).await?;
33//!     Ok(())
34//! }
35//! ```
36
37pub mod api_critique;
38pub mod artifact_freezer;
39pub mod behavioral_simulator;
40pub mod budget_manager;
41pub mod chat_orchestrator;
42pub mod config;
43pub mod contract_diff_handler;
44pub mod conversation_store;
45pub mod debug_analyzer;
46pub mod debug_context;
47pub mod debug_context_integrator;
48mod frozen_generation;
49pub mod nl_mock_generator;
50pub mod org_controls;
51#[cfg(feature = "database")]
52pub mod org_controls_db;
53pub mod persona_generator;
54pub mod system_generator;
55
56pub use api_critique::{
57    AntiPattern, ApiCritique, ApiCritiqueEngine, ConsolidationOpportunity, CritiqueRequest,
58    HierarchyImprovement, NamingIssue, Redundancy, ResourceModelingSuggestion,
59    RestructuringRecommendations, ToneAnalysis, ToneIssue,
60};
61pub use artifact_freezer::{ArtifactFreezer, FreezeMetadata, FreezeRequest, FrozenArtifact};
62pub use behavioral_simulator::{
63    AppState, BehaviorPolicy, BehavioralSimulator, BehavioralTraits, CartState, CreateAgentRequest,
64    Intention, NarrativeAgent, NextAction, PolicyRule, SimulateBehaviorRequest,
65    SimulateBehaviorResponse,
66};
67pub use budget_manager::{AiFeature, BudgetConfig, BudgetManager, FeatureUsage, UsageStats};
68pub use chat_orchestrator::{
69    ChatContext, ChatIntent, ChatMessage, ChatOrchestrator, ChatRequest, ChatResponse,
70};
71pub use config::{AiStudioConfig, FreezeMode};
72pub use contract_diff_handler::{
73    BreakingChange, ContractDiffFilters, ContractDiffHandler, ContractDiffIntent,
74    ContractDiffQueryResult,
75};
76pub use conversation_store::{
77    get_conversation_store, initialize_conversation_store, ConversationStore,
78};
79pub use debug_analyzer::{
80    DebugAnalyzer, DebugRequest, DebugResponse, DebugSuggestion, LinkedArtifact,
81};
82pub use debug_context::{
83    ChaosContext, ContractContext, DebugContext, PersonaContext, RealityContext, ScenarioContext,
84};
85pub use debug_context_integrator::{
86    ChaosAccessor, ContractAccessor, DebugContextIntegrator, PersonaAccessor, RealityAccessor,
87    ScenarioAccessor,
88};
89pub use nl_mock_generator::{MockGenerationResult, MockGenerator};
90pub use org_controls::{
91    OrgAiControlsConfig, OrgBudgetConfig, OrgControls, OrgControlsAccessor, OrgRateLimitConfig,
92};
93#[cfg(feature = "database")]
94pub use org_controls_db::DbOrgControls;
95pub use persona_generator::{
96    PersonaGenerationRequest, PersonaGenerationResponse, PersonaGenerator,
97};
98pub use system_generator::{
99    AppliedSystem, GeneratedSystem, SystemArtifact, SystemGenerationRequest, SystemGenerator,
100    SystemMetadata,
101};