Pulastya B commited on
Commit
e5a2aca
·
1 Parent(s): d36c362

Fixed Agent Reasoning-1

Browse files
Files changed (1) hide show
  1. src/orchestrator.py +5 -6
src/orchestrator.py CHANGED
@@ -2990,12 +2990,11 @@ You receive quality reports from EDA agent and deliver clean data to modeling ag
2990
  response_message = None
2991
 
2992
  # 💰 TOKEN BUDGET: Enforce context window limits before LLM call
2993
- if self.token_manager.enabled:
2994
- messages, token_count = self.token_manager.enforce_budget(
2995
- messages=messages,
2996
- system_prompt=system_prompt
2997
- )
2998
- print(f"💰 Token budget: {token_count}/{self.token_manager.max_tokens} tokens")
2999
 
3000
  # Call LLM with function calling (provider-specific)
3001
  if self.provider == "mistral":
 
2990
  response_message = None
2991
 
2992
  # 💰 TOKEN BUDGET: Enforce context window limits before LLM call
2993
+ messages, token_count = self.token_manager.enforce_budget(
2994
+ messages=messages,
2995
+ system_prompt=system_prompt
2996
+ )
2997
+ print(f"💰 Token budget: {token_count}/{self.token_manager.max_tokens} ({(token_count/self.token_manager.max_tokens*100):.1f}%)")
 
2998
 
2999
  # Call LLM with function calling (provider-specific)
3000
  if self.provider == "mistral":