CSPON-TSC/landsite_agent

SQL生成器Web应用

这是一个基于FastAPI、OpenAI和LangChain的SQL生成器Web应用，可以根据自然语言描述生成SQL查询语句，执行查询并返回分析结果。

功能特点

• 基于自然语言生成SQL查询
• 自动执行SQL查询
• 数据可视化展示
• RESTful API接口

环境要求

• Python 3.10+
• PostgreSQL数据库
• OpenAI API密钥

安装步骤

1. 克隆项目并安装依赖：

   pip install -r requirements.txt
   

   1. 配置环境变量： 创建.env文件并设置以下变量： ```

OPENAI_API_KEY=your_openai_api_key DATABASE_URL=postgresql://username:password@localhost:5432/your_database

3. 启动应用：
```bash
python main.py

API使用

生成SQL并执行查询

POST /generate-sql

请求体：

{
    "description": "查询描述"
}

响应：

{
    "sql": "生成的SQL语句",
    "data": "查询结果数据",
    "visualization": "可视化数据（如果有）"
}

示例

import requests

response = requests.post(
    "http://localhost:8000/generate-sql",
    json={
        "description": "查询所有已上架的公告地块"
    }
)
print(response.json())