27 lines
953 B
Python
27 lines
953 B
Python
from src.chat.heart_flow.heartflow import heartflow
|
|
from src.chat.heart_flow.sub_heartflow import ChatState
|
|
from src.common.logger import get_logger
|
|
|
|
logger = get_logger("api")
|
|
|
|
|
|
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
|
|
|
|
|
|
async def get_all_states():
|
|
"""获取所有状态"""
|
|
all_states = await heartflow.api_get_all_states()
|
|
logger.debug(f"所有状态: {all_states}")
|
|
return all_states
|