Skip to content

Commit 072192a

Browse files
committed
Fix adding messages
1 parent b3d6e84 commit 072192a

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

llms_wrapper/chatbot.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def __init__(self, llm, config=None, initial_message=None, message_template=None
7171
self.llm_messages = []
7272
if initial_message:
7373
self.initial_message = any2message(initial_message)
74-
self.llm_messages.append(initial_message)
74+
self.llm_messages.extend(initial_message)
7575
else:
7676
self.initial_message = None
7777
self.message_template = message_template
@@ -100,7 +100,7 @@ def reply(
100100
else:
101101
message = any2message(message, vars=metadata)
102102
# Add the message to the chat history
103-
self.llm_messages.append(message)
103+
self.llm_messages.extend(message)
104104

105105
ret = self.llm.query(self.llm_messages, return_cost=True)
106106
if ret.get("error"):
@@ -948,4 +948,4 @@ def run_sync_example(schbot: SerialChatbot):
948948
run_sync_example(schbot)
949949
print("\n" + "="*60 + "\n")
950950

951-
logger.debug("Examples finished.")
951+
logger.debug("Examples finished.")

llms_wrapper/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
import importlib.metadata
2-
__version__ = "0.4.0"
2+
__version__ = "0.4.1"
33

0 commit comments

Comments
 (0)