Merge pull request #13 from pfnet-research/mitmul/fix-rephraze-quiz

Fix rephrase_quiz
pfnet-research · Dec 9, 2024 · 381cd60 · 381cd60
2 parents cea9531 + 708c90b
commit 381cd60
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/rephrase_quiz/main.py b/examples/rephrase_quiz/main.py
@@ -1,7 +1,7 @@
 import os
 
-from langchain_core.messages import HumanMessage, SystemMessage
-from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.messages import SystemMessage
+from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
 from langchain_core.tools import tool
 from langchain_openai import ChatOpenAI
 
@@ -67,7 +67,7 @@ def wrong_answer(feedback: str) -> str:
             "---\n"
             "問題文: 台風１０号が近づいている。\n",
         ),
-        HumanMessage("{input}"),
+        HumanMessagePromptTemplate.from_template("{input}"),
     ]
 )