refactor: take type hints from user in function args

1 files changed, 4 insertions, 1 deletions

diff --git a/lexer.py b/lexer.py
index 0a634c5..ea4fefb 100644
--- a/lexer.py
+++ b/lexer.py
@@ -9,9 +9,12 @@ class LexError(BaseException):
 types = {
     ":int": TokenType.INT_TYPE,
     ":float": TokenType.FLOAT_TYPE,
+    ":number": TokenType.NUMBER_TYPE,
     ":string": TokenType.STRING_TYPE,
     ":list": TokenType.LIST_TYPE,
-    ":any": TokenType.ANY_TYPE }
+    ":any": TokenType.ANY_TYPE,
+    ":literal": TokenType.LITERAL_TYPE,
+    ":bool": TokenType.BOOL_TYPE }
 
 keywords = {
     "print": TokenType.PRINT,