add get_typed to interp.Frame (#156)

allow better error message when implementing the method table of an
interpreter.

changing the error from `KeyError` to `InterpreterError` so you can get
a stacktrace at where the statement produces the undefined error - I
think it's possible due to users' code actually having this error so it
doesn't makes sense to terminate the interpreter program right away but
instead should return the result as error.

cc: @weinbe58 @kaihsin

Author: Roger-luo

src/kirin/interp/frame.py

@@ -5,6 +5,7 @@
 from typing_extensions import Self
 
 from kirin.ir import SSAValue, Statement
+from kirin.exceptions import InterpreterError
 
 ValueType = TypeVar("ValueType")
 
@@ -66,7 +67,47 @@ def from_func_like(cls, code: Statement) -> Self:
         return cls(code=code)
 
     def get(self, key: SSAValue) -> ValueType:
-        return self.entries[key]
+        """Get the value for the given [`SSAValue`][kirin.ir.SSAValue].
+
+        Args:
+            key(SSAValue): The key to get the value for.
+
+        Returns:
+            ValueType: The value.
+
+        Raises:
+            InterpreterError: If the value is not found. This will be catched by the interpreter
+                and will be converted to an [`interp.Err`][kirin.interp.Err] in the interpretation
+                results.
+        """
+        err = InterpreterError(f"SSAValue {key} not found")
+        value = self.entries.get(key, err)
+        if isinstance(value, InterpreterError):
+            raise err
+        else:
+            return value
+
+    ExpectedType = TypeVar("ExpectedType")
+
+    def get_typed(self, key: SSAValue, type_: type[ExpectedType]) -> ExpectedType:
+        """Similar to [`get`][kirin.interp.frame.Frame.get] but also checks the type.
+
+        Args:
+            key(SSAValue): The key to get the value for.
+            type_(type): The expected type.
+
+        Returns:
+            ExpectedType: The value.
+
+        Raises:
+            InterpreterError: If the value is not of the expected type. This will be catched
+                by the interpreter and will be converted to an [`interp.Err`][kirin.interp.Err]
+                in the interpretation results.
+        """
+        value = self.get(key)
+        if not isinstance(value, type_):
+            raise InterpreterError(f"expected {type_}, got {type(value)}")
+        return value
 
     def set(self, key: SSAValue, value: ValueType) -> None:
         self.entries[key] = value