updates

yeshaokai · yeshaokai · commit 21e5b7b17c04 · 2024-07-05T15:53:09.000+02:00
diff --git a/amadeusgpt/main.py b/amadeusgpt/main.py
@@ -85,6 +85,11 @@ def step(self, user_query):
         result = self.sandbox.llm_step(user_query)
         return result
 
+    def get_analysis(self):
+        sandbox = self.sandbox
+        analysis = sandbox.exec_namespace['behavior_analysis']
+        return analysis
+
 
 if __name__ == "__main__":
     from amadeusgpt.config import Config   
diff --git a/notebook/EPM_demo.ipynb b/notebook/EPM_demo.ipynb
diff --git a/setup.cfg b/setup.cfg
@@ -31,7 +31,8 @@ install_requires =
     moviepy # 0.02
     umap-learn 
     cachetools # 0.03
-    matplotlib<3.9 # for cmap to work 
+    matplotlib<3.9 # for cmap to work
+    ipympl
 
 [options.extras_require]
 dev =
