Skip to main content

batuta/oracle/knowledge_graph/
domain_mappings.rs

1//! Problem domain to capability mappings.
2
3use super::super::types::ProblemDomain;
4use super::types::KnowledgeGraph;
5
6impl KnowledgeGraph {
7    /// Initialize problem domain to capability mappings
8    pub(crate) fn initialize_domain_mappings(&mut self) {
9        use ProblemDomain::{
10            CMigration, DataPipeline, DeepLearning, DistributedCompute, GraphAnalytics, Inference,
11            LinearAlgebra, MediaProduction, ModelServing, Profiling, PythonMigration,
12            ShellMigration, SpeechRecognition, SupervisedLearning, Testing, UnsupervisedLearning,
13            Validation, VectorSearch,
14        };
15
16        self.domain_capabilities.insert(
17            SupervisedLearning,
18            vec![
19                "linear_regression".into(),
20                "logistic_regression".into(),
21                "decision_tree".into(),
22                "random_forest".into(),
23                "gbm".into(),
24                "naive_bayes".into(),
25                "knn".into(),
26                "svm".into(),
27            ],
28        );
29
30        self.domain_capabilities.insert(
31            UnsupervisedLearning,
32            vec!["kmeans".into(), "pca".into(), "dbscan".into(), "hierarchical".into()],
33        );
34
35        self.domain_capabilities.insert(
36            DeepLearning,
37            vec!["autograd".into(), "lora".into(), "qlora".into(), "quantization".into()],
38        );
39
40        self.domain_capabilities.insert(
41            Inference,
42            vec!["model_serving".into(), "batching".into(), "moe_routing".into()],
43        );
44
45        self.domain_capabilities.insert(
46            SpeechRecognition,
47            vec![
48                "speech_recognition".into(),
49                "streaming_transcription".into(),
50                "multilingual".into(),
51                "whisper_quantization".into(),
52            ],
53        );
54
55        self.domain_capabilities.insert(
56            LinearAlgebra,
57            vec!["vector_ops".into(), "matrix_ops".into(), "simd".into(), "gpu".into()],
58        );
59
60        self.domain_capabilities.insert(
61            VectorSearch,
62            vec!["vector_store".into(), "similarity_search".into(), "knn_search".into()],
63        );
64
65        self.domain_capabilities.insert(
66            GraphAnalytics,
67            vec!["pathfinding".into(), "centrality".into(), "community_detection".into()],
68        );
69
70        self.domain_capabilities.insert(
71            PythonMigration,
72            vec!["type_inference".into(), "sklearn_to_aprender".into(), "numpy_to_trueno".into()],
73        );
74
75        self.domain_capabilities
76            .insert(CMigration, vec!["ownership_inference".into(), "unsafe_elimination".into()]);
77
78        self.domain_capabilities
79            .insert(ShellMigration, vec!["script_conversion".into(), "cli_generation".into()]);
80
81        self.domain_capabilities.insert(
82            DistributedCompute,
83            vec![
84                "work_stealing".into(),
85                "cpu_executor".into(),
86                "gpu_executor".into(),
87                "remote_executor".into(),
88            ],
89        );
90
91        self.domain_capabilities.insert(
92            DataPipeline,
93            vec!["csv".into(), "parquet".into(), "json".into(), "streaming".into()],
94        );
95
96        self.domain_capabilities.insert(
97            ModelServing,
98            vec!["model_serving".into(), "lambda".into(), "container".into(), "edge".into()],
99        );
100
101        self.domain_capabilities.insert(
102            Testing,
103            vec![
104                "coverage_check".into(),
105                "mutation_testing".into(),
106                "tdg_scoring".into(),
107                "parity_checking".into(),
108                "oracle_generation".into(),
109                "falsification_testing".into(),
110                "quantization_drift".into(),
111                "roundtrip_validation".into(),
112            ],
113        );
114
115        self.domain_capabilities.insert(
116            Profiling,
117            vec!["syscall_trace".into(), "flamegraph".into(), "golden_trace_comparison".into()],
118        );
119
120        self.domain_capabilities.insert(
121            Validation,
122            vec![
123                "privacy_audit".into(),
124                "quality_gates".into(),
125                "complexity_analysis".into(),
126                "contract_parsing".into(),
127                "scaffold_generation".into(),
128                "kani_verification".into(),
129                "probar_generation".into(),
130                "traceability_audit".into(),
131                "binding_registry".into(),
132            ],
133        );
134
135        self.domain_capabilities.insert(
136            MediaProduction,
137            vec![
138                "video_rendering".into(),
139                "mlt_xml".into(),
140                "ffmpeg_encode".into(),
141                "transition_blend".into(),
142                "course_production".into(),
143                "audio_processing".into(),
144                "subtitle_burn_in".into(),
145                "media_concat".into(),
146                "transcription_qa".into(),
147                "vocabulary_enrichment".into(),
148                "content_generation".into(),
149                "svg_asset_pipeline".into(),
150                "course_quality_scoring".into(),
151                "course_publishing".into(),
152            ],
153        );
154    }
155}