github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-29 11:09:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow patching run_name and max_concurrency

Browse Source
This commit is contained in:
Nuno Campos 2023-08-24 17:20:49 +02:00
parent 9a07032055
commit 738d93215d
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/schema/runnable/config.py
View File

@ -102,6 +102,8 @@ def patch_config(
deep_copy_locals: bool = False, deep_copy_locals: bool = False,
callbacks: Optional[BaseCallbackManager] = None, callbacks: Optional[BaseCallbackManager] = None,
recursion_limit: Optional[int] = None, recursion_limit: Optional[int] = None,
max_concurrency: Optional[int] = None,
run_name: Optional[str] = None,
) -> RunnableConfig: ) -> RunnableConfig:
config = ensure_config(config) config = ensure_config(config)
if deep_copy_locals: if deep_copy_locals:
@ -114,6 +116,10 @@ def patch_config(
config["run_id"] = None config["run_id"] = None
if recursion_limit is not None: if recursion_limit is not None:
config["recursion_limit"] = recursion_limit config["recursion_limit"] = recursion_limit
if max_concurrency is not None:
config["max_concurrency"] = max_concurrency
if run_name is not None:
config["run_name"] = run_name
return config return config