Skip to content

Commit ac471ce

Browse files
author
Smruthi Raj Mohan
committed
Adding support for system message
1 parent 748e9cd commit ac471ce

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

nemoguardrails/integrations/langchain/runnable_rails.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from typing import Any, List, Optional
1919

2020
from langchain_core.language_models import BaseLanguageModel
21-
from langchain_core.messages import AIMessage, HumanMessage
21+
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
2222
from langchain_core.prompt_values import ChatPromptValue, StringPromptValue
2323
from langchain_core.runnables import Runnable
2424
from langchain_core.runnables.config import RunnableConfig
@@ -139,6 +139,8 @@ def _transform_input_to_rails_format(self, _input):
139139
messages.append({"role": "assistant", "content": msg.content})
140140
elif isinstance(msg, HumanMessage):
141141
messages.append({"role": "user", "content": msg.content})
142+
elif isinstance(msg, SystemMessage):
143+
messages.append({"role": "system", "content": msg.content})
142144
elif isinstance(_input, StringPromptValue):
143145
messages.append({"role": "user", "content": _input.text})
144146
elif isinstance(_input, dict):

0 commit comments

Comments
 (0)