__init__.py
1 from mlflow.langchain.autolog import autolog 2 from mlflow.langchain.constants import FLAVOR_NAME 3 from mlflow.version import IS_TRACING_SDK_ONLY 4 5 __all__ = ["autolog", "FLAVOR_NAME"] 6 7 # Import model logging APIs only if mlflow skinny or full package is installed, 8 # i.e., skip if only mlflow-tracing package is installed. 9 if not IS_TRACING_SDK_ONLY: 10 from mlflow.langchain.model import ( 11 _LangChainModelWrapper, 12 _load_pyfunc, 13 load_model, 14 log_model, 15 save_model, 16 ) 17 18 __all__ += [ 19 "_LangChainModelWrapper", 20 "_load_pyfunc", 21 "load_model", 22 "log_model", 23 "save_model", 24 ]