add restaurant example

examples/restaurant_recs.py

@@ -0,0 +1,33 @@
+from controlflow import Task, flow
+from pydantic import BaseModel
+
+
+class Restaurant(BaseModel):
+    name: str
+    description: str
+
+
+@flow
+def restaurant_recs(n: int) -> list[Restaurant]:
+    """
+    An agentic workflow that asks the user for their location and
+    cuisine preference, then recommends n restaurants based on their input.
+    """
+
+    # get the user's location
+    location = Task("Get a location", user_access=True)
+
+    # get the user's preferred cuisine
+    cuisine = Task("Get a preferred cuisine", user_access=True)
+
+    # generate the recommendations from the user's input
+    recs = Task(
+        f"Recommend {n} restaurants to the user",
+        context=dict(location=location, cuisine=cuisine),
+        result_type=list[Restaurant],
+    )
+    return recs
+
+
+if __name__ == "__main__":
+    restaurant_recs(5)

src/controlflow/llm/completions.py

@@ -161,7 +161,7 @@ def _completion_generator(
                 invalid_tool_message = handle_invalid_tool_call(tool_call)
                 response_messages.append(invalid_tool_message)
                 yield CompletionEvent(
-                    type="tool_result_done", payload=dict(message=tool_result_message)
+                    type="tool_result_done", payload=dict(message=invalid_tool_message)
                 )
 
             counter += 1
@@ -272,7 +272,7 @@ async def _completion_async_generator(
                 invalid_tool_message = handle_invalid_tool_call(tool_call)
                 response_messages.append(invalid_tool_message)
                 yield CompletionEvent(
-                    type="tool_result_done", payload=dict(message=tool_result_message)
+                    type="tool_result_done", payload=dict(message=invalid_tool_message)
                 )
 
             counter += 1