From 6f8b434f687d34dc5a65d473a069e491fe079990 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 27 May 2023 17:58:00 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(llms.py):=20add=20format?= =?UTF-8?q?=5Fopenai=5Ffield=20method=20to=20format=20OpenAI=20field=20dis?= =?UTF-8?q?play=20names=20The=20`format=5Fopenai=5Ffield`=20method=20forma?= =?UTF-8?q?ts=20the=20display=20name=20of=20fields=20that=20contain=20the?= =?UTF-8?q?=20string=20"openai"=20in=20their=20name.=20It=20capitalizes=20?= =?UTF-8?q?the=20first=20letter=20of=20"OpenAI",=20replaces=20underscores?= =?UTF-8?q?=20with=20spaces,=20and=20replaces=20"Api"=20with=20"API".=20Th?= =?UTF-8?q?is=20improves=20the=20readability=20and=20consistency=20of=20th?= =?UTF-8?q?e=20display=20names=20of=20OpenAI=20fields.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/template/frontend_node/llms.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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)