-
Notifications
You must be signed in to change notification settings - Fork 2
/
suggestion_api.py
64 lines (54 loc) · 2.2 KB
/
suggestion_api.py
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
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import List, Dict
from suggestion_db import SuggestionDB
app = FastAPI()
db = SuggestionDB()
class Suggestion(BaseModel):
file: str
question: str
response: Dict
model: str
class SuggestionResponse(BaseModel):
id: int
file: str
question: str
response: Dict
model: str
timestamp: str
@app.post("/suggestions/", response_model=SuggestionResponse)
def create_suggestion(suggestion: Suggestion):
db.add_suggestion(suggestion.file, suggestion.question, suggestion.response, suggestion.model)
return db.get_suggestions(suggestion.file)[0]
@app.get("/suggestions/", response_model=List[SuggestionResponse])
def read_suggestions(file: str = None):
return db.get_suggestions(file)
@app.get("/suggestions/{suggestion_id}", response_model=SuggestionResponse)
def read_suggestion(suggestion_id: int):
suggestion = db.get_suggestion(suggestion_id)
if suggestion is None:
raise HTTPException(status_code=404, detail="Suggestion not found")
return suggestion
@app.put("/suggestions/{suggestion_id}", response_model=SuggestionResponse)
def update_suggestion(suggestion_id: int, response: Dict):
suggestion = db.get_suggestion(suggestion_id)
if suggestion is None:
raise HTTPException(status_code=404, detail="Suggestion not found")
db.update_suggestion(suggestion_id, response)
return db.get_suggestion(suggestion_id)
@app.delete("/suggestions/{suggestion_id}", response_model=Dict)
def delete_suggestion(suggestion_id: int):
suggestion = db.get_suggestion(suggestion_id)
if suggestion is None:
raise HTTPException(status_code=404, detail="Suggestion not found")
return suggestion
@app.post("/suggestions/{suggestion_id}/confirm_delete", response_model=Dict)
def confirm_delete_suggestion(suggestion_id: int):
suggestion = db.get_suggestion(suggestion_id)
if suggestion is None:
raise HTTPException(status_code=404, detail="Suggestion not found")
db.delete_suggestion(suggestion_id)
return {"message": "Suggestion deleted successfully"}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8000)