format: Update formatting and test

aws-samples · Oct 31, 2024 · 562e1e5 · 562e1e5
1 parent 3572f1e
commit 562e1e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...-interfaces/langchain/functions/request-handler/adapters/shared/prompts/system_prompts.py b/...-interfaces/langchain/functions/request-handler/adapters/shared/prompts/system_prompts.py
@@ -75,7 +75,8 @@ class Language(Enum):
         # standalone question (French-Canadian)
         "condense_question_prompt": (
             "Avec la conversation ci-dessous et la question de suivi, "
-            "reformulez la question de suivi de manière à ce qu'elle soit une question autonome."
+            "reformulez la question de suivi de manière à ce qu'elle soit "
+            "une question autonome."
         ),
         "current_conversation_word": "Conversation en cours",
         "question_word": "Question",

diff --git a/tests/model-interfaces/langchain/functions/request-handler/adapters/bedrock/base_test.py b/tests/model-interfaces/langchain/functions/request-handler/adapters/bedrock/base_test.py
@@ -53,7 +53,7 @@ def test_chat_adapter(mocker):
         input="input", chat_history=[HumanMessage(content="history")]
     )
 
-    assert "Given the conversation inside the tags" in result
+    assert "Given the following conversation and" in result
     assert "Human: history" in result
     assert "Human: input" in result
 
@@ -104,7 +104,7 @@ def test_chat_without_system_adapter(mocker):
     result = model.get_condense_question_prompt().format(
         input="input", chat_history=[HumanMessage(content="history")]
     )
-    assert "Given the conversation inside the tags" in result
+    assert "Given the following conversation and" in result
     assert "Chat History" in result
     assert "Human: history" in result
     assert "Follow Up Input: input" in result