Skip to content

querysets

django_spire.ai.chat.querysets

ChatQuerySet

Bases: HistoryQuerySet

by_user

Source code in django_spire/ai/chat/querysets.py 
 7
 8
 9
10
def by_user(self, user):
    return self.filter(
        user=user,
    )

get_empty_or_create

Source code in django_spire/ai/chat/querysets.py 
12
13
14
15
16
17
18
19
20
21
def get_empty_or_create(self, user):
    try:
        return (
            self.filter(user=user)
            .annotate(num_messages=Count('message'))
            .filter(num_messages=0)
            .earliest('-id')
        )
    except self.model.DoesNotExist:
        return self.create(user=user)

search

Source code in django_spire/ai/chat/querysets.py 
23
24
25
26
def search(self, query: str):
    return self.filter(
        Q(name__icontains=query)
    )

ChatMessageQuerySet

Bases: HistoryQuerySet

newest_by_count

Source code in django_spire/ai/chat/querysets.py 
30
31
def newest_by_count(self, count: int = 20):
    return self.order_by('-created_datetime')[:count]

newest_by_count_reversed

Source code in django_spire/ai/chat/querysets.py 
33
34
def newest_by_count_reversed(self, count: int = 20):
    return self.order_by('-created_datetime')[:count][::-1]