Improve search

core/clients/signal.py

@@ -588,7 +588,11 @@ class HandleMessage(Command):
                         None,
                     )
                     log.info("Running Signal mutate prompt")
-                    result = await ai.run_prompt(prompt, manip.ai)
+                    result = await ai.run_prompt(
+                        prompt,
+                        manip.ai,
+                        operation="signal_mutate",
+                    )
                     log.info(
                         f"Sending {len(xmpp_attachments)} attachments from Signal to XMPP."
                     )
@@ -733,7 +737,11 @@ class HandleMessage(Command):
                     )
 
                     log.info("Running context prompt")
-                    result = await ai.run_prompt(prompt, manip.ai)
+                    result = await ai.run_prompt(
+                        prompt,
+                        manip.ai,
+                        operation="signal_reply",
+                    )
                     if manip.mode == "active":
                         await history.store_own_message(
                             session=chat_session,

core/clients/xmpp.py

@@ -1313,7 +1313,11 @@ class XMPPComponent(ComponentXMPP):
                 chat_history,
             )
             self.log.debug("Running XMPP context prompt")
-            result = await ai.run_prompt(prompt, manip.ai)
+            result = await ai.run_prompt(
+                prompt,
+                manip.ai,
+                operation="xmpp_mutate",
+            )
             self.log.debug("Generated mutated response for XMPP message")
             await history.store_own_message(
                 session=session,