diff --git a/src/backend/langflow/template/frontend_node/llms.py b/src/backend/langflow/template/frontend_node/llms.py index 044025ec1..4dc32dff1 100644 --- a/src/backend/langflow/template/frontend_node/llms.py +++ b/src/backend/langflow/template/frontend_node/llms.py @@ -5,6 +5,13 @@ from langflow.template.frontend_node.base import FrontendNode class LLMFrontendNode(FrontendNode): + @staticmethod + def format_openai_field(field: TemplateField): + if "openai" in field.name.lower(): + field.display_name = ( + field.name.title().replace("Openai", "OpenAI").replace("_", " ") + ).replace("Api", "API") + @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: display_names_dict = { @@ -39,3 +46,5 @@ class LLMFrontendNode(FrontendNode): elif field.name in ["model_name", "temperature"]: field.advanced = False field.show = True + + LLMFrontendNode.format_openai_field(field)