darkfire514 commited on
Commit
ef4f876
·
verified ·
1 Parent(s): 1c9799a

Update openspace/skill_engine/store.py

Browse files
Files changed (1) hide show
  1. openspace/skill_engine/store.py +5 -4
openspace/skill_engine/store.py CHANGED
@@ -1454,11 +1454,12 @@ class SkillStore:
1454
 
1455
  for chunk in chunks(sids, 900):
1456
  placeholders = ",".join(["?"] * len(chunk))
 
1457
 
1458
  # Parents
1459
  p_rows = conn.execute(
1460
  f"SELECT skill_id, parent_skill_id FROM skill_lineage_parents WHERE skill_id IN ({placeholders})",
1461
- chunk,
1462
  ).fetchall()
1463
  for pr in p_rows:
1464
  parents_map[pr["skill_id"]].append(pr["parent_skill_id"])
@@ -1466,7 +1467,7 @@ class SkillStore:
1466
  # Tool deps
1467
  d_rows = conn.execute(
1468
  f"SELECT skill_id, tool_key, critical FROM skill_tool_deps WHERE skill_id IN ({placeholders})",
1469
- chunk,
1470
  ).fetchall()
1471
  for dr in d_rows:
1472
  deps_map[dr["skill_id"]].append(dr)
@@ -1474,7 +1475,7 @@ class SkillStore:
1474
  # Tags
1475
  t_rows = conn.execute(
1476
  f"SELECT skill_id, tag FROM skill_tags WHERE skill_id IN ({placeholders})",
1477
- chunk,
1478
  ).fetchall()
1479
  for tr in t_rows:
1480
  tags_map[tr["skill_id"]].append(tr["tag"])
@@ -1488,7 +1489,7 @@ class SkillStore:
1488
  WHERE sj.skill_id IN ({placeholders})
1489
  ORDER BY ea.timestamp DESC
1490
  """,
1491
- chunk,
1492
  ).fetchall()
1493
  for ar in a_rows:
1494
  sid = ar["sj_skill_id"]
 
1454
 
1455
  for chunk in chunks(sids, 900):
1456
  placeholders = ",".join(["?"] * len(chunk))
1457
+ chunk_tuple = tuple(chunk)
1458
 
1459
  # Parents
1460
  p_rows = conn.execute(
1461
  f"SELECT skill_id, parent_skill_id FROM skill_lineage_parents WHERE skill_id IN ({placeholders})",
1462
+ chunk_tuple,
1463
  ).fetchall()
1464
  for pr in p_rows:
1465
  parents_map[pr["skill_id"]].append(pr["parent_skill_id"])
 
1467
  # Tool deps
1468
  d_rows = conn.execute(
1469
  f"SELECT skill_id, tool_key, critical FROM skill_tool_deps WHERE skill_id IN ({placeholders})",
1470
+ chunk_tuple,
1471
  ).fetchall()
1472
  for dr in d_rows:
1473
  deps_map[dr["skill_id"]].append(dr)
 
1475
  # Tags
1476
  t_rows = conn.execute(
1477
  f"SELECT skill_id, tag FROM skill_tags WHERE skill_id IN ({placeholders})",
1478
+ chunk_tuple,
1479
  ).fetchall()
1480
  for tr in t_rows:
1481
  tags_map[tr["skill_id"]].append(tr["tag"])
 
1489
  WHERE sj.skill_id IN ({placeholders})
1490
  ORDER BY ea.timestamp DESC
1491
  """,
1492
+ chunk_tuple,
1493
  ).fetchall()
1494
  for ar in a_rows:
1495
  sid = ar["sj_skill_id"]