Spaces:
Sleeping
Sleeping
File size: 2,404 Bytes
9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b 9fcbecf 0fd0c7b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import requests
import json
import os
import uvicorn
# Configuration
TOKEN = os.getenv("POLLINATIONS_TOKEN")
POLLINATIONS_URL = "https://text.pollinations.ai/openai"
app = FastAPI(title="Pollinations OpenAI API")
# CORS
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
async def root():
return {
"message": "Pollinations OpenAI Compatible API",
"status": "running",
"endpoints": {
"models": "/models",
"chat": "/chat/completions"
}
}
@app.get("/models")
async def get_models():
"""Models endpoint - यहीं missing था!"""
return {
"object": "list",
"data": [
{"id": "openai", "object": "model", "owned_by": "pollinations"},
{"id": "llama", "object": "model", "owned_by": "pollinations"},
{"id": "mistral", "object": "model", "owned_by": "pollinations"},
{"id": "gemini", "object": "model", "owned_by": "pollinations"},
{"id": "deepseek", "object": "model", "owned_by": "pollinations"}
]
}
@app.post("/chat/completions")
async def chat_completions(data: dict):
"""Chat endpoint"""
model = data.get("model", "openai")
messages = data.get("messages", [])
temperature = data.get("temperature", 0.7)
payload = {
"model": model,
"messages": messages,
"temperature": temperature,
"referrer": "hf-api"
}
headers = {"Content-Type": "application/json"}
if TOKEN:
headers["Authorization"] = f"Bearer {TOKEN}"
try:
response = requests.post(POLLINATIONS_URL, json=payload, headers=headers, timeout=60)
result = response.json()
return {
"id": "chatcmpl-123",
"object": "chat.completion",
"model": model,
"choices": [{
"index": 0,
"message": {
"role": "assistant",
"content": result["choices"][0]["message"]["content"]
},
"finish_reason": "stop"
}]
}
except Exception as e:
return {"error": str(e)}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=7860)
|