mod.rs 859 Bytes
Newer Older
1
2
3
/// Tool parser module for handling function/tool calls in model outputs
///
/// This module provides infrastructure for parsing tool calls from various model formats.
4
// Core modules
5
6
7
8
9
10
11
pub mod errors;
pub mod partial_json;
pub mod registry;
pub mod state;
pub mod traits;
pub mod types;

12
13
14
// Parser implementations
pub mod parsers;

15
16
17
18
19
20
21
22
23
#[cfg(test)]
mod tests;

// Re-export commonly used types
pub use errors::{ToolParserError, ToolParserResult};
pub use registry::ParserRegistry;
pub use state::{ParsePhase, ParseState};
pub use traits::{PartialJsonParser, ToolParser};
pub use types::{FunctionCall, PartialToolCall, StreamResult, TokenConfig, ToolCall};
24
25

// Re-export parsers for convenience
26
pub use parsers::{
27
28
    DeepSeekParser, Glm4MoeParser, GptOssParser, JsonParser, KimiK2Parser, LlamaParser,
    MistralParser, PythonicParser, QwenParser, Step3Parser,
29
};