File size: 498 Bytes
82276a6
561dc24
82276a6
5b90a2e
 
82276a6
 
 
 
 
 
 
5b90a2e
82276a6
5b90a2e
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from fastapi import FastAPI, Depends
import os
import auth
import client
import model

app = FastAPI()

@app.get("/")
async def welcome():
    return "Hello, Welcome to AI space!"

@app.get("/test")
async def generate(api_key = Depends(auth.api_key_auth)):
    return "All good. You only get this message if you're authenticated"

@app.post("/generate")
async def generate(ticket: model.Ticket, api_key = Depends(auth.api_key_auth)):
    response = client.get_completion(ticket)
    return response