# 建表import sqlite3
# Connect to databaseconn = sqlite3.connect('demo.db')cursor = conn.cursor()
#Create tablequery = '''CREATE TABLE amazon_sales (uniqId VARCHAR(255),productName VARCHAR(255),brandName VARCHAR(255),Asin VARCHAR(255),category INTEGER,upcEanCode VARCHAR(255),listPrice VARCHAR(255),sellingPrice VARCHAR(255),Quantity VARCHAR(255),modelNum VARCHAR(255),aboutProduct VARCHAR(255));'''
cursor.execute(query)

import pandas as pd
df = pd.read_csv('marketing_sample.csv')df = df[df.columns[:11]] # 仅取前11列数据
df.columns = ['uniqId', 'productName', 'brandName', 'asin', 'category', 'upcEanCode', 'listPrice', 'sellingPrice', 'Quantity', 'modelNum', 'aboutProduct']
#Import the csv into databasedf.to_sql('amazon_sales', conn, if_exists='append', index=False)conn.close()

from langchain_community.llms import Minimax

# Load the model, 填充你自己的key和groupidMINMAXAPIKEY = ''MINMAXGROUPID = ''llm = Minimax(minimax_api_key=MINMAXAPIKEY, minimax_group_id=MINMAXGROUPID)
# Prompt the model，验证接口可用response = minimax.invoke("What is the difference between panda and bear?")print(response)
from langchain.utilities import SQLDatabasefrom langchain_experimental.sql import SQLDatabaseChain

# 数据库链接db = SQLDatabase.from_uri("sqlite:///demo.db")db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True)

db_chain.run('what is the productName for uniqId 4c69b61db1fc16e7013b43fc926e502d?')

db_chain.run('不同的productName有多少，如果数量超过10条数据，仅返回数量?')

cursor.execute('select count(distinct productName) from amazon_sales')result = cursor.fetchall()print(result)
>>> [(9930,)]

import langchainlangchain.verbose = Truelangchain.debug = True
db_chain.run('what is the productName for uniqId 4c69b61db1fc16e7013b43fc926e502d?')