2929use EventEngine \Exception \MissingAggregateCollection ;
3030use EventEngine \Exception \NoDocumentStore ;
3131use EventEngine \Exception \RuntimeException ;
32+ use EventEngine \JsonSchema \JsonSchemaAwareRecord ;
3233use EventEngine \Logger \LogEngine ;
3334use EventEngine \Messaging \CommandDispatchResult ;
3435use EventEngine \Messaging \CommandDispatchResultCollection ;
@@ -522,8 +523,8 @@ public function registerResponseType(string $nameOrImmutableRecordClass, Respons
522523 if (null === $ schema ) {
523524 $ refObj = new \ReflectionClass ($ nameOrImmutableRecordClass );
524525
525- if (! $ refObj ->implementsInterface (ImmutableRecord ::class)) {
526- throw new InvalidArgumentException ("Invalid type given. $ nameOrImmutableRecordClass does not implement " . ImmutableRecord ::class);
526+ if (! $ refObj ->implementsInterface (JsonSchemaAwareRecord ::class)) {
527+ throw new InvalidArgumentException ("Invalid type given. $ nameOrImmutableRecordClass does not implement " . JsonSchemaAwareRecord ::class);
527528 }
528529
529530 $ name = \call_user_func ([$ nameOrImmutableRecordClass , '__type ' ]);
@@ -549,8 +550,8 @@ public function registerInputType(string $nameOrImmutableRecordClass, InputTypeS
549550 if (null === $ schema ) {
550551 $ refObj = new \ReflectionClass ($ nameOrImmutableRecordClass );
551552
552- if (! $ refObj ->implementsInterface (ImmutableRecord ::class)) {
553- throw new InvalidArgumentException ("Invalid type given. $ nameOrImmutableRecordClass does not implement " . ImmutableRecord ::class);
553+ if (! $ refObj ->implementsInterface (JsonSchemaAwareRecord ::class)) {
554+ throw new InvalidArgumentException ("Invalid type given. $ nameOrImmutableRecordClass does not implement " . JsonSchemaAwareRecord ::class);
554555 }
555556
556557 $ name = \call_user_func ([$ nameOrImmutableRecordClass , '__type ' ]);
0 commit comments