nemoguardrails.integrations.langchain.providers.huggingface.pipeline

View as Markdown

Module Contents

Classes

NameDescription
HuggingFacePipelineCompatibleHackish way to add backward-compatibility functions to the Langchain class.

API

class nemoguardrails.integrations.langchain.providers.huggingface.pipeline.HuggingFacePipelineCompatible()

Bases: HuggingFacePipeline

Hackish way to add backward-compatibility functions to the Langchain class. TODO: Planning to add this fix directly to Langchain repo.

nemoguardrails.integrations.langchain.providers.huggingface.pipeline.HuggingFacePipelineCompatible._acall(
prompt: str,
stop: typing.Optional[typing.List[str]] = None,
run_manager: typing.Optional[langchain_core.callbacks.manager.AsyncCallbackManagerForLLMRun] = None,
kwargs: typing.Any = {}
) -> str
async

Hackish way to add async support

nemoguardrails.integrations.langchain.providers.huggingface.pipeline.HuggingFacePipelineCompatible._call(
prompt: str,
stop: typing.Optional[typing.List[str]] = None,
run_manager: typing.Optional[langchain_core.callbacks.manager.CallbackManagerForLLMRun] = None,
kwargs: typing.Any = {}
) -> str

Hackish way to perform a single llm call since Langchain dropped support