Package org.bsc.spring.agentexecutor.function

Class AgentFunctionCallbackWrapper<I,O>

java.lang.Object

org.bsc.spring.agentexecutor.function.AgentFunctionCallbackWrapper<I,O>

All Implemented Interfaces:

BiFunction<I,org.springframework.ai.chat.model.ToolContext,O>, org.springframework.ai.model.function.FunctionCallback

public class AgentFunctionCallbackWrapper<I,O>
extends Object
implements BiFunction<I,org.springframework.ai.chat.model.ToolContext,O>, org.springframework.ai.model.function.FunctionCallback

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	AgentFunctionCallbackWrapper.Builder<I,O>

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	AgentFunctionCallbackWrapper(String name, String description, String inputTypeSchema, Class<I> inputType, Function<O,String> responseConverter, com.fasterxml.jackson.databind.ObjectMapper objectMapper, BiFunction<I,org.springframework.ai.chat.model.ToolContext,O> function)

Method Summary

Modifier and Type	Method	Description
O	apply(I input, org.springframework.ai.chat.model.ToolContext context)
static <I, O> AgentFunctionCallbackWrapper.Builder<I,O>	builder(BiFunction<I,org.springframework.ai.chat.model.ToolContext,O> biFunction)
static <I, O> AgentFunctionCallbackWrapper.Builder<I,O>	builder(Function<I,O> function)
String	call(String functionArguments)
String	call(String functionInput, org.springframework.ai.chat.model.ToolContext toolContext)
O	convertResponse(org.springframework.ai.chat.messages.ToolResponseMessage.ToolResponse toolResponse)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface java.util.function.BiFunction

andThen

Methods inherited from interface org.springframework.ai.model.function.FunctionCallback

getDescription, getInputTypeSchema, getName

Constructor Details

AgentFunctionCallbackWrapper

protected AgentFunctionCallbackWrapper(String name,
 String description,
 String inputTypeSchema,
 Class<I> inputType,
 Function<O,String> responseConverter,
 com.fasterxml.jackson.databind.ObjectMapper objectMapper,
 BiFunction<I,org.springframework.ai.chat.model.ToolContext,O> function)

Method Details

convertResponse

public O convertResponse(org.springframework.ai.chat.messages.ToolResponseMessage.ToolResponse toolResponse)

call

public String call(String functionInput,
 org.springframework.ai.chat.model.ToolContext toolContext)

Specified by:

call in interface org.springframework.ai.model.function.FunctionCallback

call

public String call(String functionArguments)

Specified by:

call in interface org.springframework.ai.model.function.FunctionCallback

apply

public O apply(I input,
 org.springframework.ai.chat.model.ToolContext context)

Specified by:

apply in interface BiFunction<I,org.springframework.ai.chat.model.ToolContext,O>

builder

public static <I,
O>
AgentFunctionCallbackWrapper.Builder<I,O> builder(BiFunction<I,org.springframework.ai.chat.model.ToolContext,O> biFunction)

builder

public static <I,
O>
AgentFunctionCallbackWrapper.Builder<I,O> builder(Function<I,O> function)