File tree Expand file tree Collapse file tree 1 file changed +0
-35
lines changed
Expand file tree Collapse file tree 1 file changed +0
-35
lines changed Original file line number Diff line number Diff line change @@ -371,38 +371,3 @@ def test_insert_into_knowledge_base(self):
371371 ),
372372 )
373373 assert ast == expected_ast
374-
375- def test_evaluate_knowledge_base (self ):
376- sql = """
377- EVALUATE KNOWLEDGE_BASE my_knowledge_base
378- USING
379- TEST_TABLE = my_database.some_table_1,
380- SAVE_TO = my_database.some_table_2,
381- LLM = {
382- "provider": "openai",
383- "model": "gpt-3.5-turbo",
384- "api_key": "my_api_key"
385- },
386- GENERATE_DATA = {
387- "from_sql": "SELECT content FROM my_database.some_table",
388- "count": 100
389- }
390- """
391- ast = parse_sql (sql )
392- expected_ast = EvaluateKnowledgeBase (
393- name = Identifier ("my_knowledge_base" ),
394- params = {
395- "test_table" : Identifier (parts = ["my_database" , "some_table_1" ]),
396- "save_to" : Identifier (parts = ["my_database" , "some_table_2" ]),
397- "llm" : {
398- "provider" : "openai" ,
399- "model" : "gpt-3.5-turbo" ,
400- "api_key" : "my_api_key"
401- },
402- "generate_data" : {
403- "from_sql" : "SELECT content FROM my_database.some_table" ,
404- "count" : 100
405- }
406- }
407- )
408- assert ast == expected_ast
You can’t perform that action at this time.
0 commit comments