🔧 Improved error handling when type is missing

shroominic · Oct 8, 2023 · 3b439da · 3b439da
1 parent e1b6e50
commit 3b439da
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/funcchain/utils/function_frame.py b/funcchain/utils/function_frame.py
@@ -23,7 +23,10 @@ def get_output_type() -> type:
     """
     Get the output type annotation of the parent caller function.
     """
-    return (caller_frame := get_parent_frame()).frame.f_globals[caller_frame.function].__annotations__["return"]
+    try:
+        return (caller_frame := get_parent_frame()).frame.f_globals[caller_frame.function].__annotations__["return"]
+    except KeyError:
+        raise ValueError("The funcchain must have a return type annotation")
 
 
 def parser_for(output_type: type) -> BaseOutputParser: