o i}@sddlmZddlZddlmZddlmZmZzddlm Z ddl m Z Wn e y1e dwer@ddlmZdd lmZdddZdddZdS)) annotationsN)Iterable) TYPE_CHECKINGAny)SQLAlchemyInstrumentor)CommenterOptionsz`logfire.instrument_sqlalchemy()` requires the `opentelemetry-instrumentation-sqlalchemy` package. You can install this with: pip install 'logfire[sqlalchemy]')Engine AsyncEngineengineAsyncEngine | Engine | Nonereturn Engine | NonecCs ddlm}t||r|jS|S)Nrr )sqlalchemy.ext.asyncior isinstance sync_engine)r r ro/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/integrations/sqlalchemy.py_convert_to_sync_engines  rengines%Iterable[AsyncEngine | Engine] | Noneenable_commenterboolcommenter_optionsrkwargsrNonecKsdttt|}|durdd|D}Wdn1swYtjd||||d|S)zInstrument the `sqlalchemy` module so that spans are automatically created for each query. See the `Logfire.instrument_sqlalchemy` method for details. NcSsg|]}t|qSr)r).0 engine_entryrrr .sz)instrument_sqlalchemy..)r rrrr) contextlibsuppress ImportErrorrr instrument)r rrrrrrrinstrument_sqlalchemys r#)r r r r) r r rrrrrrrrr r) __future__rrcollections.abcrtypingrr(opentelemetry.instrumentation.sqlalchemyrlogfire.integrations.sqlalchemyrr! RuntimeError sqlalchemyrrr rr#rrrrs"