AttributeError: 'CrateCompilerSA20' object has no attribute 'visit_on_conflict_do_update'

[amotl]

Problem

When deriving a CrateDB adapter for LangChain from langchain-postgres, we are observing this error when invoking a test case.

pytest -vvv -k test_add_documents_documents

    def _compiler_dispatch(
        self: Visitable, visitor: Any, **kw: Any
    ) -> str:
        """Look for an attribute named "visit_<visit_name>" on the
        visitor, and call it with the same kw params.

        """
        try:
>           meth = getter(visitor)
E           AttributeError: 'CrateCompilerSA20' object has no attribute 'visit_on_conflict_do_update'

[amotl]

Can quickly be solved using a monkeypatch. Needs to be upstreamed to sqlalchemy-cratedb.

def patch_sqlalchemy_dialect():
    """
    Fixes `AttributeError: 'CrateCompilerSA20' object has no attribute 'visit_on_conflict_do_update'`

    TODO: Upstream to `sqlalchemy-cratedb`.
          https://github.com/crate/sqlalchemy-cratedb/issues/186
    """  # noqa: E501
    from sqlalchemy.dialects.postgresql.base import PGCompiler
    from sqlalchemy_cratedb.compiler import CrateCompiler

    CrateCompiler.visit_on_conflict_do_update = PGCompiler.visit_on_conflict_do_update
    CrateCompiler._on_conflict_target = PGCompiler._on_conflict_target


patch_sqlalchemy_dialect()