mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-21 08:45:22 -04:00
Tim Conkling
c28990d871
A new implementation of `StreamlitCallbackHandler`. It formats Agent thoughts into Streamlit expanders. You can see the handler in action here: https://langchain-mrkl.streamlit.app/ Per a discussion with Harrison, we'll be adding a `StreamlitCallbackHandler` implementation to an upcoming [Streamlit](https://github.com/streamlit/streamlit) release as well, and will be updating it as we add new LLM- and LangChain-specific features to Streamlit. The idea with this PR is that the LangChain `StreamlitCallbackHandler` will "auto-update" in a way that keeps it forward- (and backward-) compatible with Streamlit. If the user has an older Streamlit version installed, the LangChain `StreamlitCallbackHandler` will be used; if they have a newer Streamlit version that has an updated `StreamlitCallbackHandler`, that implementation will be used instead. (I'm opening this as a draft to get the conversation going and make sure we're on the same page. We're really excited to land this into LangChain!) #### Who can review? @agola11, @hwchase17
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.infino_callback import InfinoCallbackHandler
|
|
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,
|
|
)
|
|
from langchain.callbacks.streamlit import LLMThoughtLabeler, 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",
|
|
"InfinoCallbackHandler",
|
|
"MlflowCallbackHandler",
|
|
"OpenAICallbackHandler",
|
|
"StdOutCallbackHandler",
|
|
"StreamingStdOutCallbackHandler",
|
|
"StreamlitCallbackHandler",
|
|
"LLMThoughtLabeler",
|
|
"WandbCallbackHandler",
|
|
"WhyLabsCallbackHandler",
|
|
"get_openai_callback",
|
|
"tracing_enabled",
|
|
"wandb_tracing_enabled",
|
|
]
|