openfunctions-rs 0.1.0

A universal framework for creating and managing LLM tools and agents
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

//! Parsers for extracting tool definitions from source code.
//!
//! This module provides parsers for various programming languages to extract
//! structured `ToolDefinition`s from comments in source files. This allows
//! developers to define their tools inline with their code.

use crate::core::tool::ToolLanguage;
use crate::models::ToolDefinition;
use anyhow::Result;

pub mod bash;
pub mod javascript;
pub mod python;

/// A parser for extracting tool definitions from source code.
///
/// `ToolParser` delegates to a language-specific parser based on the
/// provided `ToolLanguage`.
pub struct ToolParser {
    language: ToolLanguage,
}

impl ToolParser {
    /// Creates a new `ToolParser` for the given language.
    pub fn new(language: ToolLanguage) -> Self {
        Self { language }
    }

    /// Parses the given source code and extracts a `ToolDefinition`.
    ///
    /// # Arguments
    ///
    /// * `source` - The source code to parse as a string slice.
    ///
    /// # Returns
    ///
    /// A `Result` containing the parsed `ToolDefinition` or an error if
    /// parsing fails.
    pub fn parse(&self, source: &str) -> Result<ToolDefinition> {
        match self.language {
            ToolLanguage::Bash => bash::parse(source),
            ToolLanguage::JavaScript => javascript::parse(source),
            ToolLanguage::Python => python::parse(source),
        }
    }
}