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
This commit is contained in:
kYLe 2023-05-12 11:48:38 -05:00 committed by GitHub
parent a5371a0fa2
commit 570d057db4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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,