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)

Files changed (1) hide show
  1. api/db/services/dialog_service.py +6 -6
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].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
 
 
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