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):
371
371
),
372
372
)
373
373
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