test: align graphrag agent expectations with strict output parsing

backend/tests/test_graphrag_agent.py

@@ -16,7 +16,7 @@ def test_generate_answer_appends_graph_context_without_changing_sources(monkeypa
 
     # Mock the executor and the tool
     mock_executor = MagicMock()
-    mock_executor.invoke.return_value = {"output": "Agent answer"}
+    mock_executor.invoke.return_value = {"output": '{"answer":"Agent answer"}'}
     
     mock_pdf_tool = MagicMock()
     mock_pdf_tool.last_sources = chunks
@@ -58,7 +58,7 @@ def test_generate_answer_stream_appends_graph_context(monkeypatch):
     mock_executor.stream.return_value = iter([
         {"actions": [MagicMock(log="Thought: I should search. Action: pdf_search")]},
         {"intermediate_steps": []}, # This triggers source yielding in my implementation if last_sources is set
-        {"output": "Final Answer: Streamed answer"}
+        {"output": 'Final Answer: {"answer":"Streamed answer"}'}
     ])
     
     mock_pdf_tool = MagicMock()
@@ -69,7 +69,7 @@ def test_generate_answer_stream_appends_graph_context(monkeypatch):
     events = list(agent.generate_answer_stream("OpenAI Microsoft", "user-1", "doc-1"))
 
     # Verify event types and data
-    assert any("Thinking" in e for e in events)
+    assert not any("Thinking" in e for e in events)
     assert any("Streamed answer" in e for e in events)
     assert any("Vector stream context" in e for e in events)
     assert events[-1] == f"data: {json.dumps({'type': 'done'})}\n\n"