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-
|
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-
|
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,
|