From 891e8a04ac99ef8f11ca25c1c49908e35539d136 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 18 Feb 2024 17:28:32 -0300 Subject: [PATCH] Add UUIDGeneratorComponent to generate unique IDs --- .../components/utilities/IDGenerator.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 src/backend/langflow/components/utilities/IDGenerator.py diff --git a/src/backend/langflow/components/utilities/IDGenerator.py b/src/backend/langflow/components/utilities/IDGenerator.py new file mode 100644 index 000000000..5b81e61e5 --- /dev/null +++ b/src/backend/langflow/components/utilities/IDGenerator.py @@ -0,0 +1,18 @@ +import uuid + +from langflow import CustomComponent + + +class UUIDGeneratorComponent(CustomComponent): + documentation: str = "http://docs.langflow.org/components/custom" + display_name = "Unique ID Generator" + description = "Generates a unique ID." + + def generate(self, *args, **kwargs): + return str(uuid.uuid4().hex) + + def build_config(self): + return {"unique_id": {"display_name": "Value", "value": self.generate}} + + def build(self, unique_id: str) -> str: + return unique_id