Fix object streaming example in README (#10)

jackmpcollins · Sep 10, 2023 · 81fadb2 · 81fadb2
1 parent c056d17
commit 81fadb2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -187,6 +187,16 @@ Structured outputs can also be streamed from the LLM by using the return type an
 from collections.abc import Iterable
 from time import time
 
+from magentic import prompt
+from pydantic import BaseModel
+
+
+class Superhero(BaseModel):
+    name: str
+    age: int
+    power: str
+    enemies: list[str]
+
 
 @prompt("Create a Superhero team named {name}.")
 def create_superhero_team(name: str) -> Iterable[Superhero]: