From 7c771c4070b9a1a7fcebcda3d05b201728cdc013 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Tue, 23 Jun 2026 00:22:38 +0200
Subject: [PATCH] feat(checker): add input function to preamble

---
 midas/checker/preamble.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/midas/checker/preamble.py b/midas/checker/preamble.py
index 96a4ef7..1dcd157 100644
--- a/midas/checker/preamble.py
+++ b/midas/checker/preamble.py
@@ -54,6 +54,11 @@ class Preamble(Environment):
             returns=self._list_of(map_out),  # TODO: replace with Iterable[U]
             type_vars=[map_in, map_out],
         )
+        self._def_function(
+            name="input",
+            pos=[Param("prompt", TopType(), required=False)],
+            returns=self._types.get_type("str"),
+        )
 
     def _list_of(self, item_type: Type) -> Type:
         return self._types.apply_generic(self._types.get_type("list"), [item_type])