morpheus.llm.services.llm_service.LLMClient

class LLMClient[source]

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

generate(input_dict) Issue a request to generate a response based on a given prompt.
generate_async(input_dict) Issue an asynchronous request to generate a response based on a given prompt.
generate_batch(inputs) Issue a request to generate a list of responses based on a list of prompts.
generate_batch_async(inputs) Issue an asynchronous request to generate a list of responses based on a list of prompts.
get_input_names() Returns the names of the inputs to the model.
abstract generate(input_dict)[source]

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

Parameters
input_dictdict

Input containing prompt data.

abstract async generate_async(input_dict)[source]

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

Parameters
input_dictdict

Input containing prompt data.

abstract generate_batch(inputs)[source]

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

Parameters
inputsdict

Inputs containing prompt data.

abstract async generate_batch_async(inputs)[source]

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

Parameters
inputsdict

Inputs containing prompt data.

abstract get_input_names()[source]

Returns the names of the inputs to the model.

Previous morpheus.llm.services.llm_service
Next morpheus.llm.services.llm_service.LLMService
© Copyright 2023, NVIDIA. Last updated on Feb 2, 2024.