Fix request and credit usage counting for datasource

Author: vegito22

llmstack/data/apis.py

@@ -317,7 +317,10 @@ def process_entry(self, request, uid):
         entry.status = DataSourceEntryStatus.READY if not document.processing_errors else DataSourceEntryStatus.FAILED
         entry.save(update_fields=["config", "size", "status", "updated_at"])
 
-        return DRFResponse(DataSourceEntrySerializer(instance=entry, context={"request_user": request.user}).data)
+        context = {}
+        if request:
+            context["request_user"] = request.user
+        return DRFResponse(DataSourceEntrySerializer(instance=entry, context=context).data)
 
     def resync(self, request, uid):
         datasource_entry_object = get_object_or_404(DataSourceEntry, uuid=uuid.UUID(uid))
@@ -569,9 +572,7 @@ def add_entry(self, request, uid):
         documents = self.process_add_entry_request(datasource, source_data)
         for document in documents:
             create_result = DataSourceEntryViewSet().create_entry(user=request.user, document=document)
-            process_result = DataSourceEntryViewSet().process_entry(
-                request=request, uid=str(create_result.data["uuid"])
-            )
+            process_result = DataSourceEntryViewSet().process_entry(request=None, uid=str(create_result.data["uuid"]))
             datasource.size += process_result.data["size"]
 
         datasource.save()

llmstack/data/models.py

@@ -169,6 +169,8 @@ def pipeline(self):
         return self.config.get("pipeline", {})
 
     def has_read_permission(self, user):
+        if not user:
+            return False
         return (
             self.owner == user
             or user.email in self.read_accessible_by
@@ -179,6 +181,8 @@ def has_read_permission(self, user):
         )
 
     def has_write_permission(self, user):
+        if not user:
+            return False
         return (
             self.owner == user
             or user.email in self.write_accessible_by
@@ -280,7 +284,7 @@ def register_data_change(sender, instance: DataSourceEntry, **kwargs):
                 "uuid": str(instance.uuid),
                 "old_size": instance.old_size,
                 "size": instance.size,
-                "owner": instance.owner_id,
+                "owner_email": instance.owner_id.email,
             },
         )
 
@@ -308,7 +312,7 @@ def register_data_delete(sender, instance: DataSourceEntry, **kwargs):
                 "uuid": str(instance.uuid),
                 "old_size": instance.size,
                 "size": 0,
-                "owner": instance.owner_id,
+                "owner_email": instance.owner_id.email,
             },
         )