hamza82 commited on
Commit
9d9f437
·
verified ·
1 Parent(s): da471a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py CHANGED
@@ -154,3 +154,23 @@ async def on_chat_start():
154
  )
155
  await cl.Message(content="Connected to Chainlit!").send()
156
  logger.info("Chat started with Chainlit.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
154
  )
155
  await cl.Message(content="Connected to Chainlit!").send()
156
  logger.info("Chat started with Chainlit.")
157
+
158
+ @cl.on_message
159
+ async def on_message(message: cl.Message):
160
+ message_history = cl.user_session.get("message_history")
161
+ message_history.append({"role": "user", "content": message.content})
162
+
163
+ msg = cl.Message(content="")
164
+ await msg.send()
165
+
166
+ stream = await client.chat.completions.create(
167
+ messages=message_history, stream=True, **settings
168
+ )
169
+
170
+ async for part in stream:
171
+ if token := part.choices[0].delta.content or "":
172
+ await msg.stream_token(token)
173
+
174
+ message_history.append({"role": "assistant", "content": msg.content})
175
+ await msg.update()
176
+ logger.info("Message processed and response sent.")