|
|
|
@ -3,8 +3,9 @@ from __future__ import annotations
|
|
|
|
|
from concurrent.futures import Executor, ThreadPoolExecutor
|
|
|
|
|
from contextlib import contextmanager
|
|
|
|
|
from copy import deepcopy
|
|
|
|
|
from typing import Any, Dict, Generator, List, Optional, TypedDict
|
|
|
|
|
from typing import TYPE_CHECKING, Any, Dict, Generator, List, Optional, TypedDict
|
|
|
|
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
|
|
|
from langchain.callbacks.base import BaseCallbackManager, Callbacks
|
|
|
|
|
from langchain.callbacks.manager import AsyncCallbackManager, CallbackManager
|
|
|
|
|
|
|
|
|
@ -87,6 +88,8 @@ def patch_config(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_callback_manager_for_config(config: RunnableConfig) -> CallbackManager:
|
|
|
|
|
from langchain.callbacks.manager import CallbackManager
|
|
|
|
|
|
|
|
|
|
return CallbackManager.configure(
|
|
|
|
|
inheritable_callbacks=config.get("callbacks"),
|
|
|
|
|
inheritable_tags=config.get("tags"),
|
|
|
|
@ -97,6 +100,8 @@ def get_callback_manager_for_config(config: RunnableConfig) -> CallbackManager:
|
|
|
|
|
def get_async_callback_manager_for_config(
|
|
|
|
|
config: RunnableConfig,
|
|
|
|
|
) -> AsyncCallbackManager:
|
|
|
|
|
from langchain.callbacks.manager import AsyncCallbackManager
|
|
|
|
|
|
|
|
|
|
return AsyncCallbackManager.configure(
|
|
|
|
|
inheritable_callbacks=config.get("callbacks"),
|
|
|
|
|
inheritable_tags=config.get("tags"),
|
|
|
|
|