-- 獲取所有表名
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 服務(wù)地址
DEESEEK_API_URL = "http://localhost:11434/api/generate"

def generate_sql_query(prompt):
    """
    調(diào)用 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"調(diào)用 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 數(shù)據(jù)庫并執(zhí)行查詢
    :param database_config: 數(shù)據(jù)庫配置字典（包含 host, dbname, user, password）
    :param query: SQL 查詢語句
    :return: 查詢結(jié)果（列表形式）
    """
    try:
        # 建立數(shù)據(jù)庫連接
        conn = psycopg2.connect(
            host=database_config["host"],
            dbname=database_config["dbname"],
            user=database_config["user"],
            password=database_config["password"]
        )
        cursor = conn.cursor()

        # 執(zhí)行查詢
        cursor.execute(query)
        results = cursor.fetchall()

        # 關(guān)閉連接
        cursor.close()
        conn.close()

        return results
    except Exception as e:
        print(f"數(shù)據(jù)庫查詢失?。簕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("數(shù)據(jù)庫中的表名：")
print(tables)

def get_table_fields(database_config, table_name):
    """
    獲取指定表的字段信息
    :param database_config: 數(shù)據(jù)庫配置字典
    :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]}，數(shù)據(jù)類型：{field[1]}")

def describe_table_in_natural_language(table_name, fields):
    """
    將表結(jié)構(gòu)信息轉(zhuǎn)換為自然語言描述
    :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）