From e8c62332239069647b93d4c2d08db6c223144e87 Mon Sep 17 00:00:00 2001
From: sanj <67624670+iodrift@users.noreply.github.com>
Date: Mon, 12 Aug 2024 22:27:39 -0700
Subject: [PATCH] Auto-update: Mon Aug 12 22:27:39 PDT 2024

---
 sijapi/database.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/sijapi/database.py b/sijapi/database.py
index f623603..fcbdc5f 100644
--- a/sijapi/database.py
+++ b/sijapi/database.py
@@ -24,7 +24,10 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.orm import sessionmaker, declarative_base
 from sqlalchemy.exc import OperationalError
 from sqlalchemy import Column, Integer, String, DateTime, JSON, Text
-from sqlalchemy.dialects.postgresql import JSONB
+import uuid
+from sqlalchemy import Column, String, DateTime, Text, ARRAY
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from sqlalchemy.sql import func
 from urllib.parse import urljoin
 import hashlib
 import random
@@ -41,15 +44,17 @@ ENV_PATH = CONFIG_DIR / ".env"
 load_dotenv(ENV_PATH)
 TS_ID = os.environ.get('TS_ID')
 
+
 class QueryTracking(Base):
     __tablename__ = 'query_tracking'
 
-    id = Column(Integer, primary_key=True)
-    ts_id = Column(String, nullable=False)
+    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    origin_ts_id = Column(String, nullable=False)
     query = Column(Text, nullable=False)
     args = Column(JSONB)
     executed_at = Column(DateTime(timezone=True), server_default=func.now())
-    completed_by = Column(JSONB, default={})
+    completed_by = Column(ARRAY(String), default=[])
+    result_checksum = Column(String(32))
 
 class Database:
     @classmethod