Add Missing Dependencies In Codegen (#557)

* Add optional 'json' dependency in smithy-aws-core.

* Include optional dependencies on 'eventstream' and 'json' for smithy-aws-core when necessary

Author: jonathan343

codegen/core/src/main/java/software/amazon/smithy/python/codegen/ClientGenerator.java

@@ -222,6 +222,7 @@ raise ExpectationNotMetError("protocol and transport MUST be set on the config t
     private void generateEventStreamOperation(PythonWriter writer, OperationShape operation) {
         writer.pushState(new OperationSection(service, operation));
         writer.addDependency(SmithyPythonDependency.SMITHY_CORE);
+        writer.addDependency(SmithyPythonDependency.SMITHY_AWS_CORE.withOptionalDependencies("eventstream"));
         var operationSymbol = symbolProvider.toSymbol(operation);
         writer.putContext("operation", operationSymbol);
         var operationMethodSymbol = operationSymbol.expectProperty(OPERATION_METHOD);

codegen/core/src/main/java/software/amazon/smithy/python/codegen/integrations/RestJsonProtocolGenerator.java

@@ -83,7 +83,7 @@ public ApplicationProtocol getApplicationProtocol(GenerationContext context) {
 
     @Override
     public void initializeProtocol(GenerationContext context, PythonWriter writer) {
-        writer.addDependency(SmithyPythonDependency.SMITHY_AWS_CORE);
+        writer.addDependency(SmithyPythonDependency.SMITHY_AWS_CORE.withOptionalDependencies("json"));
         writer.addImport("smithy_aws_core.aio.protocols", "RestJsonClientProtocol");
         var serviceSymbol = context.symbolProvider().toSymbol(context.settings().service(context.model()));
         var serviceSchema = serviceSymbol.expectProperty(SymbolProperties.SCHEMA);

packages/smithy-aws-core/pyproject.toml

@@ -47,6 +47,9 @@ path = "src/smithy_aws_core/__init__.py"
 eventstream = [
     "smithy-aws-event-stream"
 ]
+json = [
+    "smithy-json"
+]
 
 [tool.hatch.build]
 exclude = [