refactor: unify streaming

Signed-off-by: Sebastian Krupinski <krupinski01@gmail.com>

src/services/entityService.ts

@@ -22,9 +22,7 @@ import type {
   EntityTransmitResponse,
   EntityInterface,
   EntityStreamRequest,
-  EntityStreamLine,
-  EntityStreamEntityLine,
-  EntityStreamDoneLine,
+  EntityStreamResponse,
 } from '../types/entity';
 import { useIntegrationStore } from '@KTXC/stores/integrationStore';
 import { EntityObject } from '../models';
@@ -177,24 +175,13 @@ export const entityService = {
     request: EntityStreamRequest,
     onEntity: (entity: EntityObject) => void
   ): Promise<{ total: number }> {
-    let total = 0;
-
-    await transceiveStream<EntityStreamRequest, EntityStreamLine>(
+    return await transceiveStream<EntityStreamRequest, EntityStreamResponse>(
       'entity.stream',
       request,
-      (line) => {
-        if (line.type === 'entity') {
-          onEntity(createEntityObject(line as EntityStreamEntityLine));
-        } else if (line.type === 'done') {
-          total = (line as EntityStreamDoneLine).total;
-        } else if (line.type === 'error') {
-          throw new Error(`[entity.stream] ${line.message}`);
-        }
-        // 'meta' lines are silently consumed
+      (entity) => {
+        onEntity(createEntityObject(entity));
       }
     );
-
-    return { total };
   },
 };