feat: add protocol object for mcp implementation

packages/toolbox-core/src/toolbox_core/protocol.py

@@ -11,12 +11,32 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
+from enum import Enum
 from inspect import Parameter
 from typing import Any, Optional, Type, Union
 
 from pydantic import BaseModel
 
+
+class Protocol(str, Enum):
+    """Defines how the client should choose between communication protocols."""
+
+    TOOLBOX = "toolbox"
+    MCP_v20250618 = "2025-06-18"
+    MCP_v20250326 = "2025-03-26"
+    MCP_v20241105 = "2024-11-05"
+    MCP_LATEST = MCP_v20250618
+    MCP = MCP_LATEST
+
+    @classmethod
+    def get_supported_mcp_versions(cls):
+        """Returns a list of supported MCP versions, sorted from newest to oldest."""
+        versions = [member for member in cls if member.name.startswith("MCP_v")]
+        # Sort by the version date in descending order
+        versions.sort(key=lambda x: x.value, reverse=True)
+        return [v.value for v in versions]
+
+
 __TYPE_MAP = {
     "string": str,
     "integer": int,

packages/toolbox-core/tests/test_protocol.py

@@ -18,7 +18,20 @@
 
 import pytest
 
-from toolbox_core.protocol import AdditionalPropertiesSchema, ParameterSchema
+from toolbox_core.protocol import AdditionalPropertiesSchema, ParameterSchema, Protocol
+
+
+def test_get_supported_mcp_versions():
+    """
+    Tests that get_supported_mcp_versions returns the correct list of versions,
+    sorted from newest to oldest.
+    """
+    expected_versions = ["2025-06-18", "2025-03-26", "2024-11-05"]
+    supported_versions = Protocol.get_supported_mcp_versions()
+
+    assert supported_versions == expected_versions
+    # Also verify that the non-MCP members are not included
+    assert "toolbox" not in supported_versions
 
 
 def test_parameter_schema_float():