bugfix: Fix issue with toolset df description updating

llmkernel/kernel.py

@@ -38,8 +38,8 @@ def set_context(self, context, context_info):
                 dataset_id = context_info["id"]
                 print(f"Processing dataset w/id {dataset_id}")
                 self.toolset.set_dataset(dataset_id)
+                self.toolset.kernel = self.shell
                 self.context = self.agent.add_context(self.toolset.context())
-                # self.shell.ex("""import pandas as pd; import numpy as np;""")
                 self.shell.ex("""import pandas as pd; import numpy as np; import scipy;""")
                 self.shell.push({
                     "df": self.toolset.df

toolsets/dataset_toolset.py

@@ -80,14 +80,27 @@ def dataset_info(self) -> str:
         Returns:
             str: a textual representation of the dataset
         """
-        # Need to actually track things. Maybe a good idea to split this in to finer tools so certain things can be queried?
+        # Update the local dataframe to match what's in the shell.
+        # This will be factored out when we switch around to allow using multiple runtimes.
+        if self.kernel:
+            try:
+                self.df = self.kernel.ev("df")
+            except:
+                pass
+
         output = f"""
 Dataframe head:
 {self.df.head(15)}
+
+
 Columns:
 {self.df.columns}
+
+
 dtypes:
 {self.df.dtypes}
+
+
 Statistics:
 {self.df.describe()}
 """