KevinHuSh
commited on
Commit
·
eb21264
1
Parent(s):
dd8762b
fix ollama bug (#999)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
api/db/services/dialog_service.py
CHANGED
|
@@ -58,17 +58,17 @@ def message_fit_in(msg, max_length=4000):
|
|
| 58 |
if c < max_length:
|
| 59 |
return c, msg
|
| 60 |
|
| 61 |
-
ll = num_tokens_from_string(msg_[0]
|
| 62 |
-
l = num_tokens_from_string(msg_[-1]
|
| 63 |
if ll / (ll + l) > 0.8:
|
| 64 |
-
m = msg_[0]
|
| 65 |
m = encoder.decode(encoder.encode(m)[:max_length - l])
|
| 66 |
-
msg[0]
|
| 67 |
return max_length, msg
|
| 68 |
|
| 69 |
-
m = msg_[1]
|
| 70 |
m = encoder.decode(encoder.encode(m)[:max_length - l])
|
| 71 |
-
msg[1]
|
| 72 |
return max_length, msg
|
| 73 |
|
| 74 |
|
|
|
|
| 58 |
if c < max_length:
|
| 59 |
return c, msg
|
| 60 |
|
| 61 |
+
ll = num_tokens_from_string(msg_[0]["content"])
|
| 62 |
+
l = num_tokens_from_string(msg_[-1]["content"])
|
| 63 |
if ll / (ll + l) > 0.8:
|
| 64 |
+
m = msg_[0]["content"]
|
| 65 |
m = encoder.decode(encoder.encode(m)[:max_length - l])
|
| 66 |
+
msg[0]["content"] = m
|
| 67 |
return max_length, msg
|
| 68 |
|
| 69 |
+
m = msg_[1]["content"]
|
| 70 |
m = encoder.decode(encoder.encode(m)[:max_length - l])
|
| 71 |
+
msg[1]["content"] = m
|
| 72 |
return max_length, msg
|
| 73 |
|
| 74 |
|