@an5/orm 1.0.8 → 1.1.0

package/generator/src/python-generator.ts

@@ -1,114 +1,480 @@
-import fs from 'fs';
-import path from 'path';
-import { Model } from './types';
-
-export class PythonGenerator {
-  constructor(private outputPath: string) {}
-
-  public generate(models: Model[]) {
-    let pyContent = '# This file is auto-generated. Do not edit directly.\n\n';
-
-    // 1. MODEL_TO_TABLE
-    pyContent += 'MODEL_TO_TABLE = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      const fullTableName = `[${model.schemaName}].[${model.tableName}]`;
-      for (const prop of props) {
-        pyContent += `    "${prop}": "${fullTableName}",\n`;
-      }
-    }
-    pyContent += '}\n\n';
-
-    // 2. MODEL_DESCRIPTIONS
-    pyContent += 'MODEL_DESCRIPTIONS = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      for (const prop of props) {
-        pyContent += `    "${prop}": ${this.pyString(model.description)},\n`;
-      }
-    }
-    pyContent += '}\n\n';
-
-    // 3. MODEL_FIELDS
-    pyContent += 'MODEL_FIELDS = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      const fieldsStr = this.formatFields(model);
-      for (const prop of props) {
-        pyContent += `    "${prop}": ${fieldsStr},\n`;
-      }
-    }
-    pyContent += '}\n\n';
-
-    // 4. RELATION_MAP
-    pyContent += 'RELATION_MAP = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      let relationsContent = `{\n`;
-      for (const rel of model.relations) {
-        relationsContent += `        "${rel.name}": {\n`;
-        relationsContent += `            "modelName": "${this.toCamelCase(rel.type)}",\n`;
-        relationsContent += `            "relationType": "${rel.isArray ? 'many' : 'one'}",\n`;
-        relationsContent += `            "foreignKey": "${rel.foreignKey || 'id'}",\n`;
-        relationsContent += `            "localKey": "${rel.localKey || 'id'}"\n`;
-        relationsContent += `        },\n`;
-      }
-      relationsContent += `    }`;
-
-      for (const prop of props) {
-        pyContent += `    "${prop}": ${relationsContent},\n`;
-      }
-    }
-    pyContent += '}\n';
-
-    fs.writeFileSync(this.outputPath, pyContent);
-  }
-
-  private getAllPropertyVariations(modelName: string): string[] {
-    const variations = new Set<string>();
-    
-    // 1. camelCase (User -> user, McpServer -> mcpServer)
-    variations.add(this.toCamelCase(modelName));
-    
-    // 2. snake_case (User -> user, McpServer -> mcp_server)
-    variations.add(this.toSnakeCase(modelName));
-
-    const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
-    for (const acronym of acronyms) {
-      if (modelName.startsWith(acronym)) {
-        variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
-        variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
-        variations.add(this.toSnakeCase(acronym.toLowerCase() + modelName.slice(acronym.length)));
-      }
-    }
-
-    return Array.from(variations);
-  }
-
-  private formatFields(model: Model): string {
-    if (model.fields.length === 0) return '[]';
-
-    const fields = model.fields.map(f => {
-      return `        { "name": "${f.name}", "type": "${f.type}${f.isOptional ? '?' : ''}", "sql": "${f.sqlType}", "isOptional": ${f.isOptional ? 'True' : 'False'}, "hasDefault": ${f.hasDefault ? 'True' : 'False'}, "isId": ${f.isId ? 'True' : 'False'}, "description": ${this.pyString(f.description)} }`;
-    });
-
-    return `[\n${fields.join(',\n')}\n    ]`;
-  }
-
-  private pyString(value?: string): string {
-    return value ? JSON.stringify(value) : 'None';
-  }
-
-  private toCamelCase(str: string): string {
-    if (!str) return '';
-    return str[0].toLowerCase() + str.slice(1);
-  }
-
-  private toSnakeCase(str: string): string {
-    if (!str) return '';
-    return str
-      .replace(/([A-Z]+)([A-Z][a-z])/g, '$1_$2')
-      .replace(/([a-z\d])([A-Z])/g, '$1_$2')
-      .toLowerCase();
-  }
-}
+import fs from 'fs';
+import path from 'path';
+import { Model, Field, bracketedTableName } from './types';
+import { fieldKind } from './type-kinds';
+import { Provider } from './field-types';
+
+export class PythonGenerator {
+  constructor(private outputPath: string) {}
+
+  /**
+   * The metadata module name, taken from the configured file name itself.
+   *
+   * The imports used to hardcode `an5_metadata` while `outputPath` was
+   * configurable, so setting `python.metadataFile` to another name (say
+   * `an5Metadata.py`) still generated `from an5_metadata import ...` and the
+   * client broke on import — a failure that only showed up at runtime.
+   */
+  private metadataModule(): string {
+    return path.basename(this.outputPath).replace(/\.py$/i, '');
+  }
+
+  /** The database being generated for; decides types like `TIMESTAMP`. */
+  private provider: Provider | undefined;
+
+  public generate(models: Model[]) {
+    this.provider = models[0]?.provider;
+
+    const outputDir = path.dirname(this.outputPath);
+    if (!fs.existsSync(outputDir)) {
+      fs.mkdirSync(outputDir, { recursive: true });
+    }
+
+    // 1. Generate an5_metadata.py
+    this.generateMetadata(models);
+
+    // 2. Generate per-model entity files like dotnet (<Model>.py)
+    for (const model of models) {
+      this.generateModelFile(model, outputDir);
+    }
+
+    // 3. Generate an5_models.py as backward-compat aggregator
+    this.generateModelsIndex(models, outputDir);
+
+    // 4. Generate an5_orm_types.py - type-safe ORM filter/args dataclasses
+    this.generateOrmTypes(models, outputDir);
+
+    // 5. Generate an5_client.py
+    this.generateClient(models, outputDir);
+
+    // 6. Generate __init__.py
+    this.generateInit(models, outputDir);
+  }
+
+  private getPyFilterType(field: Field): string {
+    switch (fieldKind(field, this.provider)) {
+      case 'date': return 'DateTimeFilter';
+      case 'bool': return 'BoolFilter';
+      case 'int':
+      case 'bigint': return 'IntFilter';
+      case 'float': return 'NumberFilter';
+      default: return 'StringFilter';
+    }
+  }
+
+  private generateOrmTypes(models: Model[], outputDir: string) {
+    let content = '# This file is auto-generated. Do not edit directly.\n';
+    content += '"""\nAN5 ORM typed filter/args dataclasses for type-safe queries.\n\nUsage example::\n\n    db.user.find_many(where=UserWhereInput(name=StringFilter(contains="John")),\n                      order_by=UserOrderByInput(created_at="desc"),\n                      take=10)\n"""\n';
+    content += 'from __future__ import annotations\n';
+    content += 'from dataclasses import dataclass, field\n';
+    content += 'from typing import List, Optional, Any\n';
+    content += 'from datetime import datetime\n\n';
+
+    // Base filter types
+    content += '# ─── Base filter types ────────────────────────────────────────────────────────\n\n';
+    content += `@dataclass\nclass StringFilter:\n    """Type-safe filter for string fields."""\n    equals: Optional[str] = None\n    not_: Optional[str] = None\n    contains: Optional[str] = None\n    starts_with: Optional[str] = None\n    ends_with: Optional[str] = None\n    in_: Optional[List[str]] = None\n    not_in: Optional[List[str]] = None\n    gt: Optional[str] = None\n    gte: Optional[str] = None\n    lt: Optional[str] = None\n    lte: Optional[str] = None\n\n`;
+
+    content += `@dataclass\nclass IntFilter:\n    """Type-safe filter for integer fields."""\n    equals: Optional[int] = None\n    not_: Optional[int] = None\n    in_: Optional[List[int]] = None\n    not_in: Optional[List[int]] = None\n    gt: Optional[int] = None\n    gte: Optional[int] = None\n    lt: Optional[int] = None\n    lte: Optional[int] = None\n\n`;
+
+    content += `@dataclass\nclass NumberFilter:\n    """Type-safe filter for float/decimal fields."""\n    equals: Optional[float] = None\n    not_: Optional[float] = None\n    in_: Optional[List[float]] = None\n    not_in: Optional[List[float]] = None\n    gt: Optional[float] = None\n    gte: Optional[float] = None\n    lt: Optional[float] = None\n    lte: Optional[float] = None\n\n`;
+
+    content += `@dataclass\nclass BoolFilter:\n    """Type-safe filter for boolean fields."""\n    equals: Optional[bool] = None\n\n`;
+
+    content += `@dataclass\nclass DateTimeFilter:\n    """Type-safe filter for datetime fields."""\n    equals: Optional[datetime] = None\n    not_: Optional[datetime] = None\n    in_: Optional[List[datetime]] = None\n    not_in: Optional[List[datetime]] = None\n    gt: Optional[datetime] = None\n    gte: Optional[datetime] = None\n    lt: Optional[datetime] = None\n    lte: Optional[datetime] = None\n\n`;
+
+    // Per-model types
+    for (const model of models) {
+      const name = this.capitalize(model.name);
+      content += `# ─── ${name} ORM Types ────────────────────────────────────────────────────────\n\n`;
+
+      // WhereInput
+      content += `@dataclass\nclass ${name}WhereInput:\n    """Type-safe WHERE filter for ${name} queries."""\n`;
+      content += `    AND: Optional[List['${name}WhereInput']] = None\n`;
+      content += `    OR: Optional[List['${name}WhereInput']] = None\n`;
+      content += `    NOT: Optional['${name}WhereInput'] = None\n`;
+      for (const f of model.fields) {
+        const ft = this.getPyFilterType(f);
+        content += `    ${this.toSnakeCase(f.name)}: Optional[${ft}] = None\n`;
+      }
+      content += '\n';
+
+      // OrderByInput
+      content += `@dataclass\nclass ${name}OrderByInput:\n    """Type-safe ORDER BY for ${name} queries. Value: 'asc' or 'desc'."""\n`;
+      for (const f of model.fields) {
+        content += `    ${this.toSnakeCase(f.name)}: Optional[str] = None\n`;
+      }
+      content += '\n';
+
+      // CreateInput
+      const requiredFields = model.fields.filter(f => !f.isId && !f.isOptional && !f.hasDefault);
+      const optionalFields = model.fields.filter(f => !f.isId && (f.isOptional || f.hasDefault));
+      content += `@dataclass\nclass ${name}CreateInput:\n    """Typed data for creating a new ${name} record."""\n`;
+      for (const f of requiredFields) {
+        content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+      }
+      for (const f of optionalFields) {
+        content += `    ${this.toSnakeCase(f.name)}: Optional[${this.mapPyType(f)}] = None\n`;
+      }
+      if (requiredFields.length === 0 && optionalFields.length === 0) content += '    pass\n';
+      content += '\n';
+
+      // UpdateInput
+      const updateFields = model.fields.filter(f => !f.isId);
+      content += `@dataclass\nclass ${name}UpdateInput:\n    """Typed data for updating an existing ${name} record."""\n`;
+      for (const f of updateFields) {
+        content += `    ${this.toSnakeCase(f.name)}: Optional[${this.mapPyType(f)}] = None\n`;
+      }
+      if (updateFields.length === 0) content += '    pass\n';
+      content += '\n';
+
+      // FindManyArgs
+      content += `@dataclass\nclass ${name}FindManyArgs:\n    """ORM-style args for ${name}.find_many()."""\n`;
+      content += `    where: Optional[${name}WhereInput] = None\n`;
+      content += `    order_by: Optional[${name}OrderByInput] = None\n`;
+      content += `    take: Optional[int] = None\n`;
+      content += `    skip: int = 0\n`;
+      content += `    select: Optional[List[str]] = None\n\n`;
+
+      // FindFirstArgs
+      content += `@dataclass\nclass ${name}FindFirstArgs:\n    """ORM-style args for ${name}.find_first()."""\n`;
+      content += `    where: Optional[${name}WhereInput] = None\n`;
+      content += `    order_by: Optional[${name}OrderByInput] = None\n`;
+      content += `    select: Optional[List[str]] = None\n\n`;
+    }
+
+    fs.writeFileSync(path.join(outputDir, 'an5_orm_types.py'), content);
+  }
+
+  private generateMetadata(models: Model[]) {
+    let pyContent = '# This file is auto-generated. Do not edit directly.\n\n';
+
+    // 1. MODEL_TO_TABLE
+    pyContent += 'MODEL_TO_TABLE = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      const fullTableName = bracketedTableName(model);
+      for (const prop of props) {
+        pyContent += `    "${prop}": "${fullTableName}",\n`;
+      }
+    }
+    pyContent += '}\n\n';
+
+    // 2. MODEL_DESCRIPTIONS
+    pyContent += 'MODEL_DESCRIPTIONS = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      for (const prop of props) {
+        pyContent += `    "${prop}": ${this.pyString(model.description)},\n`;
+      }
+    }
+    pyContent += '}\n\n';
+
+    // 3. MODEL_FIELDS
+    pyContent += 'MODEL_FIELDS = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      const fieldsStr = this.formatFields(model);
+      for (const prop of props) {
+        pyContent += `    "${prop}": ${fieldsStr},\n`;
+      }
+    }
+    pyContent += '}\n\n';
+
+    // 4. RELATION_MAP
+    pyContent += 'RELATION_MAP = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      let relationsContent = `{\n`;
+      for (const rel of model.relations) {
+        relationsContent += `        "${rel.name}": {\n`;
+        relationsContent += `            "modelName": "${this.toCamelCase(rel.type)}",\n`;
+        relationsContent += `            "relationType": "${rel.isArray ? 'many' : 'one'}",\n`;
+        relationsContent += `            "foreignKey": "${rel.foreignKey || 'id'}",\n`;
+        relationsContent += `            "localKey": "${rel.localKey || 'id'}"\n`;
+        relationsContent += `        },\n`;
+      }
+      relationsContent += `    }`;
+
+      for (const prop of props) {
+        pyContent += `    "${prop}": ${relationsContent},\n`;
+      }
+    }
+    pyContent += '}\n';
+
+    fs.writeFileSync(this.outputPath, pyContent);
+  }
+
+  /**
+   * The Python type for a field.
+   *
+   * From the declared type, so `DECIMAL` is a `float` and `BIGINT` stays an `int`:
+   * both arrived as `number` before and took the integer branch.
+   */
+  private mapPyType(field: Field): string {
+    switch (fieldKind(field, this.provider)) {
+      case 'int':
+      case 'bigint': return 'int';
+      case 'float': return 'float';
+      case 'bool': return 'bool';
+      case 'date': return 'datetime';
+      case 'bytes': return 'bytes';
+      case 'json': return 'Any';
+      case 'vector': return 'List[float]';
+      default: return 'str';
+    }
+  }
+
+  private generateModelFile(model: Model, outputDir: string) {
+    // Single-entity file like dotnet `${model.name}.cs` / typescript `${model.name}.ts`.
+    // Contains the @dataclass entity + TypedDict row shapes for that model only.
+    let content = '# This file is auto-generated. Do not edit directly.\n';
+    content += 'from dataclasses import dataclass, field\n';
+    content += 'from typing import Optional, List, Any, TypedDict\n';
+    content += 'from datetime import datetime\n\n';
+
+    if (model.description) {
+      content += `"""${model.description}"""\n`;
+    }
+    content += `@dataclass\nclass ${model.name}:\n`;
+
+    if (model.fields.length === 0 && model.relations.length === 0) {
+      content += '    pass\n\n';
+    } else {
+      // Required fields first
+      const required = model.fields.filter(f => !f.isOptional && !f.hasDefault);
+      const optionals = model.fields.filter(f => f.isOptional || f.hasDefault);
+
+      for (const f of required) {
+        const pyType = this.mapPyType(f);
+        const snakeName = this.toSnakeCase(f.name);
+        content += `    ${snakeName}: ${pyType}\n`;
+      }
+
+      for (const f of optionals) {
+        const pyType = this.mapPyType(f);
+        const snakeName = this.toSnakeCase(f.name);
+        content += `    ${snakeName}: Optional[${pyType}] = None\n`;
+      }
+
+      // Relations
+      for (const rel of model.relations) {
+        const relName = this.toSnakeCase(rel.name);
+        if (rel.isArray) {
+          content += `    ${relName}: List[Any] = field(default_factory=list)\n`;
+        } else {
+          content += `    ${relName}: Optional[Any] = None\n`;
+        }
+      }
+
+      content += '\n';
+
+      // TypedDict row shape: mirrors what the runtime actually returns
+      // (plain dicts), so `AdapterTableClient["<Model>Row"]` delegates
+      // type-check. Required keys stay required via a total=True base;
+      // everything else is optional (total=False), keeping partial
+      // selects valid without extra dependencies.
+      const rowRequired = model.fields.filter(f => !f.isOptional && !f.hasDefault);
+      const rowOptional = model.fields.filter(f => f.isOptional || f.hasDefault);
+      content += `class _${model.name}Required(TypedDict):\n`;
+      content += `    """Required keys of a ${model.name} row."""\n`;
+      if (rowRequired.length === 0) {
+        content += '    pass\n';
+      }
+      for (const f of rowRequired) {
+        content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+      }
+      content += `\nclass ${model.name}Row(_${model.name}Required, total=False):\n`;
+      content += `    """Row shape returned for ${model.name} queries."""\n`;
+      if (rowOptional.length === 0 && model.relations.length === 0) {
+        content += '    pass\n';
+      }
+      for (const f of rowOptional) {
+        content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+      }
+      for (const rel of model.relations) {
+        const relName = this.toSnakeCase(rel.name);
+        content += rel.isArray ? `    ${relName}: List[Any]\n` : `    ${relName}: Any\n`;
+      }
+      content += '\n';
+    }
+
+    fs.writeFileSync(path.join(outputDir, `${model.name}.py`), content);
+  }
+
+  private generateModelsIndex(models: Model[], outputDir: string) {
+    // Backward-compat aggregator: `from an5_models import User` keeps working.
+    // New code can also import directly: `from .User import User, UserRow`.
+    // Supports both package-relative (`python -m`) and top-level (`sys.path`
+    // pointing at the output dir) imports via try/except fallback.
+    let content = '# This file is auto-generated. Do not edit directly.\n';
+    content += '"""Backward-compat aggregator re-exporting per-model entity files."""\n';
+    for (const model of models) {
+      content += `try:\n`;
+      content += `    from .${model.name} import ${model.name}, ${model.name}Row, _${model.name}Required\n`;
+      content += `except ImportError:\n`;
+      content += `    from ${model.name} import ${model.name}, ${model.name}Row, _${model.name}Required\n`;
+    }
+    if (models.length > 0) {
+      content += '\n__all__ = [\n';
+      for (const model of models) {
+        content += `    "${model.name}", "${model.name}Row", "_${model.name}Required",\n`;
+      }
+      content += ']\n';
+    } else {
+      content += '\n__all__ = []\n';
+    }
+
+    fs.writeFileSync(path.join(outputDir, 'an5_models.py'), content);
+  }
+
+  private generateClient(models: Model[], outputDir: string) {
+    let content = '# This file is auto-generated. Do not edit directly.\n';
+    content += 'import os\n';
+    content += 'from typing import Dict, List, Optional, Any, Callable, TYPE_CHECKING\n\n';
+
+    // Runtime adapter comes from the `an5-adapters` PyPI package
+    // (module `an5_adapter`). There is no local adapter file in the
+    // generated output dir, so no relative-import fallback here — a
+    // relative fallback would be unresolvable and trip Pylance
+    // `reportMissingImports` on every generated client.
+    content += 'from an5_adapter import An5Adapter, AdapterTableClient, create_an5_adapter, set_adapter_metadata\n\n';
+
+    const metadataModule = this.metadataModule();
+    content += 'try:\n';
+    content += `    from .${metadataModule} import MODEL_TO_TABLE, MODEL_FIELDS\n`;
+    content += 'except ImportError:\n';
+    content += `    from ${metadataModule} import MODEL_TO_TABLE, MODEL_FIELDS\n\n`;
+
+    if (models.length > 0) {
+      content += 'if TYPE_CHECKING:\n';
+      for (const m of models) {
+        content += '    try:\n';
+        content += `        from .${m.name} import ${m.name}Row\n`;
+        content += '    except ImportError:\n';
+        content += `        from .an5_models import ${m.name}Row  # fallback aggregator\n`;
+      }
+      content += '\n';
+    }
+
+    content += 'class An5Client:\n';
+    content += '    """AN5 Python ORM Client - type-safe database access.\n\n';
+    content += '    Usage:\n';
+    content += '        db = An5Client()\n';
+    content += '        users = db.user.find_many(where={"name": {"contains": "John"}}, order_by={"created_at": "asc"}, take=10)\n';
+    content += '        user  = db.user.find_first(where={"id": "abc"})\n';
+    content += '        new   = db.user.create(data={"name": "Alice", "email": "alice@example.com"})\n';
+    content += '    """\n';
+    content += '    def __init__(self, connection_string: Optional[str] = None):\n';
+    content += '        conn_str = (connection_string or \n';
+    content += '                    os.getenv("DATABASE_URL") or \n';
+    content += '                    "Server=localhost;Database=master;Trusted_Connection=True;TrustServerCertificate=True;")\n';
+    content += '        set_adapter_metadata({"model_to_table": MODEL_TO_TABLE, "model_fields": MODEL_FIELDS})\n';
+    content += '        self.adapter: An5Adapter = create_an5_adapter(conn_str)\n\n';
+
+    for (const model of models) {
+      const propName = this.toSnakeCase(model.name) + 's';
+      const singleName = this.toSnakeCase(model.name);
+      content += `        client = AdapterTableClient(self.adapter, "${model.name}")\n`;
+      content += `        self.${model.name}: AdapterTableClient["${model.name}Row"] = client\n`;
+      content += `        self.${model.name}s: AdapterTableClient["${model.name}Row"] = client\n`;
+      content += `        self.${singleName}: AdapterTableClient["${model.name}Row"] = client\n`;
+      content += `        self.${propName}: AdapterTableClient["${model.name}Row"] = client\n`;
+    }
+
+    content += '\n';
+    content += '    def __getattr__(self, name: str) -> AdapterTableClient[Any]:\n';
+    content += '        return self.adapter.table(name)\n\n';
+    content += '    def query_raw(self, sql: str, *params) -> List[Dict]:\n';
+    content += '        return self.adapter.query_raw(sql, *params)\n\n';
+    content += '    def execute_raw(self, sql: str, *params) -> int:\n';
+    content += '        return self.adapter.execute_raw(sql, *params)\n\n';
+    content += '    def transaction(self, fn: Callable) -> Any:\n';
+    content += '        return self.adapter.transaction(fn)\n';
+
+    fs.writeFileSync(path.join(outputDir, 'an5_client.py'), content);
+  }
+
+  private generateInit(models: Model[], outputDir: string) {
+    let content = '# This file is auto-generated. Do not edit directly.\n';
+    content += `from .${this.metadataModule()} import MODEL_TO_TABLE, MODEL_DESCRIPTIONS, MODEL_FIELDS, RELATION_MAP\n`;
+    content += 'from .an5_models import *\n';
+    content += 'from .an5_orm_types import (\n';
+    content += '    StringFilter, IntFilter, NumberFilter, BoolFilter, DateTimeFilter,\n';
+    for (const m of models) {
+      const name = this.capitalize(m.name);
+      content += `    ${name}WhereInput, ${name}OrderByInput,\n`;
+      content += `    ${name}CreateInput, ${name}UpdateInput,\n`;
+      content += `    ${name}FindManyArgs, ${name}FindFirstArgs,\n`;
+    }
+    content += ')\n';
+    content += 'from .an5_client import An5Client\n\n';
+    content += '__all__ = [\n';
+    content += '    "MODEL_TO_TABLE",\n';
+    content += '    "MODEL_DESCRIPTIONS",\n';
+    content += '    "MODEL_FIELDS",\n';
+    content += '    "RELATION_MAP",\n';
+    content += '    "An5Client",\n';
+    content += '    "StringFilter", "IntFilter", "NumberFilter", "BoolFilter", "DateTimeFilter",\n';
+
+    for (const m of models) {
+      const name = this.capitalize(m.name);
+      content += `    "${m.name}",\n`;
+      content += `    "${name}WhereInput", "${name}OrderByInput",\n`;
+      content += `    "${name}CreateInput", "${name}UpdateInput",\n`;
+      content += `    "${name}FindManyArgs", "${name}FindFirstArgs",\n`;
+    }
+    content += ']\n';
+
+    fs.writeFileSync(path.join(outputDir, '__init__.py'), content);
+  }
+
+  private getAllPropertyVariations(modelName: string): string[] {
+    const variations = new Set<string>();
+
+    variations.add(this.toCamelCase(modelName));
+    variations.add(this.toSnakeCase(modelName));
+
+    const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
+    for (const acronym of acronyms) {
+      if (modelName.startsWith(acronym)) {
+        variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
+        variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
+        variations.add(this.toSnakeCase(acronym.toLowerCase() + modelName.slice(acronym.length)));
+      }
+    }
+
+    return Array.from(variations);
+  }
+
+  private formatFields(model: Model): string {
+    if (model.fields.length === 0) return '[]';
+
+    const fields = model.fields.map(f => {
+      return `        { "name": "${f.name}", "type": "${f.type}${f.isOptional ? '?' : ''}", "sql": "${f.sqlType}", "isOptional": ${f.isOptional ? 'True' : 'False'}, "hasDefault": ${f.hasDefault ? 'True' : 'False'}, "isId": ${f.isId ? 'True' : 'False'}, "description": ${this.pyString(f.description)} }`;
+    });
+
+    return `[\n${fields.join(',\n')}\n    ]`;
+  }
+
+  private pyString(value?: string): string {
+    return value ? JSON.stringify(value) : 'None';
+  }
+
+  private toCamelCase(str: string): string {
+    if (!str) return '';
+    return str[0].toLowerCase() + str.slice(1);
+  }
+
+  private toSnakeCase(str: string): string {
+    if (!str) return '';
+    return str
+      .replace(/([A-Z]+)([A-Z][a-z])/g, '$1_$2')
+      .replace(/([a-z\d])([A-Z])/g, '$1_$2')
+      .toLowerCase();
+  }
+
+  private capitalize(str: string): string {
+    if (!str) return '';
+    return str[0].toUpperCase() + str.slice(1);
+  }
+}