Skip to content

Commit 3d9d288

Browse files
authored
Update test_knowledgebase.py
1 parent 3009b34 commit 3d9d288

File tree

1 file changed

+0
-35
lines changed

1 file changed

+0
-35
lines changed

tests/test_mindsdb/test_knowledgebase.py

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)