github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 16:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

don't deepcopy handlers (#3995)

Browse Source 
Co-authored-by: Sami Liedes <sami.liedes@iki.fi>
Co-authored-by: Sami Liedes <sami.liedes@rocket-science.ch>
This commit is contained in:
Ankush Gola 2023-05-02 13:53:27 -07:00 committed by GitHub
parent c26cf04110
commit 18f9d7b4f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
langchain/callbacks/base.py
View File

@ -1,7 +1,6 @@
"""Base callback handler that can be used to handle callbacks in langchain.""" """Base callback handler that can be used to handle callbacks in langchain."""
from __future__ import annotations from __future__ import annotations
import copy
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
from uuid import UUID from uuid import UUID
@ -367,15 +366,3 @@ class BaseCallbackManager(CallbackManagerMixin):
def set_handler(self, handler: BaseCallbackHandler, inherit: bool = True) -> None: def set_handler(self, handler: BaseCallbackHandler, inherit: bool = True) -> None:
"""Set handler as the only handler on the callback manager.""" """Set handler as the only handler on the callback manager."""
self.set_handlers([handler], inherit=inherit) self.set_handlers([handler], inherit=inherit)
def __copy__(self) -> "BaseCallbackManager":
return self.__class__(
self.handlers.copy(), self.inheritable_handlers.copy(), self.parent_run_id
)
def __deepcopy__(self, memo: dict) -> "BaseCallbackManager":
return self.__class__(
[copy.deepcopy(handler, memo) for handler in self.handlers],
[copy.deepcopy(handler, memo) for handler in self.inheritable_handlers],
self.parent_run_id,
)

10
langchain/callbacks/manager.py
View File

@ -1,7 +1,6 @@
from __future__ import annotations from __future__ import annotations
import asyncio import asyncio
import copy
import functools import functools
import os import os
import warnings import warnings
@ -700,8 +699,8 @@ def _configure(
if isinstance(inheritable_callbacks, list) or inheritable_callbacks is None: if isinstance(inheritable_callbacks, list) or inheritable_callbacks is None:
inheritable_callbacks_ = inheritable_callbacks or [] inheritable_callbacks_ = inheritable_callbacks or []
callback_manager = callback_manager_cls( callback_manager = callback_manager_cls(
handlers=inheritable_callbacks_, handlers=inheritable_callbacks_.copy(),
inheritable_handlers=inheritable_callbacks_, inheritable_handlers=inheritable_callbacks_.copy(),
) )
else: else:
callback_manager = callback_manager_cls( callback_manager = callback_manager_cls(
@ -709,14 +708,13 @@ def _configure(
inheritable_handlers=inheritable_callbacks.inheritable_handlers, inheritable_handlers=inheritable_callbacks.inheritable_handlers,
parent_run_id=inheritable_callbacks.parent_run_id, parent_run_id=inheritable_callbacks.parent_run_id,
) )
callback_manager = copy.deepcopy(callback_manager)
local_handlers_ = ( local_handlers_ = (
local_callbacks local_callbacks
if isinstance(local_callbacks, list) if isinstance(local_callbacks, list)
else (local_callbacks.handlers if local_callbacks else []) else (local_callbacks.handlers if local_callbacks else [])
) )
for handler in local_handlers_: for handler in local_handlers_:
callback_manager.add_handler(copy.deepcopy(handler), False) callback_manager.add_handler(handler, False)
tracer = tracing_callback_var.get() tracer = tracing_callback_var.get()
open_ai = openai_callback_var.get() open_ai = openai_callback_var.get()
@ -734,7 +732,6 @@ def _configure(
for handler in callback_manager.handlers for handler in callback_manager.handlers
): ):
callback_manager.add_handler(StdOutCallbackHandler(), False) callback_manager.add_handler(StdOutCallbackHandler(), False)
if tracing_enabled_ and not any( if tracing_enabled_ and not any(
isinstance(handler, LangChainTracer) isinstance(handler, LangChainTracer)
for handler in callback_manager.handlers for handler in callback_manager.handlers
@ -750,5 +747,4 @@ def _configure(
for handler in callback_manager.handlers for handler in callback_manager.handlers
): ):
callback_manager.add_handler(open_ai, True) callback_manager.add_handler(open_ai, True)
return callback_manager return callback_manager