Merge pull request #432 from Aggregate-Intellect/event_logging

Change event logging mechanism

src/sherpa_ai/actions/base.py

@@ -7,6 +7,7 @@
 
 from sherpa_ai.actions.utils.refinement import BaseRefinement
 from sherpa_ai.actions.utils.reranking import BaseReranking
+from sherpa_ai.events import EventType
 
 if TYPE_CHECKING:
     from sherpa_ai.memory.belief import Belief
@@ -132,13 +133,24 @@ def __call__(self, **kwargs):
                     "are 'agent' and 'belief'"
                 )
 
+        # Log to the belief
+        if self.belief is not None:
+            self.belief.update_internal(
+                EventType.action, self.name, f"Action: {self.name} starts, Args: {filtered_kwargs}"
+            )
+
+
         # Execute the action
         result = self.execute(**filtered_kwargs)
 
         # Save the result to the belief
         self.belief: Belief = self.belief
         if self.belief:
             self.belief.set(self.output_key, result)
+            self.belief.update_internal(
+                EventType.action_output, self.name, f"Action: {self.name} finishes, Observation: {result}"
+            )
+
         return result
 
     def __str__(self):

src/sherpa_ai/agents/base.py

@@ -93,12 +93,6 @@ def run(self):
                 f"🤖{self.name} is executing```" "``` {result.action.name}...```"
             )
 
-            self.belief.update_internal(
-                EventType.action,
-                self.name,
-                "Action: " + result.action.name + str(result.args),
-            )
-
             try:
                 action_output = self.act(result.action, result.args)
             except Exception as e:
@@ -114,9 +108,6 @@ def run(self):
 
             self.verbose_logger.log(f"```Action output: {action_output}```")
             logger.debug(f"```Action output: {action_output}```")
-            self.belief.update_internal(
-                EventType.action_output, self.name, "Output: " + action_output
-            )
 
         result = (
             self.validate_output()

src/sherpa_ai/agents/qa_agent.py

@@ -87,6 +87,7 @@ def create_actions(self) -> List[BaseAction]:
                 task=self.belief.current_task.content,
                 llm=self.llm,
                 config=self.config,
+                belief=self.belief,
             ),
         ]