Update app.py

app.py

@@ -289,7 +289,7 @@ class ChatRequest(BaseModel):
     history: List[Any] = []
 
 # =====================================================
-# Claude Desktop / Claude Code Compatible Endpoints
+# VERIFY API KEY
 # =====================================================
 
 def verify_api_key(request: Request):
@@ -308,6 +308,38 @@ def verify_api_key(request: Request):
 
     raise HTTPException(status_code=401, detail="Invalid API key. Use 'Authorization: Bearer KEY' or 'X-API-Key: KEY'")
 
+# =====================================================
+# CLAUDE-DESKTOP / CLAUDE-CODE COMPATIBLE ENDPOINTS
+# =====================================================
+
+@app.get("/v1/models")
+async def v1_models(request: Request):
+    """Anthropic API compatible models endpoint - NO AUTH REQUIRED"""
+    # Claude Desktop يطلب هذا أولاً بدون مفتاح
+    
+    models = []
+    for pname, pobj in REAL_PROVIDERS.items():
+        if pname not in _PROVIDER_MODEL_CACHE:
+            _PROVIDER_MODEL_CACHE[pname] = discover_provider_models(pobj, pname)
+        for model in _PROVIDER_MODEL_CACHE[pname][:5]:
+            models.append({
+                "id": model,
+                "type": "model",
+                "display_name": f"{pname} - {model}"
+            })
+    
+    # نماذج افتراضية في حال لم يتم العثور على شيء
+    if not models:
+        models = [
+            {"id": "sonar", "type": "model", "display_name": "Perplexity - Sonar"},
+            {"id": "sonar-pro", "type": "model", "display_name": "Perplexity - Sonar Pro"},
+            {"id": "gpt-4o", "type": "model", "display_name": "GPT-4o"},
+            {"id": "claude-3", "type": "model", "display_name": "Claude 3"},
+            {"id": "deepseek-chat", "type": "model", "display_name": "DeepSeek Chat"}
+        ]
+    
+    return {"data": models}
+
 @app.post("/v1/messages")
 async def v1_messages(request: Request):
     """Anthropic API compatible endpoint for Claude Desktop / Claude Code"""
@@ -315,7 +347,6 @@ async def v1_messages(request: Request):
     
     body = await request.json()
     
-    # استخراج الرسالة من تنسيق Anthropic
     messages = body.get("messages", [])
     if not messages:
         raise HTTPException(status_code=400, detail="No messages provided")
@@ -323,30 +354,23 @@ async def v1_messages(request: Request):
     last_message = messages[-1]
     user_message = last_message.get("content", "")
     
-    # استخراج النموذج
     model = body.get("model", "sonar")
-    
-    # استخراج النظام (system prompt) إن وجد
     system_prompt = body.get("system", "")
     
-    # بناء التاريخ من الرسائل السابقة
     history = []
     for msg in messages[:-1]:
         role = msg.get("role", "user")
         content = msg.get("content", "")
         history.append({"role": role, "content": content})
     
-    # إضافة system prompt إذا موجود
     full_message = user_message
     if system_prompt:
         full_message = f"[System: {system_prompt}]\n\n{user_message}"
     
-    # الحصول على الرد
     full_response = ""
     for chunk in ask(full_message, history, "Perplexity", model):
         full_response = chunk
     
-    # إرجاع الرد بتنسيق Anthropic API
     return {
         "id": f"msg_{int(time.time())}_{os.urandom(4).hex()}",
         "type": "message",
@@ -374,18 +398,26 @@ async def v1_messages_stream(request: Request):
     last_message = messages[-1]
     user_message = last_message.get("content", "")
     model = body.get("model", "sonar")
+    system_prompt = body.get("system", "")
+    
+    full_message = user_message
+    if system_prompt:
+        full_message = f"[System: {system_prompt}]\n\n{user_message}"
     
     async def generate_stream():
         message_id = f"msg_{int(time.time())}_{os.urandom(4).hex()}"
         
-        # رسالة البدء
+        # Message start
         yield f"event: message_start\ndata: {{\"message\": {{\"id\": \"{message_id}\", \"type\": \"message\", \"role\": \"assistant\", \"content\": [], \"model\": \"{model}\", \"stop_reason\": null, \"stop_sequence\": null, \"usage\": {{\"input_tokens\": 0, \"output_tokens\": 0}}}}}}\n\n"
         
-        # رسالة التدفق
-        for chunk in ask(user_message, [], "Perplexity", model):
+        # Content block start
+        yield f"event: content_block_start\ndata: {{\"type\": \"content_block_start\", \"index\": 0, \"content_block\": {{\"type\": \"text\", \"text\": \"\"}}}}\n\n"
+        
+        # Stream deltas
+        for chunk in ask(full_message, [], "Perplexity", model):
             yield f"event: content_block_delta\ndata: {{\"type\": \"content_block_delta\", \"index\": 0, \"delta\": {{\"type\": \"text_delta\", \"text\": {json.dumps(chunk, ensure_ascii=False)}}}}}\n\n"
         
-        # رسالة النهاية
+        # Message end
         yield f"event: message_delta\ndata: {{\"type\": \"message_delta\", \"delta\": {{\"stop_reason\": \"end_turn\", \"stop_sequence\": null}}, \"usage\": {{\"output_tokens\": 100}}}}\n\n"
         yield f"event: message_stop\ndata: {{}}\n\n"
     
@@ -399,27 +431,8 @@ async def v1_messages_stream(request: Request):
         }
     )
 
-@app.get("/v1/models")
-async def v1_models(request: Request):
-    """Anthropic API compatible models endpoint"""
-    verify_api_key(request)
-    
-    models = []
-    for pname, pobj in REAL_PROVIDERS.items():
-        if pname not in _PROVIDER_MODEL_CACHE:
-            _PROVIDER_MODEL_CACHE[pname] = discover_provider_models(pobj, pname)
-        for model in _PROVIDER_MODEL_CACHE[pname]:
-            models.append({
-                "id": model,
-                "type": "model",
-                "display_name": f"{pname} - {model}",
-                "created_at": "2024-01-01T00:00:00Z"
-            })
-    
-    return {"data": models}
-
 # =====================================================
-# ORIGINAL ENDPOINTS (للتوافق مع الواجهة القديمة)
+# ORIGINAL ENDPOINTS (للواجهة القديمة)
 # =====================================================
 
 @app.get("/")
@@ -429,9 +442,9 @@ async def root():
         "endpoints": {
             "GET /": "هذه الصفحة",
             "GET /health": "التحقق من صحة الخادم",
-            "GET /v1/models": "قائمة النماذج (Claude-compatible)",
-            "POST /v1/messages": "إرسال رسالة (Claude-compatible)",
-            "POST /v1/messages/stream": "إرسال رسالة متدفق (Claude-compatible)",
+            "GET /v1/models": "قائمة النماذج (Claude-compatible - NO AUTH)",
+            "POST /v1/messages": "إرسال رسالة (Claude-compatible - REQUIRES AUTH)",
+            "POST /v1/messages/stream": "إرسال رسالة متدفق (Claude-compatible - REQUIRES AUTH)",
             "GET /providers": "قائمة المزودين (يتطلب مفتاح)",
             "POST /chat": "إرسال رسالة (legacy)",
             "POST /chat/stream": "إرسال رسالة متدفق (legacy)"