Callbacks Refactor [base]

docs/ecosystem/gpt4all.md

@@ -28,13 +28,15 @@ To stream the model's predictions, add in a CallbackManager.
 
 ```python
 from langchain.llms import GPT4All
-from langchain.callbacks.base import CallbackManager
+from langchain.callbacks.manager import CallbackManager
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+
 # There are many CallbackHandlers supported, such as
 # from langchain.callbacks.streamlit import StreamlitCallbackHandler
 
 callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
-model = GPT4All(model="./models/gpt4all-model.bin", n_ctx=512, n_threads=8, callback_handler=callback_handler, verbose=True)
+model = GPT4All(model="./models/gpt4all-model.bin", n_ctx=512, n_threads=8, callback_handler=callback_handler,
+                verbose=True)
 
 # Generate text. Tokens are streamed through the callback manager.
 model("Once upon a time, ")

langchain/callbacks/__init__.py

@@ -5,24 +5,22 @@
 
 from langchain.callbacks.aim_callback import AimCallbackHandler
 from langchain.callbacks.base import (
-    AsyncCallbackManager,
     BaseCallbackHandler,
     BaseCallbackManager,
-    CallbackManager,
 )
 from langchain.callbacks.clearml_callback import ClearMLCallbackHandler
 from langchain.callbacks.comet_ml_callback import CometCallbackHandler
+from langchain.callbacks.manager import CallbackManager
 from langchain.callbacks.openai_info import OpenAICallbackHandler
-from langchain.callbacks.shared import SharedCallbackManager
 from langchain.callbacks.stdout import StdOutCallbackHandler
 from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
-from langchain.callbacks.tracers import SharedLangChainTracer
+from langchain.callbacks.tracers import LangChainTracer
 from langchain.callbacks.wandb_callback import WandbCallbackHandler
 
 
 def get_callback_manager() -> BaseCallbackManager:
     """Return the shared callback manager."""
-    return SharedCallbackManager()
+    return CallbackManager([])
 
 
 def set_handler(handler: BaseCallbackHandler) -> None:
@@ -48,7 +46,7 @@ def set_default_callback_manager() -> None:
 
 def set_tracing_callback_manager(session_name: Optional[str] = None) -> None:
     """Set tracing callback manager."""
-    handler = SharedLangChainTracer()
+    handler = LangChainTracer()
     callback = get_callback_manager()
     callback.set_handlers([handler, StdOutCallbackHandler()])
     if session_name is None:
@@ -71,10 +69,7 @@ def get_openai_callback() -> Generator[OpenAICallbackHandler, None, None]:
 
 
 __all__ = [
-    "CallbackManager",
-    "AsyncCallbackManager",
     "OpenAICallbackHandler",
-    "SharedCallbackManager",
     "StdOutCallbackHandler",
     "AimCallbackHandler",
     "WandbCallbackHandler",