Update openspace/skill_engine/store.py

openspace/skill_engine/store.py

@@ -58,19 +58,26 @@ def _db_retry(
             for attempt in range(max_retries):
                 try:
                     return func(*args, **kwargs)
-                except (sqlite3.OperationalError, sqlite3.DatabaseError) as exc:
-                    if attempt == max_retries - 1:
-                        logger.error(
-                            f"DB {func.__name__} failed after "
-                            f"{max_retries} retries: {exc}"
+                except Exception as exc:
+                    # libsql_experimental might not have standard sqlite3 exceptions
+                    # so we catch Exception and check if it's a known database error string
+                    exc_str = str(exc).lower()
+                    if "locked" in exc_str or "database error" in exc_str or "operational error" in exc_str:
+                        if attempt == max_retries - 1:
+                            logger.error(
+                                f"DB {func.__name__} failed after "
+                                f"{max_retries} retries: {exc}"
+                            )
+                            raise
+                        logger.warning(
+                            f"DB {func.__name__} retry {attempt + 1}"
+                            f"/{max_retries}: {exc}"
                         )
+                        time.sleep(delay)
+                        delay *= backoff
+                    else:
+                        # If it's not a known transient error, raise immediately
                         raise
-                    logger.warning(
-                        f"DB {func.__name__} retry {attempt + 1}"
-                        f"/{max_retries}: {exc}"
-                    )
-                    time.sleep(delay)
-                    delay *= backoff
 
         return wrapper
 
@@ -971,8 +978,8 @@ class SkillStore:
         with self._reader() as conn:
             where = " WHERE is_active=1" if active_only else ""
             total = conn.execute(
-                f"SELECT COUNT(*) FROM skill_records{where}"
-            ).fetchone()[0]
+                f"SELECT COUNT(*) as count FROM skill_records{where}"
+            ).fetchone()["count"]
 
             by_category = {
                 r["category"]: r["cnt"]
@@ -989,12 +996,12 @@ class SkillStore:
                 ).fetchall()
             }
             n_analyses = conn.execute(
-                "SELECT COUNT(*) FROM execution_analyses"
-            ).fetchone()[0]
+                "SELECT COUNT(*) as count FROM execution_analyses"
+            ).fetchone()["count"]
             n_candidates = conn.execute(
-                "SELECT COUNT(*) FROM execution_analyses "
+                "SELECT COUNT(*) as count FROM execution_analyses "
                 "WHERE candidate_for_evolution=1"
-            ).fetchone()[0]
+            ).fetchone()["count"]
             agg = conn.execute(
                 f"""
                 SELECT SUM(total_selections)  AS sel,
@@ -1007,8 +1014,8 @@ class SkillStore:
 
             # Also report total (including inactive) for context
             total_all = conn.execute(
-                "SELECT COUNT(*) FROM skill_records"
-            ).fetchone()[0]
+                "SELECT COUNT(*) as count FROM skill_records"
+            ).fetchone()["count"]
 
             return {
                 "total_skills": total,