Fix prompt

NeonGeckoCom · Dec 4, 2023 · a89b49e · a89b49e
1 parent 5e06320
commit a89b49e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neon_llm_claude/claude.py b/neon_llm_claude/claude.py
@@ -134,7 +134,7 @@ def _assemble_prompt(self, message: str, chat_history: List[List[str]], persona:
  for role, content in chat_history[-self.context_depth:]:
  role_claude = self.convert_role(role)
  prompt += f"{role_claude} {content}"
- prompt += f"{self.convert_role('user')} {message}"
+ prompt += f"{self.convert_role('user')} {message}{self.convert_role('llm')}"
  return prompt
 
  def _score(self, prompt: str, targets: List[str], persona: dict) -> List[float]: