Jelajahi Sumber

total_entries

jack 1 bulan lalu
induk
melakukan
3f7c6a33ca

+ 3 - 1
py/core/main/services/ingestion_service.py

@@ -446,6 +446,7 @@ class IngestionService:
             # If we haven't set usage yet, do so on the first chunk
             # If we haven't set usage yet, do so on the first chunk
             if current_usage is None:
             if current_usage is None:
                 user_id_for_usage_check = msg.owner_id
                 user_id_for_usage_check = msg.owner_id
+                '''
                 usage_data = (
                 usage_data = (
                     await self.providers.database.chunks_handler.list_chunks(
                     await self.providers.database.chunks_handler.list_chunks(
                         limit=1,
                         limit=1,
@@ -453,7 +454,8 @@ class IngestionService:
                         filters={"owner_id": msg.owner_id},
                         filters={"owner_id": msg.owner_id},
                     )
                     )
                 )
                 )
-                current_usage = usage_data["total_entries"]
+                '''
+                current_usage = 0
 
 
             # Figure out the user's limit
             # Figure out the user's limit
             user = await self.providers.database.users_handler.get_user_by_id(
             user = await self.providers.database.users_handler.get_user_by_id(

+ 6 - 2
py/core/main/services/management_service.py

@@ -1028,12 +1028,16 @@ class ManagementService(Service):
             )
             )
         )["total_entries"]
         )["total_entries"]
         max_chunks = await self.get_user_max_chunks(user_id)
         max_chunks = await self.get_user_max_chunks(user_id)
-        used_chunks = (
+        used_chunks = 0
+        
+        '''
+        (
             await self.providers.database.chunks_handler.list_chunks(
             await self.providers.database.chunks_handler.list_chunks(
                 limit=1, offset=0, filters={"owner_id": user_id}
                 limit=1, offset=0, filters={"owner_id": user_id}
             )
             )
         )["total_entries"]
         )["total_entries"]
-
+        '''
+        
         max_collections = await self.get_user_max_collections(user_id)
         max_collections = await self.get_user_max_collections(user_id)
         used_collections: int = (  # type: ignore
         used_collections: int = (  # type: ignore
             await self.providers.database.collections_handler.get_collections_overview(
             await self.providers.database.collections_handler.get_collections_overview(