Send context using in venv operator

[phi-friday]

origin: #41039
related: #41362

---

^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

[eladkal]

something is wrong as it shows 0 files changed
ah you need to wait till the revert PR is merged. otherwise the diff is nothing

[phi-friday]

something is wrong as it shows 0 files changed ah you need to wait till the revert PR is merged. otherwise the diff is nothing

I'm not sure what to do as I've never seen this before.
If #41362 is merged, should I just rebase it?

[phi-friday]

Based on the error message given by #41362, I checked and it only works with pydantic>=2.
I think we need to either modify the function _pydantic_model_dump or set use_pydantic_models to False.
We could also make this only available for pydantic>=2.

[phi-friday]

I don't know if this is allowed.
Modifying the model as follows did not cause the problem,
even if it was pydantic<2 or none (I checked in local, only pydantic test).

in airflow.utils.pydantic

if is_pydantic_2_installed():
    from pydantic import BaseModel, ConfigDict, PlainSerializer, PlainValidator, ValidationInfo
else:

    class BaseModel:  # type: ignore[no-redef]  # noqa: D101
        def __init__(self, *args, **kwargs):
            self.__dict__.update(kwargs)

        @classmethod
        def model_validate(cls, value: object):
            args = {name: getattr(value, name, None) for name in cls.__annotations__}
            return cls(**args)

        def model_dump(self, *args, **kwargs):
            from airflow.serialization.serialized_objects import BaseSerialization

            return {
                name: BaseSerialization.serialize(getattr(self, name, None), use_pydantic_models=True)
                for name in self.__annotations__
            }

[potiuk]

Ah yes. This is because serialization uses Pydantic currently to serialize DB models that are part of the context and that requires Pydantic v2 to be installed. I'd say we should make this feature only available if Pydantic2 is installed (with approproate documentation and error messages) - and skip the tests if Pydantic 2 is not installed.

[phi-friday]

airflow/airflow/serialization/serialized_objects.py

Line 630 in 4e62909

if use_pydantic_models and not _ENABLE_AIP_44:

If AIP44 is not enabled, an error is thrown.
So I made it so that an error is thrown if pydantic<2 or AIP44 is not enabled.
And I added test cases for each of them.