jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add advanced parameters to llamacpp node (#465)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-08 18:40:41 -03:00 • committed by GitHub
commit eb3421711f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "langflow" name = "langflow"
version = "0.0.87" version = "0.0.88"
description = "A Python package with a built-in web application" description = "A Python package with a built-in web application"
authors = ["Logspace <contact@logspace.ai>"] authors = ["Logspace <contact@logspace.ai>"]
maintainers = [ maintainers = [

18
src/backend/langflow/template/frontend_node/llms.py
View file

@ -16,20 +16,30 @@ class LLMFrontendNode(FrontendNode):
def format_azure_field(field: TemplateField): def format_azure_field(field: TemplateField):
if field.name == "model_name": if field.name == "model_name":
field.show = False # Azure uses deployment_name instead of 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.show = False
field.password = False field.password = False
field.value = "azure" field.value = "azure"
if field.name == "openai_api_version": elif field.name == "openai_api_version":
field.password = False field.password = False
field.value = "2023-03-15-preview" field.value = "2023-03-15-preview"
@staticmethod
def format_llama_field(field: TemplateField):
field.show = True
field.advanced = not field.required
@staticmethod @staticmethod
def format_field(field: TemplateField, name: Optional[str] = None) -> None: def format_field(field: TemplateField, name: Optional[str] = None) -> None:
display_names_dict = { display_names_dict = {
"huggingfacehub_api_token": "HuggingFace Hub API Token", "huggingfacehub_api_token": "HuggingFace Hub API Token",
} }
FrontendNode.format_field(field, name) 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"] SHOW_FIELDS = ["repo_id"]
if field.name in SHOW_FIELDS: if field.name in SHOW_FIELDS:
field.show = True field.show = True
@ -65,7 +75,3 @@ class LLMFrontendNode(FrontendNode):
]: ]:
field.advanced = False field.advanced = False
field.show = True field.show = True
LLMFrontendNode.format_openai_field(field)
if name and "azure" in name.lower():
LLMFrontendNode.format_azure_field(field)