feat: 添加获取所有子心流ID和强制改变子心流状态的API接口

This commit is contained in:
墨梓柒
2025-05-05 01:26:34 +08:00
parent 27212c5d43
commit 2115917580
4 changed files with 61 additions and 4 deletions

15
src/api/apiforgui.py Normal file
View File

@@ -0,0 +1,15 @@
from src.heart_flow.heartflow import heartflow
from src.heart_flow.sub_heartflow import ChatState
async def get_all_subheartflow_ids() -> list:
"""获取所有子心流的ID列表"""
all_subheartflows = heartflow.subheartflow_manager.get_all_subheartflows()
return [subheartflow.subheartflow_id for subheartflow in all_subheartflows]
async def forced_change_subheartflow_status(subheartflow_id: str, status: ChatState) -> bool:
"""强制改变子心流的状态"""
subheartflow = await heartflow.get_or_create_subheartflow(subheartflow_id)
if subheartflow:
return await heartflow.force_change_subheartflow_status(subheartflow_id, status)
return False