Skip to content

Commit 0459322

Browse files
committed
Update rq dependencies and pass request object in process entry
1 parent abe0ae3 commit 0459322

File tree

3 files changed

+22
-23
lines changed

3 files changed

+22
-23
lines changed

llmstack/data/apis.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ def create_entry(self, user, document: DataDocument):
287287

288288
def process_entry(self, request, uid):
289289
entry = get_object_or_404(DataSourceEntry, uuid=uuid.UUID(uid))
290-
if request and request.user != entry.datasource.owner:
290+
if request and request.user != entry.datasource.has_write_permission(request.user):
291291
return DRFResponse(status=404)
292292

293293
document = DataDocument(**entry.config)
@@ -569,7 +569,9 @@ def add_entry(self, request, uid):
569569
documents = self.process_add_entry_request(datasource, source_data)
570570
for document in documents:
571571
create_result = DataSourceEntryViewSet().create_entry(user=request.user, document=document)
572-
process_result = DataSourceEntryViewSet().process_entry(request=None, uid=str(create_result.data["uuid"]))
572+
process_result = DataSourceEntryViewSet().process_entry(
573+
request=request, uid=str(create_result.data["uuid"])
574+
)
573575
datasource.size += process_result.data["size"]
574576

575577
datasource.save()

poetry.lock

Lines changed: 15 additions & 19 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ django-allauth = "^0.63.3"
3838
django-environ = "^0.11.2"
3939
django-picklefield = "^3.2"
4040
django-redis = "^5.4.0"
41-
django-rq = "^2.10.2"
4241
djangorestframework = "^3.15.2"
4342
django-flags = "^5.0.13"
4443
django-jsonform = {version = "^2.17.4"}
@@ -50,7 +49,6 @@ channels = "^4.1.0"
5049
importlib-metadata = {version = "^6.0.0" }
5150
pyyaml = {version = "^6.0" }
5251
redis = {version = "^4.5.4" }
53-
rq = "^1.16.2"
5452
sendgrid = {version = "^6.10.0" }
5553
python-liquid = {version ="^1.12.1"}
5654
orjson = {version = "^3.8.14" }
@@ -72,6 +70,9 @@ daphne = "^4.1.2"
7270
striprtf = "^0.0.26"
7371
langrocks = "0.1.4"
7472
diff-match-patch = "^20230430"
73+
rq = "^2.0.0"
74+
django-rq = "^3.0.0"
75+
distlib = "^0.3.9"
7576

7677
[tool.poetry.group.faiss.dependencies]
7778
faiss-cpu = "^1.8.0"

0 commit comments

Comments
 (0)