u 6

Author: MervinPraison

praisonai_tools/docs_generator/generator.py

@@ -697,7 +697,7 @@ def _render_module(self, info: ModuleInfo) -> str:
             "---",
             f'title: "{info.short_name}"',
             f'description: "{desc}"',
-            f'icon: "file-code"',
+            f'icon: "{get_icon_for_module(info.short_name)}"',
             "---",
             "",
             f"# {info.display_name}",
@@ -754,6 +754,7 @@ def _render_module_granular(self, info: ModuleInfo) -> str:
             "---",
             f'title: "{info.short_name}"',
             f'description: "{desc}"',
+            f'icon: "{get_icon_for_module(info.short_name)}"',
             "---",
             "",
             f"# {info.display_name}",
@@ -807,6 +808,7 @@ def _render_class_page(self, cls: ClassInfo, module_info: ModuleInfo) -> str:
             "---",
             f'title: "{cls.name}"',
             f'description: "{desc}"',
+            'icon: "brackets-curly"',
             "---",
             "",
             f"# {cls.name}",
@@ -855,8 +857,9 @@ def _render_function_page(self, func: FunctionInfo, module_info: ModuleInfo, cls
 
         lines = [
             "---",
-            f'title: "{display_name}"',
+            f'title: "{func.name}"',
             f'description: "{desc}"',
+            'icon: "function"',
             "---",
             "",
             f"# {display_name}",

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "praisonai-tools"
-version = "0.2.10"
+version = "0.2.11"
 description = "Extended tools for PraisonAI Agents"
 authors = [
     {name = "Mervin Praison"}