/ mlflow / langchain / __init__.py
__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      ]