Update chatbot/plugins/chat.py
Browse files- 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=
|
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 |
)
|