dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: Fix token_count method to handle list of messages

Browse source
This commit is contained in:
Paul Gauthier 2024-08-07 13:13:10 -03:00 • committed by Paul Gauthier (aider)
parent ed38497c68
commit 0a5b5c3fd6
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aider/models.py
View file

@ -528,6 +528,9 @@ class Model:
return litellm.encode(model=self.name, text=text)
def token_count(self, messages):
if type(messages) is list:
return litellm.token_counter(model=self.name, messages=messages)
if not self.tokenizer:
return