优化项目结构,格式化代码

This commit is contained in:
雅诺狐
2025-08-13 09:49:39 +08:00
parent 26039b003c
commit fe97d897ef
24 changed files with 57 additions and 77 deletions

View File

@@ -10,7 +10,7 @@ from typing import Dict, List, Any, Union, Type, Optional
from contextlib import contextmanager
from sqlalchemy.orm import Session
from sqlalchemy.exc import SQLAlchemyError, DisconnectionError, OperationalError
from sqlalchemy import desc, asc, func, and_, or_
from sqlalchemy import desc, asc, func, and_
from src.common.logger import get_logger
from src.common.database.sqlalchemy_models import (
Base, get_db_session, Messages, ActionRecords, PersonInfo, ChatStreams,
@@ -61,7 +61,7 @@ def get_db_session():
time.sleep(retry_delay * (attempt + 1))
else:
raise
except Exception as e:
except Exception:
if session:
session.rollback()
raise

View File

@@ -68,7 +68,7 @@ class SessionProxy:
# commit后不要清除session只是刷新状态
pass # 保持session活跃
return result
except Exception as e:
except Exception:
try:
if session and hasattr(session, 'rollback'):
session.rollback()
@@ -555,7 +555,7 @@ def get_db_session():
session = SessionLocal()
yield session
session.commit()
except Exception as e:
except Exception:
if session:
session.rollback()
raise