diff --git a/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java b/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java
index bbe5a0181..eada94a19 100644
--- a/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java
+++ b/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java
@@ -156,6 +156,9 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
                 if (elements.getPackageOf(reqTypeElement).getSimpleName().toString().equals("index")) {
                     messageGenerateProcessor.messages.get(reqTypeName).addSuperinterface(Constant.STORE_REQ);
                 }
+                if (elements.getPackageOf(reqTypeElement).getSimpleName().toString().equals("document")) {
+                    messageGenerateProcessor.messages.get(reqTypeName).addSuperinterface(Constant.STORE_REQ);
+                }
                 serviceElements.add((TypeElement) element.getEnclosingElement());
                 entityElements.add((TypeElement) reqTypeElement.getEnclosingElement());
             } catch (Exception e) {