morpheus.llm.services.llm_service.LLMClient

Bases: abc.ABC

Abstract interface for clients which are able to interact with LLM models. Concrete implementations of this class will have an associated implementation of LLMService which is able to construct instances of this class.

Methods

abstract generate(**input_dict)[source]

Issue a request to generate a response based on a given prompt.

Parameters

input_dict

Input containing prompt data.

Returns

str

Generated response for prompt.

abstract async generate_async(**input_dict)[source]

Issue an asynchronous request to generate a response based on a given prompt.

Parameters

input_dict

Input containing prompt data.

Returns

str

Generated async response for prompt.

abstract generate_batch(inputs: dict[str, list], return_exceptions: Literal[True] = True) → list[str | BaseException][source]

abstract generate_batch(inputs: dict[str, list], return_exceptions: Literal[False] = False) → list[str]

Issue a request to generate a list of responses based on a list of prompts.

Parameters

inputs

Inputs containing prompt data.

return_exceptions

Whether to return exceptions in the output list or raise them immediately.

Returns

list[str] | list[str | BaseException]

List of responses or list of responses and exceptions.

abstract async generate_batch_async(inputs: dict[str, list], return_exceptions: Literal[True] = True) → list[str | BaseException][source]

abstract async generate_batch_async(inputs: dict[str, list], return_exceptions: Literal[False] = False) → list[str]

Issue an asynchronous request to generate a list of responses based on a list of prompts.

Parameters

inputs

Inputs containing prompt data.

return_exceptions

Whether to return exceptions in the output list or raise them immediately.

Returns

list[str] | list[str | BaseException]

List of responses or list of responses and exceptions.

abstract get_input_names()[source]

Returns the names of the inputs to the model.

Returns

list[str]

List of input names.