change python version

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
+from fastapi import FastAPI,Req
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 import uvicorn
@@ -7,6 +7,7 @@ from query_engine import set_query_engine
 from llama_index.core.indices.struct_store import NLSQLTableQueryEngine
 import os
 from huggingface_hub import hf_hub_download
+from models import SQL
 
 app = FastAPI()
 
@@ -31,10 +32,10 @@ def startup():
     query_file_path = hf_hub_download(repo_id=dataset_name, filename=duckdb_filename, repo_type="dataset")
 
 @app.post("/sql")
-async def query_database(query:str):
+async def query_database(query_data: SQL):
     global query_file_path
     conn = duckdb.connect(query_file_path,read_only=True)
-    df = conn.execute(query).fetch_df()
+    df = conn.execute(query_data.query).fetch_df()
     return JSONResponse(df.to_json(orient = "records"))
 
 # @app.post("/llm")

models/__init__.py

@@ -0,0 +1,5 @@
+from .sql_model import SQL
+
+__all__ = [
+    "SQL"
+]

models/sql_model.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class SQL(BaseModel):
+    query:str