File size: 10,344 Bytes
e9d4f6a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
#!/usr/bin/env python3
"""
AirMicroDrip Audit Integration
Connects the audit framework to the AirMicroDrip perpetual futures system.
Provides continuous monitoring, health checks, and compliance verification.
"""

import os
import sys
from datetime import datetime
from typing import Dict, Any, Optional

# Add parent directory to path for audit_framework import
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from audit_framework import (
    AuditFramework,
    AuditCategory,
    AuditSeverity,
    AuditStatus,
)


class AirMicroDripAuditor:
    """Audit wrapper for AirMicroDrip systems."""

    def __init__(self, db_path: str = "airmicrodrip_audit.db"):
        self.audit = AuditFramework()
        self.db_path = db_path
        self._register_airmicrodrip_checks()

    def _register_airmicrodrip_checks(self):
        """Register AirMicroDrip-specific audit checks."""
        from audit_framework import AuditCheck

        extra_checks = [
            AuditCheck(
                check_id="amd_001",
                name="Liquidity Provider Health",
                description="Verify at least one active LLM inference provider",
                category=AuditCategory.AVAILABILITY,
                severity=AuditSeverity.HIGH,
            ),
            AuditCheck(
                check_id="amd_002",
                name="Synthetic Liquidity Depth",
                description="Verify total synthetic liquidity exceeds minimum threshold",
                category=AuditCategory.ACCURACY,
                severity=AuditSeverity.HIGH,
            ),
            AuditCheck(
                check_id="amd_003",
                name="Perpetual Engine Consistency",
                description="Verify mark prices and index prices are within tolerance",
                category=AuditCategory.ACCURACY,
                severity=AuditSeverity.CRITICAL,
            ),
            AuditCheck(
                check_id="amd_004",
                name="Funding Rate Bounds",
                description="Verify funding rates are within configured min/max",
                category=AuditCategory.ACCURACY,
                severity=AuditSeverity.MEDIUM,
            ),
            AuditCheck(
                check_id="amd_005",
                name="Liquidation Backlog",
                description="Verify no positions are stuck in liquidation queue",
                category=AuditCategory.INTEGRITY,
                severity=AuditSeverity.CRITICAL,
            ),
            AuditCheck(
                check_id="amd_006",
                name="Order Book Spread",
                description="Verify bid-ask spread is within acceptable range",
                category=AuditCategory.PERFORMANCE,
                severity=AuditSeverity.MEDIUM,
            ),
        ]

        for check in extra_checks:
            self.audit.checks[check.check_id] = check

    def check_liquidity_providers(self, registry) -> Dict[str, Any]:
        """Run liquidity provider health check."""
        providers = registry.get_all_providers(status="active")
        if not providers:
            return {
                "status": AuditStatus.FAILED,
                "message": "No active liquidity providers",
                "details": {"active_count": 0},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": f"{len(providers)} active liquidity providers",
            "details": {"active_count": len(providers)},
        }

    def check_liquidity_depth(self, converter) -> Dict[str, Any]:
        """Run synthetic liquidity depth check."""
        total = converter.get_total_liquidity()
        total_usd = total.get("total_usd", 0.0)
        min_liquidity = float(os.environ.get("MIN_LIQUIDITY_USD", 10000.0))

        if total_usd < min_liquidity:
            return {
                "status": AuditStatus.FAILED,
                "message": f"Total liquidity ${total_usd:.2f} below minimum ${min_liquidity:.2f}",
                "details": {"total_usd": total_usd, "minimum": min_liquidity},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": f"Total liquidity ${total_usd:.2f} above minimum",
            "details": {"total_usd": total_usd, "by_market": total.get("by_market", {})},
        }

    def check_mark_price_consistency(self, trading_engine, tolerance: float = 0.02) -> Dict[str, Any]:
        """Verify mark prices are close to index prices."""
        inconsistent = []
        for market, state in trading_engine.market_states.items():
            if state.index_price == 0:
                continue
            deviation = abs(state.mark_price - state.index_price) / state.index_price
            if deviation > tolerance:
                inconsistent.append({
                    "market": market,
                    "mark": state.mark_price,
                    "index": state.index_price,
                    "deviation": deviation,
                })

        if inconsistent:
            return {
                "status": AuditStatus.FAILED,
                "message": f"{len(inconsistent)} market(s) with price deviation > {tolerance:.1%}",
                "details": {"inconsistent": inconsistent},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": "Mark prices consistent with index prices",
            "details": {"markets_checked": len(trading_engine.market_states)},
        }

    def check_funding_rate_bounds(self, funding_engine) -> Dict[str, Any]:
        """Verify funding rates within bounds."""
        from funding_rate_engine import FUNDING_CONFIG

        out_of_bounds = []
        for market in funding_engine.trading_engine.market_states:
            rate = funding_engine.calculate_funding_rate(market)
            if rate < FUNDING_CONFIG["min_funding_rate"] or rate > FUNDING_CONFIG["max_funding_rate"]:
                out_of_bounds.append({"market": market, "rate": rate})

        if out_of_bounds:
            return {
                "status": AuditStatus.FAILED,
                "message": f"{len(out_of_bounds)} funding rate(s) out of bounds",
                "details": {"out_of_bounds": out_of_bounds},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": "All funding rates within bounds",
            "details": {"markets_checked": len(funding_engine.trading_engine.market_states)},
        }

    def check_liquidation_backlog(self, liq_system) -> Dict[str, Any]:
        """Check for stuck liquidations."""
        at_risk = liq_system.get_at_risk_positions()
        if len(at_risk) > 10:
            return {
                "status": AuditStatus.WARNING,
                "message": f"{len(at_risk)} positions at risk — possible backlog",
                "details": {"at_risk_count": len(at_risk)},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": f"Liquidation queue healthy ({len(at_risk)} at risk)",
            "details": {"at_risk_count": len(at_risk)},
        }

    def check_orderbook_spread(self, trading_engine, max_spread_bps: float = 50.0) -> Dict[str, Any]:
        """Verify bid-ask spreads are within tolerance."""
        wide_spreads = []
        for market, ob in trading_engine.order_books.items():
            best_bid = ob.get_best_bid()
            best_ask = ob.get_best_ask()
            if best_bid and best_ask and best_bid > 0:
                spread_bps = ((best_ask - best_bid) / best_bid) * 10000
                if spread_bps > max_spread_bps:
                    wide_spreads.append({"market": market, "spread_bps": spread_bps})

        if wide_spreads:
            return {
                "status": AuditStatus.WARNING,
                "message": f"{len(wide_spreads)} market(s) with wide spread",
                "details": {"wide_spreads": wide_spreads},
            }
        return {
            "status": AuditStatus.PASSED,
            "message": "Order book spreads within tolerance",
            "details": {"markets_checked": len(trading_engine.order_books)},
        }

    def run_airmicrodrip_audit(
        self,
        registry=None,
        converter=None,
        trading_engine=None,
        funding_engine=None,
        liq_system=None,
    ) -> Dict[str, Any]:
        """Run the full AirMicroDrip audit suite."""
        ctx: Dict[str, Any] = {}

        if registry:
            ctx["liquidity_providers"] = self.check_liquidity_providers(registry)
        if converter:
            ctx["liquidity_depth"] = self.check_liquidity_depth(converter)
        if trading_engine:
            ctx["price_consistency"] = self.check_mark_price_consistency(trading_engine)
        if funding_engine:
            ctx["funding_bounds"] = self.check_funding_rate_bounds(funding_engine)
        if liq_system:
            ctx["liquidation_backlog"] = self.check_liquidation_backlog(liq_system)
        if trading_engine:
            ctx["orderbook_spread"] = self.check_orderbook_spread(trading_engine)

        # Log all results
        for check_name, result in ctx.items():
            status = result.get("status", AuditStatus.SKIPPED)
            self.audit.log(
                category=AuditCategory.INTEGRITY,
                severity=AuditSeverity.HIGH if status == AuditStatus.FAILED else AuditSeverity.INFO,
                status=status,
                message=result.get("message", f"{check_name} check completed"),
                details=result.get("details", {}),
                actor="airmicrodrip_auditor",
                component=check_name,
            )

        # Run base framework checks too
        base_report = self.audit.run_audit(context=ctx)

        return {
            "base_report_id": base_report.report_id,
            "overall_score": base_report.overall_score,
            "airmicrodrip_checks": ctx,
            "system_health": self.audit.get_system_health(),
        }


if __name__ == "__main__":
    # Standalone demo
    auditor = AirMicroDripAuditor()
    print("AirMicroDrip Auditor initialized with checks:")
    for cid, check in auditor.audit.checks.items():
        print(f"  {cid}: {check.name} ({check.category.value}, {check.severity.value})")
    print(f"\nTotal checks registered: {len(auditor.audit.checks)}")