feat: 支持 agent 运行取消功能
This commit is contained in:
@@ -30,6 +30,7 @@ from schemas.domain.chat_message import (
|
||||
from v1.agent.schemas import (
|
||||
AgentRepositoryLike,
|
||||
AttachmentStorageLike,
|
||||
CancelRequested,
|
||||
EventStreamLike,
|
||||
HistorySnapshotResponse,
|
||||
QueueClientLike,
|
||||
@@ -157,6 +158,26 @@ class AgentService:
|
||||
created=created,
|
||||
)
|
||||
|
||||
async def cancel_run(
|
||||
self,
|
||||
*,
|
||||
thread_id: str,
|
||||
run_id: str,
|
||||
current_user: CurrentUser,
|
||||
) -> CancelRequested:
|
||||
owner = await self._repository.get_session_owner(session_id=thread_id)
|
||||
ensure_session_owner(owner_id=owner, current_user=current_user)
|
||||
await self._queue.request_cancel(
|
||||
thread_id=thread_id,
|
||||
run_id=run_id,
|
||||
requested_by=str(current_user.id),
|
||||
)
|
||||
return CancelRequested(
|
||||
thread_id=thread_id,
|
||||
run_id=run_id,
|
||||
accepted=True,
|
||||
)
|
||||
|
||||
async def _append_context_cache_user_message(
|
||||
self,
|
||||
*,
|
||||
|
||||
Reference in New Issue
Block a user