Verified Commit e9d0df5f authored by Richard Glosner's avatar Richard Glosner
Browse files

optimize db calls

parent 3d423ef7

llm/serializer/template_suggestor.py

+6 −3
Original line number Diff line number Diff line
@@ -16,10 +16,13 @@ class TemplateSuggestor: 
    def __init__(self, thread: EmailThread):

        self.thread = thread

        definition_participant: EmailParticipant = ensure_exists(

            thread.participants.filter(definition_address__isnull=False)

            .select_related("definition_address")

            thread.participants.select_related(

                "definition_address",

                "definition_address__activity",

                "definition_address__activity__objective",

            )

            .prefetch_related("definition_address__templates")

            .prefetch_related("definition_address__activity")

            .filter(definition_address__isnull=False)

        )

        self.activity = definition_participant.definition_address.activity