Store type origin info for traceback display

trans/transpiler/phases/typing/expr.py

@@ -42,6 +42,8 @@ class ScoperExprVisitor(ScoperVisitor):
             TB_SKIP = False
             raise NotImplementedError(f"`{ast.unparse(node)}` {type(node)}")
         res = res.resolve()
+        if True or not hasattr(res, "from_node"):
+            res.from_node = node
         node.type = res
         return res
 

trans/transpiler/phases/typing/types.py

@@ -174,6 +174,8 @@ class TypeOperator(BaseType, ABC):
 
     def unify_internal(self, other: BaseType):
         from transpiler.phases.typing.exceptions import TypeMismatchError, TypeMismatchKind
+        if from_node := next(filter(None, (getattr(x, "from_node", None) for x in (other, self))), None):
+            TB_NODE = from_node
         if not isinstance(other, TypeOperator):
             raise TypeMismatchError(self, other, TypeMismatchKind.DIFFERENT_TYPE)
         if other.is_protocol and not self.is_protocol: