Skip to content

Commit f338cfb

Browse files
committed
🐛 Throws an exception when there are more tokens available #954
1 parent 5534686 commit f338cfb

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

sdk/nexent/core/models/openai_long_context_model.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,12 @@ def prepare_long_text_message(self, text_content: str, system_prompt: str, user_
161161
# Reserve tokens for text content
162162
available_tokens = self.max_context_tokens - system_tokens - user_prompt_tokens - 100 # Reserve 100 tokens as buffer
163163

164+
# Check if there are sufficient tokens available
165+
if available_tokens <= 0:
166+
error_msg = f"Insufficient tokens available. Required: {system_tokens + user_prompt_tokens + 100}, Available: {self.max_context_tokens}, Shortage: {abs(available_tokens)}"
167+
logger.error(error_msg)
168+
raise ValueError(error_msg)
169+
164170
# Truncate the text content
165171
truncated_text = self.truncate_text(text_content, available_tokens)
166172
final_content_tokens = self.count_tokens(truncated_text)

0 commit comments

Comments
 (0)