Delete database.py

database.py

@@ -1,136 +0,0 @@
-import json
-import datetime
-import httpx
-from config import SUPABASE_URL, SUPABASE_KEY, user_tokens
-
-async def save_token_to_supabase(user_id: str, service: str, token_data: dict):
-    google_services = ["google", "gmail", "calendar", "drive"]
-    target_services = google_services if service in google_services else [service]
-    now_iso = datetime.datetime.now(datetime.timezone.utc).isoformat()
-    
-    for s in target_services:
-        user_tokens[(user_id, s)] = token_data
-        try:
-            url = f"{SUPABASE_URL}/rest/v1/user_connectors"
-            headers = {
-                "apikey": SUPABASE_KEY,
-                "Authorization": f"Bearer {SUPABASE_KEY}",
-                "Content-Type": "application/json",
-                "Prefer": "resolution=merge-duplicates"
-            }
-            payload = {
-                "user_id": user_id,
-                "service": s,
-                "token_data": token_data,
-                "updated_at": now_iso
-            }
-            async with httpx.AsyncClient(timeout=15.0) as c:
-                await c.post(url, headers=headers, json=payload)
-        except Exception:
-            pass
-
-async def get_token_from_supabase(user_id: str, service: str):
-    google_services = ["google", "gmail", "calendar", "drive"]
-    check_services = [service] + [x for x in google_services if x != service] if service in google_services else [service]
-
-    for s in check_services:
-        if (user_id, s) in user_tokens:
-            return user_tokens[(user_id, s)]
-        try:
-            url = f"{SUPABASE_URL}/rest/v1/user_connectors?user_id=eq.{user_id}&service=eq.{s}&select=token_data"
-            headers = {
-                "apikey": SUPABASE_KEY,
-                "Authorization": f"Bearer {SUPABASE_KEY}"
-            }
-            async with httpx.AsyncClient(timeout=15.0) as c:
-                res = await c.get(url, headers=headers)
-                if res.status_code == 200:
-                    data = res.json()
-                    if data and len(data) > 0:
-                        raw_t = data[0].get("token_data")
-                        if isinstance(raw_t, dict):
-                            t_data = raw_t
-                        elif isinstance(raw_t, str):
-                            t_data = json.loads(raw_t)
-                        else:
-                            t_data = {}
-                        if t_data:
-                            user_tokens[(user_id, s)] = t_data
-                            return t_data
-        except Exception:
-            pass
-    return None
-
-async def delete_token_from_supabase(user_id: str, service: str):
-    google_services = ["google", "gmail", "calendar", "drive"]
-    target_services = google_services if service in google_services else [service]
-
-    for s in target_services:
-        user_tokens.pop((user_id, s), None)
-        try:
-            url = f"{SUPABASE_URL}/rest/v1/user_connectors?user_id=eq.{user_id}&service=eq.{s}"
-            headers = {
-                "apikey": SUPABASE_KEY,
-                "Authorization": f"Bearer {SUPABASE_KEY}"
-            }
-            async with httpx.AsyncClient(timeout=15.0) as c:
-                await c.delete(url, headers=headers)
-        except Exception:
-            pass
-
-async def fetch_session_messages(session_id: str):
-    try:
-        url = f"{SUPABASE_URL}/rest/v1/chat_messages?session_id=eq.{session_id}&order=created_at.asc"
-        headers = {
-            "apikey": SUPABASE_KEY,
-            "Authorization": f"Bearer {SUPABASE_KEY}"
-        }
-        async with httpx.AsyncClient(timeout=20.0) as c:
-            res = await c.get(url, headers=headers)
-            if res.status_code == 200:
-                return res.json()
-    except Exception:
-        pass
-    return []
-
-async def save_assistant_message_to_db(session_id: str, user_id: str, content: str, thought_reasoning: str = "", sources = None, tool_calls = None):
-    try:
-        url = f"{SUPABASE_URL}/rest/v1/chat_messages"
-        headers = {
-            "apikey": SUPABASE_KEY,
-            "Authorization": f"Bearer {SUPABASE_KEY}",
-            "Content-Type": "application/json",
-            "Prefer": "return=minimal"
-        }
-        payload = {
-            "session_id": session_id,
-            "user_id": user_id,
-            "sender": "assistant",
-            "content": content,
-            "thought_reasoning": thought_reasoning
-        }
-        if sources is not None:
-            payload["sources"] = sources
-        if tool_calls is not None:
-            payload["tool_calls"] = tool_calls
-
-        async with httpx.AsyncClient(timeout=20.0) as c:
-            await c.post(url, headers=headers, json=payload)
-    except Exception:
-        pass
-
-async def update_user_message_file_content(session_id: str, file_name: str, file_content: str):
-    try:
-        url = f"{SUPABASE_URL}/rest/v1/chat_messages?session_id=eq.{session_id}&sender=eq.user&file_name=eq.{file_name}&order=created_at.desc&limit=1"
-        headers = {
-            "apikey": SUPABASE_KEY,
-            "Authorization": f"Bearer {SUPABASE_KEY}",
-            "Content-Type": "application/json"
-        }
-        payload = {
-            "file_content": file_content
-        }
-        async with httpx.AsyncClient(timeout=20.0) as c:
-            await c.patch(url, headers=headers, json=payload)
-    except Exception:
-        pass