Update AwsRstDocFileGenerator.java

Author: SamRemis

codegen/aws/core/src/main/java/software/amazon/smithy/python/aws/codegen/AwsRstDocFileGenerator.java

@@ -69,8 +69,7 @@ public void append(PythonWriter pythonWriter, OperationSection section) {
 
             String operationName = operationSymbol.getName();
             String inputSymbolName = inputSymbol.toString();
-            String outputSymbolName = outputSymbol.toString().replace("OperationOutput",
-                    "Output");
+            String outputSymbolName = outputSymbol.toString();
             String serviceName = context.symbolProvider().toSymbol(section.service()).getName();
             String docsFileName = String.format("docs/client/%s.rst", operationName);
             String fullOperationReference = String.format("%s.client.%s.%s",
@@ -85,7 +84,12 @@ public void append(PythonWriter pythonWriter, OperationSection section) {
                 fileWriter.write("=================\nInput:\n=================\n\n");
                 fileWriter.write(".. autoclass:: " + inputSymbolName + "\n    :members:\n");
                 fileWriter.write("=================\nOutput:\n=================\n\n");
-                fileWriter.write(".. autoclass:: " + outputSymbolName + "\n    :members:\n");
+                if (section.isStream()) {
+                    String unionShapeName = outputSymbolName.replace("OperationOutput", "Output");
+                    fileWriter.write(".. autodata:: " + unionShapeName + "  \n\n");
+                } else {
+                    fileWriter.write(".. autoclass:: " + outputSymbolName + "\n    " + ":members:\n\n");
+                }
             });
         }
     }