Update prompt_compiler.py

prompt_compiler.py

@@ -8,10 +8,10 @@ class PromptCompiler:
         self.token_engine = TokenEngine()
         self.mode = mode
         self.system_prompt = system_prompt or prompt_templates.get(mode, "")
-    
+
     def compile(self, user_input: str, context: dict = None) -> str:
         """
-        Combines system prompt, user input, and context into a final prompt for the model.
+        Compile the final prompt from the system prompt, context, and user input.
         """
         context_block = self._format_context(context or {})
         final_prompt = f"{self.system_prompt}\n\n{context_block}\n\nUser: {user_input}\n\nLORIEN:"
@@ -23,11 +23,10 @@ class PromptCompiler:
         """
         if not context:
             return ""
-        formatted = "\n".join([f"[{k.upper()}]: {v}" for k, v in context.items()])
-        return f"Context:\n{formatted}"
+        return "\n".join([f"[{k.upper()}]: {v}" for k, v in context.items()])
 
     def estimate_tokens(self, prompt: str) -> int:
         """
-        Estimates token count using the TokenEngine.
+        Estimate token usage of the compiled prompt.
         """
         return self.token_engine.count_tokens(prompt)