msg

model/custom_agent.py

@@ -22,7 +22,12 @@ Based on https://github.com/huggingface/transformers/blob/main/src/transformers/
 import time
 import requests
 from transformers import Agent
-from transformers.prompts import CHAT_MESSAGE_PROMPT
+#from transformers.tools.prompts import CHAT_MESSAGE_PROMPT
+
+CHAT_MESSAGE_PROMPT = """
+Human: <<task>>
+
+Assistant: """
 
 from utils.logger import log_response