refactor: change class name

Author: fshabashev

mindsdb_sql_parser/ast/mindsdb/knowledge_base.py

@@ -106,7 +106,7 @@ 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):
+class CreateKnowledgeBaseIndex(ASTNode):
     """
     Create a new index in the knowledge base
     """

mindsdb_sql_parser/parser.py

@@ -17,7 +17,7 @@
 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, \
-    CreateKnowledgeBaseIndexBase
+    CreateKnowledgeBaseIndex
 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
@@ -138,7 +138,7 @@ def create_kb(self, p):
 
     @_("CREATE INDEX ON KNOWLEDGE_BASE if_exists_or_empty identifier")
     def create_index(self, p):
-        return CreateKnowledgeBaseIndexBase(name=p.identifier)
+        return CreateKnowledgeBaseIndex(name=p.identifier)
 
     @_('DROP KNOWLEDGE_BASE if_exists_or_empty identifier')
     def drop_kb(self, p):

tests/test_mindsdb/test_knowledgebase_index.py

@@ -1,5 +1,5 @@
 from mindsdb_sql_parser import parse_sql
-from mindsdb_sql_parser.ast.mindsdb.knowledge_base import CreateKnowledgeBaseIndexBase
+from mindsdb_sql_parser.ast.mindsdb.knowledge_base import CreateKnowledgeBaseIndex
 from mindsdb_sql_parser.ast import *
 
 class TestKB:
@@ -10,7 +10,7 @@ def test_create_knowledge_base_index(self):
         sql = """CREATE INDEX ON KNOWLEDGE_BASE my_index"""
         ast = parse_sql(sql)
         print(ast)
-        expected_ast = CreateKnowledgeBaseIndexBase(
+        expected_ast = CreateKnowledgeBaseIndex(
             name=Identifier('my_index'),
         )
         assert str(ast).lower() == sql.lower()