Spaces:
Sleeping
Sleeping
File size: 2,430 Bytes
4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b 4db8795 99bbd9b | 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 65 66 67 68 | # # from fastapi import APIRouter, HTTPException
# # from app.models import SQLQueryRequest, SQLQueryResponse
# # from app.services.sql_agent import execute_query
# # router = APIRouter()
# # @router.post("/query", response_model=SQLQueryResponse)
# # async def query_database(request: SQLQueryRequest):
# # try:
# # result = execute_query(request.query)
# # return SQLQueryResponse(result=result)
# # except ValueError as e:
# # raise HTTPException(status_code=400, detail=str(e))
# # except Exception as e:
# # raise HTTPException(status_code=500, detail=str(e))
# # app/api/v1/endpoints/sql_query.py
# from fastapi import APIRouter, HTTPException
# from pydantic import BaseModel
# from app.services.sql_agent_instance import sql_agent
# router = APIRouter()
# class SQLQueryRequest(BaseModel):
# query: str
# class SQLQueryResponse(BaseModel):
# result: str
# @router.post("/query", response_model=SQLQueryResponse)
# async def query_database(request: SQLQueryRequest):
# try:
# result = sql_agent.execute_query(request.query)
# return SQLQueryResponse(result=result)
# except ValueError as e:
# raise HTTPException(status_code=400, detail=str(e))
# except Exception as e:
# raise HTTPException(status_code=500, detail=str(e))
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from app.services.sql_agent_instance import sql_agent
from typing import Optional
import uuid
router = APIRouter()
class SQLQueryRequest(BaseModel):
query: str
thread_id: Optional[str] = None
class SQLQueryResponse(BaseModel):
result: str
thread_id: str ## client can use this to continue the conversation
@router.post("/query", response_model=SQLQueryResponse)
async def query_database(request: SQLQueryRequest):
try:
## generate if not provided thread id
thread_id = request.thread_id or str(uuid.uuid4())
## add debug
print(f"Thread ID: {thread_id}, Query: {request.query}")
result = sql_agent.execute_query(request.query, config={"configurable": {"thread_id": thread_id}})
print(f"Result: {result}")
return SQLQueryResponse(result=result, thread_id=thread_id)
except ValueError as e:
raise HTTPException(status_code=400, detail=str(e))
except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) |