From f3908627ed039b954dcd323580ab49cb566e5e98 Mon Sep 17 00:00:00 2001 From: rajib Date: Sun, 23 Jul 2023 23:30:46 -0700 Subject: [PATCH] changed to mlflow-ai-gateway in llms/__init__.py (#8114) - Description: In the llms/__init__.py, the key name is wrong for mlflowaigateway. It should be mlflow-ai-gateway - Issue: NA - Dependencies: NA - Tag maintainer: @hwchase17, @baskaryan - Twitter handle: na Without this fix, when we run the code for mlflowaigateway, we will get error as below ValueError: Loading mlflow-ai-gateway LLM not supported --------- Co-authored-by: rajib76 Co-authored-by: Bagatur --- libs/langchain/langchain/llms/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langchain/langchain/llms/__init__.py b/libs/langchain/langchain/llms/__init__.py index 40836b404db..31bcc8c5d5a 100644 --- a/libs/langchain/langchain/llms/__init__.py +++ b/libs/langchain/langchain/llms/__init__.py @@ -150,7 +150,7 @@ type_to_cls_dict: Dict[str, Type[BaseLLM]] = { "koboldai": KoboldApiLLM, "llamacpp": LlamaCpp, "textgen": TextGen, - "mlflow-gateway": MlflowAIGateway, + "mlflow-ai-gateway": MlflowAIGateway, "modal": Modal, "mosaic": MosaicML, "nlpcloud": NLPCloud,