vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-04-28 11:40:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

update multi turn

Browse source
This commit is contained in:
puzhen 2025-10-18 14:33:02 +01:00
parent 3f0e0c36f4
commit bc114f39ce
8 changed files with 492 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

20
backend/app/controller/chat_controller.py
View file

@ -70,10 +70,26 @@ async def post(data: Chat, request: Request):
def improve(id: str, data: SupplementChat):
chat_logger.info(f"Improving chat for task_id: {id} with question: {data.question}")
task_lock = get_task_lock(id)
# Allow continuing conversation even after task is done
# This supports multi-turn conversation after complex task completion
if task_lock.status == Status.done:
raise UserException(code.error, "Task was done")
chat_logger.info(f"[CONTEXT] Task {id} was done, restarting for context-aware conversation")
# Reset status to allow processing new messages
task_lock.status = Status.confirming
# Clear any existing background tasks since workforce was stopped
if hasattr(task_lock, 'background_tasks'):
task_lock.background_tasks.clear()
# Note: conversation_history and last_task_result are preserved
# Log context preservation
if hasattr(task_lock, 'conversation_history'):
chat_logger.info(f"[CONTEXT] Preserved {len(task_lock.conversation_history)} conversation entries")
if hasattr(task_lock, 'last_task_result'):
chat_logger.info(f"[CONTEXT] Preserved task result: {len(task_lock.last_task_result)} chars")
asyncio.run(task_lock.put_queue(ActionImproveData(data=data.question)))
chat_logger.info(f"Improvement request queued for task_id: {id}")
chat_logger.info(f"Improvement request queued with preserved context")
return Response(status_code=201)