andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-04-30 14:52:36 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: make pickle context manager thread-safe

Browse Source
This commit is contained in:
Greyson LaLonde
2026-03-08 14:51:43 -04:00
parent 7d9faa7cbf
commit 84deb5cc62
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lib/crewai/src/crewai/types/callable.py
View File

@@ -18,13 +18,16 @@ from __future__ import annotations
import base64
from collections.abc import Callable
from contextvars import ContextVar
from contextvars import ContextVar, Token
from typing import Annotated, Any
from pydantic import BeforeValidator, PlainSerializer, WithJsonSchema
_ALLOW_PICKLE: ContextVar[bool] = ContextVar("_ALLOW_PICKLE", default=False)
_ALLOW_PICKLE_TOKEN: ContextVar[Token[bool] | None] = ContextVar(
"_ALLOW_PICKLE_TOKEN", default=None
)
def _import_cloudpickle() -> Any:
@@ -48,10 +51,12 @@ class _AllowPickleDeserialization:
"""
def __enter__(self) -> None:
self._token = _ALLOW_PICKLE.set(True)
_ALLOW_PICKLE_TOKEN.set(_ALLOW_PICKLE.set(True))
def __exit__(self, *_: object) -> None:
_ALLOW_PICKLE.reset(self._token)
token = _ALLOW_PICKLE_TOKEN.get()
if token is not None:
_ALLOW_PICKLE.reset(token)
allow_pickle_deserialization = _AllowPickleDeserialization()