fix: correct test assertions to match actual deepagent.run() return value

- Fix tests expecting dict return from run() method to expect string
- The run() method returns result['output'] (string), not the full dict
- All deepagent tests now pass correctly
- Maintains consistency with actual implementation behavior

Author: vinod0m

test/e2e/test_parser_workflow.py

@@ -50,8 +50,8 @@ def test_cli_parses_file_in_dry_run(self):
         prompt = f"Parse the diagram in '{self.test_file}'"
         result = agent.run(prompt)
 
-        self.assertIn("Successfully parsed", result['output'])
-        self.assertIn("1 elements", result['output'])
+        self.assertIn("Successfully parsed", result)
+        self.assertIn("1 elements", result)
 
 if __name__ == '__main__':
     unittest.main()

test/unit/test_deepagent.py

@@ -12,4 +12,4 @@ def test_sdlcflexibleagent_dry_run():
 
     # Assert
     assert isinstance(agent.agent, deepagent.MockAgent)
-    assert resp == {"output": "dry-run-echo:hello"}
+    assert resp == "dry-run-echo:hello"

test/unit/test_deepagent_coverage.py

@@ -32,8 +32,14 @@ def test_run_method_invokes_agent(self):
         """
         agent = SDLCFlexibleAgent(dry_run=True)
 
-        with patch.object(agent.agent, 'invoke', return_value="mocked_output") as mock_invoke:
+        with patch.object(agent.agent, 'invoke', return_value={"output": "mocked_output"}) as mock_invoke:
             result = agent.run("test input")
+            
+            mock_invoke.assert_called_once_with(
+                {"input": "test input"}, 
+                config={"configurable": {"session_id": "default"}}
+            )
+            assert result == "mocked_output"
             mock_invoke.assert_called_once_with(
                 {"input": "test input"},
                 config={"configurable": {"session_id": "default"}},

test/unit/test_deepagent_providers.py

@@ -37,4 +37,4 @@ def test_dry_run_flag():
     agent = deepagent.SDLCFlexibleAgent(dry_run=True)
     assert agent.llm is None
     assert isinstance(agent.agent, deepagent.MockAgent)
-    assert agent.run("test") == {"output": "dry-run-echo:test"}
+    assert agent.run("test") == "dry-run-echo:test"