pctx_codegen 0.3.0

Code generation utilities for pctx
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use pctx_codegen::{RootSchema, Tool, ToolSet};
use pctx_type_check_runtime::type_check;
use serde::Deserialize;

const BASIC_TOOL: &str = include_str!("./fixtures/tools/basic.yml");
const NESTED_TYPES_TOOL: &str = include_str!("./fixtures/tools/nested_types.yml");
const NO_OUTPUT_TOOL: &str = include_str!("./fixtures/tools/no_output.yml");
const NO_INPUT_TOOL: &str = include_str!("./fixtures/tools/no_input.yml");
const NO_INPUT_OR_OUTPUT_TOOL: &str = include_str!("./fixtures/tools/no_input_or_output.yml");
const ALL_OPTIONAL_INPUT_TOOL: &str = include_str!("./fixtures/tools/all_optional_input.yml");

#[derive(Debug, Deserialize)]
struct ToolFixture {
    pub name: String,
    pub description: Option<String>,
    pub input_schema: Option<RootSchema>,
    pub output_schema: Option<RootSchema>,
}

impl ToolFixture {
    fn to_tool(&self) -> Tool {
        Tool::new(
            &self.name,
            self.description.clone(),
            self.input_schema.clone(),
            self.output_schema.clone(),
        )
        .expect("Tool::new failed")
    }
}

fn load_fixture(yml: &str) -> ToolFixture {
    serde_yaml::from_str(yml).expect("Failed to parse tool fixture YAML")
}

// --- Tool tests ---

macro_rules! tool_test {
    ($test_name:ident, $fixture:expr) => {
        #[tokio::test]
        async fn $test_name() {
            let fixture = load_fixture($fixture);
            let tool = fixture.to_tool();

            let impl_code =
                pctx_codegen::format::format_ts(&tool.ts_fn_impl(Some("test_server".into())));
            let check_res = type_check(&impl_code).expect("failed typecheck");

            assert!(
                check_res.success,
                "tool fn_impl failed typecheck: {check_res:?}"
            );
            insta::assert_snapshot!(format!("{}__fn_impl.ts", stringify!($test_name)), impl_code);
        }
    };
}

tool_test!(test_basic, BASIC_TOOL);
tool_test!(test_nested_types, NESTED_TYPES_TOOL);
tool_test!(test_no_output, NO_OUTPUT_TOOL);
tool_test!(test_no_input, NO_INPUT_TOOL);
tool_test!(test_no_input_or_output, NO_INPUT_OR_OUTPUT_TOOL);
tool_test!(test_all_optional_input, ALL_OPTIONAL_INPUT_TOOL);

// --- ToolSet tests ---

#[test]
fn test_toolset_namespace() {
    let basic = load_fixture(BASIC_TOOL);
    let notif = load_fixture(NESTED_TYPES_TOOL);

    let toolset = ToolSet::new(
        Some("my_tools".into()),
        "A collection of utility tools",
        vec![basic.to_tool(), notif.to_tool()],
    );

    insta::assert_snapshot!(
        "toolset__namespace_interface.ts",
        toolset.ts_namespace_declaration(true)
    );
}