Update app.py

app.py

@@ -1,18 +1,20 @@
 #!/usr/bin/env python3
 #
-#  san_integration_script.py (v6 - Simplified Audio Processing)
+#  san_integration_script.py (v6 - Simplified Audio Processing - Fixed)
 #  ============================================================
 #  Description:
 #  - Establishes a real-time, two-way audio bridge between a SAN system
 #    and the Millis AI platform.
 #  - Direct audio forwarding at 16kHz PCM format (no conversion needed).
 #  - Removed buffer logic for direct streaming.
+#  - Fixed credential extraction logic.
 #
 #  Changes in this version:
 #  - Removed audio format conversion logic (resampling)
 #  - Removed buffer management and chunking
 #  - Simplified audio flow for direct 16kHz PCM streaming
 #  - Removed scipy dependency
+#  - Fixed comprehensive credential extraction fallback
 #  -------------------------------------------------------------------
 
 import os
@@ -185,10 +187,13 @@ async def fetch_call_credentials_with_fallback(call_id: str, fallback_msg: dict)
     extra_params = fallback_msg.get("extraParams", {})
     custom_field = fallback_msg.get("custom_field", {})
     
+    # Comprehensive agent_id extraction
     agent_id = (extra_params.get("platform_agent_id") or 
                 custom_field.get("agentId") or 
-                fallback_msg.get("agentId"))
+                fallback_msg.get("agentId") or
+                extra_params.get("agentId"))
     
+    # Comprehensive public_key extraction
     public_key = (extra_params.get("publicKey") or extra_params.get("public_key") or 
                   custom_field.get("publicKey") or custom_field.get("public_key") or 
                   fallback_msg.get("publicKey") or fallback_msg.get("public_key"))
@@ -407,10 +412,24 @@ async def media_socket(ws: WebSocket):
                 if processor is None:
                     logger.info(f"Starting processor for call: {new_call_id}")
                     
-                    # Extract credentials
+                    # Extract credentials with comprehensive fallback
                     extra_params = msg.get("extraParams", {})
-                    agent_id = extra_params.get("platform_agent_id")
-                    public_key = PUBLIC_KEY
+                    custom_field = msg.get("custom_field", {})
+                    
+                    # Try multiple sources for agent_id (restored original logic)
+                    agent_id = (extra_params.get("platform_agent_id") or 
+                               custom_field.get("agentId") or 
+                               msg.get("agentId") or
+                               extra_params.get("agentId"))
+                    
+                    # Try multiple sources for public_key
+                    public_key = (PUBLIC_KEY or 
+                                 extra_params.get("publicKey") or 
+                                 extra_params.get("public_key") or 
+                                 custom_field.get("publicKey") or 
+                                 custom_field.get("public_key") or 
+                                 msg.get("publicKey") or 
+                                 msg.get("public_key"))
                     
                     if not agent_id or not public_key:
                         credentials = await fetch_call_credentials_with_fallback(new_call_id, msg)
@@ -419,6 +438,7 @@ async def media_socket(ws: WebSocket):
                     
                     if not agent_id:
                         logger.error(f"No agent_id found for call_id: {new_call_id}")
+                        logger.error("Tried: extraParams.platform_agent_id, custom_field.agentId, msg.agentId, extraParams.agentId, MongoDB")
                         await ws.close(code=1008, reason="Missing agent_id")
                         return
                     
@@ -548,7 +568,7 @@ async def test_mongodb():
         }
 
 if __name__ == "__main__":
-    print("Starting SAN to Millis AI Integration Server (v6 - Simplified)...")
+    print("Starting SAN to Millis AI Integration Server (v6 - Simplified - Fixed)...")
     logger.info("=== SERVER STARTING ===")
     logger.info(f"Public Key: {PUBLIC_KEY[:10] + '...' if PUBLIC_KEY else 'NOT_SET'}")
     logger.info(f"Millis URI: {MILLIS_WS_URI}")