randydev commited on
Commit
c81be17
·
verified ·
1 Parent(s): 0d14b43

Update chatbot/plugins/chat.py

Browse files
Files changed (1) hide show
  1. chatbot/plugins/chat.py +5 -1
chatbot/plugins/chat.py CHANGED
@@ -51,9 +51,13 @@ async def process_stream(message, prompt):
51
  # backup_chat = await db._get_openai_chat_from_db(message.from_user.id)
52
  # backup_chat.append({"role": "system", "content": SYSTEM_PROMPT})
53
  # backup_chat.append({"role": "user", "content": prompt})
 
 
 
 
54
  stream = client_hf.chat.completions.create(
55
  model="mistralai/Mixtral-8x7B-Instruct-v0.1",
56
- messages=backup_chat,
57
  max_tokens=500,
58
  stream=True
59
  )
 
51
  # backup_chat = await db._get_openai_chat_from_db(message.from_user.id)
52
  # backup_chat.append({"role": "system", "content": SYSTEM_PROMPT})
53
  # backup_chat.append({"role": "user", "content": prompt})
54
+ messages = [
55
+ {"role": "system", "content": SYSTEM_PROMPT},
56
+ {"role": "user", "content": prompt}
57
+ ]
58
  stream = client_hf.chat.completions.create(
59
  model="mistralai/Mixtral-8x7B-Instruct-v0.1",
60
+ messages=messages,
61
  max_tokens=500,
62
  stream=True
63
  )