alter agent syntax

Author: ea-rus

mindsdb_sql_parser/parser.py

@@ -199,6 +199,7 @@ def drop_agent(self, p):
         return DropAgent(name=p.identifier, if_exists=p.if_exists_or_empty)
 
     @_('UPDATE AGENT identifier SET kw_parameter_list')
+    @_('ALTER AGENT identifier USING kw_parameter_list')
     def update_agent(self, p):
         return UpdateAgent(name=p.identifier, updated_params=p.kw_parameter_list)
 

tests/test_mindsdb/test_agents.py

@@ -44,6 +44,14 @@ def test_update_agent(self):
         assert str(ast) == str(expected_ast)
         assert ast.to_tree() == expected_ast.to_tree()
 
+        sql = '''
+            alter agent my_agent
+            USING
+            model = 'new_model',
+            skills = ['new_skill1', 'new_skill2']
+        '''
+        ast = parse_sql(sql)
+
         # Parse again after rendering to catch problems with rendering.
         ast = parse_sql(str(ast))
         assert str(ast) == str(expected_ast)