misc: updated tests

Author: jublankeviadee

tests/utest/test_base_locator_agent.py

@@ -131,6 +131,8 @@ def __init__(self) -> None:
             self.use_llm_for_locator_generation: bool = True
             self.locator_agent_provider: str = "prov"
             self.locator_agent_model: str = "mod"
+            self.locator_agent_temperature: float = 0.1
+            self.orchestrator_agent_temperature: float = 0.1
 
     cfg_mod.Cfg = Cfg
     _force_module("SelfhealingAgents.utils", aid_utils)
@@ -332,6 +334,8 @@ class Cfg:
             use_llm_for_locator_generation: bool = use_llm
             locator_agent_provider: str = "prov"
             locator_agent_model: str = "mod"
+            locator_agent_temperature: float = 0.1
+            orchestrator_agent_temperature: float = 0.1
 
         return Impl(Cfg(), dom)
 
@@ -452,4 +456,4 @@ def test_sort_and_filter_helpers(mod_and_cls: Tuple[Any, Any, Any, Any]) -> None
     sorted_list = inst._sort_locators(["b", "a", "c"])
     assert sorted_list[:2] == ["a", "c"]
     filtered = inst._filter_clickable_locators(["a", "b", "c"])
-    assert filtered == ["a", "c"]
+    assert filtered == ["a", "c"]

tests/utest/test_orchestrator_agent.py

@@ -247,6 +247,7 @@ class FakeCfg:
         total_tokens_limit: int = 1000
         orchestrator_agent_provider: str = "prov"
         orchestrator_agent_model: str = "mod"
+        orchestrator_agent_temperature: float = 0.1
 
     orch = OrchestratorAgent(FakeCfg(), _FakeLocatorAgent(is_failed=False))
     payload = PromptPayload(
@@ -277,6 +278,7 @@ class FakeCfg:
         total_tokens_limit: int = 1000
         orchestrator_agent_provider: str = "prov"
         orchestrator_agent_model: str = "mod"
+        orchestrator_agent_temperature: float = 0.1
 
     orch = OrchestratorAgent(FakeCfg(), _FakeLocatorAgent(is_failed=True))
     payload = PromptPayload(
@@ -316,6 +318,7 @@ class FakeCfg:
         total_tokens_limit: int = 1000
         orchestrator_agent_provider: str = "prov"
         orchestrator_agent_model: str = "mod"
+        orchestrator_agent_temperature: float = 0.1
 
     orch = OrchestratorAgent(FakeCfg(), _FakeLocatorAgent(is_failed=True))
     payload = PromptPayload(
@@ -353,6 +356,7 @@ class FakeCfg:
         total_tokens_limit: int = 1000
         orchestrator_agent_provider: str = "prov"
         orchestrator_agent_model: str = "mod"
+        orchestrator_agent_temperature: float = 0.1
 
     orch = OrchestratorAgent(
         FakeCfg(),
@@ -373,6 +377,7 @@ class FakeCfg:
         total_tokens_limit: int = 1000
         orchestrator_agent_provider: str = "prov"
         orchestrator_agent_model: str = "mod"
+        orchestrator_agent_temperature: float = 0.1
 
     orch = OrchestratorAgent(
         FakeCfg(), _FakeLocatorAgent(is_failed=True, raise_on_heal=True)
@@ -390,4 +395,4 @@ def test_catch_token_limit_exceedance_logs(
     assert logger.infos == ["error: out of tokens"]
     logger.infos.clear()
     OrchestratorAgent._catch_token_limit_exceedance("ok")
-    assert logger.infos == []
+    assert logger.infos == []