feat: update featurs

Author: gusye1234

drive_flow/__init__.py

@@ -1,7 +1,7 @@
 from .core import EventEngineCls
-from .types import EventInput
+from .types import EventInput, ReturnBehavior
 
-__version__ = "0.0.1.alpha3"
+__version__ = "0.0.1.alpha4"
 __author__ = "Jianbai Ye"
 __url__ = "https://github.com/memodb-io/drive-flow"
 

drive_flow/types.py

@@ -16,6 +16,7 @@ class ReturnBehavior(Enum):
     DISPATCH = "dispatch"
     GOTO = "goto"
     ABORT = "abort"
+    INPUT = "input"
 
 
 class TaskStatus(Enum):
@@ -44,6 +45,12 @@ class EventGroupInput:
 class EventInput(EventGroupInput):
     task_id: str = field(default_factory=generate_uuid)
 
+    @classmethod
+    def from_input(cls: "EventInput", input_data: dict[str, Any]) -> "EventInput":
+        return cls(
+            group_name="user_input", results=input_data, behavior=ReturnBehavior.INPUT
+        )
+
 
 @dataclass
 class _SpecialEventReturn:

examples/6_llm_agent_ReAct.py

@@ -1 +1,45 @@
-# TODO
+import asyncio
+from openai import AsyncOpenAI
+from drive_flow import default_drive, EventInput, ReturnBehavior
+from drive_flow.dynamic import goto_events, abort_this
+
+openai_client = AsyncOpenAI()
+
+
+@default_drive.make_event
+async def plan(event: EventInput, global_ctx):
+    print("Planning")
+    if event.behavior == ReturnBehavior.INPUT:
+        query = event.results["query"]
+        global_ctx["user_query"] = query
+
+
+@default_drive.listen_group([plan])
+async def action(event: EventInput, global_ctx):
+    print("Executing")
+
+
+@default_drive.listen_group([action])
+async def observate(event: EventInput, global_ctx):
+    print("Observing")
+
+
+# make it a loop
+plan = default_drive.listen_group([observate])(plan)
+
+if __name__ == "__main__":
+    question = "What is the answer to the ultimate question of life, the universe, and everything?"
+    storage_results = {}
+    print(observate.debug_string())
+    # asyncio.run(
+    #     default_drive.invoke_event(
+    #         plan,
+    #         event_input=EventInput.from_input({"query": question}),
+    #         global_ctx=storage_results,
+    #     )
+    # )
+
+    # if "answer" not in storage_results:
+    #     print(f"Failed to get answer {question}")
+    #     exit(1)
+    # print(storage_results)

readme.md

@@ -62,11 +62,15 @@ async def hello(event: EventInput, global_ctx):
 async def world(event: EventInput, global_ctx):
     print("world")
 
+# display the dependencies of 'world' event
+print(world.debug_string()) 
 asyncio.run(default_drive.invoke_event(hello))
 ```
 
 In this example, The return of `hello` event will trigger `world` event.
 
+### Break-down
+
 To make an event function, there are few elements:
 
 * Input Signature: must be `(event: EventInput, global_ctx)`. `EventInput` is the returns of the listening groups. `global_ctx` is set by you when invoking events, it can be anything and default to `None`.
@@ -116,9 +120,13 @@ async def adding(event: EventInput, global_ctx):
 results = asyncio.run(default_drive.invoke_event(start))
 assert results[adding.id] == 3
 ```
+
+`adding` will be triggered at first time as long as `hello` and `world` are done.
 </details>
 
-`drive_flow` suppports different behaviors for multi-event triggering:
+#### Re-trigger the event
+
+`drive_flow` suppports different behaviors for multi-event retriggering:
 
 - `all`: retrigger this event only when all the listening events are updated.
 - `any`: retrigger this event as long as one of the listening events is updated.
@@ -150,7 +158,6 @@ async def world(event: EventInput, global_ctx):
     await asyncio.sleep(0.2)
     print(datetime.now(), "world done")
 
-
 asyncio.run(default_drive.invoke_event(start))
 ```
 

tests/test_types.py

@@ -1,4 +1,11 @@
-from drive_flow.types import BaseEvent, EventGroup
+from drive_flow.types import BaseEvent, EventGroup, EventInput, ReturnBehavior
+
+
+def test_user_input():
+    fake_input = {"query": "Hello World"}
+    a = EventInput.from_input(fake_input)
+    assert a.results == fake_input
+    assert a.behavior == ReturnBehavior.INPUT
 
 
 def test_node_hash():