nemoguardrails.integrations.langchain.providers.providers

View as Markdown

Module that exposes all the supported LLM providers.

Currently, this module automatically discovers all the LLM providers available in LangChain and registers them.

Additional providers can be registered using the register_llm_provider function.

Module Contents

Functions

NameDescription
_acallAdds asynchronous support to LLM providers that only have synchronous methods.
_discover_langchain_community_chat_providersCreates a mapping from provider name to chat model class.
_discover_langchain_community_llm_providersAutomatically discover all LLM providers from LangChain.
_discover_langchain_partner_chat_providers-
_get_all_chat_provider_namesConsolidates all chat provider names.
_get_chat_completion_provider-
_get_text_completion_provider-
_parse_version-
_patch_acall_method_to-
discover_langchain_providersAutomatically discover all LLM providers from LangChain.
get_chat_provider_namesReturns the list of supported chat providers.
get_community_chat_provider_namesReturns the list of supported chat providers.
get_llm_provider_namesReturns the list of supported LLM providers.
register_chat_providerRegister an additional chat provider.
register_llm_providerRegister an additional LLM provider.

Data

DEPRECATED_LLM_PROVIDERS

_CUSTOM_CHAT_PROVIDERS

__all__

_chat_providers

_llm_providers

log

API

nemoguardrails.integrations.langchain.providers.providers._acall(
self,
args = (),
kwargs = {}
)
async

Adds asynchronous support to LLM providers that only have synchronous methods.

nemoguardrails.integrations.langchain.providers.providers._discover_langchain_community_chat_providers()

Creates a mapping from provider name to chat model class. The provider name is defined as the last segment of the module path. For example, for module path “langchain_community.chat_models.google_palm”, the provider name is “google_palm”.

nemoguardrails.integrations.langchain.providers.providers._discover_langchain_community_llm_providers()

Automatically discover all LLM providers from LangChain.

nemoguardrails.integrations.langchain.providers.providers._discover_langchain_partner_chat_providers() -> typing.Set[str]
nemoguardrails.integrations.langchain.providers.providers._get_all_chat_provider_names() -> typing.List[str]

Consolidates all chat provider names.

nemoguardrails.integrations.langchain.providers.providers._get_chat_completion_provider(
provider_name: str
) -> typing.Type[langchain_core.language_models.BaseChatModel]
nemoguardrails.integrations.langchain.providers.providers._get_text_completion_provider(
provider_name: str
) -> typing.Type[langchain_core.language_models.BaseLLM]
nemoguardrails.integrations.langchain.providers.providers._parse_version(
version_str
)
nemoguardrails.integrations.langchain.providers.providers._patch_acall_method_to(
llm_providers: typing.Dict[str, typing.Type[langchain_core.language_models.BaseLLM]]
)
nemoguardrails.integrations.langchain.providers.providers.discover_langchain_providers()

Automatically discover all LLM providers from LangChain.

This function is deprecated and will be removed in a future release. Use _discover_langchain_providers directly instead.

nemoguardrails.integrations.langchain.providers.providers.get_chat_provider_names() -> typing.List[str]

Returns the list of supported chat providers.

nemoguardrails.integrations.langchain.providers.providers.get_community_chat_provider_names() -> typing.List[str]

Returns the list of supported chat providers.

nemoguardrails.integrations.langchain.providers.providers.get_llm_provider_names() -> typing.List[str]

Returns the list of supported LLM providers.

nemoguardrails.integrations.langchain.providers.providers.register_chat_provider(
name: str,
provider_cls: typing.Type[langchain_core.language_models.BaseChatModel]
)

Register an additional chat provider.

nemoguardrails.integrations.langchain.providers.providers.register_llm_provider(
name: str,
provider_cls: typing.Type[langchain_core.language_models.BaseLLM]
)

Register an additional LLM provider.

nemoguardrails.integrations.langchain.providers.providers.DEPRECATED_LLM_PROVIDERS = ['mlflow-chat', 'databricks-chat']
nemoguardrails.integrations.langchain.providers.providers._CUSTOM_CHAT_PROVIDERS = {'nim'}
nemoguardrails.integrations.langchain.providers.providers.__all__ = ['_llm_providers', '_parse_version', 'get_llm_provider_names', 'get_community_ch...
nemoguardrails.integrations.langchain.providers.providers._chat_providers: Dict[str, Type[BaseChatModel]] = _discover_langchain_community_chat_providers()
nemoguardrails.integrations.langchain.providers.providers._llm_providers: Dict[str, Type[BaseLLM]] = {'trt_llm': TRTLLM}
nemoguardrails.integrations.langchain.providers.providers.log = logging.getLogger(__name__)