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

Co-authored-by: Fangyin Cheng <[email protected]>
eosphoros-ai · Jan 24, 2024 · be67188 · be67188
1 parent 9a2b0e3
commit be67188
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/dbgpt/core/interface/operators/message_operator.py b/dbgpt/core/interface/operators/message_operator.py
@@ -416,6 +416,18 @@ def _filter_round_messages(
             ...     ],
             ... ]
 
+            # 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 @@ def _filter_round_messages(
 
         """
         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

diff --git a/dbgpt/storage/vector_store/milvus_store.py b/dbgpt/storage/vector_store/milvus_store.py
@@ -133,8 +133,9 @@ def __init__(self, vector_store_config: MilvusVectorConfig) -> None:
         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]: