diff --git a/core/services/nodes/inflight.go b/core/services/nodes/inflight.go index 85b10f71c..b51ef6001 100644 --- a/core/services/nodes/inflight.go +++ b/core/services/nodes/inflight.go @@ -218,6 +218,12 @@ func (c *InFlightTrackingClient) Score(ctx context.Context, in *pb.ScoreRequest, return res, c.reconcile(err) } +func (c *InFlightTrackingClient) SoundDetection(ctx context.Context, in *pb.SoundDetectionRequest, opts ...ggrpc.CallOption) (*pb.SoundDetectionResponse, error) { + defer c.track(ctx)() + res, err := c.Backend.SoundDetection(ctx, in, opts...) + return res, c.reconcile(err) +} + func (c *InFlightTrackingClient) AudioEncode(ctx context.Context, in *pb.AudioEncodeRequest, opts ...ggrpc.CallOption) (*pb.AudioEncodeResult, error) { defer c.track(ctx)() res, err := c.Backend.AudioEncode(ctx, in, opts...) diff --git a/core/services/nodes/inflight_test.go b/core/services/nodes/inflight_test.go index 2eb90f9c6..5fc9820e7 100644 --- a/core/services/nodes/inflight_test.go +++ b/core/services/nodes/inflight_test.go @@ -408,6 +408,13 @@ var _ = Describe("InFlightTrackingClient", func() { return err }) }) + + It("SoundDetection", func() { + assertTracked(func() error { + _, err := client.SoundDetection(context.Background(), &pb.SoundDetectionRequest{}) + return err + }) + }) }) Describe("stale model reload (self-heal)", func() {