Update app/arbitrage_analyzer.py

app/arbitrage_analyzer.py

@@ -1,7 +1,6 @@
 """
-The Oracle Discrepancy Analyzer.
-Uses Gemini to provide risk assessment and strategic plans for
-discrepancies between the Pyth and Chainlink oracles.
+The Multi-Asset Oracle Discrepancy Analyzer.
+Uses Gemini to provide context-aware risk assessment for various assets.
 """
 import os
 import logging
@@ -21,12 +20,12 @@ class ArbitrageAnalyzer:
         self.params = {"key": self.api_key}
         self.headers = {"Content-Type": "application/json"}
 
-    def _build_prompt(self, opportunity: Dict) -> dict:
+    def _build_prompt(self, asset_symbol: str, opportunity: Dict) -> dict:
         return {
             "contents": [{
                 "parts": [{
                     "text": f"""
-                    You are a high-frequency DeFi strategist. A price dislocation for Bitcoin (BTC) has been detected between the Pyth and Chainlink oracle networks.
+                    You are a high-frequency DeFi strategist. A price dislocation for {asset_symbol} has been detected between the Pyth and Chainlink oracle networks.
                     Provide a concise "Alpha Briefing" as a single, minified JSON object with NO markdown formatting.
 
                     The JSON object must have these exact keys: "risk", "strategy", "rationale".
@@ -35,7 +34,7 @@ class ArbitrageAnalyzer:
                     - "strategy": A brief, one-sentence action plan.
                     - "rationale": A short explanation for the risk assessment.
 
-                    DISLOCATION DETAILS:
+                    DISLOCATION DETAILS for {asset_symbol}:
                     - Pyth Network Price: ${opportunity['pyth_price']:,.2f}
                     - Chainlink Aggregated Price: ${opportunity['chainlink_price']:,.2f}
                     - Discrepancy: {opportunity['spread_pct']:.3f}%
@@ -44,8 +43,8 @@ class ArbitrageAnalyzer:
             }]
         }
 
-    async def get_alpha_briefing(self, opportunity: Dict) -> Optional[Dict]:
-        prompt = self._build_prompt(opportunity)
+    async def get_alpha_briefing(self, asset_symbol: str, opportunity: Dict) -> Optional[Dict]:
+        prompt = self._build_prompt(asset_symbol, opportunity)
         try:
             response = await self.client.post(self.API_URL, json=prompt, params=self.params, headers=self.headers, timeout=20)
             response.raise_for_status()
@@ -54,5 +53,5 @@ class ArbitrageAnalyzer:
                 content = content.strip("```json\n")
             return json.loads(content)
         except Exception as e:
-            logger.error(f"❌ Gemini Alpha Briefing Error: {e}")
+            logger.error(f"❌ Gemini Alpha Briefing Error for {asset_symbol}: {e}")
             return None