Skip to content

Commit

Permalink
feat: use manager auth when getting the external token (#326)
Browse files Browse the repository at this point in the history
  • Loading branch information
helllllllder authored Jan 11, 2024
1 parent ca4ddb1 commit 8ef32ef
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,7 @@ dmypy.json

# Static files
media/
static/

# Gunicorn access
gunicorn.access
7 changes: 5 additions & 2 deletions chats/apps/projects/models/models.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from django.core.exceptions import ObjectDoesNotExist
from django.core.exceptions import MultipleObjectsReturned, ObjectDoesNotExist
from django.db import models
from django.db.models import Q
from django.utils.translation import gettext_lazy as _
Expand Down Expand Up @@ -94,7 +94,10 @@ def openai_token(self):

@property
def external_token(self):
return self.permissions.get_or_create(user=None, role=1)[0]
try:
return self.permissions(manager="auth").get_or_create(user=None, role=1)[0]
except MultipleObjectsReturned:
return self.permissions(manager="auth").first()

def add_contact_to_history_blocklist(self, contact_external_id: str):
config = self.config or {}
Expand Down

0 comments on commit 8ef32ef

Please sign in to comment.