github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-13 13:10:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:default setting prompt with history messages (#1117)

Browse Source 
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
Aries-ckt
2024-01-24 23:03:08 +08:00
committed by GitHub
parent 9a2b0e355e
commit be6718849f
2 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
dbgpt/core/interface/operators/message_operator.py
View File

@@ -416,6 +416,18 @@ class BufferedConversationMapperOperator(ConversationMapperOperator):
... ],
... ]
# Test end rounds is zero
>>> operator = BufferedConversationMapperOperator(
... keep_start_rounds=1, keep_end_rounds=0
... )
>>> assert operator._filter_round_messages(messages) == [
... [
... HumanMessage(content="Hi", round_index=1),
... AIMessage(content="Hello!", round_index=1),
... ]
... ]
Args:
messages_by_round (List[List[BaseMessage]]):
The messages grouped by round.
@@ -425,7 +437,12 @@ class BufferedConversationMapperOperator(ConversationMapperOperator):
"""
total_rounds = len(messages_by_round)
if self._keep_start_rounds is not None and self._keep_end_rounds is not None:
if (
self._keep_start_rounds is not None
and self._keep_end_rounds is not None
and self._keep_start_rounds > 0
and self._keep_end_rounds > 0
):
if self._keep_start_rounds + self._keep_end_rounds > total_rounds:
# Avoid overlapping when the sum of start and end rounds exceeds total
# rounds

5
dbgpt/storage/vector_store/milvus_store.py
View File

@@ -133,8 +133,9 @@ class MilvusStore(VectorStoreBase):
connections.connect(
host=self.uri or "127.0.0.1",
port=self.port or "19530",
alias="default"
# secure=self.secure,
username=self.username,
password=self.password,
alias="default",
)
def init_schema_and_load(self, vector_name, documents) -> List[str]: