Parcourir la source

fix(follow_ups): handle empty LLM responses in context (#18237)

tags/1.3.0
Novice il y a 6 mois
Parent
révision
6d66e3f680
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2
    1
      api/core/memory/token_buffer_memory.py

+ 2
- 1
api/core/memory/token_buffer_memory.py Voir le fichier

Message.created_at, Message.created_at,
Message.workflow_run_id, Message.workflow_run_id,
Message.parent_message_id, Message.parent_message_id,
Message.answer_tokens,
) )
.filter( .filter(
Message.conversation_id == self.conversation.id, Message.conversation_id == self.conversation.id,
thread_messages = extract_thread_messages(messages) thread_messages = extract_thread_messages(messages)


# for newly created message, its answer is temporarily empty, we don't need to add it to memory # for newly created message, its answer is temporarily empty, we don't need to add it to memory
if thread_messages and not thread_messages[0].answer:
if thread_messages and not thread_messages[0].answer and thread_messages[0].answer_tokens == 0:
thread_messages.pop(0) thread_messages.pop(0)


messages = list(reversed(thread_messages)) messages = list(reversed(thread_messages))

Chargement…
Annuler
Enregistrer