"""AI Agent Work Shift Scheduler & Anti-Spam Rate Limiting Service for DSCons.""" from __future__ import annotations import asyncio import logging import random import time from datetime import datetime from typing import Any from app.core.postgres.erp_client import ErpDatabaseClient logger = logging.getLogger("dscons.agent_scheduler") class AgentSchedulerService: """Manages business work shifts and enforces anti-spam rate limiting across AI Personas.""" # Phân ca làm việc thông minh so le / song song theo giờ trong ngày (Tối ưu hóa 11 Đặc vụ AI) WORK_SHIFTS: list[dict[str, Any]] = [ { "shift_id": "SHIFT-01-EARLY-SITE", "agent_code": "phuc", "agent_name": "Phúc", "role_title": "Tiến độ & Nhắc việc công trường", "time_window": "06:00 - 08:30", "start_hour": 6.0, "end_hour": 8.5, "primary_task": "Thu thập nhật trình xe Howo 4 chân, sổ ca máy đào Kobelco và đôn đốc chấm công hiện trường.", "preferred_provider": "lmstudio", "preferred_model": "qwen2.5-7b-instruct", "rate_limit_per_min": 60, }, { "shift_id": "SHIFT-02-LOGISTICS", "agent_code": "tuan", "agent_name": "Tuấn", "role_title": "Kỹ sư Định mức & Bảng giá Logistics", "time_window": "07:00 - 09:30", "start_hour": 7.0, "end_hour": 9.5, "primary_task": "Phân tích biến động giá thép Hòa Phát, xi măng Hà Tiên và điều phối vận tải vật tư đến công trường.", "preferred_provider": "openrouter", "preferred_model": "anthropic/claude-3-haiku", "rate_limit_per_min": 20, }, { "shift_id": "SHIFT-03-QAQC", "agent_code": "hung", "agent_name": "Hùng", "role_title": "Kỹ thuật hiện trường & Ca máy", "time_window": "08:30 - 10:30", "start_hour": 8.5, "end_hour": 10.5, "primary_task": "Thẩm định kết quả nén mẫu bê tông R28, chứng chỉ xuất xưởng CO/CQ vật tư cát đá và máy móc.", "preferred_provider": "openrouter", "preferred_model": "anthropic/claude-fable-5", "rate_limit_per_min": 10, }, { "shift_id": "SHIFT-04-METHODS", "agent_code": "lam", "agent_name": "Lâm", "role_title": "Kỹ sư Biện pháp thi công & Công trình tạm", "time_window": "09:00 - 11:30", "start_hour": 9.0, "end_hour": 11.5, "primary_task": "Thiết kế biện pháp chống văng tầng hầm, tính toán khả năng chịu lực cừ Larsen và giàn giáo bao che.", "preferred_provider": "openrouter", "preferred_model": "deepseek/deepseek-r1", "rate_limit_per_min": 5, }, { "shift_id": "SHIFT-05-QS-BOQ", "agent_code": "quynh", "agent_name": "Quỳnh", "role_title": "Kế toán chi phí & Dòng tiền", "time_window": "10:30 - 12:30", "start_hour": 10.5, "end_hour": 12.5, "primary_task": "Bóc tách khối lượng thi công thực tế, tính toán đơn giá định mức 2026 và lập phụ lục 03a.", "preferred_provider": "openrouter", "preferred_model": "deepseek/deepseek-v4-pro", "rate_limit_per_min": 8, }, { "shift_id": "SHIFT-06-ACCOUNTING", "agent_code": "nam", "agent_name": "Nam", "role_title": "Tài chính & Quyết toán", "time_window": "13:30 - 15:30", "start_hour": 13.5, "end_hour": 15.5, "primary_task": "Đối soát 3 bên Hóa đơn VAT điện tử (MISA/GDT), khấu trừ tạm ứng 10% và giữ lại 5% bảo hành.", "preferred_provider": "openrouter", "preferred_model": "anthropic/claude-opus-4.8", "rate_limit_per_min": 15, }, { "shift_id": "SHIFT-07-CLIENT", "agent_code": "lan", "agent_name": "Lan", "role_title": "Chăm sóc khách hàng & CĐT", "time_window": "14:00 - 16:30", "start_hour": 14.0, "end_hour": 16.5, "primary_task": "Gửi báo cáo tiến độ bằng hình ảnh flycam cho Ban Quản lý Dự án, giải đáp thắc mắc CĐT về chất lượng.", "preferred_provider": "openrouter", "preferred_model": "openai/gpt-4o-mini", "rate_limit_per_min": 25, }, { "shift_id": "SHIFT-08-LEGAL", "agent_code": "thao", "agent_name": "Thảo", "role_title": "Hồ sơ pháp lý & Định mức BoQ", "time_window": "15:00 - 17:00", "start_hour": 15.0, "end_hour": 17.0, "primary_task": "Rà soát điều khoản Hợp đồng FIDIC Clause 14.3 và quy định Luật Xây dựng 2026 chống rủi ro thanh tra.", "preferred_provider": "openrouter", "preferred_model": "deepseek/deepseek-v4-pro", "rate_limit_per_min": 6, }, { "shift_id": "SHIFT-09-DIRECTOR", "agent_code": "tung", "agent_name": "Tùng", "role_title": "Kiểm toán nội bộ & Soát lỗi", "time_window": "16:30 - 18:00", "start_hour": 16.5, "end_hour": 18.0, "primary_task": "Đánh giá chỉ số tiến độ SPI / chi phí CPI, ký duyệt phát hành hồ sơ thanh toán IPC xuất ra Chủ Đầu Tư.", "preferred_provider": "openrouter", "preferred_model": "stealth/ox-alpha", "rate_limit_per_min": 8, }, { "shift_id": "SHIFT-10-SECRETARY", "agent_code": "thuy", "agent_name": "Thủy", "role_title": "Trợ lý Giám đốc & Điều hành", "time_window": "17:30 - 19:30", "start_hour": 17.5, "end_hour": 19.5, "primary_task": "Theo dõi đơn đặt hàng PO, phiếu nhập kho GRN, tập hợp báo cáo ngày và đối chiếu phiếu cân trạm trộn.", "preferred_provider": "openrouter", "preferred_model": "mistralai/mistral-small-24b-instruct-2501:free", "rate_limit_per_min": 10, }, { "shift_id": "SHIFT-11-OPS", "agent_code": "minh", "agent_name": "Minh", "role_title": "Điều phối tổng & Pháp lý Luật 2026", "time_window": "19:00 - 21:00", "start_hour": 19.0, "end_hour": 21.0, "primary_task": "Tổng kết hoạt động toàn công ty trong ngày, quét tự trị (Autonomous Sweep) và sao lưu CSDL nội bộ.", "preferred_provider": "openrouter", "preferred_model": "stealth/ox-alpha", "rate_limit_per_min": 8, }, ] def __init__(self, postgres_client: ErpDatabaseClient | None = None) -> None: self.postgres_client = postgres_client or ErpDatabaseClient() self._last_request_timestamp: dict[str, float] = {} def get_daily_schedule(self) -> list[dict[str, Any]]: """Lấy lịch làm việc phân bổ so le trong ngày của 9 Nhân sự AI.""" now = datetime.now() current_time_dec = now.hour + (now.minute / 60.0) result = [] for s in self.WORK_SHIFTS: is_active = s["start_hour"] <= current_time_dec < s["end_hour"] item = dict(s) item["is_current_shift"] = is_active item["status"] = "ACTIVE_DUTY" if is_active else "STANDBY" item["protection_strategy"] = "ANTI_SPAM_RATE_LIMIT_GUARD" result.append(item) return result def get_anti_spam_shield_status(self) -> dict[str, Any]: """Báo cáo trạng thái bộ đệm bảo vệ chống Rate Limit & Spam Detection của OpenRouter.""" return { "strategy": "Distributed Time-Shifted Queue with Multi-Provider Failover", "openrouter_free_rpm_cap": 8, "minimum_request_gap_seconds": 2.5, "jitter_range_seconds": "0.5 - 2.0s", "fallback_routes": { "openrouter_429": "Tự động chuyển tiếp sang Local LM Studio (Offline 0% Cost) hoặc Gemini API Direct", "stealth_expired": "Tự động hạ cấp sang Gemini 2.0 Flash / Qwen 32B an toàn", }, "active_guards": [ "Token Bucket Rate Limiter", "Dynamic Artificial Jitter", "Payload Batch Aggregation", "Local LM Studio Zero-Cost Offloading", ], "system_health": "OPTIMAL_PROTECTED", } async def throttle_agent_call( self, agent_code: str, provider: str = "openrouter" ) -> float: """Kiểm soát tần suất gọi API của từng AI Agent, tự động chèn jitter để chống bị OpenRouter gắn cờ spam.""" key = f"{agent_code}:{provider}" now = time.time() last_t = self._last_request_timestamp.get(key, 0.0) elapsed = now - last_t min_gap = 2.0 # Giữ khoảng cách tối thiểu 2.0 giây giữa 2 lần gọi cùng agent jitter = random.uniform(0.5, 1.8) required_wait = (min_gap + jitter) - elapsed if required_wait > 0: logger.debug( "[ANTI-SPAM GUARD] Giãn cách gọi API cho %s trong %.2fs", key, required_wait, ) await asyncio.sleep(required_wait) waited = required_wait else: waited = 0.0 self._last_request_timestamp[key] = time.time() return waited