Skip to content

Commit 7ee26c3

Browse files
Update llm_agent.py
1 parent 362cdd0 commit 7ee26c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llama_cpp_agent/llm_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,12 +429,12 @@ def get_response_role_and_completion(
429429
)
430430

431431
if system_prompt:
432-
if messages[0]["role"] != Roles.system:
432+
if messages[0]["role"] != Roles.system and (messages[0]["role"] != Roles.system.value):
433433
messages.insert(0, {"role": Roles.system, "content": system_prompt})
434434
else:
435435
messages[0]["content"] = system_prompt
436436
else:
437-
if messages[0]["role"] != Roles.system:
437+
if messages[0]["role"] != Roles.system and (messages[0]["role"] != Roles.system.value):
438438
messages.insert(0, {"role": Roles.system, "content": self.system_prompt})
439439
else:
440440
messages[0]["content"] = self.system_prompt

0 commit comments

Comments
 (0)