Change template to private attribute and simplify imports (#101)

---------

Co-authored-by: ian <ian@cinnamon.is>

knowledgehub/base/__init__.py

@@ -1,4 +1,4 @@
-from .component import BaseComponent
+from .component import BaseComponent, Node, Param, lazy
 from .schema import (
     AIMessage,
     BaseMessage,
@@ -22,4 +22,7 @@ __all__ = [
     "RetrievedDocument",
     "LLMInterface",
     "ExtractorOutput",
+    "Param",
+    "Node",
+    "lazy",
 ]

knowledgehub/base/component.py

@@ -1,6 +1,6 @@
 from abc import abstractmethod
 
-from theflow.base import Function
+from theflow import Function, Node, Param, lazy
 
 from kotaemon.base.schema import Document
 
@@ -35,3 +35,6 @@ class BaseComponent(Function):
     def run(self, *args, **kwargs) -> Document | list[Document] | None:
         """Run the component."""
         ...
+
+
+__all__ = ["BaseComponent", "Param", "Node", "lazy"]