golem_rust_macro/
lib.rs

1// Copyright 2024-2026 Golem Cloud
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use proc_macro::TokenStream;
16use proc_macro_crate::{FoundCrate, crate_name};
17use proc_macro2::Span;
18use syn::DeriveInput;
19
20use crate::transaction::golem_operation_impl;
21
22mod agentic;
23pub(crate) mod recursion;
24mod transaction;
25mod value;
26
27#[proc_macro_derive(IntoValue, attributes(flatten_value, unit_case))]
28pub fn derive_into_value(input: TokenStream) -> TokenStream {
29    let ast: DeriveInput = syn::parse(input).expect("derive input");
30    let golem_rust_crate_ident = get_golem_rust_crate_ident();
31
32    value::derive_into_value(&ast, &golem_rust_crate_ident)
33}
34
35#[proc_macro_derive(FromValueAndType, attributes(flatten_value, unit_case))]
36pub fn derive_from_value_and_type(input: TokenStream) -> TokenStream {
37    let ast: DeriveInput = syn::parse(input).expect("derive input");
38    let golem_rust_crate_ident = get_golem_rust_crate_ident();
39
40    value::derive_from_value_and_type(&ast, &golem_rust_crate_ident)
41}
42
43#[proc_macro_derive(MultimodalSchema)]
44pub fn derive_multimodal(input: TokenStream) -> TokenStream {
45    agentic::derive_multimodal(input)
46}
47
48#[proc_macro_derive(Schema)]
49pub fn derive_schema(input: TokenStream) -> TokenStream {
50    let golem_rust_crate_ident = get_golem_rust_crate_ident();
51
52    agentic::derive_schema(input, &golem_rust_crate_ident)
53}
54
55#[proc_macro_derive(ConfigSchema, attributes(config_schema))]
56pub fn derive_config_schema(input: TokenStream) -> TokenStream {
57    let golem_rust_crate_ident = get_golem_rust_crate_ident();
58
59    agentic::derive_config_schema(input, &golem_rust_crate_ident)
60}
61
62#[proc_macro_derive(AllowedLanguages, attributes(code))]
63pub fn derive_allowed_languages(input: TokenStream) -> TokenStream {
64    let golem_rust_crate_ident = get_golem_rust_crate_ident();
65
66    agentic::derive_allowed_languages(input, &golem_rust_crate_ident)
67}
68
69#[proc_macro_derive(AllowedMimeTypes, attributes(mime_type))]
70pub fn derive_allowed_mimetypes(input: TokenStream) -> TokenStream {
71    let golem_rust_crate_ident = get_golem_rust_crate_ident();
72
73    agentic::derive_allowed_mime_types(input, &golem_rust_crate_ident)
74}
75
76#[proc_macro_attribute]
77pub fn description(_attr: TokenStream, item: TokenStream) -> TokenStream {
78    item
79}
80
81#[proc_macro_attribute]
82pub fn prompt(_attr: TokenStream, item: TokenStream) -> TokenStream {
83    item
84}
85
86#[proc_macro_attribute]
87pub fn endpoint(_attr: TokenStream, item: TokenStream) -> TokenStream {
88    item
89}
90
91/// Defines a function as an `Operation` that can be used in transactions
92#[proc_macro_attribute]
93pub fn golem_operation(attr: TokenStream, item: TokenStream) -> TokenStream {
94    golem_operation_impl(attr, item)
95}
96
97#[proc_macro_attribute]
98pub fn agent_definition(attr: TokenStream, item: TokenStream) -> TokenStream {
99    agentic::agent_definition_impl(attr, item)
100}
101
102#[proc_macro_attribute]
103pub fn agent_implementation(attr: TokenStream, item: TokenStream) -> TokenStream {
104    agentic::agent_implementation_impl(attr, item)
105}
106
107// get the identifier of golem_rust crate to use for referencing the `golem-rust` crate
108// within the macros. This handles the case where the crate is renamed in Cargo.toml
109// or when the macro is used within the `golem-rust` crate itself.
110fn get_golem_rust_crate_ident() -> syn::Ident {
111    match crate_name("golem-rust") {
112        Ok(FoundCrate::Itself) => syn::Ident::new("crate", Span::call_site()),
113        Ok(FoundCrate::Name(name)) => syn::Ident::new(&name, Span::call_site()),
114        Err(_) => syn::Ident::new("golem_rust", Span::call_site()),
115    }
116}
golem_rust_macro/lib.rs

golem_rust_macro/
lib.rs
