Package com.google.genkit.ai

Class GenerateAction

java.lang.Object

com.google.genkit.ai.GenerateAction

All Implemented Interfaces:

Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>, Registerable

public class GenerateAction
extends Object
implements Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

GenerateAction is a utility action that provides a unified interface for generating content from AI models. It's registered at /util/generate and is used by the Dev UI.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	GenerateAction.GenerateActionOptions	Options for the generate utility action.

Constructor Summary

Constructors

Constructor	Description
GenerateAction(Registry registry)

Method Summary

Modifier and Type	Method	Description
static GenerateAction	define(Registry registry)	Defines and registers the generate utility action.
ActionDesc	getDesc()	Returns the descriptor of the action containing metadata, schemas, etc.
Map<String,Object>	getInputSchema()	Returns the JSON schema for the action's input type.
Map<String,Object>	getMetadata()	Returns additional metadata for the action.
String	getName()	Returns the name of the action.
Map<String,Object>	getOutputSchema()	Returns the JSON schema for the action's output type.
ActionType	getType()	Returns the type of the action.
void	register(Registry registry)	Registers this primitive with the given registry.
ModelResponse	run(ActionContext ctx, GenerateAction.GenerateActionOptions options)	Runs the action with the given input.
ModelResponse	run(ActionContext ctx, GenerateAction.GenerateActionOptions options, Consumer<ModelResponseChunk> streamCallback)	Runs the action with the given input and streaming callback.
com.fasterxml.jackson.databind.JsonNode	runJson(ActionContext ctx, com.fasterxml.jackson.databind.JsonNode input, Consumer<com.fasterxml.jackson.databind.JsonNode> streamCallback)	Runs the action with JSON input and returns JSON output.
ActionRunResult<com.fasterxml.jackson.databind.JsonNode>	runJsonWithTelemetry(ActionContext ctx, com.fasterxml.jackson.databind.JsonNode input, Consumer<com.fasterxml.jackson.databind.JsonNode> streamCallback)	Runs the action with JSON input and returns the result with telemetry information.

Methods inherited from class java.lang.Object

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

Constructor Details

GenerateAction

public GenerateAction(Registry registry)

Method Details

define

public static GenerateAction define(Registry registry)

Defines and registers the generate utility action.

Parameters:

registry - the registry to register with

Returns:

the generate action

getName

public String getName()

Description copied from interface: Action

Returns the name of the action.

Specified by:

getName in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the action name

getType

public ActionType getType()

Description copied from interface: Action

Returns the type of the action.

Specified by:

getType in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the action type

getDesc

public ActionDesc getDesc()

Description copied from interface: Action

Returns the descriptor of the action containing metadata, schemas, etc.

Specified by:

getDesc in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the action descriptor

run

public ModelResponse run(ActionContext ctx,
 GenerateAction.GenerateActionOptions options)
                  throws GenkitException

Description copied from interface: Action

Runs the action with the given input.

Specified by:

run in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Parameters:

ctx - the action context

options - the input to the action

Returns:

the output of the action

Throws:

GenkitException - if the action fails

run

public ModelResponse run(ActionContext ctx,
 GenerateAction.GenerateActionOptions options,
 Consumer<ModelResponseChunk> streamCallback)
                  throws GenkitException

Description copied from interface: Action

Runs the action with the given input and streaming callback.

Specified by:

run in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Parameters:

ctx - the action context

options - the input to the action

streamCallback - callback for receiving streaming chunks, may be null

Returns:

the output of the action

Throws:

GenkitException - if the action fails

runJson

public com.fasterxml.jackson.databind.JsonNode runJson(ActionContext ctx,
 com.fasterxml.jackson.databind.JsonNode input,
 Consumer<com.fasterxml.jackson.databind.JsonNode> streamCallback)
                                                throws GenkitException

Description copied from interface: Action

Runs the action with JSON input and returns JSON output.

Specified by:

runJson in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Parameters:

ctx - the action context

input - the JSON input

streamCallback - callback for receiving streaming JSON chunks, may be null

Returns:

the JSON output

Throws:

GenkitException - if the action fails

runJsonWithTelemetry

public ActionRunResult<com.fasterxml.jackson.databind.JsonNode> runJsonWithTelemetry(ActionContext ctx,
 com.fasterxml.jackson.databind.JsonNode input,
 Consumer<com.fasterxml.jackson.databind.JsonNode> streamCallback)
                                                                              throws GenkitException

Description copied from interface: Action

Runs the action with JSON input and returns the result with telemetry information.

Specified by:

runJsonWithTelemetry in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Parameters:

ctx - the action context

input - the JSON input

streamCallback - callback for receiving streaming JSON chunks, may be null

Returns:

the action result including telemetry data

Throws:

GenkitException - if the action fails

getInputSchema

public Map<String,Object> getInputSchema()

Description copied from interface: Action

Returns the JSON schema for the action's input type.

Specified by:

getInputSchema in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the input schema as a map, or null if not defined

getOutputSchema

public Map<String,Object> getOutputSchema()

Description copied from interface: Action

Returns the JSON schema for the action's output type.

Specified by:

getOutputSchema in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the output schema as a map, or null if not defined

getMetadata

public Map<String,Object> getMetadata()

Description copied from interface: Action

Returns additional metadata for the action.

Specified by:

getMetadata in interface Action<GenerateAction.GenerateActionOptions,ModelResponse,ModelResponseChunk>

Returns:

the metadata map

register

public void register(Registry registry)

Description copied from interface: Registerable

Registers this primitive with the given registry.

Specified by:

register in interface Registerable

Parameters:

registry - the registry to register with