[python-fastapi] Fix additionalProperties support & support pydantic v2
- Don't let generated model code extend the type specified within the
additionalPropertiesschema (or the array item schema) but the pydanticBaseModel. - Support
additionalPropertiesthe pydantic v2 way within__dict__: Previously, the generatedadditional_propertiesfield showed up within the response of the generated API as opposed marshaling the model so that its fields are added to/embedded into the root object. Apparently that is because pydantic v2 does not honour the generatedto_dictmethods (which would have mapped the object to the correct representation) but, instead, supports additional properties natively by specifyingextra=allowwithin themodel_config. Correspondingly, the following changes have been applied:- To allow additional fields, specify
extra=allowwithin themodel_config. - Don't generate the
additional_propertiesfield but use__dict__- users can use pydantic's built-inmodel.model_extrainstead. - ~~Don't generate the
{to|from}_{dict|json}methods since pydantic is taking care of the model mapping based on the declared fields andmodel_config- users can use pydantic'smodel.model_dump[_json]instead.~~ - Let the generated
{to|from}_{dict|json}methods delegate to Pydantic'smodel_dump[_json]methods. - Let the generated API endpoints enable the
exclude_unsetresponse marshalling option in order to omit fields from the response that weren't explicitly set by the code. This is so that non-required fields don't show up withnullvalues within the response (which would be invalid according to the OpenAPI spec, unless those fields are explicitly marked asnullablewithin the OpenAPI schema).
- To allow additional fields, specify
- Support
oneOfandanyOfschemas the Pydantic v2 way:- Let the generated model extend the pydantic
RootModel. - Generate (discriminated)
Unions and leave it to Pydantic to validate and figure out the type. - Generate model constructors that forcefully set the discriminator field in order to ensure it is included in the marshalled representation (otherwise it would be ignored when not explicitly set due to the
exclude_unsetmarshalling option being enabled).
- Let the generated model extend the pydantic
Closes #19311 Relates to #17703 (might also close that one) Relates to #19454
cc @cbornet @tomplus @krjakbrjak @fa0311 @multani
PR checklist
- [x] Read the contribution guidelines.
- [x] Pull Request title clearly describes the work in the pull request and Pull Request description provides details about how to validate the work. Missing information here may result in delayed response from the community.
- [x] Run the following to build the project and update samples:
(For Windows users, please run the script in Git BASH) Commit all changed files. This is important, as CI jobs will verify all generator outputs of your HEAD commit as it would merge with master. These must match the expectations made by your contribution. You may regenerate an individual generator by passing the relevant config(s) as an argument to the script, for example./mvnw clean package ./bin/generate-samples.sh ./bin/configs/*.yaml ./bin/utils/export_docs_generators.sh./bin/generate-samples.sh bin/configs/java*. IMPORTANT: Do NOT purge/delete any folders/files (e.g. tests) when regenerating the samples as manually written tests may be removed. - [x] File the PR against the correct branch:
master(upcoming 7.6.0 minor release - breaking changes with fallbacks),8.0.x(breaking changes without fallbacks) - [x] If your PR is targeting a particular programming language, @mention the technical committee members, so they are more likely to review the pull request.
Thanks for the PR! I'll need something like this PR to move off python-pydantic-v1.
I have a question about this bit:
Don't generate the
{to|from}_{dict|json}methods since pydantic is taking care of the model mapping based on the declared fields andmodel_config- users can use pydantic'smodel.model_dump[_json]instead.
Why remove {to|from}_{dict|json} (and to_str) rather than replace the body of those methods with the right pydantic calls? I imagine this could be a substantial breaking change for some folks.
Why remove
{to|from}_{dict|json}(andto_str) rather than replace the body of those methods with the right pydantic calls? I imagine this could be a substantial breaking change for some folks.
The idea is to let the generated code fully leverage the fastapi/pydantic platform idioms, which include letting the route methods return typed objects that pydantic then maps and validates based on the type annotation.
I didn't want to bloat the generated code unnecessarily and I was thinking the python-fastapi generator is young enough so that I could still make such a breaking change. However, to make it backward-compatible, I'll adjust the PR to let it generate those methods again but letting them delegate to pydantic's methods...
I updated the PR correspondingly but unfortunately this is still a breaking change since
nullablefield semantics are not honoured anymore - would need to be generated into the type annotation like this maybe or applied via theexclude_unsetmarshalling option.- Discriminator logic for
oneOfschemas is not generated - would need to be generated into the type annotation like this.
I can look into making the adjustments some other evening.
I like it! I anticipate using this as a client to replace python-pydantic-v1. Hopefully a maintainer can chime in/approve when you are ready!
I also made the generator generate oneOf and anyOf schemas as (discriminated) Unions the Pydantic v2 way and enabled the exclude_unset response option for generated API routes/endpoints (PR description updated correspondingly). The PR is ready for review now.
Epic! I hope a maintainer can review it for you soon.
ping @cbornet @tomplus @krjakbrjak @fa0311 @multani
Came across this while looking to convert an OpenAPI spec to a Python client. Would really like to try this out, but not going to bother until Pydantic V2 support is released. Fingers crossed this PR gets merged soon.
Same here !I'd also be interested by the support of pydantic v2 :-)
I just merged the master branch into this one in order to resolve the merge conflict (an import within a Java test class).
However, now the Python test of the generated pydantic v1 client fails due to the DELETE /v2/pet/{petId} endpoint (delete_pet) returning a 404 response when being called during the test data setup of the test_404_error test (prior to the actual 404 response test):
=========================== short test summary info ============================
FAILED tests/test_api_exception.py::ApiExceptionTests::test_404_error - petstore_api.exceptions.NotFoundException: (404)
Reason: Not Found
HTTP response headers: HTTPHeaderDict({'Date': 'Sat, 08 Mar 2025 16:39:45 GMT', 'Content-Length': '0', 'Connection': 'keep-alive', 'Access-Control-Allow-Origin': '*', 'Access-Control-Allow-Methods': 'GET, POST, DELETE, PUT', 'Access-Control-Allow-Headers': 'Content-Type, api_key, Authorization', 'Server': 'Jetty(9.2.9.v20150224)'})
============ 1 failed, 265 passed, 2 skipped, 18 warnings in 5.69s =============
That doesn't seem to be caused by the changes introduced within this PR as the build of #20830 confirms which fails with the same error.
Turns out the failed python client test was fixed here now. Correspondingly, I applied the fix also to the python-pydantic-v1 client test which failed for the same reason. Also, I rebased the PR.
Now that the merge conflicts are resolved and the CI build succeeds again, this PR is ready for review again.
Thanks for your efforts with this MR, it would be really great to see this get merged (along with a few of the other python-fastapi fixes).
Thank you for your work - as many I probably found this PR because I ran into similar problems with the python-fastapi generator. It would be really nice if the python-fastapi fix PRs could be merged.
I rebased the PR now again to resolve the merge conflicts (which dropped my last commit that fixed the python-pydantic-v1 test since the fix is on the master branch now).
However, now the Python client deserialization test fails with the following error:
tests/test_deserialization.py:301: error: Item "None" of "Union[Cat, Dog, None]" has no attribute "class_name" [union-attr]
tests/test_deserialization.py:302: error: Item "Dog" of "Union[Cat, Dog, None]" has no attribute "declawed" [union-attr]
tests/test_deserialization.py:302: error: Item "None" of "Union[Cat, Dog, None]" has no attribute "declawed" [union-attr]
tests/test_deserialization.py:303: error: Item "None" of "Union[Cat, Dog, None]" has no attribute "to_json" [union-attr]
Though, that problem is not related to this PR which the master branch build of #21415 shows which also fails with the same error! (The same problem also makes the build of PR #21402 fail.)
Thus, this PR is ready for review again!
Rebased the PR again to include the python unit test fix of #21423. Now the docs and samples are reported to be outdated but that has nothing to do with this PR but is the case within the master branch.
This PR is ready for review again!
I rebased the PR again to include the fixes from the main branch.
Please review!
@wing328 can you review the PR by any chance?
@mgoltzsche thanks for the PR 🙏
it's a huge change. do you mind pinging me via Slack (https://app.slack.com/client/TLQFRCNJZ/dms) when you've time tomorrow or Friday for a few quick questions on this PR?
Hey guys bumping the pull request would be great to get it in the mainstream branch
Unfortunately it doesn't look like this PR will be merged. @wing328 told me via Slack that it is too big to be reviewed and that the actual Pydantic v2 support should be implemented in the client generator first in form of multiple small incremental PRs since there are more tests for that. Afterwards the changes could be applied to the FastAPI server generator similarly. However, I don't have time for that any time soon.
thanks for your effort so far!
I can confirm that anyOf and oneOf schemas are also not supported for the client generator, which generates erroneous code for those.