From abb91124a4372b0efe5ab1b7aa25859c635d30eb Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 10 四月 2025 13:49:37 +0800 Subject: [PATCH] merge --- app/models/token_model.py | 21 ++++++++++++--------- 1 files changed, 12 insertions(+), 9 deletions(-) diff --git a/app/models/token_model.py b/app/models/token_model.py index 764d999..bfd088d 100644 --- a/app/models/token_model.py +++ b/app/models/token_model.py @@ -1,9 +1,6 @@ from datetime import datetime -from typing import Type - -from sqlalchemy import Column, Integer, DateTime, Text +from sqlalchemy import Column, Integer, DateTime, Text, String from sqlalchemy.orm import Session - from app.models.base_model import Base @@ -11,9 +8,9 @@ __tablename__ = "token" id = Column(Integer, primary_key=True, index=True) user_id = Column(Integer, index=True) - token = Column(Text(10000), unique=True, index=True) - bisheng_token = Column(Text(10000), unique=True, index=True) - ragflow_token = Column(Text(10000), unique=True, index=True) + token = Column(Text(10000)) + bisheng_token = Column(Text(10000)) + ragflow_token = Column(Text(10000)) created_at = Column(DateTime, default=datetime.utcnow) @@ -52,5 +49,11 @@ db.rollback() # 鍥炴粴浜嬪姟 -def get_token(db: Session, user_id: int) -> Type[TokenModel] | None: - return db.query(TokenModel).filter_by(user_id=user_id).first() +class ApiTokenModel(Base): + __tablename__ = "chat_api_tokens" + id = Column(String(36), primary_key=True) + app_id = Column(String(36), index=True) + type = Column(String(16)) + token = Column(String(255)) + created_at = Column(DateTime, default=datetime.now()) + last_used_at = Column(DateTime, default=datetime.now()) -- Gitblit v1.8.0