Merge pull request #75 from plasma-umass/fix-ipython9-compat

Add IPython 9 compatibility

Author: emeryberger

.github/workflows/code-quality.yml

@@ -10,7 +10,7 @@ jobs:
   code-quality:
     strategy:
       matrix:
-        python: ["3.9","3.10", "3.11", "3.12", "3.13"]
+        python: ["3.11", "3.12", "3.13"]
 
     runs-on: ubuntu-latest
 

.github/workflows/sanity.yml

@@ -11,7 +11,7 @@ jobs:
     strategy:
       matrix:
         platform: ["ubuntu-latest", "macos-latest", "windows-latest"]
-        python: ["3.9","3.10", "3.11", "3.12", "3.13"]
+        python: ["3.11", "3.12", "3.13"]
 
     runs-on: ${{ matrix.platform }}
 

pyproject.toml

@@ -18,15 +18,16 @@ dependencies = [
   "ansicolors>=1.1.8",
   "traitlets>=5.14.1",
   "ipdb>=0.13.13",
-  "ipython==8.18.1",
+  "ipython>=9.0.0",
+  "pygments>=2.0.0",
   "litellm==1.55.9",
   "PyYAML>=6.0.1",
   "ipyflow>=0.0.130",
   "numpy>=1.26.3",
 ]
 description = "AI-assisted debugging. Uses AI to answer 'why'."
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.11"
 classifiers = [
   "Programming Language :: Python :: 3",
   "License :: OSI Approved :: Apache Software License",

src/chatdbg/chatdbg_pdb.py

@@ -12,6 +12,7 @@
 from pathlib import Path
 
 import IPython
+from pygments.token import Token
 
 import pdb
 
@@ -473,9 +474,7 @@ def _hidden_predicate(self, frame):
         return False
 
     def print_stack_trace(self, context=None, locals=None):
-        # override to print the skips into stdout instead of stderr...
-        Colors = self.color_scheme_table.active_colors
-        ColorsNormal = Colors.Normal
+        # override to print the skips into stdout instead of stderr and to add locals
         if context is None:
             context = self.context
         try:
@@ -497,19 +496,24 @@ def print_stack_trace(self, context=None, locals=None):
                     skipped += 1
                     continue
                 if skipped:
-                    print(
-                        f"{Colors.excName}    [... skipping {skipped} hidden frame(s)]{ColorsNormal}\n",
-                        file=self.stdout,
+                    msg = self.theme.format(
+                        [
+                            (
+                                Token.ExcName,
+                                f"    [... skipping {skipped} hidden frame(s)]",
+                            )
+                        ]
                     )
+                    print(f"{msg}\n", file=self.stdout)
                     skipped = 0
-                self.print_stack_entry(frame_lineno, context=context)
+                self.print_stack_entry(frame_lineno)
                 if locals:
                     print_locals(self.stdout, frame_lineno[0])
             if skipped:
-                print(
-                    f"{Colors.excName}    [... skipping {skipped} hidden frame(s)]{ColorsNormal}\n",
-                    file=self.stdout,
+                msg = self.theme.format(
+                    [(Token.ExcName, f"    [... skipping {skipped} hidden frame(s)]")]
                 )
+                print(f"{msg}\n", file=self.stdout)
         except KeyboardInterrupt:
             pass