morpheus.llm.nodes.llm_generate_node.LLMGenerateNode

Bases: morpheus._lib.llm.LLMNodeBase

Generates responses from an LLM using the provided llm_client instance based on prompts provided as input from upstream nodes.

Parameters

llm_client

The client instance to use to generate responses.

input_names

The names of the inputs to this node. Defaults to ["prompt"].

Methods

async execute(self: morpheus._lib.llm.LLMNodeBase, context: morpheus._lib.llm.LLMContext) → Awaitable[morpheus._lib.llm.LLMContext][source]

Execute the current node with the given context instance.

All inputs for the given node should be fetched from the context, typically by calling either context.get_inputs to fetch all inputs as a dict, or context.get_input to fetch a specific input.

Similarly the output of the node is written to the context using context.set_output.

Parameters

context : morpheus._lib.llm.LLMContext

Context instance to use for the execution

get_input_names(self: morpheus._lib.llm.LLMNodeBase) → List[str][source]

Get the input names for the node.

Returns

list[str]

The input names for the node