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

--------- Co-authored-by: ian <ian@cinnamon.is>
2023-12-08 18:10:34 +07:00
parent 1f927d3391
commit da0ac1d69f
13 changed files with 31 additions and 39 deletions
--- a/knowledgehub/base/init.py
+++ b/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",
 ]