mod.rs 947 Bytes
Newer Older
1
2
3
// SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

4
pub mod config;
5
pub mod dsml;
6
7
pub mod harmony;
pub mod json;
8
pub mod parsers;
9
pub mod pythonic;
10
pub mod response;
11
12
#[cfg(test)]
pub mod tests;
13
pub mod tools;
14
pub mod xml;
15
16

// Re-export main types and functions for convenience
17
pub use config::{JsonParserConfig, ParserConfig, ToolCallConfig, XmlParserConfig};
18
pub use dsml::try_tool_call_parse_dsml;
19
pub use harmony::parse_tool_calls_harmony_complete;
20
pub use json::try_tool_call_parse_json;
21
22
23
24
pub use parsers::{
    detect_and_parse_tool_call, detect_tool_call_start, find_tool_call_end_position,
    try_tool_call_parse,
};
25
pub use pythonic::try_tool_call_parse_pythonic;
26
27
pub use response::{CalledFunction, ToolCallResponse, ToolCallType};
pub use tools::{try_tool_call_parse_aggregate, try_tool_call_parse_stream};
28
pub use xml::try_tool_call_parse_xml;