helloparthshah commited on
Commit
7d69384
·
1 Parent(s): 3fa3bab

updated budget

Browse files
src/manager/budget_manager.py CHANGED
@@ -46,7 +46,7 @@ class BudgetManager():
46
  ram_mem = mem.total/ 1024 ** 3
47
  print("No GPU detected. Using CPU.")
48
  print(f"Total RAM: {ram_mem:.2f} GB")
49
- total_mem = gpu_mem + ram_mem
50
  return round((total_mem / 16) * 100)
51
 
52
  def get_total_resource_budget(self):
 
46
  ram_mem = mem.total/ 1024 ** 3
47
  print("No GPU detected. Using CPU.")
48
  print(f"Total RAM: {ram_mem:.2f} GB")
49
+ total_mem = gpu_mem + ram_mem/4
50
  return round((total_mem / 16) * 100)
51
 
52
  def get_total_resource_budget(self):
src/tools/default_tools/agent_cost_manager.py CHANGED
@@ -30,12 +30,12 @@ class AgentCostManager():
30
  "create_resource_cost": 28,
31
  "invoke_resource_cost": 35,
32
  },
33
- "gemini-2.5-flash-preview-04-17": {
34
  "description": "Avg Accuracy: 75.8%, 82.8% on LegalBench, 81.6% on multi-task understanding, 91.6% on Math",
35
  "create_expense_cost": 0,
36
  "invoke_expense_cost": 0.15,
37
  },
38
- "gemini-2.5-pro-preview-05-06": {
39
  "description": "Avg Accuracy: 64.3%, 83.6% on LegalBench, 84.1% on multi-task understanding, 95.2% on Math, 63.8% on Coding",
40
  "create_expense_cost": 0,
41
  "invoke_expense_cost": 1.25,
 
30
  "create_resource_cost": 28,
31
  "invoke_resource_cost": 35,
32
  },
33
+ "gemini-2.5-flash-preview-05-20": {
34
  "description": "Avg Accuracy: 75.8%, 82.8% on LegalBench, 81.6% on multi-task understanding, 91.6% on Math",
35
  "create_expense_cost": 0,
36
  "invoke_expense_cost": 0.15,
37
  },
38
+ "gemini-2.5-pro-exp-03-25": {
39
  "description": "Avg Accuracy: 64.3%, 83.6% on LegalBench, 84.1% on multi-task understanding, 95.2% on Math, 63.8% on Coding",
40
  "create_expense_cost": 0,
41
  "invoke_expense_cost": 1.25,