fix system_prompt param that llm not support from issue 725

This commit is contained in:
better629 2024-01-10 11:26:23 +08:00
parent 16e3f94b44
commit 17479a2360
2 changed files with 6 additions and 2 deletions

View file

@ -43,7 +43,9 @@ class BaseLLM(ABC):
if system_msgs:
message = self._system_msgs(system_msgs)
else:
message = [self._default_system_msg()] if self.use_system_prompt else []
message = [self._default_system_msg()]
if not self.use_system_prompt:
message = []
if format_msgs:
message.extend(format_msgs)
message.append(self._user_msg(msg))

View file

@ -41,7 +41,9 @@ class MockLLM(OpenAILLM):
if system_msgs:
message = self._system_msgs(system_msgs)
else:
message = [self._default_system_msg()] if self.use_system_prompt else []
message = [self._default_system_msg()]
if not self.use_system_prompt:
message = []
if format_msgs:
message.extend(format_msgs)
message.append(self._user_msg(msg))