update in schema

app/api/v1/endpoints/schema.py

@@ -46,7 +46,7 @@ async def get_schema():
             detail="An error occurred while fetching the schema."
         )
     
-router.post("schema/indexing")
+@router.post("schema/indexing")
 async def update_indexing(request: TableIndexingRequest):
     if not sql_agent.db:
         raise HTTPException(

app/api/v1/endpoints/sql_query.py

@@ -42,6 +42,7 @@ from pydantic import BaseModel
 from app.services.sql_agent_instance import sql_agent
 from typing import Optional
 import uuid
+from langchain_core.messages import AIMessage
 router = APIRouter()
 
 class SQLQueryRequest(BaseModel):
@@ -60,9 +61,11 @@ async def query_database(request: SQLQueryRequest):
         ## 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))
+        raise HTTPException(status_code=500, detail=str(e))
+