Add /instruct command to apply instruction to next response.

src/simulacrum.py

@@ -1,4 +1,5 @@
 import re
+import textwrap
 
 from .context import Context
 from .lm_executors import ChatExecutor
@@ -9,11 +10,12 @@ def __init__(self, context_file):
         self.context = Context(context_file)
         self.last_completion = None
         self.cost_warning_sent = False
+        self.instruction_text = None
 
     async def chat(self, user_input, user_name, image_url):
         self.context.load()
         if user_input:
-            user_input = self._apply_attribution(user_input, user_name)
+            user_input = self._inject_instruction(user_input)
             self.context.add_message("user", user_input, image_url)
         self.context.save()
         completion = await ChatExecutor(self.context).execute()
@@ -37,11 +39,14 @@ def reset_conversation(self):
         self.context.save()
         self.cost_warning_sent = False
 
-    def add_conversation_fact(self, fact):
+    def add_conversation_fact(self, fact_text):
         self.context.load()
-        self.context.add_conversation_fact(fact)
+        self.context.add_conversation_fact(fact_text)
         self.context.save()
 
+    def apply_instruction(self, instruction_text):
+        self.instruction_text = instruction_text
+
     def undo_last_messages_by_role(self, role):
         self.context.load()
         num_messages = len(self.context.conversation_messages)
@@ -69,10 +74,16 @@ def _strip_tag(self, content, tag):
         content = re.sub(r"\n{3,}", "\n\n", content)
         return content.strip()
 
-    def _apply_attribution(self, input, name):
-        attribute_messages = self.context.vars.get("attribute_messages")
-        if attribute_messages and name:
-            if isinstance(attribute_messages, dict) and name in attribute_messages:
-                name = attribute_messages[name]
-            input = f"{name}: {input}"
-        return input
+    def _inject_instruction(self, text):
+        if self.instruction_text:
+            text = textwrap.dedent(
+                f"""
+                {text}
+
+                <instruct>
+                {self.instruction_text}
+                </instruct>
+                """
+            )
+            self.instruction_text = None
+        return text

src/telegram/telegram_bot.py

@@ -37,6 +37,7 @@ def __init__(self, context_filepath, telegram_token, authorized_users):
             (["continue", "co"], self.continue_command_handler),
             (["undo", "u"], self.undo_command_handler),
             (["fact", "f"], self.add_fact_command_handler),
+            (["instruct", "i"], self.apply_instruction_command_handler),
             (["stats", "s"], self.stats_command_handler),
             (["clear", "cl"], self.clear_command_handler),
             (["help", "h"], self.help_command_handler),
@@ -119,12 +120,15 @@ async def clear_command_handler(self, ctx):
 
     @message_handler
     async def add_fact_command_handler(self, ctx):
-        fact_text = re.search(r"/fact (.*)", ctx.message.text)
-        if fact_text:
-            self.sim.add_conversation_fact(fact_text.group(1))
-            await ctx.send_message("`✅ Fact added to conversation`")
-        else:
-            await ctx.send_message("`❌ No text provided`")
+        await self._process_text_after_command(
+            ctx, self.sim.add_conversation_fact, "`✅ Fact added to conversation`"
+        )
+
+    @message_handler
+    async def apply_instruction_command_handler(self, ctx):
+        await self._process_text_after_command(
+            ctx, self.sim.apply_instruction, "`✅ Instruction applied to next response`"
+        )
 
     @message_handler
     async def help_command_handler(self, ctx):
@@ -181,3 +185,11 @@ async def _warn_cost(self, ctx, threshold_high=0.15, threshold_elevated=0.10):
             await ctx.send_message(
                 "🟡 Cost is elevated. Start a new conversation when ready."
             )
+
+    async def _process_text_after_command(self, ctx, action_method, success_message):
+        command_text = re.search(r"/\w+\s+(.*)", ctx.message.text)
+        if command_text:
+            action_method(command_text.group(1))
+            await ctx.send_message(success_message)
+        else:
+            await ctx.send_message("`❌ No text provided`")