Update app.py

app.py

@@ -32,7 +32,7 @@ def format_chatml_prompt(state):
     for message in state.value["history"]:
         chatml_prompt += f"<|im_start|> "+message['role']+"- "+ message['content']+"<|im_end|>\n"
     return chatml_prompt
-def get_bot_response(url, prompt):
+def get_bot_response(url, prompt,state):
     chatml_prompt = format_chatml_prompt(state)
     payload = {
         "input": {
@@ -59,8 +59,8 @@ def chat_with_bots(user_input, state):
     # Update the state with the names of the last bots
     state.update({'last_bots': [bot_names[0], bot_names[1]]})
 
-    bot1_response = get_bot_response(bot1_url, user_input)
-    bot2_response = get_bot_response(bot2_url, user_input)
+    bot1_response = get_bot_response(bot1_url, user_input,state)
+    bot2_response = get_bot_response(bot2_url, user_input,state)
 
     return bot1_response, bot2_response