update to_tree

Author: ea-rus

mindsdb_sql_parser/ast/mindsdb/knowledge_base.py

@@ -37,16 +37,15 @@ def __init__(
 
     def to_tree(self, *args, level=0, **kwargs):
         ind = indent(level)
-        storage_str = f"{ind} storage={self.storage.to_string()},\n" if self.storage else ""
-        model_str = f"{ind} model={self.model.to_string()},\n" if self.model else ""
-        out_str = f"""
-        {ind}CreateKnowledgeBase(
-        {ind}    if_not_exists={self.if_not_exists},
-        {ind}    name={self.name.to_string()},
-        {ind}    from_query={self.from_query.to_tree(level=level + 1) if self.from_query else None},
-        {model_str}{storage_str}{ind}    params={self.params}
-        {ind})
-        """
+        storage_str = f"{ind}  storage={self.storage.to_string()},\n" if self.storage else ""
+        model_str = f"{ind}  model={self.model.to_string()},\n" if self.model else ""
+        out_str = f"{ind}CreateKnowledgeBase(\n" \
+        f"{ind}  if_not_exists={self.if_not_exists},\n" \
+        f"{ind}  name={self.name.to_string()},\n" \
+        f"{ind}  from_query={self.from_query.to_tree(level=level + 1) if self.from_query else None},\n" \
+        f"{ind}{model_str}{storage_str}{ind}  params={self.params}\n" \
+        f"{ind})"
+
         return out_str
 
     def get_string(self, *args, **kwargs):
@@ -101,12 +100,10 @@ def __init__(
         self.params = params
     def to_tree(self, *args, level=0, **kwargs):
         ind = indent(level)
-        out_str = f"""
-        {ind}AlterKnowledgeBase(
-        {ind}    name={self.name.to_string()},
-        {ind}    params={self.params}
-        {ind})
-        """
+        out_str = f"{ind}AlterKnowledgeBase(\n" \
+        f"{ind}  name={self.name.to_string()},\n" \
+        f"{ind}  params={self.params}\n" \
+        f"{ind})"
         return out_str
 
     def get_string(self, *args, **kwargs):