From 3aaa5d45159aa15af62ee7f0431cc3d5e38d42ee Mon Sep 17 00:00:00 2001 From: Euiseo Cha Date: Tue, 7 May 2024 01:28:08 +0900 Subject: [PATCH] feat: append to extend --- HakaseCore/llm/llama3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/HakaseCore/llm/llama3.py b/HakaseCore/llm/llama3.py index 25a9f95..46292f3 100644 --- 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)