raxit-core 0.1.2

Core security scanning engine for AI agent applications
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

//! Framework-specific extractors
//!
//! Extract agent assets from different AI agent frameworks

pub mod autogen;
pub mod crewai;
pub mod langgraph;
pub mod pydantic_ai;
pub mod swarm;
// Future framework extractors will be added here:
// etc.

use crate::error::Result;
use crate::schema::{Agent, Memory, Model, Tool};
use std::path::Path;

/// Extracted assets from a single file
#[derive(Debug, Default)]
pub struct ExtractedAssets {
    pub agents: Vec<Agent>,
    pub tools: Vec<Tool>,
    pub models: Vec<Model>,
    pub memory: Vec<Memory>,
}

/// Extract assets from a Python file based on detected framework
pub fn extract_from_file(path: &Path, framework: &str) -> Result<ExtractedAssets> {
    match framework {
        "pydantic-ai" => pydantic_ai::extract(path),
        "langgraph" => langgraph::extract(path),
        "crewai" => crewai::extract(path),
        "autogen" => autogen::extract(path),
        "swarm" => swarm::extract(path),
        _ => {
            // Unknown framework, return empty results
            Ok(ExtractedAssets::default())
        }
    }
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_extractor_module() {
        // Placeholder test - module compiles successfully
    }
}