Spaces:
Runtime error
Runtime error
Commit ·
3fcc106
1
Parent(s): 011faec
refactor logging mechanism in run_concurrent_tasks to use get_session_logs
Browse files- app/app.py +3 -2
app/app.py
CHANGED
|
@@ -103,9 +103,10 @@ async def start_chat():
|
|
| 103 |
|
| 104 |
async def run_concurrent_tasks(llm, message):
|
| 105 |
asyncio.create_task(llm.acomplete(message))
|
|
|
|
| 106 |
# Stream logger output while LLM is running
|
| 107 |
-
while True:
|
| 108 |
-
async for chunk in _logger.
|
| 109 |
yield chunk # Yield each chunk directly
|
| 110 |
|
| 111 |
@cl.on_message
|
|
|
|
| 103 |
|
| 104 |
async def run_concurrent_tasks(llm, message):
|
| 105 |
asyncio.create_task(llm.acomplete(message))
|
| 106 |
+
asyncio.create_task(_logger.distribute())
|
| 107 |
# Stream logger output while LLM is running
|
| 108 |
+
while True:
|
| 109 |
+
async for chunk in _logger.get_session_logs(llm.session_id):
|
| 110 |
yield chunk # Yield each chunk directly
|
| 111 |
|
| 112 |
@cl.on_message
|