fix: broken relative imports for output types generation

danielgtaylor · Jan 31, 2024 · 2dd20e3 · 2dd20e3
1 parent 5666393
commit 2dd20e3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/betterproto/plugin/models.py b/src/betterproto/plugin/models.py
@@ -725,6 +725,9 @@ def __post_init__(self) -> None:
         self.parent.methods.append(self)
 
         # Check for imports
+        if "Optional" in self.py_input_message_type:
+            self.output_file.typing_imports.add("Optional")
+
         if "Optional" in self.py_output_message_type:
             self.output_file.typing_imports.add("Optional")