mod.rs 830 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
pub mod errors;
pub mod partial_json;
7
pub mod python_literal_parser;
8
9
10
11
12
pub mod registry;
pub mod state;
pub mod traits;
pub mod types;

13
14
15
// Parser implementations
pub mod parsers;

16
17
18
19
20
21
22
23
24
#[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};
25
26

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