diff --git a/pyproject.toml b/pyproject.toml index 380466567..436c0684c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langflow" -version = "0.0.87" +version = "0.0.88" description = "A Python package with a built-in web application" authors = ["Logspace "] maintainers = [ diff --git a/src/backend/langflow/template/frontend_node/llms.py b/src/backend/langflow/template/frontend_node/llms.py index 637169544..ac17cf8ed 100644 --- a/src/backend/langflow/template/frontend_node/llms.py +++ b/src/backend/langflow/template/frontend_node/llms.py @@ -16,20 +16,30 @@ class LLMFrontendNode(FrontendNode): def format_azure_field(field: TemplateField): if field.name == "model_name": field.show = False # Azure uses deployment_name instead of model_name. - if field.name == "openai_api_type": + elif field.name == "openai_api_type": field.show = False field.password = False field.value = "azure" - if field.name == "openai_api_version": + elif field.name == "openai_api_version": field.password = False field.value = "2023-03-15-preview" + @staticmethod + def format_llama_field(field: TemplateField): + field.show = True + field.advanced = not field.required + @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: display_names_dict = { "huggingfacehub_api_token": "HuggingFace Hub API Token", } FrontendNode.format_field(field, name) + LLMFrontendNode.format_openai_field(field) + if name and "azure" in name.lower(): + LLMFrontendNode.format_azure_field(field) + if name and "llama" in name.lower(): + LLMFrontendNode.format_llama_field(field) SHOW_FIELDS = ["repo_id"] if field.name in SHOW_FIELDS: field.show = True @@ -65,7 +75,3 @@ class LLMFrontendNode(FrontendNode): ]: field.advanced = False field.show = True - - LLMFrontendNode.format_openai_field(field) - if name and "azure" in name.lower(): - LLMFrontendNode.format_azure_field(field)