ai-agent 0.13.4

Idiomatic agent sdk inspired by the claude code source leak
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

// Source: /data/home/swei/claudecode/openclaudecode/src/types/notebook.ts
//! Jupyter notebook utilities.

use serde::{Deserialize, Serialize};

/// A Jupyter notebook cell
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct NotebookCell {
    pub cell_type: String,
    pub source: Vec<String>,
    pub metadata: serde_json::Value,
    pub outputs: Option<Vec<serde_json::Value>>,
    pub execution_count: Option<u32>,
}

/// A Jupyter notebook
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Notebook {
    pub nbformat: u32,
    pub nbformat_minor: u32,
    pub metadata: NotebookMetadata,
    pub cells: Vec<NotebookCell>,
}

/// Notebook metadata
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct NotebookMetadata {
    pub kernelspec: Option<KernelSpec>,
    pub language_info: Option<LanguageInfo>,
}

/// Kernel specification
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct KernelSpec {
    pub name: String,
    pub display_name: String,
}

/// Language info
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct LanguageInfo {
    pub name: String,
    pub version: Option<String>,
}

/// Check if a file is a Jupyter notebook
pub fn is_notebook_file(path: &str) -> bool {
    path.ends_with(".ipynb")
}

/// Parse a notebook from JSON
pub fn parse_notebook(json: &str) -> Result<Notebook, serde_json::Error> {
    serde_json::from_str(json)
}

/// Extract code from notebook cells
pub fn extract_code_cells(notebook: &Notebook) -> Vec<String> {
    notebook
        .cells
        .iter()
        .filter(|c| c.cell_type == "code")
        .map(|c| c.source.join(""))
        .collect()
}