File tree Expand file tree Collapse file tree 3 files changed +87
-0
lines changed
Expand file tree Collapse file tree 3 files changed +87
-0
lines changed Original file line number Diff line number Diff line change @@ -70,3 +70,32 @@ def test_drop_agent(self):
7070 # Parse again after rendering to catch problems with rendering.
7171 ast = parse_sql (str (ast ))
7272 assert str (ast ) == str (expected_ast )
73+
74+ def test_create_agent_case_handling (self ):
75+ sql = """
76+ CREATE AGENT my_agent
77+ USING
78+ MoDeL = 'my_model',
79+ SkIlLs = ['skill1', 'skill2']
80+ """
81+ ast = parse_sql (sql )
82+ expected_ast = CreateAgent (
83+ name = Identifier ('my_agent' ),
84+ model = 'my_model' ,
85+ params = {'skills' : ['skill1' , 'skill2' ]}
86+ )
87+ assert ast .to_tree () == expected_ast .to_tree ()
88+
89+ def test_update_agent_case_handling (self ):
90+ sql = """
91+ UPDATE AGENT my_agent
92+ SET
93+ MoDeL = 'new_model',
94+ SkIlLs = ['new_skill1', 'new_skill2']
95+ """
96+ ast = parse_sql (sql )
97+ expected_ast = UpdateAgent (
98+ name = Identifier ('my_agent' ),
99+ updated_params = {'model' : 'new_model' , 'skills' : ['new_skill1' , 'new_skill2' ]}
100+ )
101+ assert ast .to_tree () == expected_ast .to_tree ()
Original file line number Diff line number Diff line change @@ -78,3 +78,33 @@ def test_drop_chatbot(self):
7878 )
7979 assert str (ast ) == str (expected_ast )
8080 assert ast .to_tree () == expected_ast .to_tree ()
81+
82+ def test_create_chatbot_case_handling (self ):
83+ sql = """
84+ CREATE CHATBOT my_chatbot
85+ USING
86+ DaTaBaSe = 'my_db',
87+ MoDeL = 'my_model',
88+ AgEnT = 'my_agent'
89+ """
90+ ast = parse_sql (sql )
91+ expected_ast = CreateChatBot (
92+ name = Identifier ('my_chatbot' ),
93+ database = Identifier ('my_db' ),
94+ model = Identifier ('my_model' ),
95+ agent = Identifier ('my_agent' )
96+ )
97+ assert ast .to_tree () == expected_ast .to_tree ()
98+
99+ def test_update_chatbot_case_handling (self ):
100+ sql = """
101+ UPDATE CHATBOT my_chatbot
102+ SET
103+ MoDeL = 'new_model'
104+ """
105+ ast = parse_sql (sql )
106+ expected_ast = UpdateChatBot (
107+ name = Identifier ('my_chatbot' ),
108+ updated_params = {'model' : 'new_model' }
109+ )
110+ assert ast .to_tree () == expected_ast .to_tree ()
Original file line number Diff line number Diff line change @@ -60,3 +60,31 @@ def test_drop_skill(self):
6060 # Parse again after rendering to catch problems with rendering.
6161 ast = parse_sql (str (ast ))
6262 assert str (ast ) == str (expected_ast )
63+
64+ def test_create_skill_case_handling (self ):
65+ sql = """
66+ CREATE SKILL my_skill
67+ USING
68+ TyPe = 'knowledge_base',
69+ SoUrCe = 'my_knowledge_base'
70+ """
71+ ast = parse_sql (sql )
72+ expected_ast = CreateSkill (
73+ name = Identifier ('my_skill' ),
74+ type = 'knowledge_base' ,
75+ params = {'source' : 'my_knowledge_base' }
76+ )
77+ assert ast .to_tree () == expected_ast .to_tree ()
78+
79+ def test_update_skill_case_handling (self ):
80+ sql = """
81+ UPDATE SKILL my_skill
82+ SET
83+ SoUrCe = 'new_source'
84+ """
85+ ast = parse_sql (sql )
86+ expected_ast = UpdateSkill (
87+ name = Identifier ('my_skill' ),
88+ updated_params = {'source' : 'new_source' }
89+ )
90+ assert ast .to_tree () == expected_ast .to_tree ()
You can’t perform that action at this time.
0 commit comments