Browse Source

Fix: truncate message issue. (#5776)

### What problem does this PR solve?

Close #5761
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.17.1
Kevin Hu 7 months ago
parent
commit
2ad852d8df
No account linked to committer's email address
1 changed files with 2 additions and 2 deletions
  1. 2
    2
      rag/prompts.py

+ 2
- 2
rag/prompts.py View File

@@ -86,9 +86,9 @@ def message_fit_in(msg, max_length=4000):
msg[0]["content"] = m
return max_length, msg

m = msg_[1]["content"]
m = msg_[-1]["content"]
m = encoder.decode(encoder.encode(m)[:max_length - ll2])
msg[1]["content"] = m
msg[-1]["content"] = m
return max_length, msg



Loading…
Cancel
Save