@@ -1,7 +1,7 @@
from enum import Enum
from sqlalchemy import Column, Integer, Text, BigInteger, String, SmallInteger, Boolean, TIMESTAMP
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
Base = declarative_base()
@@ -1,5 +1,5 @@
from sqlalchemy import Column, Integer, Text, BigInteger
from pqai_agent import configs