From a89b49e3c34075a7eb522b69f6ad10101dbf544c Mon Sep 17 00:00:00 2001
From: NeonBohdan <bohdan@neon.ai>
Date: Mon, 4 Dec 2023 15:15:41 +0200
Subject: [PATCH] Fix prompt

---
 neon_llm_claude/claude.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/neon_llm_claude/claude.py b/neon_llm_claude/claude.py
index cb5f031..54d04bd 100644
--- 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]: