fixes ChatSnowflakeCortex parsing issues because of special characters in message_json (#337)

This PR fixes parsing issues of special characters in message_json when
using ChatSnowflakeCortex.

Fixes: langchain-ai/langchain#30115

---------

Co-authored-by: Marcus Rehm <[email protected]>
Co-authored-by: graphite-app[bot] <96075541+graphite-app[bot]@users.noreply.github.com>
Co-authored-by: Mason Daugherty <[email protected]>

Author: 4 people

libs/community/langchain_community/chat_models/snowflake.py

@@ -277,8 +277,8 @@ def _generate(
         sql_stmt = f"""
             select snowflake.cortex.{self.cortex_function}(
                 '{self.model}',
-                parse_json('{message_json}'),
-                parse_json('{options_json}')
+                parse_json($${message_json}$$),
+                parse_json($${options_json}$$)
             ) as llm_response;
         """
 
@@ -364,8 +364,8 @@ def _stream(
         sql_stmt = f"""
             select snowflake.cortex.{self.cortex_function}(
                 '{self.model}',
-                parse_json('{message_json}'),
-                parse_json('{options_json}')
+                parse_json($${message_json}$$),
+                parse_json($${options_json}$$)
             ) as llm_stream_response;
         """
 

libs/community/tests/integration_tests/chat_models/test_snowflake.py

@@ -57,3 +57,18 @@ def test_chat_snowflake_cortex_generate(chat: ChatSnowflakeCortex) -> None:
             assert isinstance(generation, ChatGeneration)
             assert isinstance(generation.text, str)
             assert generation.text == generation.message.content
+
+
+def test_chat_snowflake_cortex_message_with_special_characters(
+    chat: ChatSnowflakeCortex,
+) -> None:
+    """Test ChatSnowflakeCortex with messages containing special characters.
+
+    Args:
+        chat: The ChatSnowflakeCortex instance to test with.
+    """
+    system_message = SystemMessage(content="You are to chat with the user.")
+    human_message = HumanMessage(content="Can you give me the weather in Tokyo?\n\n")
+    response = chat([system_message, human_message])
+    assert isinstance(response, BaseMessage)
+    assert isinstance(response.content, str)