Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.")
|