feat: 添加获取子心流循环信息和所有状态的API接口

src/api/apiforgui.py

@@ -17,3 +17,20 @@ async def forced_change_subheartflow_status(subheartflow_id: str, status: ChatSt
     if subheartflow:
         return await heartflow.force_change_subheartflow_status(subheartflow_id, status)
     return False
+
+async def get_subheartflow_cycle_info(subheartflow_id: str, history_len: int) -> dict:
+    """获取子心流的循环信息"""
+    subheartflow_cycle_info =  await heartflow.api_get_subheartflow_cycle_info(subheartflow_id, history_len)
+    logger.debug(f"子心流 {subheartflow_id} 循环信息: {subheartflow_cycle_info}")
+    if subheartflow_cycle_info:
+        return subheartflow_cycle_info
+    else:
+        logger.warning(f"子心流 {subheartflow_id} 循环信息未找到")
+        return None
+    
+
+async def get_all_states():
+    """获取所有状态"""
+    all_states = await heartflow.api_get_all_states()
+    logger.debug(f"所有状态: {all_states}")
+    return all_states

src/api/main.py

@@ -2,13 +2,18 @@ from fastapi import APIRouter
 from strawberry.fastapi import GraphQLRouter
 import os
 import sys
-
+# from src.heart_flow.heartflow import heartflow
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
 # from src.config.config import BotConfig
 from src.common.logger_manager import get_logger
 from src.api.reload_config import reload_config as reload_config_func
 from src.common.server import global_server
-from src.api.apiforgui import get_all_subheartflow_ids, forced_change_subheartflow_status
+from src.api.apiforgui import (
+    get_all_subheartflow_ids, 
+    forced_change_subheartflow_status,
+    get_subheartflow_cycle_info,
+    get_all_states,
+)
 from src.heart_flow.sub_heartflow import ChatState
 
 # import uvicorn
@@ -67,7 +72,26 @@ async def force_stop_maibot():
     else:
         logger.error("MAI Bot强制停止失败")
         return {"status": "failed"}
+    
+@router.get("/gui/subheartflow/cycleinfo")
+async def get_subheartflow_cycle_info_api(subheartflow_id: str, history_len: int):
+    """获取子心流的循环信息"""
+    cycle_info = await get_subheartflow_cycle_info(subheartflow_id, history_len)
+    if cycle_info:
+        return {"status": "success", "data": cycle_info}
+    else:
+        logger.warning(f"子心流 {subheartflow_id} 循环信息未找到")
+        return {"status": "failed", "reason": "subheartflow not found"}
 
+@router.get("/gui/get_all_states")
+async def get_all_states_api():
+    """获取所有状态"""
+    all_states = await get_all_states()
+    if all_states:
+        return {"status": "success", "data": all_states}
+    else:
+        logger.warning("获取所有状态失败")
+        return {"status": "failed", "reason": "failed to get all states"}
 
 def start_api_server():
     """启动API服务器"""