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,