mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-18 18:34:27 -04:00
Leonid Ganeline
c7ca350cd3
In LangChain, all module classes are enumerated in the `__init__.py` file of the correspondent module. But some classes were missed and were not included in the module `__init__.py` This PR: - added the missed classes to the module `__init__.py` files - `__init__.py:__all_` variable value (a list of the class names) was sorted - `langchain.tools.sql_database.tool.QueryCheckerTool` was renamed into the `QuerySQLCheckerTool` because it conflicted with `langchain.tools.spark_sql.tool.QueryCheckerTool` - changes to `pyproject.toml`: - added `pgvector` to `pyproject.toml:extended_testing` - added `pandas` to `pyproject.toml:[tool.poetry.group.test.dependencies]` - commented out the `streamlit` from `collbacks/__init__.py`, It is because now the `streamlit` requires Python >=3.7, !=3.9.7 - fixed duplicate names in `tools` - fixed correspondent ut-s #### Who can review? @hwchase17 @dev2049
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
"""Callback handlers that allow listening to events in LangChain."""
|
|
|
|
from langchain.callbacks.aim_callback import AimCallbackHandler
|
|
from langchain.callbacks.argilla_callback import ArgillaCallbackHandler
|
|
from langchain.callbacks.arize_callback import ArizeCallbackHandler
|
|
from langchain.callbacks.clearml_callback import ClearMLCallbackHandler
|
|
from langchain.callbacks.comet_ml_callback import CometCallbackHandler
|
|
from langchain.callbacks.file import FileCallbackHandler
|
|
from langchain.callbacks.human import HumanApprovalCallbackHandler
|
|
from langchain.callbacks.manager import (
|
|
get_openai_callback,
|
|
tracing_enabled,
|
|
wandb_tracing_enabled,
|
|
)
|
|
from langchain.callbacks.mlflow_callback import MlflowCallbackHandler
|
|
from langchain.callbacks.openai_info import OpenAICallbackHandler
|
|
from langchain.callbacks.stdout import StdOutCallbackHandler
|
|
from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
|
|
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
|
|
from langchain.callbacks.streaming_stdout_final_only import (
|
|
FinalStreamingStdOutCallbackHandler,
|
|
)
|
|
|
|
# now streamlit requires Python >=3.7, !=3.9.7 So, it is commented out here.
|
|
# from langchain.callbacks.streamlit import StreamlitCallbackHandler
|
|
from langchain.callbacks.wandb_callback import WandbCallbackHandler
|
|
from langchain.callbacks.whylabs_callback import WhyLabsCallbackHandler
|
|
|
|
__all__ = [
|
|
"AimCallbackHandler",
|
|
"ArgillaCallbackHandler",
|
|
"ArizeCallbackHandler",
|
|
"AsyncIteratorCallbackHandler",
|
|
"ClearMLCallbackHandler",
|
|
"CometCallbackHandler",
|
|
"FileCallbackHandler",
|
|
"FinalStreamingStdOutCallbackHandler",
|
|
"HumanApprovalCallbackHandler",
|
|
"MlflowCallbackHandler",
|
|
"OpenAICallbackHandler",
|
|
"StdOutCallbackHandler",
|
|
"StreamingStdOutCallbackHandler",
|
|
# now streamlit requires Python >=3.7, !=3.9.7 So, it is commented out here.
|
|
# "StreamlitCallbackHandler",
|
|
"WandbCallbackHandler",
|
|
"WhyLabsCallbackHandler",
|
|
"get_openai_callback",
|
|
"tracing_enabled",
|
|
"wandb_tracing_enabled",
|
|
]
|