LLM을 활용한 간단한 SQL 쿼리 생성기 구축 가이드

def format_schema_for_llm(self, target_tables: Optional[List[str]] = None) -> str:
    tables = self.get_tables(include_tables=target_tables)
    schema_description = "# Database Schema (MSSQL)

"
    for table in tables:
        schema = self.get_table_schema(table)
        fks = self.get_foreign_keys(table)
        schema_description += f"## Table: {table}
"
        schema_description += "Columns:
"
        for col in schema['columns']:
            pk_marker = " (PRIMARY KEY)" if col['name'] in schema['primary_keys'] else ""
            nullable = "NULL" if col['nullable'] else "NOT NULL"
            schema_description += f"- {col['name']}: {col['type']} {nullable}{pk_marker}
"
        if fks:
            schema_description += "
Foreign Keys:
"
            for fk in fks:
                schema_description += f"- {fk['column']} → {fk['references_table']}.{fk['references_column']}
"
    return schema_description

def create_system_prompt(self, schema: str, dialect: str = "postgresql") -> str:
    return f"""You are an expert T-SQL (Microsoft SQL Server) query generator.
Database Schema: {schema}
Rules:
1. Generate ONLY the SQL query. No explanations.
2. Use proper JOIN syntax and table aliases.
3. Use TOP for limiting results instead of LIMIT.
4. For temporal queries, use YEAR(order_date) = YEAR(GETDATE()).
"""