Package com.google.genkit.ai.agent.internal

Class InProcessTransport<S>

java.lang.Object
com.google.genkit.ai.agent.internal.InProcessTransport<S>
Type Parameters:
S - the type of custom session state
All Implemented Interfaces:
AgentTransport<S>
public final class InProcessTransport<S> extends Object implements AgentTransport<S>
In-process AgentTransport that drives a locally-defined Agent via its bidi action.

Each runTurn(com.google.genkit.ai.agent.AgentInput, com.google.genkit.ai.agent.AgentInit<S>, java.util.function.Consumer<com.google.genkit.ai.agent.AgentStreamChunk>) call feeds exactly one input (offer + end) into a fresh BufferedInputSource and invokes agent.runBidiJson, deserializing the returned node into an AgentOutput. Snapshot retrieval and abort delegate to the agent's typed facades.

  • Constructor Details

    • InProcessTransport

      public InProcessTransport(Agent<S> agent, ActionContext ctx)
      Constructs an in-process transport.
      Parameters:
      agent - the agent to drive (must not be null)
      ctx - the action context used for each turn invocation (must not be null)

  • Method Details

    • runTurn

      public AgentOutput<S> runTurn(AgentInput input, AgentInit<S> init, Consumer<AgentStreamChunk> onChunk)
      Description copied from interface: AgentTransport
      Runs ONE turn: feeds input with init, collects stream chunks via onChunk, and returns the final output.
      Specified by:
      runTurn in interface AgentTransport<S>
      Parameters:
      input - the turn input (user message, resume, or detach)
      init - the initialization carrying resume context (snapshotId / sessionId / inline state); may be null for the very first turn of a fresh session
      onChunk - consumer invoked for each streamed chunk; may be a no-op
      Returns:
      the final AgentOutput for the turn

    • getSnapshot

      public SessionSnapshot<S> getSnapshot(GetSnapshotRequest req)
      Description copied from interface: AgentTransport
      Retrieves a session snapshot.
      Specified by:
      getSnapshot in interface AgentTransport<S>
      Parameters:
      req - the request specifying which snapshot to retrieve
      Returns:
      the matching snapshot, or null if not found / not supported

    • abort

      public SnapshotStatus abort(String snapshotId)
      Description copied from interface: AgentTransport
      Attempts to abort a pending snapshot.
      Specified by:
      abort in interface AgentTransport<S>
      Parameters:
      snapshotId - the snapshot to abort
      Returns:
      the resulting status, or null if not found / not supported

    • serverManaged

      public boolean serverManaged()
      Description copied from interface: AgentTransport
      Returns whether the underlying agent is server-managed (state persisted server-side) versus client-managed (state round-tripped by the caller).
      Specified by:
      serverManaged in interface AgentTransport<S>
      Returns:
      true if server-managed