forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
682 B
Python
21 lines
682 B
Python
"""Callback handlers that allow listening to events in LangChain."""
|
|
from langchain.callbacks.base import BaseCallbackHandler, BaseCallbackManager
|
|
from langchain.callbacks.shared import SharedCallbackManager
|
|
from langchain.callbacks.stdout import StdOutCallbackHandler
|
|
|
|
|
|
def get_callback_manager() -> BaseCallbackManager:
|
|
"""Return the shared callback manager."""
|
|
return SharedCallbackManager()
|
|
|
|
|
|
def set_handler(handler: BaseCallbackHandler) -> None:
|
|
"""Set handler."""
|
|
callback = get_callback_manager()
|
|
callback.set_handler(handler)
|
|
|
|
|
|
def set_default_callback_manager() -> None:
|
|
"""Set default callback manager."""
|
|
set_handler(StdOutCallbackHandler())
|