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

feat: get_or_create_task_lock

Browse source
This commit is contained in:
a7m-1st 2025-10-09 03:22:08 +03:00
parent 68305ed808
commit f31b9d1ad9
2 changed files with 9 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -19,7 +19,7 @@ from app.service.task import (
ActionSupplementData,
ActionAddTaskData,
ActionRemoveTaskData,
create_task_lock,
get_or_create_task_lock,
get_task_lock,
)
from app.component.environment import set_user_env_path
@ -37,7 +37,7 @@ chat_logger = traceroot.get_logger('chat_controller')
@traceroot.trace()
async def post(data: Chat, request: Request):
chat_logger.info(f"Starting new chat session for project_id: {data.project_id}, user: {data.email}")
task_lock = create_task_lock(data.project_id)
task_lock = get_or_create_task_lock(data.project_id)
# Set user-specific environment path for this thread
set_user_env_path(data.env_path)
@ -119,25 +119,6 @@ def install_mcp(id: str, data: McpServers):
return Response(status_code=201)
@router.get("/chat/{id}/independent-task-queue", name="get independent task queue")
@traceroot.trace()
def get_independent_task_queue(id: str):
"""Get the current independent task queue for a workforce"""
chat_logger.info(f"Getting independent task queue for task_id: {id}")
task_lock = get_task_lock(id)
try:
# Queue the action to get independent task queue
asyncio.run(task_lock.put_queue(ActionGetIndependentQueue()))
chat_logger.info(f"Independent task queue request queued for task_id: {id}")
return Response(status_code=201)
except Exception as e:
chat_logger.error(f"Error getting independent task queue for task_id: {id}: {e}")
raise UserException(code.error, f"Failed to get independent task queue: {str(e)}")
@router.post("/chat/{id}/add-task", name="add task to workforce")
@traceroot.trace()
def add_task(id: str, data: AddTaskRequest):