From 3b1bf7da6771bd1d6852d3dcc1f906d5ae5c95d1 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 12 三月 2025 13:49:12 +0800
Subject: [PATCH] 增加内容优化
---
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