From d42a4f7764a06c7f2b3bdf2d106b8d2cdf09dd90 Mon Sep 17 00:00:00 2001
From: stephenfreund <freund@cs.williams.edu>
Date: Mon, 11 Mar 2024 07:07:20 -0400
Subject: [PATCH] info supports vars/code/classes/packages now, and tries to
 clean up bad qualified names

---
 src/chatdbg/chatdbg_pdb.py | 47 ++++++++++++++++++++++++++------------
 1 file changed, 32 insertions(+), 15 deletions(-)

diff --git a/src/chatdbg/chatdbg_pdb.py b/src/chatdbg/chatdbg_pdb.py
index 5c1f395..54fa546 100644
--- a/src/chatdbg/chatdbg_pdb.py
+++ b/src/chatdbg/chatdbg_pdb.py
@@ -167,9 +167,11 @@ def _hide_lib_frames(self):
             self.curframe_locals = self.curframe.f_locals
 
         # Assume assertions are correct and the code leading to them is not!
-        current_line = linecache.getline(self.curframe.f_code.co_filename, self.lineno)
-        if current_line.strip().startswith('assert'):
-            self._error_specific_prompt += f"The code `{current_line.strip()}` is correct and MUST remain unchanged in your fix.\n"
+        if self.curframe.f_lineno != None:
+            current_line = linecache.getline(self.curframe.f_code.co_filename,
+                                             self.curframe.f_lineno)
+            if current_line.strip().startswith('assert'):
+                self._error_specific_prompt += f"The code `{current_line.strip()}` is correct and MUST remain unchanged in your fix.\n"
 
     def execRcLines(self):
 
@@ -286,23 +288,36 @@ def do_info(self, arg):
         Print the pydoc string (and source code, if available) for a name.
         """
         try:
-            obj = self._getval(arg)
+            # try both given and unqualified form incase LLM biffs
+            args_to_try = [ arg, arg.split('.')[-1] ]
+            obj = None
+            for x in args_to_try:
+                try:
+                    obj = eval(x, self.curframe.f_globals, self.curframe_locals)
+                    break  # found something so we're good
+                except:
+                    # fail silently, try the next name
+                    pass
+
+            # didn't find anything
+            if obj == None:
+                self.message(f"No name `{arg}` is visible in the current frame.")
+                return
+
             if self._is_user_file(inspect.getfile(obj)):
-                self.do_source(arg)
+                self.do_source(x)
             else:
-                self.do_pydoc(arg)
-                self.message(
-                    f"You MUST assume that `{arg}` is specified and implemented correctly."
-                )
-        except NameError:
-            # message already printed in _getval
-            pass
+                self.do_pydoc(x)
+                self.message(f"You MUST assume that `{x}` is specified and implemented correctly.")
         except OSError:
             raise
+        except NameError:
+            # alread handled
+            pass
         except Exception:
-            self.do_pydoc(arg)
+            self.do_pydoc(x)
             self.message(
-                f"You MUST assume that `{arg}` is specified and implemented correctly."
+                f"You MUST assume that `{x}` is specified and implemented correctly."
             )
 
     def do_slice(self, arg):
@@ -316,6 +331,7 @@ def do_slice(self, arg):
 
             index = self.curindex
             _x = None
+            cell = None
             while index > 0:
                 # print(index)
                 pos, _ = singletons.flow().get_position(self.stack[index][0])
@@ -338,7 +354,8 @@ def do_slice(self, arg):
                                                         blacken=True,
                                                         format_type=None)).rstrip()
             else:
-                result = f"*** No information avaiable for {arg}, only {cell.used_symbols}.  Run the command `p {arg}` to see its value."
+                used_symbols = set() if cell == None else cell.used_symbols
+                result = f"*** No information avaiable for {arg}, only {used_symbols}.  Run the command `p {arg}` to see its value."
         except OSError:
             raise
         except Exception as e: