Expose AnyScale LLM in langchain.llms (#4585)

# Expose AnyScale LLM in  langchain.llms

Fixes # update init.py so we can from langchain.llms import Anyscale
parallel_dir_loader
kYLe 1 year ago committed by GitHub
parent a5371a0fa2
commit 570d057db4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -4,6 +4,7 @@ from typing import Dict, Type
from langchain.llms.ai21 import AI21 from langchain.llms.ai21 import AI21
from langchain.llms.aleph_alpha import AlephAlpha from langchain.llms.aleph_alpha import AlephAlpha
from langchain.llms.anthropic import Anthropic from langchain.llms.anthropic import Anthropic
from langchain.llms.anyscale import Anyscale
from langchain.llms.bananadev import Banana from langchain.llms.bananadev import Banana
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from langchain.llms.cerebriumai import CerebriumAI from langchain.llms.cerebriumai import CerebriumAI
@ -37,6 +38,7 @@ from langchain.llms.writer import Writer
__all__ = [ __all__ = [
"Anthropic", "Anthropic",
"AlephAlpha", "AlephAlpha",
"Anyscale",
"Banana", "Banana",
"CerebriumAI", "CerebriumAI",
"Cohere", "Cohere",
@ -75,6 +77,7 @@ type_to_cls_dict: Dict[str, Type[BaseLLM]] = {
"ai21": AI21, "ai21": AI21,
"aleph_alpha": AlephAlpha, "aleph_alpha": AlephAlpha,
"anthropic": Anthropic, "anthropic": Anthropic,
"anyscale": Anyscale,
"bananadev": Banana, "bananadev": Banana,
"cerebriumai": CerebriumAI, "cerebriumai": CerebriumAI,
"cohere": Cohere, "cohere": Cohere,

Loading…
Cancel
Save