updated the parser to handle EVALUATE KNOLWEGE_BASE ...

Author: MinuraPunchihewa

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, \
-    CreateKnowledgeBaseIndex, DropKnowledgeBaseIndex
+    CreateKnowledgeBaseIndex, DropKnowledgeBaseIndex, EvaluateKnowledgeBase
 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
@@ -89,6 +89,7 @@ class MindsDBParser(Parser):
        'drop_trigger',
        'create_kb',
        'drop_kb',
+       'evaluate_kb',
        'create_skill',
        'drop_skill',
        'update_skill',
@@ -149,6 +150,19 @@ def  drop_index(self, p):
     def drop_kb(self, p):
         return DropKnowledgeBase(name=p.identifier, if_exists=p.if_exists_or_empty)
 
+    @_('EVALUATE KNOWLEDGE_BASE identifier USING kw_parameter_list')
+    def evaluate_kb(self, p):
+        params = p.kw_parameter_list
+
+        return EvaluateKnowledgeBase(
+            name=p.identifier,
+            test_table=params.pop('test_table'),
+            llm=params.pop('llm'),
+            save_to= params.pop('save_to'),
+            generate_data=params.pop('generate_data'),
+            params=params
+        )
+
     # -- Skills --
     @_('CREATE SKILL if_not_exists_or_empty identifier USING kw_parameter_list')
     def create_skill(self, p):