llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Start in context approval state

Browse Source
This commit is contained in:
geens
2024-11-02 12:49:39 +01:00
parent 1974769eb4
commit a58ee40c2c
2 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sia/web_agent.py
View File

@@ -45,14 +45,14 @@ class WebAgent(BaseAgent):
Initialize web agent with required components. Initialize web agent with required components.
""" """
super().__init__(action_schema, working_memory, system_metrics, llm, validator, parser) super().__init__(action_schema, working_memory, system_metrics, llm, validator, parser)
self.context = ""
self.response = "" self.response = ""
self._system_prompt = system_prompt self._system_prompt = system_prompt
self._current_state = WebAgentState.UPDATE self._current_state = WebAgentState.CONTEXT_APPROVAL
self._validation_error: Optional[str] = None self._validation_error: Optional[str] = None
self._state_change_handlers: List[Callable[[WebAgentState], None]] = [] self._state_change_handlers: List[Callable[[WebAgentState], None]] = []
self._response_change_handlers: List[Callable[[str], None]] = [] self._response_change_handlers: List[Callable[[str], None]] = []
self._command_result: Optional[CommandResult] = None self._command_result: Optional[CommandResult] = None
self.context = self._compile_context()
@property @property

15
test/web_agent_test.py
View File

@@ -59,6 +59,12 @@ class WebAgentTest(unittest.TestCase):
self.mock_metrics.generate_context.side_effect = generate_context self.mock_metrics.generate_context.side_effect = generate_context
# Mock LLM response generator
def mock_infer(prompt: str, context: str):
yield "<reasoning>test reasoning</reasoning>"
self.mock_llm.infer.side_effect = mock_infer
# Create agent with all components # Create agent with all components
self.agent = WebAgent( self.agent = WebAgent(
system_prompt="test prompt", system_prompt="test prompt",
@@ -87,9 +93,9 @@ class WebAgentTest(unittest.TestCase):
def test_initialization(self): def test_initialization(self):
"""Test initial state of web agent.""" """Test initial state of web agent."""
self.assertEqual(self.agent.current_state, WebAgentState.UPDATE) self.assertEqual(self.agent.current_state, WebAgentState.CONTEXT_APPROVAL)
self.assertEqual(self.agent.context, "")
self.assertEqual(self.agent.response, "") self.assertEqual(self.agent.response, "")
self.assertIsNotNone(self.agent.context)
self.assertIsNone(self.agent._validation_error) self.assertIsNone(self.agent._validation_error)
self.assertEqual(len(self.agent._state_change_handlers), 0) self.assertEqual(len(self.agent._state_change_handlers), 0)
self.assertEqual(len(self.agent._response_change_handlers), 0) self.assertEqual(len(self.agent._response_change_handlers), 0)
@@ -112,7 +118,10 @@ class WebAgentTest(unittest.TestCase):
self.assertEqual(len(self.agent._response_change_handlers), 1) self.assertEqual(len(self.agent._response_change_handlers), 1)
def test_approve_context_state_error(self): def test_approve_context_state_error(self):
"""Test approving context in wrong state raises error.""" """Test approving context in wrong state."""
# Force agent into UPDATE state
self.agent._current_state = WebAgentState.UPDATE
with self.assertRaises(Exception) as context: with self.assertRaises(Exception) as context:
self.agent.approve_context() self.agent.approve_context()
self.assertIn("Not in CONTEXT_APPROVAL state", str(context.exception)) self.assertIn("Not in CONTEXT_APPROVAL state", str(context.exception))