[全量] 初始化项目代码、配置、文档及Agent协同harness

langchain-chat/server/db/session.py

@@ -0,0 +1,46 @@
+from functools import wraps
+from contextlib import contextmanager
+from server.db.base import SessionLocal
+from sqlalchemy.orm import Session
+
+
+@contextmanager
+def session_scope() -> Session:
+    """上下文管理器用于自动获取 Session, 避免错误"""
+    session = SessionLocal()
+    try:
+        yield session
+        session.commit()
+    except:
+        session.rollback()
+        raise
+    finally:
+        session.close()
+
+
+def with_session(f):
+    @wraps(f)
+    def wrapper(*args, **kwargs):
+        with session_scope() as session:
+            try:
+                result = f(session, *args, **kwargs)
+                session.commit()
+                return result
+            except:
+                session.rollback()
+                raise
+
+    return wrapper
+
+
+def get_db() -> SessionLocal:
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+
+def get_db0() -> SessionLocal:
+    db = SessionLocal()
+    return db