Update identity related fields in protocol move request (#36662)

* Update source identity type in move request

* make protobuf

* Update type for identity upgrade

* make protobuf

docs/plugin-protocol/tfplugin5.9.proto

@@ -476,7 +476,7 @@ message UpgradeResourceIdentity {
         // not have access to the identity schema of prior_version, so it's the
         // provider's responsibility to interpret this value using the
         // appropriate older schema. The raw_identity will be json encoded.
-        bytes raw_identity = 3;
+        RawState raw_identity = 3;
     }
 
     message Response {
@@ -677,7 +677,14 @@ message MoveResourceState {
         // The private state of the resource being moved.
         bytes source_private = 6;
 
-        ResourceIdentityData source_identity = 7;
+        // The raw identity of the resource being moved. Only the json field is
+        // populated, as there should be no legacy providers using the flatmap
+        // format that support newly introduced RPCs.
+        RawState source_identity = 7;
+
+        // The identity schema version of the resource type that the resource
+        // is being moved from.
+        int64 source_identity_schema_version = 8;
     }
 
     message Response {

docs/plugin-protocol/tfplugin6.9.proto

@@ -493,7 +493,7 @@ message UpgradeResourceIdentity {
         // not have access to the identity schema of prior_version, so it's the
         // provider's responsibility to interpret this value using the
         // appropriate older schema. The raw_identity will be json encoded.
-        bytes raw_identity = 3;
+        RawState raw_identity = 3;
     }
 
     message Response {
@@ -693,7 +693,14 @@ message MoveResourceState {
         // The private state of the resource being moved.
         bytes source_private = 6;
 
-        ResourceIdentityData source_identity = 7;
+        // The raw identity of the resource being moved. Only the json field is
+        // populated, as there should be no legacy providers using the flatmap
+        // format that support newly introduced RPCs.
+        RawState source_identity = 7;
+
+        // The identity schema version of the resource type that the resource
+        // is being moved from.
+        int64 source_identity_schema_version = 8;
     }
 
     message Response {