diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py index caa70a79b..801abbe3a 100644 --- a/src/backend/langflow/main.py +++ b/src/backend/langflow/main.py @@ -226,6 +226,13 @@ template_node = { "type": "int", "list": False }, + "base_classes": [ + "BaseChatModel", + "Serializable", + "BaseLanguageModel", + "ChatOpenAI" + ], + "description": "Wrapper around OpenAI Chat large language models.", "_type": "ChatOpenAI" } }