feat: append to extend

HakaseAI · May 6, 2024 · 3aaa5d4 · 3aaa5d4
1 parent 9fbb6e4
commit 3aaa5d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HakaseCore/llm/llama3.py b/HakaseCore/llm/llama3.py
@@ -59,7 +59,7 @@ def load_prompt(self) -> list[dict[str, str]]:
 
     def generate_instruction(self, instruction: str) -> None:
         self.prompt = self.load_prompt()
-        self.prompt.append({"role": "user", "content": f"{instruction}"})
+        self.prompt.extend({"role": "user", "content": f"{instruction}"})
 
     def generate_text(self, instruction: str) -> str:
         self.generate_instruction(instruction=instruction)