fixed additional classes

langchain-ai · Apr 9, 2024 · 86e91d1 · 86e91d1
1 parent 7c2ff12
commit 86e91d1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/libs/community/langchain_community/chat_models/__init__.py b/libs/community/langchain_community/chat_models/__init__.py
@@ -114,6 +114,9 @@
     from langchain_community.chat_models.mlflow_ai_gateway import (
         ChatMLflowAIGateway,  # noqa: F401
     )
+    from langchain_community.chat_models.mlx import (
+        ChatMLX,  # noqa: F401
+    )
     from langchain_community.chat_models.ollama import (
         ChatOllama,  # noqa: F401
     )
@@ -177,6 +180,7 @@
     "ChatKonko",
     "ChatLiteLLM",
     "ChatLiteLLMRouter",
+    "ChatMLX",
     "ChatMLflowAIGateway",
     "ChatMaritalk",
     "ChatMlflow",
@@ -200,11 +204,12 @@
     "MiniMaxChat",
     "PaiEasChatEndpoint",
     "PromptLayerChatOpenAI",
-    "SolarChat",
     "QianfanChatEndpoint",
+    "SolarChat",
     "VolcEngineMaasChat",
 ]
 
+
 _module_lookup = {
     "AzureChatOpenAI": "langchain_community.chat_models.azure_openai",
     "BedrockChat": "langchain_community.chat_models.bedrock",