Update requests_service.py

requests_service.py

@@ -1,40 +1,42 @@
 import requests
 import os
+import sys
 
+# Fetch the key from HF Secrets
+ONYX_API_KEY = os.getenv("PR_KEY")
 ONYX_BASE_URL = "https://cloud.onyx.app/api"
-# Token should be stored in HF Secrets as ONYX_API_KEY
-ONYX_API_KEY = os.getenv("ONYX_API_KEY")
 
 def get_headers():
+    # If your key already starts with 'Bearer', this logic will break. 
+    # This assumes ONYX_API_KEY is JUST the token.
     return {
         "Authorization": f"Bearer {ONYX_API_KEY}",
         "Content-Type": "application/json"
     }
 
-def get_provider_mapping(m: str):
-    """Maps model name to (Provider, Version)"""
-    if m == "gpt-5.2":
-        return "OpenAI", m
-    if m in ("claude-opus-4.5", "claude-sonnet-4.5"):
-        return "Anthropic", m
-    if m in ("gemini-3-pro-preview", "gemini-3-flash-preview"):
-        return "Google Vertex AI", m
-    return "OpenRouter", m
-
 def create_onyx_session():
     url = f"{ONYX_BASE_URL}/chat/create-chat-session"
-    payload = {"persona_id": 0, "description": "HF Bridge Session"}
+    # Ensure persona_id is 0 (default) or a valid ID from your Onyx dashboard
+    payload = {"persona_id": 0, "description": "Hugging Face Bridge"}
+    
     try:
+        print(f"DEBUG: Attempting session create with key ending in ...{ONYX_API_KEY[-5:] if ONYX_API_KEY else 'NONE'}")
         resp = requests.post(url, json=payload, headers=get_headers(), timeout=10)
-        return resp.json().get("id") if resp.status_code == 200 else None
-    except:
+        
+        if resp.status_code == 200:
+            return resp.json().get("id")
+        
+        # This will print to your Hugging Face Logs
+        print(f"ONYX AUTH REJECTION: {resp.status_code} - {resp.text}", file=sys.stderr)
+        return None
+    except Exception as e:
+        print(f"CONNECTION ERROR: {str(e)}", file=sys.stderr)
         return None
 
-def send_onyx_message(message, model_name, session_id):
+def send_onyx_message(message, model_name, session_id, provider, version):
     url = f"{ONYX_BASE_URL}/chat/send-chat-message"
-    provider, version = get_provider_mapping(model_name)
     
-    # Your exact payload structure with dynamic overrides
+    # EXACT payload structure you requested
     payload = {
         "message": message,
         "llm_override": {
@@ -44,9 +46,7 @@ def send_onyx_message(message, model_name, session_id):
         },
         "allowed_tool_ids": [],
         "file_descriptors": [],
-        "internal_search_filters": {
-            "source_type": ["ingestion_api"]
-        },
+        "internal_search_filters": {"source_type": ["ingestion_api"]},
         "deep_research": False,
         "origin": "unset",
         "parent_message_id": None,