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 |
|