diff --git a/app/db/models/__init__.py b/app/db/models/__init__.py new file mode 100644 index 0000000..41069b0 --- /dev/null +++ b/app/db/models/__init__.py @@ -0,0 +1,3 @@ +from app.db.models.tool import Tool + +__all__ = ["Tool"] diff --git a/app/db/models.py b/app/db/models/tool.py similarity index 74% rename from app/db/models.py rename to app/db/models/tool.py index 05cb370..74dbead 100644 --- a/app/db/models.py +++ b/app/db/models/tool.py @@ -1,5 +1,6 @@ -from sqlalchemy import Column, Integer, String, Text, JSON, TIMESTAMP +from sqlalchemy import JSON, TIMESTAMP, Column, Integer, String, Text from sqlalchemy.sql import func + from app.db.database import Base @@ -7,20 +8,17 @@ class Tool(Base): __tablename__ = "tools" id = Column(Integer, primary_key=True, index=True) - name = Column(String(100), unique=True, nullable=False) - description = Column(Text, nullable=False) - parameters = Column(JSON, nullable=False) created_at = Column( TIMESTAMP, - server_default=func.current_timestamp() + server_default=func.current_timestamp(), ) updated_at = Column( TIMESTAMP, server_default=func.current_timestamp(), - onupdate=func.current_timestamp() + onupdate=func.current_timestamp(), )