Slight bug fix

AutoGPTFlow.py

@@ -95,7 +95,7 @@ class AutoGPTFlow(ControllerExecutorFlow):
             keys_to_select = ["human_feedback"],
         )
             
-        self.next_flow_to_call = {
+        self.next_state = {
             None: "MemoryRead",
             "MemoryRead": "Controller",
             "Controller": "Executor",
@@ -111,6 +111,8 @@ class AutoGPTFlow(ControllerExecutorFlow):
         
     def prepare_memory_read_output(self,data_dict: Dict[str, Any],**kwargs):
         """ This method prepares the output of the Memory Flow to be used by the Controller Flow."""
+        if len(data_dict["retrieved"]) == 0:
+            return {"memory": ""}
         retrieved_memories = data_dict["retrieved"][0][1:]
         return {"memory": "\n".join(retrieved_memories)}
     
@@ -228,16 +230,16 @@ class AutoGPTFlow(ControllerExecutorFlow):
         # either programmatically or using the _state_update_dict and 
         # input and ouput interface methods
         
-        last_called = self.flow_state["last_called"]
+        last_state = self.flow_state["last_state"]
         
-        if last_called is None:            
+        if last_state is None:            
             self.flow_state["input_message"] = input_message
             self.flow_state["goal"] = input_message.data["goal"]
         
-        elif last_called == "Executor":
+        elif last_state == "Executor":
             self.flow_state["observation"] = input_message.data
         
-        elif last_called == "Controller":
+        elif last_state == "Controller":
             self._state_update_dict(
                 {
                     "command": input_message.data["command"],
@@ -258,12 +260,12 @@ class AutoGPTFlow(ControllerExecutorFlow):
                 self.flow_state["early_exit_flag"] = True
                 
                 
-        elif last_called == "MemoryRead":
+        elif last_state == "MemoryRead":
             self._state_update_dict(
                 self.memory_read_ouput_interface(input_message).data
             )
             
-        elif last_called == "HumanFeedback":
+        elif last_state == "HumanFeedback":
             self._state_update_dict(
                 self.human_feedback_ouput_interface(input_message).data
             )
@@ -289,24 +291,24 @@ class AutoGPTFlow(ControllerExecutorFlow):
         """
         self.register_data_to_state(input_message)
         
-        flow_to_call = self.get_next_flow_to_call()
+        current_state = self.get_next_state()
 
         if self.flow_state.get("early_exit_flag",False):
             self.generate_reply()
             
-        elif flow_to_call == "MemoryRead":
+        elif current_state == "MemoryRead":
             self.call_memory_read()
         
-        elif flow_to_call == "Controller":
+        elif current_state == "Controller":
             self.call_controller()
         
-        elif flow_to_call == "Executor":
+        elif current_state == "Executor":
             self.call_executor()
         
-        elif flow_to_call == "HumanFeedback":
+        elif current_state == "HumanFeedback":
             self.call_human_feedback()
         
-        elif flow_to_call == "MemoryWrite":
+        elif current_state == "MemoryWrite":
             self.call_memory_write()
             self.flow_state["current_round"] += 1
             
@@ -314,6 +316,9 @@ class AutoGPTFlow(ControllerExecutorFlow):
             self._on_reach_max_round()
             self.generate_reply()
             
-        self.flow_state["last_called"] = self.get_next_flow_to_call()
+        if self.flow_state.get("early_exit_flag",False) or current_state is None:
+            self.flow_state["last_state"] = None
+        else:
+            self.flow_state["last_state"] = current_state
         
     

demo_db_dir/chroma-collections.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dad8d76313bc0a34d32374510832b6b28abe6539ce1b154732e09fa46614a6ee
+size 559

demo_db_dir/chroma-embeddings.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16e451ee9045b5e317ebfa8856a8478e1b33a9e04caf34f200efaf137b2f5ce1
+size 174

demo_db_dir/index/id_to_uuid_42e9cea0-eda4-4767-bb24-e634530872e5.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fd682353cb15f94132bc161e4ef161067cf4e618717ec84bc93806116b2cdb4
+size 64

demo_db_dir/index/index_42e9cea0-eda4-4767-bb24-e634530872e5.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3408b2e390cf695ccb65ae29a20526deb32ebb5872d450895ee7b163c50a0936
+size 6384

demo_db_dir/index/index_metadata_42e9cea0-eda4-4767-bb24-e634530872e5.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88eefc0a355ab39a170b7a60b8143a504f312a038e6db539cdffdc80960f2c75
+size 103

demo_db_dir/index/uuid_to_id_42e9cea0-eda4-4767-bb24-e634530872e5.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d76ef92df773960d674a3aa3e0765d99cfee2751597a6227a59ca4e24f9bb973
+size 52

run.py

@@ -83,7 +83,7 @@ if __name__ == "__main__":
         "goal": "Answer the following question: What is the profession and date of birth of Michael Jordan?",
     }
    
-    
+
     #option1: use the FlowMessage class
     input_message = FlowMessage(
         data=data,