SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 10:45:42 +01:00
Code Issues Projects Releases Packages Activity

fix: when two user messages were sent

Browse source
This commit is contained in:
Javier Martinez 2024-07-29 08:46:35 +02:00
parent 05a986231c
commit 0eb24a54ec
No known key found for this signature in database
1 changed files with 13 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

29
private_gpt/ui/ui.py
View file

@ -1,6 +1,4 @@
"""This file should be imported if and only if you want to run the UI locally."""
import itertools
import logging
import time
from collections.abc import Iterable
@ -113,21 +111,20 @@ class PrivateGptUi:
yield full_response
def build_history() -> list[ChatMessage]:
history_messages: list[ChatMessage] = list(
itertools.chain(
*[
[
ChatMessage(content=interaction[0], role=MessageRole.USER),
ChatMessage(
# Remove from history content the Sources information
content=interaction[1].split(SOURCES_SEPARATOR)[0],
role=MessageRole.ASSISTANT,
),
]
for interaction in history
]
history_messages: list[ChatMessage] = []
for interaction in history:
history_messages.append(
ChatMessage(content=interaction[0], role=MessageRole.USER)
)
)
if len(interaction) > 1 and interaction[1] is not None:
history_messages.append(
ChatMessage(
# Remove from history content the Sources information
content=interaction[1].split(SOURCES_SEPARATOR)[0],
role=MessageRole.ASSISTANT,
)
)
# max 20 messages to try to avoid context overflow
return history_messages[:20]