minor fix: removing unused imports

Author: fshabashev

mindsdb_sql_parser/ast/mindsdb/knowledge_base.py

@@ -105,3 +105,20 @@ def to_tree(self, *args, level=0, **kwargs):
     def get_string(self, *args, **kwargs):
         out_str = f'DROP KNOWLEDGE_BASE {"IF EXISTS " if self.if_exists else ""}{self.name.to_string()}'
         return out_str
+
+class CreateKnowledgeBaseIndexBase(ASTNode):
+    """
+    Create a new index in the knowledge base
+    """
+    def __init__(self, name, *args, **kwargs):
+        """
+        Args:
+            name: Identifier -- name of the knowledge base
+        """
+        super().__init__(*args, **kwargs)
+        self.name = name
+
+    def to_tree(self, *args, level=0, **kwargs):
+        ind = indent(level)
+        out_str = f"{ind}CreateIndexBase(name={self.name.to_string()})"
+        return out_str

mindsdb_sql_parser/parser.py

@@ -16,7 +16,8 @@
 from mindsdb_sql_parser.ast.mindsdb.trigger import CreateTrigger, DropTrigger
 from mindsdb_sql_parser.ast.mindsdb.latest import Latest
 from mindsdb_sql_parser.ast.mindsdb.evaluate import Evaluate
-from mindsdb_sql_parser.ast.mindsdb.knowledge_base import CreateKnowledgeBase, DropKnowledgeBase
+from mindsdb_sql_parser.ast.mindsdb.knowledge_base import CreateKnowledgeBase, DropKnowledgeBase, \
+    CreateKnowledgeBaseIndexBase
 from mindsdb_sql_parser.ast.mindsdb.skills import CreateSkill, DropSkill, UpdateSkill
 from mindsdb_sql_parser.exceptions import ParsingException
 from mindsdb_sql_parser.ast.mindsdb.retrain_predictor import RetrainPredictor
@@ -134,6 +135,10 @@ def create_kb(self, p):
             if_not_exists=if_not_exists
         )
 
+    @_("CREATE INDEX ON KNOWLEDGE_BASE if_exists_or_empty identifier")
+    def create_index(self, p):
+        return CreateKnowledgeBaseIndexBase(name=p.identifier)
+
     @_('DROP KNOWLEDGE_BASE if_exists_or_empty identifier')
     def drop_kb(self, p):
         return DropKnowledgeBase(name=p.identifier, if_exists=p.if_exists_or_empty)

tests/test_mindsdb/test_knowledgebase_index.py

@@ -0,0 +1,12 @@
+from mindsdb_sql_parser import parse_sql
+
+
+class TestKB:
+
+    def test_create_knowledge_base_index(self):
+        # create without select
+        sql = """
+            CREATE INDEX ON KNOWLEDGE_BASE my_knowledge_base;
+        """
+        ast = parse_sql(sql)
+        print(ast)