@roll-agent/octopus-agent 0.0.10 → 0.0.11

package/src/octopus_skill/query_sql_cache.py

@@ -1,72 +0,0 @@
-from __future__ import annotations
-
-import hashlib
-import json
-import re
-import time
-from dataclasses import dataclass
-from pathlib import Path
-from typing import Any
-
-
-SQL_GENERATION_CACHE_VERSION = "v9"
-
-
-@dataclass
-class QuerySqlCache:
-    path: Path
-    ttl_minutes: int
-
-    def get(self, *, question: str, schema_version: str | None) -> str | None:
-        if self.ttl_minutes <= 0:
-            return None
-        entry = self._entries().get(_cache_key(question, schema_version))
-        if not isinstance(entry, dict):
-            return None
-        created_at = entry.get("createdAt")
-        sql = entry.get("sql")
-        if not isinstance(created_at, (int, float)) or not isinstance(sql, str):
-            return None
-        if time.time() - float(created_at) > self.ttl_minutes * 60:
-            return None
-        return sql
-
-    def delete(self, *, question: str, schema_version: str | None) -> None:
-        entries = self._entries()
-        key = _cache_key(question, schema_version)
-        if key not in entries:
-            return
-        del entries[key]
-        self.path.parent.mkdir(parents=True, exist_ok=True)
-        self.path.write_text(json.dumps(entries, ensure_ascii=False, indent=2, sort_keys=True) + "\n", encoding="utf-8")
-
-    def set(self, *, question: str, schema_version: str | None, sql: str) -> None:
-        if self.ttl_minutes <= 0:
-            return
-        entries = self._entries()
-        entries[_cache_key(question, schema_version)] = {
-            "questionKey": _normalize_question(question),
-            "schemaVersion": schema_version,
-            "sql": sql,
-            "createdAt": time.time(),
-        }
-        self.path.parent.mkdir(parents=True, exist_ok=True)
-        self.path.write_text(json.dumps(entries, ensure_ascii=False, indent=2, sort_keys=True) + "\n", encoding="utf-8")
-
-    def _entries(self) -> dict[str, Any]:
-        if not self.path.exists():
-            return {}
-        try:
-            data = json.loads(self.path.read_text(encoding="utf-8"))
-        except json.JSONDecodeError:
-            return {}
-        return data if isinstance(data, dict) else {}
-
-
-def _cache_key(question: str, schema_version: str | None) -> str:
-    raw = f"{SQL_GENERATION_CACHE_VERSION}:{schema_version or ''}:{_normalize_question(question)}"
-    return hashlib.sha256(raw.encode("utf-8")).hexdigest()
-
-
-def _normalize_question(question: str) -> str:
-    return re.sub(r"\s+", "", question.strip().lower())