fixed type errors

Author: kiranandcode

effectful/handlers/llm/evaluation.py

@@ -1,15 +1,14 @@
 import ast
 import builtins
 import linecache
-from collections.abc import MutableMapping
 from types import CodeType
 from typing import Any
 
 from effectful.ops.syntax import ObjectInterpretation, defop, implements
 
 
 @defop
-def parse(source: str, filename: str) -> ast.AST:
+def parse(source: str, filename: str) -> ast.Module:
     """
     Parse source text into an AST.
 
@@ -22,7 +21,7 @@ def parse(source: str, filename: str) -> ast.AST:
 
 
 @defop
-def compile(module: ast.AST, filename: str) -> CodeType:
+def compile(module: ast.Module, filename: str) -> CodeType:
     """
     Compile an AST into a Python code object.
 
@@ -37,7 +36,7 @@ def compile(module: ast.AST, filename: str) -> CodeType:
 @defop
 def exec(
     bytecode: CodeType,
-    env: MutableMapping[str, Any],
+    env: dict[str, Any],
 ) -> None:
     """
     Execute a compiled code object.
@@ -53,7 +52,7 @@ class UnsafeEvalProvider(ObjectInterpretation):
     by shelling out to python *without* any further checks. Only use for testing."""
 
     @implements(parse)
-    def parse(self, source: str, filename: str) -> ast.AST:
+    def parse(self, source: str, filename: str) -> ast.Module:
         # Cache source under `filename` so inspect.getsource() can retrieve it later.
         # inspect uses f.__code__.co_filename -> linecache.getlines(filename)
         linecache.cache[filename] = (
@@ -66,14 +65,14 @@ def parse(self, source: str, filename: str) -> ast.AST:
         return ast.parse(source, filename=filename, mode="exec")
 
     @implements(compile)
-    def compile(self, module: ast.AST, filename: str, *rags) -> CodeType:
+    def compile(self, module: ast.Module, filename: str, *rags) -> CodeType:
         return builtins.compile(module, filename, "exec")
 
     @implements(exec)
     def exec(
         self,
         bytecode: CodeType,
-        env: MutableMapping[str, Any],
+        env: dict[str, Any],
     ) -> None:
         # Ensure builtins exist in the execution environment.
         env.setdefault("__builtins__", __builtins__)