From 570d057db425205755c46cd9678083145c685c2e Mon Sep 17 00:00:00 2001 From: kYLe Date: Fri, 12 May 2023 11:48:38 -0500 Subject: [PATCH] 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 --- langchain/llms/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/langchain/llms/__init__.py b/langchain/llms/__init__.py index a7f5980c..221a0d3e 100644 --- a/langchain/llms/__init__.py +++ b/langchain/llms/__init__.py @@ -4,6 +4,7 @@ from typing import Dict, Type from langchain.llms.ai21 import AI21 from langchain.llms.aleph_alpha import AlephAlpha from langchain.llms.anthropic import Anthropic +from langchain.llms.anyscale import Anyscale from langchain.llms.bananadev import Banana from langchain.llms.base import BaseLLM from langchain.llms.cerebriumai import CerebriumAI @@ -37,6 +38,7 @@ from langchain.llms.writer import Writer __all__ = [ "Anthropic", "AlephAlpha", + "Anyscale", "Banana", "CerebriumAI", "Cohere", @@ -75,6 +77,7 @@ type_to_cls_dict: Dict[str, Type[BaseLLM]] = { "ai21": AI21, "aleph_alpha": AlephAlpha, "anthropic": Anthropic, + "anyscale": Anyscale, "bananadev": Banana, "cerebriumai": CerebriumAI, "cohere": Cohere,