feat: added v1 versioning segment updates tools (#7489)

Author: Zaimwa9

api/features/feature_segments/views.py

@@ -27,7 +27,28 @@
 
 @method_decorator(
     name="list",
-    decorator=extend_schema(parameters=[FeatureSegmentQuerySerializer]),
+    decorator=extend_schema(
+        tags=["mcp"],
+        parameters=[FeatureSegmentQuerySerializer],
+        extensions={
+            "x-gram": {
+                "name": "list_feature_segments",
+                "description": "Lists segment overrides for a feature in an environment.",
+            },
+        },
+    ),
+)
+@method_decorator(
+    name="destroy",
+    decorator=extend_schema(
+        tags=["mcp"],
+        extensions={
+            "x-gram": {
+                "name": "delete_feature_segment",
+                "description": "Deletes a segment override. Use this tool for environments without v2 feature versioning (use_v2_feature_versioning: false).",
+            },
+        },
+    ),
 )
 class FeatureSegmentViewSet(
     viewsets.ModelViewSet,  # type: ignore[type-arg]

api/features/versioning/views.py

@@ -54,7 +54,7 @@
         extensions={
             "x-gram": {
                 "name": "get_environment_feature_versions",
-                "description": "Retrieves version information for a feature flag in a specific environment. Use this for environments with v2 feature versioning.",
+                "description": "Retrieves version information for a feature flag in a specific environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     ),
@@ -66,7 +66,7 @@
         extensions={
             "x-gram": {
                 "name": "create_environment_feature_version",
-                "description": "Creates a new version for a feature flag in a specific environment. Use this for environments with v2 feature versioning.",
+                "description": "Creates a new version for a feature flag in a specific environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     ),
@@ -163,7 +163,7 @@ def perform_destroy(self, instance: EnvironmentFeatureVersion) -> None:
         extensions={
             "x-gram": {
                 "name": "publish_environment_feature_version",
-                "description": "Publishes a feature version to make it live in the environment. Use this for environments with v2 feature versioning.",
+                "description": "Publishes a feature version to make it live in the environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     )
@@ -226,7 +226,7 @@ def get_queryset(self):  # type: ignore[no-untyped-def]
         extensions={
             "x-gram": {
                 "name": "get_environment_feature_version_states",
-                "description": "Retrieves feature state information for a specific version in an environment. Use this for environments with v2 feature versioning.",
+                "description": "Retrieves feature state information for a specific version in an environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     ),
@@ -238,7 +238,7 @@ def get_queryset(self):  # type: ignore[no-untyped-def]
         extensions={
             "x-gram": {
                 "name": "create_environment_feature_version_state",
-                "description": "Creates a new feature state for a specific version in an environment. Use this for environments with v2 feature versioning.",
+                "description": "Creates a new feature state for a specific version in an environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     ),
@@ -250,7 +250,7 @@ def get_queryset(self):  # type: ignore[no-untyped-def]
         extensions={
             "x-gram": {
                 "name": "update_environment_feature_version_state",
-                "description": "Updates an existing feature state for a specific version in an environment. Use this for environments with v2 feature versioning.",
+                "description": "Updates an existing feature state for a specific version in an environment. Use this tool for environments with v2 feature versioning (use_v2_feature_versioning: true).",
             },
         },
     ),

api/features/views.py

@@ -829,7 +829,7 @@ def update_feature_state_value(self, value, feature_state):  # type: ignore[no-u
         extensions={
             "x-gram": {
                 "name": "update_environment_feature_state",
-                "description": "Updates a feature state in an environment, including enabled status and value. Use this for environments without v2 feature versioning.",
+                "description": "Updates a feature state in an environment, including enabled status and value. Use this tool for environments without v2 feature versioning (use_v2_feature_versioning: false).",
             },
         },
     ),
@@ -920,6 +920,18 @@ def clone_from_given_identity(self, request, *args, **kwargs) -> Response:  # ty
         ]
     ),
 )
+@method_decorator(
+    name="update",
+    decorator=extend_schema(
+        tags=["mcp"],
+        extensions={
+            "x-gram": {
+                "name": "update_feature_state",
+                "description": "Updates a feature state, including its enabled status and value. Use this tool to update a segment override's value for environments without v2 feature versioning (use_v2_feature_versioning: false).",
+            },
+        },
+    ),
+)
 class SimpleFeatureStateViewSet(
     mixins.CreateModelMixin,
     mixins.UpdateModelMixin,
@@ -1155,8 +1167,15 @@ def organisation_has_got_feature(request, organisation):  # type: ignore[no-unty
 
 
 @extend_schema(
+    tags=["mcp"],
     request=CustomCreateSegmentOverrideFeatureStateSerializer(),
     responses={201: CustomCreateSegmentOverrideFeatureStateSerializer()},
+    extensions={
+        "x-gram": {
+            "name": "create_segment_override",
+            "description": "Creates a segment override for a feature in an environment in a single call, setting both the segment binding and its value. Use this tool for environments without v2 feature versioning (use_v2_feature_versioning: false).",
+        },
+    },
 )
 @api_view(["POST"])
 @permission_classes([CreateSegmentOverridePermissions])