diff --git a/src/api/__init__.py b/src/api/__init__.py new file mode 100644 index 000000000..359eff96e --- /dev/null +++ b/src/api/__init__.py @@ -0,0 +1 @@ +# This file makes src/api a Python package. \ No newline at end of file diff --git a/src/api/message_router.py b/src/api/message_router.py new file mode 100644 index 000000000..5a2dd2d0a --- /dev/null +++ b/src/api/message_router.py @@ -0,0 +1,13 @@ +from fastapi import APIRouter +from src.plugin_system.apis import message_api + +router = APIRouter() + +@router.get("/messages/recent") +async def get_recent_messages(chat_id: str, limit: int = 10): + """ + 获取最近的聊天记录 + """ + # 假设 message_api.get_recent_messages 是一个异步函数 + messages = await message_api.get_recent_messages(chat_id=chat_id, limit=limit) + return {"chat_id": chat_id, "messages": messages} \ No newline at end of file diff --git a/src/main.py b/src/main.py index 68d8e0801..8236f7472 100644 --- a/src/main.py +++ b/src/main.py @@ -245,6 +245,16 @@ MoFox_Bot(第三方修改版) # start_api_server() # logger.info("API服务器启动成功") + # 注册API路由 + try: + from src.api.message_router import router as message_router + self.server.register_router(message_router, prefix="/api") + logger.info("API路由注册成功") + except ImportError as e: + logger.error(f"导入API路由失败: {e}") + except Exception as e: + logger.error(f"注册API路由时发生错误: {e}") + # 加载所有actions,包括默认的和插件的 plugin_manager.load_all_plugins()