in utils/prompt_loader.py [0:0]
def test_prompt_loading():
loader = PromptLoader()
# TEST MySQL
mysql_context = {
'meta_data': 'mock_metadata',
'query': 'mock_query',
'db_type': 'hologres'
}
mysql_prompt = loader.get_prompt('mysql', mysql_context)
print("MySQL Prompt Output:\n", mysql_prompt)
assert "LIMIT n" in mysql_prompt