From 716b48c543ae9d3802db481999f022270ddbdbc1 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 3 Sep 2025 16:48:59 +0100
Subject: [PATCH] Fix contract tests

---
 tests/contract/conftest.py               | 2 +-
 tests/contract/test_message_contracts.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/contract/conftest.py b/tests/contract/conftest.py
index 5c5b82cb..3d184d3d 100644
--- a/tests/contract/conftest.py
+++ b/tests/contract/conftest.py
@@ -82,8 +82,8 @@ def sample_message_data():
         },
         "AgentRequest": {
             "question": "What is machine learning?",
-            "plan": "",
             "state": "",
+            "group": [],
             "history": []
         },
         "AgentResponse": {
diff --git a/tests/contract/test_message_contracts.py b/tests/contract/test_message_contracts.py
index 861e5368..e4779d8b 100644
--- a/tests/contract/test_message_contracts.py
+++ b/tests/contract/test_message_contracts.py
@@ -198,8 +198,8 @@ class TestAgentMessageContracts:
         # Test required fields
         request = AgentRequest(**request_data)
         assert hasattr(request, 'question')
-        assert hasattr(request, 'plan')
         assert hasattr(request, 'state')
+        assert hasattr(request, 'group')
         assert hasattr(request, 'history')
 
     def test_agent_response_schema_contract(self, sample_message_data):