Spaces:
Sleeping
Sleeping
File size: 1,119 Bytes
e6b0b52 8132b6d e6b0b52 |
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 |
from fastapi import FastAPI, Request
from pydantic import BaseModel
from infinity_ai import AI
from fastapi.middleware.cors import CORSMiddleware
# Initialize the AI client
client = AI()
# FastAPI app
app = FastAPI()
# CORS Middleware (so JS from browser can access it too)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Change "*" to your frontend URL for better security
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Request body model
class Question(BaseModel):
question: str
@app.post("/ask")
async def ask(question: Question):
try:
response = client.chat.completions.create(
model="Orion",
messages=[
{"role": "user", "content": question.question},
{"role": "system", "content": "You are a helpful AI assistant created by abdullah ali who is very intelegent and he is 13 years old and live in lahore."}
],
web_search=False
)
return {"answer": response.choices[0].message.content}
except Exception as e:
return {"error": str(e)}
|