rwitz commited on
Commit
8e00642
·
verified ·
1 Parent(s): c17a206

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,18 +14,18 @@ import sqlite3
14
  from pymongo.mongo_client import MongoClient
15
  from pymongo.server_api import ServerApi
16
  async def direct_regenerate(model, user_input, chatbot):
 
17
  temp_state = {
18
  "history": [
19
- [{"role": "user", "content": user_input}],
20
  [{"role": "user", "content": user_input}]
21
  ]
22
  }
23
- response = await get_bot_response(model, user_input, temp_state, 0)
24
  chatbot[-1] = (user_input, response)
25
  return "", chatbot
26
 
27
  async def regenerate_responses(state, chatbot1, chatbot2):
28
- user_input = state["history"][0][-2]["content"]
29
  bot1_response = await get_bot_response(state['last_bots'][0], user_input, state, 0)
30
  bot2_response = await get_bot_response(state['last_bots'][1], user_input, state, 1)
31
  chatbot1[-1] = (user_input, bot1_response)
 
14
  from pymongo.mongo_client import MongoClient
15
  from pymongo.server_api import ServerApi
16
  async def direct_regenerate(model, user_input, chatbot):
17
+ adapter = next(entry['adapter'] for entry in chatbots_data if entry['original_model'] == model)
18
  temp_state = {
19
  "history": [
 
20
  [{"role": "user", "content": user_input}]
21
  ]
22
  }
23
+ response = await get_bot_response(adapter, user_input, temp_state, 0)
24
  chatbot[-1] = (user_input, response)
25
  return "", chatbot
26
 
27
  async def regenerate_responses(state, chatbot1, chatbot2):
28
+ user_input = state["history"][0][-1]["content"]
29
  bot1_response = await get_bot_response(state['last_bots'][0], user_input, state, 0)
30
  bot2_response = await get_bot_response(state['last_bots'][1], user_input, state, 1)
31
  chatbot1[-1] = (user_input, bot1_response)