fix: regenerate models with proper class names from OpenAPI spec

Enable `use_title_as_name` in datamodel-codegen config to use schema
titles as class names. Add post-processing script to fix discriminator
field name bug. Update client code for renamed models (TaskOptions,
ScheduleCreateAction, etc.).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: vdusek

.github/workflows/manual_regenerate_models.yaml

@@ -84,6 +84,9 @@ jobs:
             uv run datamodel-codegen
           fi
 
+      - name: Post-process generated models
+        run: python scripts/fix_generated_models.py
+
       - name: Commit model changes
         id: commit
         uses: EndBug/add-and-commit@v10

pyproject.toml

@@ -224,6 +224,7 @@ wrap_string_literal = true
 snake_case_field = true
 use_subclass_enum = true
 extra_fields = "allow"
+use_title_as_name = true
 allow_population_by_field_name = true
 aliases = "datamodel_codegen_aliases.json"
 formatters = ["ruff-check", "ruff-format"]
@@ -269,4 +270,4 @@ shell = "./build_api_reference.sh && corepack enable && yarn && uv run yarn star
 cwd = "website"
 
 [tool.poe.tasks.generate-models]
-shell = "uv run datamodel-codegen"
+shell = "uv run datamodel-codegen && python scripts/fix_generated_models.py"

scripts/fix_generated_models.py

@@ -0,0 +1,46 @@
+"""Post-process the generated _models.py to fix known datamodel-codegen issues.
+
+Currently fixes:
+- Discriminator field names: datamodel-codegen sometimes emits the JSON property name (camelCase)
+  instead of the Python field name (snake_case) in `Field(discriminator='...')` annotations,
+  particularly when the discriminator is on a schema referenced inside array items.
+"""
+
+from __future__ import annotations
+
+import re
+from pathlib import Path
+
+MODELS_PATH = Path(__file__).resolve().parent.parent / 'src' / 'apify_client' / '_models.py'
+
+# Map of camelCase discriminator values to their snake_case equivalents.
+# Add new entries here as needed when the OpenAPI spec introduces new discriminators.
+DISCRIMINATOR_FIXES: dict[str, str] = {
+    'pricingModel': 'pricing_model',
+}
+
+
+def fix_discriminators(content: str) -> str:
+    """Replace camelCase discriminator values with their snake_case equivalents."""
+    for camel, snake in DISCRIMINATOR_FIXES.items():
+        content = re.sub(
+            rf"discriminator='{camel}'",
+            f"discriminator='{snake}'",
+            content,
+        )
+    return content
+
+
+def main() -> None:
+    content = MODELS_PATH.read_text()
+    fixed = fix_discriminators(content)
+
+    if fixed != content:
+        MODELS_PATH.write_text(fixed)
+        print(f'Fixed discriminator values in {MODELS_PATH}')
+    else:
+        print('No discriminator fixes needed')
+
+
+if __name__ == '__main__':
+    main()