Spaces:
Sleeping
Sleeping
Ken Sang Tang commited on
Update app.py
Browse files
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 |
# 检查字符串是否包含两个 "|" 且以 " - " 分割前缀和内容
|