如何用DeepSeek获取数据库中的表信息(表名和字段名称)

-- 获取所有表名
SELECT table_name 
FROM information_schema.tables 
WHERE table_schema = 'public';

-- 获取指定表的字段信息
SELECT column_name, data_type 
FROM information_schema.columns 
WHERE table_name = 'your_table_name';

-- 获取所有表名
SHOW TABLES;

-- 获取指定表的字段信息
DESCRIBE your_table_name;

-- 获取所有表名
SELECT name 
FROM sqlite_master 
WHERE type='table';

-- 获取指定表的字段信息
PRAGMA table_info(your_table_name);

import requests

# 本地 DeepSeek 服务地址
DEESEEK_API_URL = "http://localhost:11434/api/generate"

def generate_sql_query(prompt):
    """
    调用 DeepSeek 模型生成 SQL 查询语句
    :param prompt: 自然语言描述
    :return: DeepSeek 生成的 SQL 查询语句
    """
    payload = {
        "model": "deepseek-r1:1.5b",
        "prompt": prompt
    }
    headers = {"Content-Type": "application/json"}

    try:
        response = requests.post(DEESEEK_API_URL, json=payload, headers=headers)
        response.raise_for_status()
        result = response.json()
        return result.get("response", "").strip()
    except requests.exceptions.RequestException as e:
        print(f"调用 DeepSeek API 失败: {e}")
        return None

# 示例：生成查询所有表名的 SQL
prompt = "生成一个查询所有表名的 SQL 语句（PostgreSQL）"
sql_query = generate_sql_query(prompt)
print("生成的 SQL 查询语句：")
print(sql_query)

SELECT table_name FROM information_schema.tables WHERE table_schema = 'public';

import psycopg2

def execute_query(database_config, query):
    """
    连接 PostgreSQL 数据库并执行查询
    :param database_config: 数据库配置字典（包含 host, dbname, user, password）
    :param query: SQL 查询语句
    :return: 查询结果（列表形式）
    """
    try:
        # 建立数据库连接
        conn = psycopg2.connect(
            host=database_config["host"],
            dbname=database_config["dbname"],
            user=database_config["user"],
            password=database_config["password"]
        )
        cursor = conn.cursor()

        # 执行查询
        cursor.execute(query)
        results = cursor.fetchall()

        # 关闭连接
        cursor.close()
        conn.close()

        return results
    except Exception as e:
        print(f"数据库查询失败：{e}")
        return None

# 示例：查询所有表名
database_config = {
    "host": "localhost",
    "dbname": "test_db",
    "user": "postgres",
    "password": "password"
}
query = "SELECT table_name FROM information_schema.tables WHERE table_schema = 'public';"
tables = execute_query(database_config, query)
print("数据库中的表名：")
print(tables)

def get_table_fields(database_config, table_name):
    """
    获取指定表的字段信息
    :param database_config: 数据库配置字典
    :param table_name: 表名
    :return: 字段信息（列表形式）
    """
    query = f"SELECT column_name, data_type FROM information_schema.columns WHERE table_name = '{table_name}';"
    fields = execute_query(database_config, query)
    return fields

# 示例：获取表 users 的字段信息
table_name = "users"
fields = get_table_fields(database_config, table_name)
print(f"表 {table_name} 的字段信息：")
for field in fields:
    print(f"- 字段名：{field[0]}，数据类型：{field[1]}")

def describe_table_in_natural_language(table_name, fields):
    """
    将表结构信息转换为自然语言描述
    :param table_name: 表名
    :param fields: 字段信息（列表形式）
    :return: 自然语言描述
    """
    description = f"表 {table_name} 有 {len(fields)} 个字段：\n"
    for field in fields:
        description += f"- {field[0]}（{field[1]}）\n"
    return description

# 示例：生成自然语言描述
description = describe_table_in_natural_language(table_name, fields)
print("自然语言描述：")
print(description)

表 users 有 3 个字段：
- id（integer）
- name（text）
- email（text）