Ken Sang Tang commited on
Commit
44cb235
·
verified ·
1 Parent(s): d89287d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -10
app.py CHANGED
@@ -219,6 +219,28 @@ async def create_task(prompt: str = Body(..., embed=True)):
219
  async def run_task(task_id: str, prompt: str):
220
  try:
221
  task_manager.tasks[task_id].status = "running"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
222
 
223
  async def on_think(thought):
224
  await task_manager.update_task_step(task_id, 0, thought, "think")
@@ -271,16 +293,6 @@ async def run_task(task_id: str, prompt: str):
271
  self.task_id, 0, cleaned_message, event_type
272
  )
273
 
274
- def make_sse_handler(task_id):
275
- handler = SSELogHandler(task_id)
276
-
277
- def sink(message):
278
- asyncio.create_task(handler(str(message)))
279
-
280
- return sink
281
-
282
- logger.add(make_sse_handler(task_id))
283
-
284
  import re
285
  def has_log_prefix(message):
286
  # 检查字符串是否包含两个 "|" 且以 " - " 分割前缀和内容
 
219
  async def run_task(task_id: str, prompt: str):
220
  try:
221
  task_manager.tasks[task_id].status = "running"
222
+ logger.info("✨ Manus's thoughts: testing SSE event emission")
223
+
224
+ # Attach SSE logger
225
+ def make_sse_handler(task_id):
226
+ handler = SSELogHandler(task_id)
227
+
228
+ def sink(message):
229
+ asyncio.create_task(handler(str(message)))
230
+
231
+ return sink
232
+
233
+ logger.add(make_sse_handler(task_id))
234
+
235
+ # Simulate external call delay and emit result manually
236
+ await asyncio.sleep(2)
237
+ logger.info("🎉 Manus result: Hello from mock")
238
+
239
+ await task_manager.update_task_step(task_id, 1, "Hello from mock", "result")
240
+ await task_manager.complete_task(task_id)
241
+
242
+ except Exception as e:
243
+ await task_manager.fail_task(task_id, str(e))
244
 
245
  async def on_think(thought):
246
  await task_manager.update_task_step(task_id, 0, thought, "think")
 
293
  self.task_id, 0, cleaned_message, event_type
294
  )
295
 
 
 
 
 
 
 
 
 
 
 
296
  import re
297
  def has_log_prefix(message):
298
  # 检查字符串是否包含两个 "|" 且以 " - " 分割前缀和内容