68 lines
3.4 KiB
Python
68 lines
3.4 KiB
Python
"""核对运行中AI查数接口与独立SQL；使用已有管理员会话，仅产生查询历史。"""
|
||
import argparse
|
||
import json
|
||
import sys
|
||
from decimal import Decimal
|
||
from pathlib import Path
|
||
|
||
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser()
|
||
parser.add_argument('--local', action='store_true')
|
||
parser.add_argument('--url', default='http://127.0.0.1:18888')
|
||
args = parser.parse_args()
|
||
if args.local:
|
||
from test_wsgi import app
|
||
else:
|
||
from app import app
|
||
import requests
|
||
from services.db import query
|
||
from services.inventory_calc import SPECIAL_COST_PRICES
|
||
|
||
price_case = 'CASE i.product_code ' + ' '.join('WHEN %s THEN %s' for _ in SPECIAL_COST_PRICES) + ' ELSE i.cost END'
|
||
bindings = [v for pair in SPECIAL_COST_PRICES.items() for v in pair]
|
||
expected = query(f"""
|
||
SELECT COUNT(*) AS sku_count, COALESCE(SUM(i.qty),0) AS stock_qty,
|
||
COALESCE(SUM(ROUND(i.qty * CAST(({price_case}) AS DECIMAL(24,10)),2)),0) AS inventory_value
|
||
FROM (
|
||
SELECT product_code, SUM(available_qty) AS qty, MAX(cost_price) AS cost
|
||
FROM inventory_detail
|
||
WHERE data_date=(SELECT MAX(data_date) FROM inventory_detail)
|
||
AND product_code NOT LIKE '%%-NL%%' AND product_name NOT LIKE '%%内料%%'
|
||
GROUP BY product_code
|
||
) i JOIN products p ON p.product_code=i.product_code
|
||
WHERE p.deleted_at IS NULL AND COALESCE(p.product_type,'') != '包材'
|
||
AND COALESCE(p.product_name,'') NOT LIKE '%%内料%%' AND TRIM(p.brand)=%s
|
||
""", [*bindings, 'Aromat Herbs'])[0]
|
||
user = query("SELECT id,username,role FROM dashboard_users WHERE status=1 AND role='super_admin' ORDER BY id LIMIT 1")[0]
|
||
serializer = app.session_interface.get_signing_serializer(app)
|
||
cookie = serializer.dumps({'user_id': user['id'], 'username': user['username'],
|
||
'role': user['role'], 'must_change_password': False})
|
||
client = requests.Session()
|
||
client.cookies.set(app.config.get('SESSION_COOKIE_NAME', 'session'), cookie)
|
||
questions = ['查询品牌为Aromat Herbs的产品，库存*单价的总成本是多少',
|
||
'Aromat Herbs品牌库存总成本的具体数值是多少？',
|
||
'Aromat Herbs查一下这个品牌的商品数据目前的库存和库存金额分别是多少？']
|
||
summaries = []
|
||
for question in questions:
|
||
response = client.post(args.url + '/api/ai/ask', json={'question': question}, timeout=90)
|
||
response.raise_for_status()
|
||
payload = response.json()
|
||
assert payload.get('code') == 200, payload.get('msg')
|
||
data = payload['data']
|
||
summary = data['datasets'][0]['summary']
|
||
for key in ('sku_count', 'stock_qty', 'inventory_value'):
|
||
assert Decimal(str(summary[key])) == Decimal(str(expected[key])), (key, summary[key], str(expected[key]))
|
||
assert data['data_date'] == summary['date']
|
||
summaries.append(data['summary'])
|
||
print(json.dumps({'question': question, 'summary': data['summary'],
|
||
'detail_rows': len(data['rows']), 'unpriced_stock_sku_count': summary['unpriced_stock_sku_count']}, ensure_ascii=False))
|
||
assert len(set(summaries)) == 1, summaries
|
||
print('PASS: three live answers agree with independent SQL')
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|