fix calc

SciPhi-AI · Jan 8, 2025 · 6092ad8 · 6092ad8
1 parent 60f9cfd
commit 6092ad8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/py/core/pipes/ingestion/vector_storage_pipe.py b/py/core/pipes/ingestion/vector_storage_pipe.py
@@ -67,9 +67,12 @@ async def _run_logic(  # type: ignore
         for msg in input.message:
             if not current_usage:
                 # ASSUME ONLY ONE OWNER PER MESSAGE
-                current_usage = await self.database_provider.chunks_handler.get_user_chunk_count(
-                    msg.owner_id
-                )
+                current_usage = (
+                    await self.database_provider.chunks_handler.list_chunks(
+                        limit=1, offset=0, filters={"owner_id": msg.owner_id}
+                    )
+                )["page_info"]["total_entries"]
+
             user = await self.database_provider.users_handler.get_user_by_id(
                 msg.owner_id
             )