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

Update IDGenerator.py to generate a unique ID in the build_config

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-05 22:05:57 -03:00
commit abf9bbd4a2
1 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/langflow/components/helpers/IDGenerator.py
View file

@ -1,5 +1,5 @@
import uuid
from typing import Text
from typing import Any, Text
from langflow import CustomComponent
@ -9,11 +9,20 @@ class UUIDGeneratorComponent(CustomComponent):
display_name = "Unique ID Generator"
description = "Generates a unique ID."
def generate(self, *args, **kwargs):
return Text(uuid.uuid4().hex)
def update_build_config(
self, build_config: dict, field_name: Text, field_value: Any
):
if field_name == "unique_id":
build_config[field_name]["value"] = str(uuid.uuid4())
return build_config
def build_config(self):
return {"unique_id": {"display_name": "Value", "value": self.generate}}
return {
"unique_id": {
"display_name": "Value",
"refresh": True,
}
}
def build(self, unique_id: str) -> str:
return unique_id