enh: tools & functions frontmatter

backend/apps/webui/utils.py

@@ -1,19 +1,41 @@
 from importlib import util
 import os
+import re
 
 from config import TOOLS_DIR, FUNCTIONS_DIR
 
 
+def extract_frontmatter(file_path):
+    """
+    Extract frontmatter as a dictionary from the specified file path.
+    """
+    frontmatter = {}
+    frontmatter_pattern = re.compile(r"^([a-z_]+):\s*(.*)\s*$", re.IGNORECASE)
+
+    with open(file_path, "r", encoding="utf-8") as file:
+        for line in file:
+            if line.strip() == '"""':
+                # End of frontmatter section
+                break
+            match = frontmatter_pattern.match(line)
+            if match:
+                key, value = match.groups()
+                frontmatter[key] = value
+
+    return frontmatter
+
+
 def load_toolkit_module_by_id(toolkit_id):
     toolkit_path = os.path.join(TOOLS_DIR, f"{toolkit_id}.py")
     spec = util.spec_from_file_location(toolkit_id, toolkit_path)
     module = util.module_from_spec(spec)
+    frontmatter = extract_frontmatter(toolkit_path)
 
     try:
         spec.loader.exec_module(module)
         print(f"Loaded module: {module.__name__}")
         if hasattr(module, "Tools"):
-            return module.Tools()
+            return module.Tools(), frontmatter
         else:
             raise Exception("No Tools class found")
     except Exception as e:
@@ -28,14 +50,15 @@ def load_function_module_by_id(function_id):
 
     spec = util.spec_from_file_location(function_id, function_path)
     module = util.module_from_spec(spec)
+    frontmatter = extract_frontmatter(function_path)
 
     try:
         spec.loader.exec_module(module)
         print(f"Loaded module: {module.__name__}")
         if hasattr(module, "Pipe"):
-            return module.Pipe(), "pipe"
+            return module.Pipe(), "pipe", frontmatter
         elif hasattr(module, "Filter"):
-            return module.Filter(), "filter"
+            return module.Filter(), "filter", frontmatter
         else:
             raise Exception("No Function class found")
     except Exception as e: