feat: inheritance mode, body-field descriptions, keep soft keywords

[K1rL3s]

Three additions, driven by generating a hand-written client's shape:

• --inheritance renders allOf: [{$ref: Base}, ...] as a real base class instead of merging the parent's fields into every subtype. A discriminated base stays a model (its own properties survive) and its mapped subtypes inherit from it, re-declaring only the discriminator tag. Declarations are emitted parent-first so a class Sub(Base) statement resolves, and model constructors become keyword-only because a subclass may pin an inherited field to a default while adding required fields of its own.

• IRBodyField.description carries the schema description of a spread request body field, which was silently dropped. IRParameter already had it.

• sanitize_identifier no longer suffixes soft keywords: type is a legal attribute name and an extremely common spec field, so type_ was noise. _ stays reserved.

Verified on a real 130-schema spec: both file layouts x all three serializers generate code that passes ruff --isolated and mypy --strict.

Description

Please include a summary of the change and specify which issue is being addressed. Additionally, provide relevant motivation and context.

Fixes # (issue number)

Type of change

Please delete options that are not relevant.

• Documentation (typos, code examples, or any documentation updates)
• Bug fix (a non-breaking change that resolves an issue)
• New feature (a non-breaking change that adds functionality)
• Breaking change (a fix or feature that would disrupt existing functionality)
• This change requires a documentation update

Checklist

• My code adheres to the style guidelines of this project (uv run ruff check and uv run ruff format --check show no errors)
• I have conducted a self-review of my own code
• I have made the necessary changes to the documentation
• My changes do not generate any new warnings
• I have added tests to validate the effectiveness of my fix or the functionality of my new feature
• I have ensured that type checking passes by running uv run mypy
• I have included code examples to illustrate the modifications

[K1rL3s]

Review of 566dc7c → fixes in 53f0765

Went through the diff for recall. Found 12 issues, all fixed in a follow-up commit. What was breaking, and what it does now.

Critical: silently corrupted decoded data

1. oneOf + discriminator was turned into an empty class

The most common polymorphism shape in OpenAPI is a union holder with no properties of its own:

Button:
  oneOf: [CallbackButton, LinkButton]
  discriminator: {propertyName: type, mapping: {...}}

The _build_named branch went into _build_object unconditionally, so --inheritance produced:

@dataclass(kw_only=True)
class Button:
    pass

@dataclass(kw_only=True)
class Keyboard:
    buttons: list[Button]     # ← everything decoded into this, and everything was lost

There is nothing to inherit from such a schema. Only a base that declares its own properties (_is_object) becomes a class now; the rest stays a union alias. Verified: pydantic again returns CallbackButton(payload='P'), while isinstance(x, ButtonBase) still holds — both goals of the feature coexist.

2. The discriminator tag was overwritten by a sibling field

_apply_discriminator_tag inserted the field bypassing the model's field_names registry. If a subtype declared Type while propertyName was type, both snake-cased to the same identifier:

class A(B):
    type: Literal['a'] = 'a'
    type: int | None = None      # ← the second one wins

ruff --isolated lets this through (F811 does not fire on annotated class attributes) and dataclasses.fields(A) shows only int | None. The tag was destroyed, and name_mapping mapped type -> 'Type' — the wrong wire name. The name is now reserved against the fields already on the model.

3. Re-declaring an inherited field broke mypy --strict

Specs routinely restate a base property to attach prose or to allow null. That produced an unsound override:

error: Incompatible types in assignment (expression has type "str | None",
base class "P" defined the type as "str")  [assignment]

So --inheritance --check failed on ordinary specs. Pure restatements and widenings are now inherited; genuine narrowings (a Literal over a str) are kept. _is_narrowing is deliberately conservative: a false yes emits code that does not type-check, a false no merely inherits a slightly less precise type.

4. Inheritance depended on graph traversal order

The base was looked up in self._declarations, which has no entry yet while the base itself is being built. If the base's own body refers back to its subtype (a recursive hierarchy), the entry was missing:

before: LeafNode base=None  fields=['id', 'child', 'value']   ← merged, --inheritance did nothing
after:  LeafNode base=Node  fields=['value']

The decision is now made from the schema (_declares_model), not from a half-built registry.

The rest

5. kw_only is now per-model. A single allOf subtype used to flip every model in the package to keyword-only, breaking positional construction for models unrelated to any hierarchy. Only hierarchy members are marked now; Keyboard from the example stays a plain @dataclass.

6. IRModel.discriminator was read by nobody. For a base kept as a class, no serializer resolves the concrete subtype from a base-class annotation on its own. The mapping is now emitted as a comment — exactly what you need to wire tagged decoding in _serialization.py. The limitation is documented in the README.

7. IRBodyField.description was not rendered. The commit message says it "was silently dropped" — but the drop point had only moved from the builder to the renderer: no template reads field descriptions. Both IRBodyField.description and IRParameter.description now reach the generated code as PEP 258 attribute docstrings, without touching the constructor signature:

title: Body[str]
"""Doc title."""
type: Query[Omittable[str]] = Omitted()
"""Kind of send."""

8. IRModel.base → base_model. IREnum.base is the enum's value type ("str"/"int"). Both new call sites already guarded with isinstance(...), but any future getattr(decl, "base", None) loop would read an enum's "str" as a superclass name.

9. Soft keywords use an allow-list. frozenset({"_"}) in place of keyword.issoftkeyword() would stop catching whatever word the next Python release promotes (that is how type arrived in 3.12). It is now issoftkeyword(c) and c not in {"type", "match", "case"} — the reviewed names are hardcoded, not the forbidden ones.

10-12. _ordered_declarations breaks an inheritance cycle explicitly (marking visited before recursing prevented infinite recursion but not the wrong emit order); the two discriminator.mapping traversals share one _convert_mapped_subtypes helper; _inherited_ref is computed once and passed into _flatten_object instead of a keep_base flag, so the caller's decision and the merge cannot disagree.

Verification

• 208 tests green (+6 regression tests, one per bug 1-4, +1 for per-model kw_only, +1 for mapping visibility, +1 for attribute docstrings).
• ruff format --check, ruff check, mypy --strict clean across the repo.
• End-to-end generation: both file layouts × all three serializers → the output passes ruff check --isolated and mypy --strict. The only remaining error is BaseMethod.__init_subclass__ [no-untyped-call], which comes from unihttp and predates this PR.

One call worth a second opinion

For the widening case (item 3) I chose to inherit the base's declaration, which means a subtype's nullable: true is dropped. The alternative is to keep the override and accept the mypy failure. I went with code that type-checks, but if annotation fidelity matters more for your specs, that is a one-line flip in _is_narrowing.

[K1rL3s]

The pinned tag bypasses _is_narrowing, so --inheritance --check fails on the most common discriminator shape.

_drop_unsafe_overrides runs inside _build_object; this insert runs later, from _convert_ref -> _apply_discriminator_tag. Nothing re-checks the field it adds, so Literal[tag] is emitted over whatever the base declared. When the base types the discriminator property as a $ref to an enum (the idiomatic OpenAPI form) that is not a narrowing.

Worse, the two paths fight each other: if the subtype does restate type, _drop_unsafe_overrides correctly drops it, and then this branch puts it right back unchecked.

Reproduced with Button.type: {$ref: ButtonKind} + mapping: {callback: CallbackButton, link: LinkButton}:

@dataclass(kw_only=True)
class Button:
    type: ButtonKind
    text: str

@dataclass(kw_only=True)
class CallbackButton(Button):
    payload: str
    type: Literal["callback"] = "callback"

models.py:26: error: Incompatible types in assignment (expression has type "Literal['callback']",
  base class "Button" defined the type as "ButtonKind")  [assignment]
models.py:32: error: ... "Literal['link']" ... "ButtonKind"  [assignment]

The same happens in a 3-level hierarchy where the middle class already pinned the tag to its own Literal.

The conservative rule the rest of the module follows applies here too: only pin when _is_narrowing(LiteralType((value,)), inherited_type) holds, otherwise inherit the base's declaration.

[K1rL3s]

Regression in the default (non---inheritance) path.

This fires for any IRModel whose discriminator is set, and in merge mode _flatten_object copies the base's discriminator down into every allOf subtype (discriminator = discriminator or d). So concrete subtypes now carry a header announcing them as tagged-union bases, with a partial, self-referential mapping - the mapping is resolved while the subtype is being built, so _ref_to_name only knows the subtypes converted so far.

No flag needed to hit it; --serializer adaptix on a plain Pet/Dog/Cat spec:

# discriminator: petType (dog=Dog)          <- claims Dog is a base that maps "dog" to itself
# subtype resolution is left to the serializer config
@dataclass
class Dog:
    ...

# discriminator: petType (cat=Cat, dog=Dog) <- and Cat is a base of Dog?
@dataclass
class Cat:
    ...

The root cause is in _flatten_object: a discriminator belongs to the schema that declares it - the same reasoning the PR already applies to the inherited member. Dropping the discriminator or d merge entirely fixes this and is safe: IRModel.discriminator has no other reader, and the inheritance base re-resolves its own in _build_named.

[K1rL3s]

wire_name is the wrong key: shadowing happens on the python name.

Type/type and packSize/pack_size are distinct wire names that produce one identifier. The lookup misses, the field is kept, and the subclass attribute shadows an inherited one of an unrelated type. _build_object's field_names registry only dedups within the model's own fields, so nothing else catches it either.

Reproduced (base B has type: string + packSize: integer; subtype A adds Type: integer + pack_size: string):

@dataclass(kw_only=True)
class A(B):
    type2: Literal["a"] = "a"
    type: int | None = None        # wire "Type"  -> shadows B.type: str
    pack_size: str | None = None   # wire "pack_size" -> shadows B.pack_size: int

models.py:20: error: Incompatible types in assignment (expression has type "int | None",
  base class "B" defined the type as "str")  [assignment]
models.py:21: error: ... "str | None" ... "int"  [assignment]

Dropping is also the wrong remedy here - the subtype's property is genuinely its own, it just needs a non-colliding identifier. Reserving the subtype's field names against the inherited ones (and letting only a same-wire-name re-declaration keep the name) preserves both fields and both aliases.

[K1rL3s]

Only one level of the base chain is checked.

parent.fields holds a subclass's own fields only - by construction, that is the whole point of this mode. So in A <- B <- C, A's fields are invisible when C is checked, and an incompatible re-declaration of an A field goes straight through.

Reproduced with A{v: string}, B: allOf[A] + {b}, C: allOf[B] + {v: integer, c}:

@dataclass(kw_only=True)
class C(B):
    v: int
    c: str | None = None

models.py:20: error: Incompatible types in assignment (expression has type "int",
  base class "A" defined the type as "str")  [assignment]

The inherited set has to be accumulated by walking base_model to the root (nearest declaration winning, since that is what mypy compares against).

[K1rL3s]

The half-built-registry problem _resolve_base_model fixes is reintroduced here.

_resolve_base_model goes to some length to avoid reading self._declarations for a base that has no entry yet, and documents exactly why:

a base whose own body refers back to this subtype (a recursive hierarchy) is still mid-build and has no entry yet, which would silently downgrade the subtype [...] based on nothing but graph traversal order

This method then does that read anyway and returns silently. In the Node / LeafNode shape covered by test_inheritance_recursive_base_still_subclasses, LeafNode is built while Node is mid-flight, so parent is None, no override is pruned, and a widening restatement (v: str | None over v: str) reaches the output and fails mypy --strict.

Since the decision needs every base's final field set anyway (see the grandparent case), the natural place for this pass is after the whole graph is built - build() already computes a base-before-subclass ordering, so walking that list makes the inherited set final and both this and the multi-level gap disappear.

[K1rL3s]

Wasted recursion, and it makes the new cycle guard dead code.

The recursive _flatten_object call happens before the sub is not inherited test, so in inheritance mode the base's entire property set is merged and then thrown away on the next line (only required survives). That is O(depth) redundant work per subtype.

It also has a correctness consequence for the guard added in _ordered_declarations. A base_model cycle can only come from an allOf cycle, and an allOf cycle dies here first:

A: allOf: [{$ref: B}, {properties: {a}}]
B: allOf: [{$ref: A}, {properties: {b}}]

  File "ir/builder.py", line 622, in _flatten_object
    p, r, a, d = self._flatten_object(sub_schema, sub_base)
  [Previous line repeated 980 more times]
RecursionError: maximum recursion depth exceeded

So _ordered_declarations' grey-marking branch, the logger.warning and the assert are unreachable. (The RecursionError itself predates this PR - merge mode crashes identically - but the guard is presented as handling it.)

Worth noting what that branch would do if it were reachable: by the time it runs, _flatten_object has already skipped the base's properties and _drop_unsafe_overrides has already removed the subtype's re-declarations. Clearing base_model at that point emits a class that has lost every inherited field rather than one that fails to import.

[K1rL3s]

A restatement with the same annotation is dropped wholesale, but annotation is not all a restatement can carry. default and description ride on the same IRField, and specs restate a property to change exactly those:

Base:  {properties: {mode: {type: string, default: "fast"}}}
Sub:
  allOf:
    - $ref: '#/components/schemas/Base'
    - properties: {mode: {type: string, default: "slow", description: "Slower here."}}

Sub silently inherits mode = "fast". The description loss is the more awkward one given this same PR adds descriptions to body fields - the prose the spec author attached to the subtype never reaches the output.

Keeping the field when only default/description differ is sound (identical annotation is always a legal override), so the return False could be conditioned on the restatement adding nothing at all.

[K1rL3s]

docstring(text, indent) uses indent for two things: the prefix it writes, and the wrap width (88 - len(indent)). Passing "" here and letting render_method_class add the four spaces gets the prefix right and the width wrong.

 16 |    q: Query[str]|
 17 |    r"""First paragraph that is long enough to need wrapping across several lines in the output|  <- 95 cols
 18 |    file.|
 19 |    |  <- trailing whitespace
 20 |    Second paragraph with a backslash: curl \ -H "X: y".|

Model field docstrings pass " " and come out right. E501/W291 are outside ruff's default select so --check stays green, but the 88-column target this helper exists to hit is missed. Rendering with the real indent and stripping it back off before render_method_class re-adds it fixes the width; skipping the prefix on empty lines fixes the trailing whitespace.

[K1rL3s]

The docstring is candid that this mirrors _build_named ("kept next to nothing else so the two stay reviewable side by side"), but the mirror is already off by one branch.

_build_named, for a discriminated base that is not _is_object, falls through to _build_discriminated_base, which ends with:

if not members:
    self._declarations[name] = self._build_object(name, schema, base_uri)
    return

So an empty or fully unresolvable mapping yields an IRModel, while this predicate returns False for it - a subtype pointing at that base silently downgrades to a merge.

The drift is minor today, but the shape is the concern: two hand-synchronised copies of a dispatch, in a mode where a wrong answer changes the class hierarchy. Having _build_named record its decision (or splitting the classification out and calling it from both) removes the class of bug rather than this instance.

[K1rL3s]

_kw_only_models starts empty, so is_kw_only answers False for every model until bind_document has run. Both production paths happen to bind (render_models_module does it internally, _write_per_object_layout does it explicitly), but render_declaration / render_model are public and neither documents the requirement.

Get it wrong and the failure is not a missing keyword - it is @dataclass on a subclass whose base ends in a defaulted field while the subclass declares a required one:

TypeError: non-default argument 'payload' follows default argument

at import time of the generated package.

The property is a fact about the IR ("this model is in a hierarchy"), not about a rendering session; computing it from model.base_model plus the document once in the builder, or storing it on IRModel, would make the ordering unrepresentable.

[K1rL3s]

Pydantic merges model_config down the MRO, so on a subclass this line is a no-op:

class CallbackButton(Button):
    model_config = ConfigDict(populate_by_name=True)   # already true via Button
    type: Literal["callback"] = "callback"

It also reads as if the subclass were deliberately overriding the parent's config. Guarding on model.base_model is None keeps the emitted hierarchy honest.

[K1rL3s]

_discriminator_comment emits

# discriminator: type (callback=CallbackButton, link=LinkButton)
# subtype resolution is left to the serializer config

and render_alias, just below, emits

# discriminator: type (tagged-union wiring is left to the serializer config)

Same information, same audience, two wordings and two layouts - and the alias form drops the value->class mapping even though IRAlias.discriminator.mapping carries it, which is the part a reader actually needs to wire tagged decoding. Routing render_alias through the new helper gives both forms the mapping and leaves one string to maintain.

[K1rL3s]

_convert_ref on the line above already resolved this $ref (it needs the pointer to key _ref_to_name); this resolves it again for the same pointer. RefResolver.resolve_ref has no memoisation and re-splits/re-walks the document each call, and this now runs for every allOf subtype in the spec.

Only the key and resolved.value are needed here, both of which _convert_ref had in hand - returning them (or looking the pointer up from _ref_to_name) drops one full resolution per subtype.

[K1rL3s]

Correction to the scope claim above: the extra resolve_ref is guarded by if declared is not None: return ..., so it only runs when the base has no _declarations entry yet - the recursive-hierarchy case, not every allOf subtype. The duplicate resolution is real but rare, so I left it alone.

[K1rL3s]

Pushed 2847eee addressing the review above.

Fixed

#	Finding	Fix
1	Pinned tag bypassed _is_narrowing (enum-typed discriminator property)	override pruning moved into _reconcile_inheritance, a whole-graph pass, so the tag is checked like any other override
2	# discriminator: header on concrete subtypes in default mode	_flatten_object no longer merges a discriminator in from allOf
3	Inherited fields keyed by wire_name, not the python name	_rename_shadowed_fields: the subtype's field is renamed and aliased instead of shadowing
4	Only the direct parent was consulted	_inherited_fields walks the chain to the root, nearest declaration winning
5	Check no-opped on a mid-build base	same whole-graph pass - every ancestor is final by then
7	Restatement dropped its own default	kept when the annotation is identical and the default differs
8	Attribute docstrings wrapped for the wrong column, trailing whitespace	rendered at the real indent, blank separators left unindented
11	model_config repeated on pydantic subclasses	emitted only on a root model (with a pass fallback for an empty subclass body)
12	Two formats for the discriminator note	render_alias reuses _discriminator_comment, which also gives the alias form the value → class mapping it was dropping

One extra bug surfaced while fixing #3: renaming the sibling freed the name but left the tag on type2, so A had two attributes for wire key type and adaptix refused to build the retort ("fields point to the same path"). _align_override_names now puts a re-declared wire name back on the inherited attribute.

Skipped

• fix: resolve ruff and mypy from the generator's own environment #6 (redundant _flatten_object recursion / dead cycle guard) — the recursion cannot be dropped without regressing the required merge: a narrowing restatement that relies on the base's required would flip to optional and then be pruned as a widening. The underlying RecursionError predates this PR (merge mode dies identically), so the guard stays as-is rather than being reworked here.
• #9 (_declares_model mirrors _build_named) — the real fix is having _build_named record its decision, which is a wider restructuring than this PR's surface.
• #10 (is_kw_only depends on bind_document) — moving the flag onto IRModel touches the IR contract and every strategy; both production paths do bind today.
• #13 — overstated, see the correction in-thread.

Verification: 7 specs × 3 serializers × 2 layouts. Every package passes mypy --strict (only the pre-existing BaseMethod.__init_subclass__ no-untyped-call remains), adaptix builds a dumper for every generated model, and all three serializers round-trip the collision cases. 213 unit tests pass, including 5 new regression tests.

[K1rL3s]

Review of 566dc7c..2847eee — ready to merge, with one commit added

Verdict: the inheritance mode is correct. I could not break it. What I did add is the safety net the two follow-up commits argue for.

What I verified independently

Built two adversarial specs (beyond the ones in the tests) and generated 3 serializers x 2 layouts from each:

• discriminated base with its own properties, plus a bare oneOf holder (stays a union alias);
• a subtype that is itself a discriminated base with properties (three levels of polymorphism);
• tag typed as $ref to an enum vs. as a plain string;
• restatements that only add prose / relax to nullable / change the default;
• packSize + pack_size on one class, in both declaration orders, and type + Type;
• a three-level chain re-declaring the grandparent's fields;
• a base whose own body refers back to its subtype;
• allOf with two refs (mixin) and with a ref to an enum — both keep the merge;
• --optional omitted on top of --inheritance.

Every package: ruff check clean, mypy --strict clean, generation deterministic. Runtime round-trip through pydantic / msgspec / adaptix keeps all seven colliding wire keys distinct, isinstance holds across the hierarchy, the default-only override wins over the base's value, and the recursive hierarchy constructs. _align_override_names running after _rename_shadowed_fields is load-bearing and provably collision-free — the only field that can hold an inherited python name after the rename pass is the override itself.

Gaps closed (712779d)

1. --inheritance had no compile gate. Both follow-up commits fixed generated code that failed mypy --strict; the unit tests assert on the IR, and a subclass declaration is exactly where a slightly-wrong IR becomes an [assignment] error. Added to test_compile_gate.py: the hierarchy spec through all three serializers x both layouts. Per-object is not redundant — the base class is the one model reference that must be imported at runtime rather than deferred into TYPE_CHECKING.

2. No round-trip test. ruff and mypy are both blind to the failure that 53f0765 fixed: two attributes collapsing onto one identifier silently drops a value, and two attributes on one wire key makes adaptix refuse to build the retort. Added to test_behavior.py, parametrized over all three serializers. Shared hierarchy_spec fixture in conftest.py, one schema per rule.

3. _ordered_declarations' cycle break was unreachable. An inheritance cycle requires an allOf cycle, and _flatten_object followed one until the stack gave out — pre-existing on main, in both modes. _flatten_object now tracks the schemas open on the current path and skips a repeated member with a warning, so the cycle break actually runs and emits resolvable classes. Output for every acyclic spec is byte-identical; verified by diffing the full generated tree before and after.

Known limitations, deliberate and documented

• A discriminated base that stays a class decodes into the base, not the subtype — the # discriminator: comment carries the mapping. Documented in the README, correct call.
• When the base types the tag as an enum, the subtype's Literal tag is dropped rather than pinned. Type-safe, and the base still requires the field. Also documented.
• An allOf cycle still crashes before the builder ever runs, inside openapi-spec-validator — third-party, out of scope here.

223 passed, ruff + mypy --strict clean on the repo itself.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 100.00%. Comparing base (83590bd) to head (9ca381d).

Additional details and impacted files

@@            Coverage Diff             @@
##              main        #5    +/-   ##
==========================================
  Coverage   100.00%   100.00%            
==========================================
  Files           29        29            
  Lines         2256      2557   +301     
==========================================
+ Hits          2256      2557   +301     

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.