diff --git a/deploy/profile-scripts/batch_reassess.py b/deploy/profile-scripts/batch_reassess.py index 80bc84d3..834e8e01 100644 --- a/deploy/profile-scripts/batch_reassess.py +++ b/deploy/profile-scripts/batch_reassess.py @@ -129,7 +129,7 @@ def collect_data(code): if (not _sector_ctx) or _sector_ctx.startswith('大盘上涨比') or len(_sector_ctx) < 4: try: _sdb = sqlite3.connect(DB) - _sr = _sdb.execute("SELECT sector FROM stock_sectors WHERE code=? LIMIT 1", (code,)).fetchone() + _sr = _sdb.execute("SELECT sector_name FROM stock_sectors WHERE code=? LIMIT 1", (code,)).fetchone() _sdb.close() if _sr and _sr[0]: _sector_ctx = f"行业{_sr[0]}" @@ -184,7 +184,7 @@ def build_prompt(data): _sector_name = "" try: _sr = _db.execute( - "SELECT sector FROM stock_sectors WHERE code=? LIMIT 1", (data['code'],)).fetchone() + "SELECT sector_name FROM stock_sectors WHERE code=? LIMIT 1", (data['code'],)).fetchone() _sector_name = _sr[0] if _sr else "" except Exception: pass