@agentunion/kite 1.3.2 → 1.5.0

package/extensions/services/evol/auth_manager.py

@@ -0,0 +1,443 @@
+"""
+认证管理模块 - Token 缓存、验证、吊销
+
+使用 JSONL 格式存储 Token 历史记录，每次变更追加一条记录。
+查询时使用每个 Token 的最新记录。
+"""
+
+import json
+import os
+import secrets
+import time
+from datetime import datetime, timezone
+from typing import Optional
+
+
+class AuthManager:
+    """认证管理器"""
+
+    def __init__(self, data_dir: str):
+        """
+        初始化认证管理器
+
+        Args:
+            data_dir: 数据目录（$KITE_DATA/evol）
+        """
+        self.data_dir = data_dir
+        self.auth_dir = os.path.join(data_dir, "auth")
+        os.makedirs(self.auth_dir, exist_ok=True)
+
+        self.evol_tokens_file = os.path.join(self.auth_dir, "evol_tokens.jsonl")
+        self.kite_tokens_file = os.path.join(self.auth_dir, "kite_tokens.jsonl")
+
+        self.kite_token_ttl = 30 * 24 * 3600  # 30 days
+        self.evol_token_ttl = 7 * 24 * 3600   # 7 days
+        self.token_update_interval = 24 * 3600  # 1 day - Token 更新间隔
+
+        # 内存缓存
+        self._kite_tokens_cache = None
+        self._kite_tokens_cache_time = 0
+        self._evol_tokens_cache = {}  # 改为字典：{phone: record}
+        self._evol_tokens_cache_time = 0
+        self.cache_ttl = 60 * 60  # 1 小时缓存
+
+    def save_evol_token(self, phone: str, token: str, data: dict):
+        """保存 Evol Token 到 JSONL"""
+        now = time.time()
+        record = {
+            "phone": phone,
+            "token": token,
+            "action": "login",
+            "obtainedAt": now,
+            "obtainedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "expiresAt": now + self.evol_token_ttl,
+            "expiresAt_human": datetime.fromtimestamp(now + self.evol_token_ttl, timezone.utc).isoformat(),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "userInfo": data.get("userInfo", {}),
+            "accountInfo": data.get("accountInfo", {}),
+            "teamInfo": data.get("teamInfo", {}),
+            "gatewayInfo": data.get("gatewayInfo", {}),
+            "timestamp": now
+        }
+
+        self._append_evol_record(record)
+        # 清除缓存
+        self._evol_tokens_cache = {}
+        self._evol_tokens_cache_time = 0
+
+    def get_evol_token(self, phone: str) -> Optional[dict]:
+        """获取指定手机号的最新 Evol Token（带缓存）"""
+        now = time.time()
+
+        # 检查缓存是否有效
+        if self._evol_tokens_cache and (now - self._evol_tokens_cache_time < self.cache_ttl):
+            return self._evol_tokens_cache.get(phone)
+
+        # 从文件读取所有 Evol Token
+        if not os.path.exists(self.evol_tokens_file):
+            return None
+
+        try:
+            # 按 phone 分组，保留每个 phone 的最新记录
+            phone_tokens = {}
+            with open(self.evol_tokens_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    line = line.strip()
+                    if not line:
+                        continue
+                    try:
+                        record = json.loads(line)
+                        record_phone = record.get("phone")
+                        if not record_phone:
+                            continue
+                        # 使用最新的记录（后面的覆盖前面的）
+                        phone_tokens[record_phone] = record
+                    except json.JSONDecodeError:
+                        continue
+
+            # 过滤掉已过期和已登出的 token
+            valid_tokens = {}
+            for p, record in phone_tokens.items():
+                # 检查是否登出
+                if record.get("action") == "logout":
+                    continue
+                # 检查是否过期
+                if time.time() > record.get("expiresAt", 0):
+                    continue
+                valid_tokens[p] = record
+
+            # 更新缓存
+            self._evol_tokens_cache = valid_tokens
+            self._evol_tokens_cache_time = now
+
+            return valid_tokens.get(phone)
+
+        except Exception:
+            return None
+
+    def update_evol_token_usage(self, phone: str):
+        """更新 Evol Token 使用时间（超过 1 天才记录）"""
+        latest_record = self.get_evol_token(phone)
+        if not latest_record:
+            return
+
+        now = time.time()
+        last_used = latest_record.get("lastUsedAt", 0)
+
+        # 如果距离上次使用不到 1 天，不记录
+        if now - last_used < self.token_update_interval:
+            return
+
+        # 追加使用记录
+        record = {
+            "phone": phone,
+            "token": latest_record.get("token"),
+            "action": "used",
+            "obtainedAt": latest_record.get("obtainedAt"),
+            "obtainedAt_human": latest_record.get("obtainedAt_human"),
+            "expiresAt": latest_record.get("expiresAt"),
+            "expiresAt_human": latest_record.get("expiresAt_human"),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "userInfo": latest_record.get("userInfo", {}),
+            "accountInfo": latest_record.get("accountInfo", {}),
+            "teamInfo": latest_record.get("teamInfo", {}),
+            "gatewayInfo": latest_record.get("gatewayInfo", {}),
+            "timestamp": now
+        }
+
+        self._append_evol_record(record)
+        # 清除缓存
+        self._evol_tokens_cache = {}
+        self._evol_tokens_cache_time = 0
+
+    def _append_evol_record(self, record: dict):
+        """追加 Evol Token 记录到 JSONL"""
+        try:
+            with open(self.evol_tokens_file, "a", encoding="utf-8") as f:
+                f.write(json.dumps(record, ensure_ascii=False) + "\n")
+        except Exception as e:
+            print(f"[auth] Failed to append evol token record: {e}")
+
+    def revoke_evol_token(self, phone: str):
+        """吊销 Evol Token（追加 logout 记录）"""
+        latest_record = self.get_evol_token(phone)
+        if not latest_record:
+            return
+
+        now = time.time()
+        record = {
+            "phone": phone,
+            "token": latest_record.get("token"),
+            "action": "logout",
+            "obtainedAt": latest_record.get("obtainedAt"),
+            "obtainedAt_human": latest_record.get("obtainedAt_human"),
+            "expiresAt": latest_record.get("expiresAt"),
+            "expiresAt_human": latest_record.get("expiresAt_human"),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "userInfo": latest_record.get("userInfo", {}),
+            "accountInfo": latest_record.get("accountInfo", {}),
+            "teamInfo": latest_record.get("teamInfo", {}),
+            "gatewayInfo": latest_record.get("gatewayInfo", {}),
+            "timestamp": now
+        }
+
+        self._append_evol_record(record)
+        # 清除缓存
+        self._evol_tokens_cache = {}
+        self._evol_tokens_cache_time = 0
+
+    def generate_kite_token(self, device_info: dict) -> str:
+        """生成 Kite Token 并追加到 JSONL"""
+        token = "kite_" + secrets.token_urlsafe(32)
+        now = time.time()
+
+        record = {
+            "token": token,
+            "deviceId": device_info.get("deviceId", "unknown"),
+            "deviceName": device_info.get("deviceName", "Unknown Device"),
+            "phone": None,  # 初始未绑定手机号
+            "action": "created",
+            "createdAt": now,
+            "createdAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "expiresAt": now + self.kite_token_ttl,
+            "expiresAt_human": datetime.fromtimestamp(now + self.kite_token_ttl, timezone.utc).isoformat(),
+            "isValid": True,
+            "timestamp": now
+        }
+
+        # 追加到 JSONL
+        self._append_token_record(record)
+        # 清除缓存
+        self._kite_tokens_cache = None
+        self._kite_tokens_cache_time = 0
+
+        return token
+
+    def verify_kite_token(self, token: str) -> bool:
+        """验证 Kite Token 并更新使用时间（超过 1 天才记录）"""
+        latest_tokens = self._get_latest_tokens()
+        now = time.time()
+
+        token_info = latest_tokens.get(token)
+        if not token_info:
+            return False
+
+        if not token_info.get("isValid", True):
+            return False
+
+        if now > token_info.get("expiresAt", 0):
+            return False
+
+        # 检查距离上次使用是否超过 1 天
+        last_used = token_info.get("lastUsedAt", 0)
+        if now - last_used < self.token_update_interval:
+            # 不到 1 天，不记录
+            return True
+
+        # 追加使用记录
+        expires_at = now + self.kite_token_ttl
+        record = {
+            "token": token,
+            "deviceId": token_info.get("deviceId"),
+            "deviceName": token_info.get("deviceName"),
+            "phone": token_info.get("phone"),  # 保留绑定的手机号
+            "action": "used",
+            "createdAt": token_info.get("createdAt"),
+            "createdAt_human": token_info.get("createdAt_human"),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "expiresAt": expires_at,
+            "expiresAt_human": datetime.fromtimestamp(expires_at, timezone.utc).isoformat(),
+            "isValid": True,
+            "timestamp": now
+        }
+        self._append_token_record(record)
+        # 清除缓存
+        self._kite_tokens_cache = None
+        self._kite_tokens_cache_time = 0
+
+        return True
+
+    def revoke_kite_token(self, token: str):
+        """吊销 Kite Token"""
+        latest_tokens = self._get_latest_tokens()
+        token_info = latest_tokens.get(token)
+
+        if not token_info:
+            return
+
+        now = time.time()
+        record = {
+            "token": token,
+            "deviceId": token_info.get("deviceId"),
+            "deviceName": token_info.get("deviceName"),
+            "phone": token_info.get("phone"),  # 保留绑定的手机号
+            "action": "revoked",
+            "createdAt": token_info.get("createdAt"),
+            "createdAt_human": token_info.get("createdAt_human"),
+            "lastUsedAt": token_info.get("lastUsedAt"),
+            "lastUsedAt_human": token_info.get("lastUsedAt_human"),
+            "expiresAt": token_info.get("expiresAt"),
+            "expiresAt_human": token_info.get("expiresAt_human"),
+            "isValid": False,
+            "timestamp": now
+        }
+        self._append_token_record(record)
+        # 清除缓存
+        self._kite_tokens_cache = None
+        self._kite_tokens_cache_time = 0
+
+    def bind_kite_token_to_phone(self, token: str, phone: str):
+        """绑定 Kite Token 到手机号"""
+        latest_tokens = self._get_latest_tokens()
+        token_info = latest_tokens.get(token)
+
+        if not token_info:
+            raise ValueError(f"Token not found: {token}")
+
+        # 如果已经绑定到同一个手机号，不需要重复记录
+        if token_info.get("phone") == phone:
+            return
+
+        now = time.time()
+        record = {
+            "token": token,
+            "deviceId": token_info.get("deviceId"),
+            "deviceName": token_info.get("deviceName"),
+            "phone": phone,  # 绑定手机号
+            "action": "bound",
+            "createdAt": token_info.get("createdAt"),
+            "createdAt_human": token_info.get("createdAt_human"),
+            "lastUsedAt": now,
+            "lastUsedAt_human": datetime.fromtimestamp(now, timezone.utc).isoformat(),
+            "expiresAt": token_info.get("expiresAt"),
+            "expiresAt_human": token_info.get("expiresAt_human"),
+            "isValid": token_info.get("isValid", True),
+            "timestamp": now
+        }
+        self._append_token_record(record)
+        # 清除缓存
+        self._kite_tokens_cache = None
+        self._kite_tokens_cache_time = 0
+
+    def get_phone_by_kite_token(self, token: str) -> Optional[str]:
+        """根据 Kite Token 获取绑定的手机号"""
+        latest_tokens = self._get_latest_tokens()
+        token_info = latest_tokens.get(token)
+
+        if not token_info:
+            return None
+
+        return token_info.get("phone")
+
+    def list_all_evol_tokens(self) -> list:
+        """列出所有有效的 Evol Token（用于前端显示）"""
+        if not os.path.exists(self.evol_tokens_file):
+            return []
+
+        try:
+            # 按 phone 分组，保留每个 phone 的最新记录
+            phone_tokens = {}
+            with open(self.evol_tokens_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    line = line.strip()
+                    if not line:
+                        continue
+                    try:
+                        record = json.loads(line)
+                        record_phone = record.get("phone")
+                        if not record_phone:
+                            continue
+                        # 使用最新的记录（后面的覆盖前面的）
+                        phone_tokens[record_phone] = record
+                    except json.JSONDecodeError:
+                        continue
+
+            # 过滤掉已过期和已登出的 token，返回列表
+            valid_tokens = []
+            for phone, record in phone_tokens.items():
+                # 检查是否登出
+                if record.get("action") == "logout":
+                    continue
+                # 检查是否过期
+                if time.time() > record.get("expiresAt", 0):
+                    continue
+                valid_tokens.append(record)
+
+            return valid_tokens
+
+        except Exception:
+            return []
+
+    def list_devices(self, current_token: str) -> list:
+        """列出所有已配对设备"""
+        latest_tokens = self._get_latest_tokens()
+        devices = []
+
+        for token, info in latest_tokens.items():
+            devices.append({
+                "deviceId": info.get("deviceId"),
+                "deviceName": info.get("deviceName"),
+                "createdAt": info.get("createdAt"),
+                "lastUsedAt": info.get("lastUsedAt"),
+                "expiresAt": info.get("expiresAt"),
+                "isValid": info.get("isValid", True),
+                "isCurrent": token == current_token
+            })
+
+        return devices
+
+    def _get_latest_tokens(self) -> dict:
+        """
+        获取所有 Token 的最新状态（带缓存）
+
+        Returns:
+            {token: latest_record} 字典
+        """
+        now = time.time()
+
+        # 检查缓存是否有效
+        if self._kite_tokens_cache and (now - self._kite_tokens_cache_time < self.cache_ttl):
+            return self._kite_tokens_cache
+
+        # 从文件读取
+        if not os.path.exists(self.kite_tokens_file):
+            return {}
+
+        tokens = {}
+        try:
+            with open(self.kite_tokens_file, "r", encoding="utf-8") as f:
+                for line in f:
+                    line = line.strip()
+                    if not line:
+                        continue
+
+                    try:
+                        record = json.loads(line)
+                        token = record.get("token")
+                        if token:
+                            # 使用最新的记录（后面的覆盖前面的）
+                            tokens[token] = record
+                    except json.JSONDecodeError:
+                        continue
+        except Exception:
+            return {}
+
+        # 更新缓存
+        self._kite_tokens_cache = tokens
+        self._kite_tokens_cache_time = now
+
+        return tokens
+
+    def _append_token_record(self, record: dict):
+        """追加 Token 记录到 JSONL"""
+        try:
+            with open(self.kite_tokens_file, "a", encoding="utf-8") as f:
+                f.write(json.dumps(record, ensure_ascii=False) + "\n")
+        except Exception as e:
+            print(f"[auth] Failed to append token record: {e}")

package/extensions/services/evol/config.yaml

@@ -0,0 +1,149 @@
+user:
+  phone_number: ""       # 当前 AI 使用的号码
+  phone_numbers: []      # AI 号码列表，可在配置页面管理
+  owners: []             # 主人列表，每项包含 phone/name 等基础信息
+  # 示例:
+  # phone_numbers:
+  #   - "13800138000"
+  #   - "13800138001"
+  # owners:
+  #   - phone: "13900139000"
+  #     name: "张三"
+  #     note: "公司CEO"
+  #   - phone: "13700137000"
+  #     name: "李四"
+
+server:
+  host: "0.0.0.0"
+  port: 18766
+  ssl: true
+
+bluetooth:
+  auto_connect: true
+  reconnect_interval: 5
+  reconnect_max_interval: 60
+  device_address: ""
+
+audio:
+  sample_rate: 16000
+  channels: 1
+  format: "s16le"
+  record_calls: true
+
+llm:
+  active_provider: "openai"
+  providers:
+    openai:
+      base_url: "https://api.openai.com/v1"
+      api_key: ""
+      model: "gpt-4o"
+      temperature: 0.7
+      max_tokens: 1024
+    claude:
+      base_url: "https://api.anthropic.com/v1"
+      api_key: ""
+      model: "claude-sonnet-4-20250514"
+      temperature: 0.7
+      max_tokens: 1024
+    gemini:
+      base_url: "https://generativelanguage.googleapis.com/v1beta"
+      api_key: ""
+      model: "gemini-2.0-flash"
+      temperature: 0.7
+      max_tokens: 1024
+
+asr:
+  provider: "whisper"
+  whisper:
+    base_url: "https://api.openai.com/v1"
+    api_key: ""
+    model: "whisper-1"
+    language: "zh"
+  xunfei:
+    app_id: ""
+    api_key: ""
+    api_secret: ""
+  volcengine:
+    app_id: ""
+    access_token: ""
+    resource_id: "volc.bigasr.sauc.duration"
+    ws_url: "wss://openspeech.bytedance.com/api/v3/sauc/bigmodel_async"
+  tencent:
+    app_id: ""
+    secret_id: ""
+    secret_key: ""
+    engine_model_type: "16k_zh_large"
+
+tts:
+  provider: "edge-tts"
+  edge_tts:
+    voice: "zh-CN-XiaoxiaoNeural"
+    rate: "+0%"
+    volume: "+0%"
+  volcengine:
+    app_id: ""
+    access_token: ""
+    cluster: "volcano_tts"
+    voice_type: "BV001_streaming"
+    speed_ratio: 1.0
+    volume_ratio: 1.0
+    pitch_ratio: 1.0
+    encoding: "mp3"
+    sample_rate: 24000
+  tencent:
+    app_id: ""
+    secret_id: ""
+    secret_key: ""
+    voice_type: 101001
+    codec: "pcm"
+    sample_rate: 16000
+    speed: 0
+    volume: 0
+  azure:
+    api_key: ""
+    region: "eastasia"
+    voice: "zh-CN-XiaoxiaoNeural"
+    rate: "1.0"
+    volume: "100"
+  openai:
+    api_key: ""
+    base_url: "https://api.openai.com/v1"
+    model: "tts-1"
+    voice: "alloy"
+    speed: 1.0
+  aliyun:
+    access_key_id: ""
+    access_key_secret: ""
+    app_key: ""
+    voice: "zhixiaobai"
+    sample_rate: 16000
+    speech_rate: 0
+    volume: 50
+  local:
+    engine: "sherpa-onnx"
+    model_path: ""
+    voice: ""
+    sample_rate: 16000
+
+vad:
+  mode: 3
+  energy_threshold: 300
+  silence_threshold_ms: 800
+  min_speech_ms: 250
+
+call:
+  max_duration_seconds: 300
+  no_response_timeout: 15
+  no_response_max_retries: 3
+  incoming_confirm_timeout: 15
+  incoming_default_action: "reject"
+  opening_message: ""
+
+webhook:
+  default_url: ""
+  retry_count: 3
+  retry_delays: [1, 5, 30]
+  timeout: 10
+
+data:
+  base_dir: "./data"

package/extensions/services/evol/config_loader.py

@@ -0,0 +1,117 @@
+"""
+配置加载工具
+
+用于加载模块的业务配置。支持从 module.md 读取业务配置块，
+并动态加载对应的 JSON5 配置文件。
+
+零共享代码依赖 - 此文件可以独立拷贝到其他模块使用。
+"""
+
+import os
+import re
+import json5
+import yaml
+
+
+def load_module_metadata(module_dir: str) -> dict:
+    """
+    读取 module.md 的 YAML frontmatter。
+
+    Args:
+        module_dir: 模块目录路径
+
+    Returns:
+        模块元数据字典
+    """
+    md_path = os.path.join(module_dir, "module.md")
+
+    if not os.path.exists(md_path):
+        return {}
+
+    try:
+        with open(md_path, "r", encoding="utf-8") as f:
+            text = f.read()
+
+        # 提取 YAML frontmatter (--- ... ---)
+        m = re.match(r'^---\s*\n(.*?)\n---', text, re.DOTALL)
+        if m:
+            return yaml.safe_load(m.group(1)) or {}
+    except Exception as e:
+        print(f"[config_loader] Error loading module.md: {e}")
+
+    return {}
+
+
+def load_business_configs(module_dir: str) -> dict:
+    """
+    加载所有业务配置。
+
+    从 module.md 的 businesses 块读取业务列表，
+    然后加载每个业务的配置文件（JSON5 格式）。
+
+    Args:
+        module_dir: 模块目录路径
+
+    Returns:
+        业务配置字典，格式：
+        {
+            "business_name": {
+                "metadata": {
+                    "name": "business_name",
+                    "type": "business_type",
+                    "description": "...",
+                    "config_file": "config.json5"
+                },
+                "config": { ... }  # JSON5 配置内容
+            }
+        }
+    """
+    metadata = load_module_metadata(module_dir)
+    businesses = metadata.get('businesses', [])
+
+    if not businesses:
+        print(f"[config_loader] No businesses configured in module.md")
+        return {}
+
+    configs = {}
+    for business in businesses:
+        name = business.get('name')
+        config_file = business.get('config_file')
+
+        if not name or not config_file:
+            print(f"[config_loader] Warning: Invalid business entry (missing name or config_file): {business}")
+            continue
+
+        config_path = os.path.join(module_dir, config_file)
+
+        if os.path.exists(config_path):
+            try:
+                with open(config_path, "r", encoding="utf-8") as f:
+                    configs[name] = {
+                        'metadata': business,
+                        'config': json5.load(f)
+                    }
+                print(f"[config_loader] Loaded business config: {name} from {config_file}")
+            except json5.JSON5DecodeError as e:
+                print(f"[config_loader] Error: Invalid JSON5 syntax in {config_file}: {e}")
+            except Exception as e:
+                print(f"[config_loader] Error loading {config_file}: {e}")
+        else:
+            print(f"[config_loader] Warning: Config file not found: {config_path}")
+
+    return configs
+
+
+def get_business_config(module_dir: str, business_name: str) -> dict | None:
+    """
+    获取指定业务的配置。
+
+    Args:
+        module_dir: 模块目录路径
+        business_name: 业务名称
+
+    Returns:
+        业务配置字典，如果不存在则返回 None
+    """
+    configs = load_business_configs(module_dir)
+    return configs.get(business_name)