schema feature endpoint added

app/api/v1/endpoints/database_connection.py

@@ -47,7 +47,6 @@ from pydantic import BaseModel
 from app.services.sql_agent_instance import sql_agent
 from sqlalchemy.exc import OperationalError, DatabaseError
 from urllib.parse import urlparse
-
 router = APIRouter()
 
 class DatabaseConnectionRequest(BaseModel):
@@ -85,4 +84,6 @@ async def setup_connection(request: DatabaseConnectionRequest):
         raise HTTPException(
             status_code=500,
             detail=f"Unexpected error occurred while setting up database connection: {str(e)}"
-        )
+        )
+
+

app/api/v1/endpoints/schema.py

@@ -0,0 +1,43 @@
+from fastapi import APIRouter, HTTPException 
+from sqlalchemy import inspect 
+from app.services.sql_agent_instance import sql_agent
+
+
+router = APIRouter()
+
+@router.get("/schema")
+async def get_schema():
+    if not sql_agent.db: 
+        raise HTTPException(
+            status_code=400,
+            detail="Database connection is not established. Please set up the connection first."
+        )
+    try: 
+        inspector = inspect(sql_agent.db._engine)
+        schema_data = {}
+        for table_name in inspector.get_table_names():
+            columns = []
+            for col in inspector.get_columns(table_name):
+                columns.append({
+                    "name": col['name'],
+                    "type": str(col['type']),
+                    "nullable": col.get('nullable', True)
+                })
+            fks = []
+            for fk in inspector.get_foreign_keys(table_name):
+                fks.append({
+                    "constrained_columns": fk['constrained_columns'],
+                    "referred_table": fk['referred_table'],
+                    "referred_columns": fk['referred_columns']
+                })
+            schema_data[table_name] = {
+                "columns": columns,
+                "foreign_keys": fks
+            }
+        return schema_data
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail="An error occurred while fetching the schema."
+        )
+