Spaces:
Sleeping
Sleeping
Download app/utils/helpers.py from arthikrangan/text-to-sql-agent-api: direct link, hf CLI and curl.
- Browser
- Download file 708 Bytes
-
https://huggingface.co/spaces/arthikrangan/text-to-sql-agent-api/resolve/main/app/utils/helpers.py
- Command line
-
hf download hf://spaces/arthikrangan/text-to-sql-agent-api/app/utils/helpers.py
-
curl -L -o helpers.py https://huggingface.co/spaces/arthikrangan/text-to-sql-agent-api/resolve/main/app/utils/helpers.py
708 Bytes
| from typing import Any, Dict, List | |
| import pandas as pd | |
| def process_data(data: List[Dict[str, Any]]) -> pd.DataFrame: | |
| """Convert list of dictionaries to pandas DataFrame.""" | |
| return pd.DataFrame(data) | |
| def generate_sql_query(natural_language_query: str) -> str: | |
| """Convert natural language query to SQL query.""" | |
| # TODO: Implement query generation logic | |
| return "SELECT * FROM data LIMIT 10" | |
| def format_response(data: pd.DataFrame) -> Dict[str, Any]: | |
| """Format DataFrame response for API.""" | |
| return { | |
| "columns": data.columns.tolist(), | |
| "data": data.to_dict(orient="records"), | |
| "summary": {"row_count": len(data), "column_count": len(data.columns)}, | |
| } | |