Fix
This commit is contained in:
parent
67779b8335
commit
543f8229d2
2 changed files with 7 additions and 3 deletions
|
@ -27,6 +27,9 @@ The bot can be run with `python -m gptbot`. If required, activate a venv first.
|
|||
You may want to run the bot in a screen or tmux session, or use a process
|
||||
manager like systemd.
|
||||
|
||||
Once it is running, just invite it to a room and it will start responding to
|
||||
messages.
|
||||
|
||||
## License
|
||||
|
||||
This project is licensed under the terms of the MIT license.
|
|
@ -77,7 +77,7 @@ async def fetch_last_n_messages(room_id, n=20):
|
|||
room = await client.join(room_id)
|
||||
messages = []
|
||||
|
||||
logging(f"Fetching last {n} messages from room {room_id}...")
|
||||
logging(f"Fetching last {n} messages from room {room_id} (starting at {SYNC_TOKEN})...")
|
||||
|
||||
response = await client.room_messages(
|
||||
room_id=room_id,
|
||||
|
@ -116,7 +116,7 @@ def truncate_messages_to_fit_tokens(messages, max_tokens=MAX_TOKENS):
|
|||
total_tokens = len(SYSTEM_MESSAGE) + 1
|
||||
truncated_messages = []
|
||||
|
||||
for message in messages[0] + reversed(messages[1:]):
|
||||
for message in [messages[0]] + list(reversed(messages[1:])):
|
||||
content = message["content"]
|
||||
tokens = len(encoding.encode(content)) + 1
|
||||
if total_tokens + tokens > max_tokens:
|
||||
|
@ -124,7 +124,7 @@ def truncate_messages_to_fit_tokens(messages, max_tokens=MAX_TOKENS):
|
|||
total_tokens += tokens
|
||||
truncated_messages.append(message)
|
||||
|
||||
return list(truncated_messages[0] + reversed(truncated_messages[1:]))
|
||||
return [truncated_messages[0]] + list(reversed(truncated_messages[1:]))
|
||||
|
||||
|
||||
async def message_callback(room: MatrixRoom, event: RoomMessageText):
|
||||
|
@ -220,6 +220,7 @@ async def accept_pending_invites():
|
|||
|
||||
|
||||
async def sync_cb(response):
|
||||
global SYNC_TOKEN
|
||||
logging(f"Sync response received (next batch: {response.next_batch})")
|
||||
SYNC_TOKEN = response.next_batch
|
||||
|
||||
|
|
Loading…
Reference in a new issue