Spaces:
Running
Running
Pulastya B commited on
Commit ·
e5a2aca
1
Parent(s): d36c362
Fixed Agent Reasoning-1
Browse files- 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 |
-
|
| 2994 |
-
messages
|
| 2995 |
-
|
| 2996 |
-
|
| 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":
|