Kris
a6fa7245e5
新增飞书机器人适配器全套功能，包括： - 基础适配器入口与工具导出 - 消息格式化、卡片渲染、回复调度逻辑 - 会话ID生成、模型覆盖策略 - 消息发送缓存、顺序队列管理 - 飞书签名验证、加解密webhook请求 - 审批权限校验、机器人菜单事件处理 - 文档评论、钉消息、语音转码处理 - 静态/动态目录管理、子代理生命周期管理 - 各类工具集：聊天、云盘、文档、知识库API封装
34 lines
827 B
Python
34 lines
827 B
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections import defaultdict
|
|
|
|
|
|
|
|
class FeishuSequentialQueue:
|
|
def __init__(self):
|
|
self._queues: dict[str, asyncio.Queue] = {}
|
|
self._locks: dict[str, asyncio.Lock] = defaultdict(asyncio.Lock)
|
|
|
|
async def acquire(self, key: str) -> None:
|
|
lock = self._locks[key]
|
|
await lock.acquire()
|
|
|
|
def release(self, key: str) -> None:
|
|
lock = self._locks.get(key)
|
|
if lock and lock.locked():
|
|
lock.release()
|
|
|
|
async def run_sequential(self, key: str, coro) -> None:
|
|
await self.acquire(key)
|
|
try:
|
|
await coro
|
|
finally:
|
|
self.release(key)
|
|
|
|
def remove(self, key: str) -> None:
|
|
self._locks.pop(key, None)
|
|
|
|
def clear(self) -> None:
|
|
self._locks.clear()
|