Merge pull request #1016 from PrefectHQ/generate-n

marvin.generate: Enforce n in typing
PrefectHQ · Jan 17, 2025 · 2616b9c · 2616b9c
2 parents 5e04439 + 44b56a9
commit 2616b9c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/marvin/fns/generate.py b/src/marvin/fns/generate.py
@@ -1,5 +1,7 @@
 from typing import TypeVar, cast
 
+from pydantic import conlist
+
 import marvin
 from marvin.agents.agent import Agent
 from marvin.thread import Thread
@@ -67,7 +69,7 @@ async def generate_async(
         name="Generation Task",
         instructions=PROMPT,
         context=task_context,
-        result_type=list[target],
+        result_type=conlist(target, min_length=n, max_length=n),
         agents=[agent] if agent else None,
     )