[DOP-23921] Include sum inputs & outputs to lineage response

Author: dolfinus

data_rentgen/db/repositories/input.py

@@ -70,6 +70,7 @@ async def create_or_update_bulk(self, inputs: list[InputDTO]) -> list[Input]:
     async def list_by_operation_ids(
         self,
         operation_ids: Sequence[UUID],
+        granularity: Literal["JOB", "RUN", "OPERATION"],
     ) -> list[Input]:
         if not operation_ids:
             return []
@@ -85,14 +86,14 @@ async def list_by_operation_ids(
             Input.operation_id == any_(operation_ids),  # type: ignore[arg-type]
         ]
 
-        return await self._get_inputs(where, "OPERATION")
+        return await self._get_inputs(where, granularity)
 
     async def list_by_run_ids(
         self,
         run_ids: Sequence[UUID],
         since: datetime,
         until: datetime | None,
-        granularity: Literal["RUN", "OPERATION"],
+        granularity: Literal["JOB", "RUN", "OPERATION"],
     ) -> list[Input]:
         if not run_ids:
             return []

data_rentgen/db/repositories/output.py

@@ -71,6 +71,7 @@ async def create_or_update_bulk(self, outputs: list[OutputDTO]) -> list[Output]:
     async def list_by_operation_ids(
         self,
         operation_ids: Sequence[UUID],
+        granularity: Literal["JOB", "RUN", "OPERATION"],
     ) -> list[Output]:
         if not operation_ids:
             return []
@@ -86,14 +87,14 @@ async def list_by_operation_ids(
             Output.operation_id == any_(operation_ids),  # type: ignore[arg-type]
         ]
 
-        return await self._get_outputs(where, "OPERATION")
+        return await self._get_outputs(where, granularity)
 
     async def list_by_run_ids(
         self,
         run_ids: Sequence[UUID],
         since: datetime,
         until: datetime | None,
-        granularity: Literal["RUN", "OPERATION"],
+        granularity: Literal["JOB", "RUN", "OPERATION"],
     ) -> list[Output]:
         if not run_ids:
             return []