File size: 880 Bytes
c0ed0f9 d9d3f03 e85eecf 69f1eb2 e85eecf 69f1eb2 e85eecf 69f1eb2 c0ed0f9 | 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 | from fastapi import FastAPI, Request
import requests
import os
app = FastAPI()
# OpenRouter API key
OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
# Root route
@app.get("/")
def home():
return {"message": "Molbot is running 🚀"}
# Chat route
@app.post("/chat")
async def chat(request: Request):
data = await request.json()
user_message = data.get("message")
if not user_message:
return {"error": "No message provided"}
url = "https://openrouter.ai/api/v1/chat/completions"
headers = {
"Authorization": f"Bearer {OPENROUTER_API_KEY}",
"Content-Type": "application/json"
}
payload = {
"model": "deepseek/deepseek-chat",
"messages": [
{"role": "user", "content": user_message}
]
}
response = requests.post(url, headers=headers, json=payload)
return response.json() |