Initial commit

app.py

@@ -4,10 +4,14 @@ from database import init_storage
 
 @cl.on_chat_start
 async def start():
-    # Инициализируем таблицы в БД при старте
-    init_storage()
+    try:
+        # Добавили await!
+        await init_storage() 
+    except Exception as e:
+        print(f"Ошибка при создании таблиц памяти: {e}")
+    
     cl.user_session.set("thread_id", cl.user_session.get("id"))
-    await cl.Message(content="Привет! Я твой аналитик WB. База подключена, контекст сохраняется.").send()
+    await cl.Message(content="Система запущена.").send()
 
 @cl.on_message
 async def main(message: cl.Message):

database.py

@@ -1,18 +1,19 @@
 import os
-from langgraph.checkpoint.postgres import PostgresSaver
-from psycopg_pool import ConnectionPool
+from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
+from psycopg_pool import AsyncConnectionPool
 
-# Используем проверенную строку (лучше через переменные окружения)
-DB_URI = os.getenv("DATABASE_URL", "postgresql://postgres.fdkvbrxnmsfctzoaewtn:pulXdHRsYb6ff2gO@aws-1-eu-west-1.pooler.supabase.com:6543/postgres?sslmode=require")
 
-# Пул соединений для высокой нагрузки
-pool = ConnectionPool(conninfo=DB_URI, max_size=20, kwargs={"autocommit": True})
+# Используем новое имя переменной, чтобы Chainlit не путался!
+DB_URI = os.getenv("SUPABASE_URL", "postgresql://postgres.fdkvbrxnmsfctzoaewtn:pulXdHRsYb6ff2gO@aws-1-eu-west-1.pooler.supabase.com:6543/postgres?sslmode=require")
 
-# Объект для сохранения состояния графа
-checkpointer = PostgresSaver(pool)
+# АСИНХРОННЫЙ пул соединений
+pool = AsyncConnectionPool(conninfo=DB_URI, max_size=20, kwargs={"autocommit": True})
 
-# Функция для первой инициализации таблиц (запустить один раз)
-def init_storage():
-    # Чекпоинтер сам возьмет коннект из пула, ничего передавать не нужно
-    checkpointer.setup()
-    print("✅ Хранилище контекста в Supabase готово.")
+# АСИНХРОННЫЙ объект для сохранения состояния графа
+checkpointer = AsyncPostgresSaver(pool)
+
+# Асинхронная инициализация
+async def init_storage():
+    # Теперь мы просто вызываем setup(), чекпоинтер сам возьмет коннект из пула
+    await checkpointer.setup()
+    print("✅ Асинхронное хранилище контекста в Supabase готово.")