andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix type issue

Browse Source
This commit is contained in:
Brandon Hancock
2024-12-09 15:27:36 -05:00
parent 0c5789f1e8
commit c89f475c4c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/llm.py
View File

@@ -131,7 +131,7 @@ class LLM:
self.api_version = api_version self.api_version = api_version
self.api_key = api_key self.api_key = api_key
self.callbacks = callbacks self.callbacks = callbacks
self.context_window_size = None self.context_window_size = 0
self.kwargs = kwargs self.kwargs = kwargs
litellm.drop_params = True litellm.drop_params = True
@@ -199,7 +199,7 @@ class LLM:
def get_context_window_size(self) -> int: def get_context_window_size(self) -> int:
# Only using 75% of the context window size to avoid cutting the message in the middle # Only using 75% of the context window size to avoid cutting the message in the middle
if self.context_window_size is not None: if self.context_window_size != 0:
return self.context_window_size return self.context_window_size
self.context_window_size = int( self.context_window_size = int(