ssdfasd
4f9571b21c
- 基于 FastAPI 的 Web API 服务 - OpenCode API 客户端封装 - 会话管理器（同步/异步任务执行） - APScheduler 定时任务调度 - 完整的 REST API 端点
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import uuid
|
|
from app.models.session import SessionType, Task, TaskStatus, CreateSessionRequest
|
|
from app.services.opencode_client import opencode_client
|
|
from app.core.logging import logger
|
|
|
|
|
|
class SessionManager:
|
|
def __init__(self):
|
|
self.tasks: dict[str, Task] = {}
|
|
|
|
async def create_task(self, request: CreateSessionRequest) -> Task:
|
|
task_id = str(uuid.uuid4())
|
|
task = Task(
|
|
id=task_id,
|
|
type=request.type,
|
|
prompt=request.prompt or "",
|
|
)
|
|
self.tasks[task_id] = task
|
|
logger.info(f"Created task: {task_id}")
|
|
return task
|
|
|
|
async def execute_task(self, task_id: str) -> dict:
|
|
task = self.tasks.get(task_id)
|
|
if not task:
|
|
raise ValueError(f"Task {task_id} not found")
|
|
|
|
task.status = TaskStatus.RUNNING
|
|
session = await opencode_client.create_session(
|
|
title=task.prompt[:50] if task.prompt else None
|
|
)
|
|
session_id = session["id"]
|
|
task.session_id = session_id
|
|
|
|
logger.info(f"Executing task {task_id} with session {session_id}")
|
|
|
|
try:
|
|
result = await opencode_client.send_message(session_id, task.prompt)
|
|
task.status = TaskStatus.COMPLETED
|
|
return result
|
|
except Exception as e:
|
|
task.status = TaskStatus.FAILED
|
|
logger.error(f"Task {task_id} failed: {e}")
|
|
raise
|
|
|
|
async def execute_task_async(self, task_id: str) -> None:
|
|
task = self.tasks.get(task_id)
|
|
if not task:
|
|
raise ValueError(f"Task {task_id} not found")
|
|
|
|
task.status = TaskStatus.RUNNING
|
|
session = await opencode_client.create_session(
|
|
title=task.prompt[:50] if task.prompt else None
|
|
)
|
|
session_id = session["id"]
|
|
task.session_id = session_id
|
|
|
|
logger.info(f"Executing async task {task_id} with session {session_id}")
|
|
|
|
await opencode_client.send_message_async(session_id, task.prompt)
|
|
|
|
async def abort_task(self, task_id: str) -> bool:
|
|
task = self.tasks.get(task_id)
|
|
if not task or not task.session_id:
|
|
return False
|
|
|
|
result = await opencode_client.abort_session(task.session_id)
|
|
task.status = TaskStatus.FAILED
|
|
return result
|
|
|
|
async def get_task(self, task_id: str) -> Task | None:
|
|
return self.tasks.get(task_id)
|
|
|
|
async def list_tasks(self) -> list[Task]:
|
|
return list(self.tasks.values())
|
|
|
|
|
|
session_manager = SessionManager()
|