Merge pull request #155 from Cloud-Code-AI/153-work-summarizer-token-…

…counter-fix fix: Token counter fix
Cloud-Code-AI · May 29, 2024 · c194726 · c194726
2 parents 373635c + c6723c0
commit c194726
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/kaizen/llms/provider.py b/kaizen/llms/provider.py
@@ -74,7 +74,7 @@ def available_tokens(self, message, percentage=0.8):
         )
 
     def get_token_count(self, message):
-        return litellm.token_counter(self.model, message)
+        return litellm.token_counter(model=self.model, text=message)
 
     def update_usage(self, total_usage, current_usage):
         if total_usage is not None:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kaizen-cloudcode"
-version = "0.1.14"
+version = "0.1.15"
 description = "An intelligent coding companion that accelerates your development workflow by providing efficient assistance, enabling you to craft high-quality code more rapidly."
 authors = ["Saurav Panda <saurav.panda@cloudcode.ai>"]
 license = "Apache2.0"