#language-model #procedural #proc-macro #definition #i32 #string #ai

macro ai_functions

Procedural macro for creating text from a function for use in Large Language Models

2 releases

0.1.1 Jun 7, 2023
0.1.0 Jun 7, 2023

#1111 in Procedural macros

Download history 99/week @ 2024-07-22 107/week @ 2024-07-29 88/week @ 2024-08-05 178/week @ 2024-08-12 119/week @ 2024-08-19 112/week @ 2024-08-26 214/week @ 2024-09-02 130/week @ 2024-09-09 157/week @ 2024-09-16 155/week @ 2024-09-23 144/week @ 2024-09-30 19/week @ 2024-10-07 106/week @ 2024-10-14 109/week @ 2024-10-21 151/week @ 2024-10-28 114/week @ 2024-11-04

481 downloads per month

MIT license

5KB

AI Function Crate

The ai_function crate provides a Rust procedural macro that allows you to transform any function into a function that returns its own definition as a string. This is useful for sending the function's code to a large language model for further processing.

Installation

Add the following to your Cargo.toml file:

[dependencies]
ai_function = "0.1.0"

Then run cargo build to download and compile the ai_function crate.

Usage

To use the ai_function macro, simply annotate your function with #[ai_function].

use ai_function::ai_function;

#[ai_function]
fn example_function(arg: i32) -> i32 {
  arg \* 2
}

When you call example_function(), instead of returning arg * 2, it will return a string containing the source code of the example_function.

Limitations

The #[ai_function] macro currently does not support functions with complex control flow like loops or conditionals. It only supports simple function definitions.

use ai_functions::ai_function;

#[ai_function]
fn add(a: i32, b: i32) -> i32 {
    a + b
}

fn main() {
    assert_eq!(add(2, 3), "fn add(a: i32, b: i32) -> i32 {\n    a + b\n}");
}

Dependencies

~250–700KB
~17K SLoC