aiteamutils 0.2.20__tar.gz → 0.2.21__tar.gz

{aiteamutils-0.2.20 → aiteamutils-0.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.20
+Version: 0.2.21
 Summary: AI Team Utilities
 Project-URL: Homepage, https://github.com/yourusername/aiteamutils
 Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues

{aiteamutils-0.2.20 → aiteamutils-0.2.21}/aiteamutils/database.py

@@ -105,10 +105,10 @@ class DatabaseService:
     @asynccontextmanager
     async def get_session(self) -> AsyncGenerator[AsyncSession, None]:
         """데이터베이스 세션을 생성하고 반환하는 비동기 컨텍스트 매니저."""
-        if self.async_session is None:
+        if self.session_factory is None:
             raise RuntimeError("Session factory not initialized")
         
-        async with self.async_session() as session:
+        async with self.session_factory() as session:
             try:
                 yield session
             finally:
@@ -792,13 +792,12 @@ class DatabaseService:
             CustomException: 데이터베이스 작업 중 오류 발생 시
         """
         try:
-            if self.db:
-                result = await self.db.execute(query)
+            if self.db is not None:
+                return await self.db.execute(query)
+            
+            async with self.get_session() as session:
+                result = await session.execute(query)
                 return result
-            else:
-                async with self.session_factory() as session:
-                    result = await session.execute(query)
-                    return result
         except Exception as e:
             raise CustomException(
                 ErrorCode.DB_QUERY_ERROR,

aiteamutils-0.2.21/aiteamutils/version.py

@@ -0,0 +1,2 @@
+"""버전 정보"""
+__version__ = "0.2.21" 

aiteamutils-0.2.20/aiteamutils/version.py

@@ -1,2 +0,0 @@
-"""버전 정보"""
-__version__ = "0.2.20"