helloparthshah commited on
Commit
53ee793
·
2 Parent(s): 1476939 6115f71

Merge branch 'main' of github.com:kunpai/HASHIRU

Browse files
src/tools/default_tools/agent_creater_tool.py CHANGED
@@ -1,6 +1,4 @@
1
  from src.manager.agent_manager import AgentManager
2
- from src.manager.config.model_selector import choose_best_model
3
- from src.manager.utils.runtime_selector import detect_runtime_environment
4
  from src.tools.default_tools.agent_cost_manager import AgentCostManager
5
  __all__ = ['AgentCreator']
6
 
@@ -40,10 +38,8 @@ class AgentCreator():
40
  def run(self, **kwargs):
41
  print("Running Agent Creator")
42
  agent_name = kwargs.get("agent_name")
43
- #base_model = kwargs.get("base_model")
44
- base_model = kwargs.get("base_model") or choose_best_model()
45
- env = detect_runtime_environment()
46
- print(f"\n[DEBUG] Detected Runtime Environment: {env}")
47
  print(f"[DEBUG] Selected Model: {base_model}")
48
 
49
  system_prompt = kwargs.get("system_prompt")
 
1
  from src.manager.agent_manager import AgentManager
 
 
2
  from src.tools.default_tools.agent_cost_manager import AgentCostManager
3
  __all__ = ['AgentCreator']
4
 
 
38
  def run(self, **kwargs):
39
  print("Running Agent Creator")
40
  agent_name = kwargs.get("agent_name")
41
+ base_model = kwargs.get("base_model")
42
+
 
 
43
  print(f"[DEBUG] Selected Model: {base_model}")
44
 
45
  system_prompt = kwargs.get("system_prompt")