Skip to content

querysets

django_spire.ai.chat.querysets

ChatQuerySet

Bases: HistoryQuerySet

by_user

Source code in django_spire/ai/chat/querysets.py 
13
14
def by_user(self, user: User) -> Self:
    return self.filter(user=user)

get_empty_or_create

Source code in django_spire/ai/chat/querysets.py 
16
17
18
19
20
21
22
23
24
25
def get_empty_or_create(self, user: User) -> Self:
    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 
27
28
def search(self, query: str) -> Self:
    return self.filter(Q(name__icontains=query))

ChatMessageQuerySet

Bases: HistoryQuerySet

newest_by_count

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

newest_by_count_reversed

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