mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-18 07:19:35 +00:00
6d797dc0d2
Root cause: sqlite_patch checked `if "sqlite" in db_url` but db_url was empty string when DATABASE_URL env var not set. Patch was skipped, then models used PostgreSQL types (JSONB/Vector) with SQLite compiler causing crash: "can't render element of type JSONB". Fix: `if not db_url or "sqlite" in db_url` — apply patch when URL is empty (defaults to SQLite anyway in database.py). Also: - Dockerfile: add libxml2/libxslt1.1 for lxml runtime - Dockerfile: increase healthcheck start-period to 120s - start.sh: log DATABASE_URL prefix for debugging https://claude.ai/code/session_01W1rJthWDkasijTdXCfxVHs
157 lines
4.1 KiB
Python
157 lines
4.1 KiB
Python
"""
|
|
SQLite Compatibility Patch for Dealix
|
|
Proper TypeDecorator subclasses — fully compatible with SQLAlchemy Column().
|
|
"""
|
|
import os
|
|
import json
|
|
from sqlalchemy import String, Text, TypeDecorator
|
|
|
|
|
|
def _get_db_url() -> str:
|
|
url = os.environ.get("DATABASE_URL", "")
|
|
if not url:
|
|
for env_file in [".env", "../.env"]:
|
|
try:
|
|
with open(env_file) as f:
|
|
for line in f:
|
|
if line.strip().startswith("DATABASE_URL="):
|
|
url = line.strip().split("=", 1)[1]
|
|
break
|
|
except FileNotFoundError:
|
|
continue
|
|
return url
|
|
|
|
|
|
class _FakeUUID(TypeDecorator):
|
|
"""UUID stored as VARCHAR(36) for SQLite."""
|
|
impl = String
|
|
cache_ok = True
|
|
|
|
def __init__(self, as_uuid=True, **kw):
|
|
super().__init__(36)
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
return str(value) if value is not None else None
|
|
|
|
def process_result_value(self, value, dialect):
|
|
return value
|
|
|
|
|
|
class _FakeJSONB(TypeDecorator):
|
|
"""JSONB stored as TEXT for SQLite."""
|
|
impl = Text
|
|
cache_ok = True
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
return value
|
|
return json.dumps(value, ensure_ascii=False)
|
|
|
|
def process_result_value(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
try:
|
|
return json.loads(value)
|
|
except Exception:
|
|
return value
|
|
return value
|
|
|
|
|
|
class _FakeINET(TypeDecorator):
|
|
"""IP address stored as VARCHAR for SQLite."""
|
|
impl = String
|
|
cache_ok = True
|
|
|
|
def __init__(self, *a, **kw):
|
|
super().__init__(45) # max IPv6 length
|
|
|
|
|
|
class _FakeARRAY(TypeDecorator):
|
|
impl = Text
|
|
cache_ok = True
|
|
|
|
def __init__(self, *a, **kw):
|
|
super().__init__()
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
return json.dumps(value, ensure_ascii=False)
|
|
|
|
def process_result_value(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
try:
|
|
return json.loads(value)
|
|
except Exception:
|
|
return value
|
|
|
|
|
|
class _FakePGModule:
|
|
"""Fake postgresql module — all common PG types mapped to SQLite-compatible types."""
|
|
UUID = _FakeUUID
|
|
JSONB = _FakeJSONB
|
|
INET = _FakeINET
|
|
ARRAY = _FakeARRAY
|
|
# Additional types as simple String fallbacks
|
|
TSVECTOR = String
|
|
TSQUERY = String
|
|
CIDR = String
|
|
MACADDR = String
|
|
HSTORE = _FakeJSONB
|
|
JSON = _FakeJSONB
|
|
|
|
|
|
class _FakeVector(TypeDecorator):
|
|
"""Vector stored as TEXT for SQLite (no pgvector needed)."""
|
|
impl = Text
|
|
cache_ok = True
|
|
|
|
def __init__(self, dim=None, *a, **kw):
|
|
super().__init__()
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
return json.dumps(value if isinstance(value, list) else list(value))
|
|
|
|
def process_result_value(self, value, dialect):
|
|
if value is None:
|
|
return None
|
|
try:
|
|
return json.loads(value)
|
|
except Exception:
|
|
return value
|
|
|
|
|
|
class _FakePGVectorSQLAlchemy:
|
|
Vector = _FakeVector
|
|
|
|
|
|
class _FakePGVectorRoot:
|
|
sqlalchemy = _FakePGVectorSQLAlchemy()
|
|
|
|
|
|
def apply_patch():
|
|
import sys
|
|
import types
|
|
db_url = _get_db_url()
|
|
if not db_url or "sqlite" in db_url.lower():
|
|
# Patch PostgreSQL dialect
|
|
sys.modules["sqlalchemy.dialects.postgresql"] = _FakePGModule() # type: ignore
|
|
|
|
# Patch pgvector
|
|
pgvector_root = types.ModuleType("pgvector")
|
|
pgvector_sa = types.ModuleType("pgvector.sqlalchemy")
|
|
pgvector_sa.Vector = _FakeVector # type: ignore
|
|
pgvector_root.sqlalchemy = pgvector_sa # type: ignore
|
|
sys.modules["pgvector"] = pgvector_root
|
|
sys.modules["pgvector.sqlalchemy"] = pgvector_sa
|
|
|
|
print("[sqlite_patch] SQLite patch applied: UUID/JSONB/Vector -> SQLite types")
|
|
else:
|
|
print(f"[sqlite_patch] DB: {db_url.split(':')[0]} - no patch needed")
|