From 5e88a61456ef4b6fa19706289839c69495d0eb16 Mon Sep 17 00:00:00 2001 From: zzstoatzz Date: Sun, 4 Aug 2024 14:52:49 -0500 Subject: [PATCH] fix flow thread kwarg --- examples/memory_between_flows.py | 20 ++++++++++++++++++++ src/controlflow/decorators.py | 2 +- 2 files changed, 21 insertions(+), 1 deletion(-) create mode 100644 examples/memory_between_flows.py diff --git a/examples/memory_between_flows.py b/examples/memory_between_flows.py new file mode 100644 index 00000000..4ec6ec90 --- /dev/null +++ b/examples/memory_between_flows.py @@ -0,0 +1,20 @@ +import controlflow as cf + +thread_id = "test-thread" + + +@cf.flow(thread=thread_id) +def flow_1(): + task = cf.Task("get the user's name", result_type=str, user_access=True) + return task + + +@cf.flow(thread=thread_id) +def flow_2(): + task = cf.Task("write the user's name backwards, if you don't know it, say so") + return task + + +if __name__ == "__main__": + flow_1() + flow_2() diff --git a/src/controlflow/decorators.py b/src/controlflow/decorators.py index 005fa419..cbac12b5 100644 --- a/src/controlflow/decorators.py +++ b/src/controlflow/decorators.py @@ -92,7 +92,7 @@ def wrapper( flow_kwargs = kwargs | (flow_kwargs or {}) if thread is not None: - flow_kwargs.setdefault("thread", thread) + flow_kwargs.setdefault("thread_id", thread) if tools is not None: flow_kwargs.setdefault("tools", tools) if agents is not None: