chore(core/protocols): make error lookup in runtime protocol consistent with existing codegen (#7478)

Author: kuhe

packages/core/src/submodules/protocols/ProtocolLib.ts

@@ -1,4 +1,4 @@
-import { ErrorSchema, NormalizedSchema, TypeRegistry } from "@smithy/core/schema";
+import { NormalizedSchema, TypeRegistry } from "@smithy/core/schema";
 import type { HttpResponse as IHttpResponse, MetadataBearer, ResponseMetadata, StaticErrorSchema } from "@smithy/types";
 
 /**

packages/core/src/submodules/protocols/query/AwsQueryProtocol.ts

@@ -158,10 +158,15 @@ export class AwsQueryProtocol extends RpcProtocol {
       response,
       errorData,
       metadata,
-      (registry: TypeRegistry, errorName: string) =>
-        registry.find(
-          (schema) => (NormalizedSchema.of(schema).getMergedTraits().awsQueryError as any)?.[0] === errorName
-        ) as StaticErrorSchema
+      (registry: TypeRegistry, errorName: string) => {
+        try {
+          return registry.getSchema(errorName) as StaticErrorSchema;
+        } catch (e) {
+          return registry.find(
+            (schema) => (NormalizedSchema.of(schema).getMergedTraits().awsQueryError as any)?.[0] === errorName
+          ) as StaticErrorSchema;
+        }
+      }
     );
 
     const ns = NormalizedSchema.of(errorSchema);