Spaces:
Sleeping
Sleeping
updated
Browse files- backend/app/api/endpoints.py +1 -21
backend/app/api/endpoints.py
CHANGED
|
@@ -22,7 +22,7 @@ class QueryRequest(BaseModel):
|
|
| 22 |
|
| 23 |
from fastapi import HTTPException
|
| 24 |
|
| 25 |
-
@router.post("/
|
| 26 |
def ask_query(req: QueryRequest):
|
| 27 |
try:
|
| 28 |
sql = generate_sql_query(req.question)
|
|
@@ -60,23 +60,3 @@ def ask_query(req: QueryRequest):
|
|
| 60 |
except Exception as e:
|
| 61 |
raise HTTPException(status_code=400, detail=f"An error occurred: {str(e)}")
|
| 62 |
|
| 63 |
-
@router.get("/schema")
|
| 64 |
-
def read_schema():
|
| 65 |
-
return get_schema()
|
| 66 |
-
|
| 67 |
-
@router.get("/query")
|
| 68 |
-
def run_query(question: str):
|
| 69 |
-
try:
|
| 70 |
-
sql_query = generate_sql_query(question)
|
| 71 |
-
results = execute_sql_query(sql_query)
|
| 72 |
-
return {
|
| 73 |
-
"question": question,
|
| 74 |
-
"sql_query": sql_query,
|
| 75 |
-
"results": results
|
| 76 |
-
}
|
| 77 |
-
except Exception as e:
|
| 78 |
-
raise HTTPException(status_code=400, detail=f"An error occurred: {str(e)}")
|
| 79 |
-
|
| 80 |
-
@router.get("/")
|
| 81 |
-
def root():
|
| 82 |
-
return {"message": "Welcome to AI SQL Query Generator!"}
|
|
|
|
| 22 |
|
| 23 |
from fastapi import HTTPException
|
| 24 |
|
| 25 |
+
@router.post("/process-text")
|
| 26 |
def ask_query(req: QueryRequest):
|
| 27 |
try:
|
| 28 |
sql = generate_sql_query(req.question)
|
|
|
|
| 60 |
except Exception as e:
|
| 61 |
raise HTTPException(status_code=400, detail=f"An error occurred: {str(e)}")
|
| 62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|