from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import openai
app = FastAPI(title="AI Service")
class ChatRequest(BaseModel):
message: str
model: str = "gpt-4o-mini"
@app.post("/chat")
async def chat(request: ChatRequest):
try:
response = openai.ChatCompletion.create(
model=request.model,
messages=[{"role": "user", "content": request.message}]
)
return {"response": response.choices[0].message.content}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.get("/health")
async def health():
return {"status": "ok"}