in internal/enricher/enricher.go [873:896]
func (mc *MetadataCollector) generateSchemaContext() (string, error) {
tables, err := mc.db.ListTables()
if err != nil {
return "", fmt.Errorf("failed to list tables for schema context: %w", err)
}
var schemaContext strings.Builder
schemaContext.WriteString("Tables:\n")
for _, table := range tables {
columns, err := mc.db.ListColumns(table)
if err != nil {
log.Printf("WARN: Failed to list columns for table %s in schema context: %v", table, err)
continue
}
schemaContext.WriteString(fmt.Sprintf("- %s: [", table))
columnNames := []string{}
for _, col := range columns {
columnNames = append(columnNames, col.Name)
}
schemaContext.WriteString(strings.Join(columnNames, ", "))
schemaContext.WriteString("]\n")
}
return schemaContext.String(), nil
}