Updated app.py

app.py

@@ -764,6 +764,9 @@ def build_onyx_payload(messages, model_provider, model_version, temperature, cha
         "include_citations": False
     }
 
+    # Remove keys with None values — Onyx rejects/errors on null parent_message_id
+    payload = {k: v for k, v in payload.items() if v is not None}
+
     return payload
 
 
@@ -858,6 +861,9 @@ def build_anthropic_payload_from_messages(messages, system_prompt, model_provide
         "include_citations": False
     }
 
+    # Remove keys with None values — Onyx rejects/errors on null parent_message_id
+    payload = {k: v for k, v in payload.items() if v is not None}
+
     return payload
 
 
@@ -1011,7 +1017,7 @@ def stream_onyx_response(payload, model, session_key, has_tools=False):
                         if msg_id:
                             last_message_id = msg_id
 
-                        if content and packet_type in ['content', 'legacy', 'raw']:
+                        if content and packet_type in ['content', 'legacy', 'raw', 'error']:
                             if has_tools:
                                 # Buffer content for tool call detection
                                 full_content_buffer += content
@@ -1093,14 +1099,8 @@ def collect_full_response(payload, model, session_key, has_tools=False):
 
                 if response.status_code != 200:
                     error_text = response.text
-                    print(f"Error response: {error_text}")
-                    return {
-                        "error": {
-                            "message": f"Onyx API error {response.status_code}: {error_text}",
-                            "type": "api_error",
-                            "code": response.status_code
-                        }
-                    }, response.status_code
+                    print(f"Error response from {url}: {response.status_code} - {error_text}")
+                    continue  # Try next endpoint
 
                 if not is_streaming_request:
                     try:
@@ -1401,13 +1401,8 @@ def collect_anthropic_full_response(payload, model, session_key, has_tools=False
                     continue
 
                 if response.status_code != 200:
-                    return {
-                        "type": "error",
-                        "error": {
-                            "type": "api_error",
-                            "message": f"Onyx API error {response.status_code}: {response.text}"
-                        }
-                    }, response.status_code
+                    print(f"Error response from {url}: {response.status_code} - {response.text}")
+                    continue  # Try next endpoint
 
                 if not is_streaming_request:
                     try: