From 54a18a58c13ecf255f313889b52c97f30eb2dd8f Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 29 Dec 2025 12:16:00 +0000
Subject: [PATCH 1/2] Initial plan


From e14e757c0340b8204619a7f73b3c683e0356e4e6 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 29 Dec 2025 12:20:01 +0000
Subject: [PATCH 2/2] Fix DataRepository to return proper trait implementation
 instead of lambda

Co-authored-by: akreit <64101884+akreit@users.noreply.github.com>
---
 .../database/DataRepository.scala             | 20 ++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/src/main/scala/com/async2databricks/database/DataRepository.scala b/src/main/scala/com/async2databricks/database/DataRepository.scala
index 10c5714..c1100c5 100644
--- a/src/main/scala/com/async2databricks/database/DataRepository.scala
+++ b/src/main/scala/com/async2databricks/database/DataRepository.scala
@@ -17,15 +17,17 @@ trait DataRepository[F[_]] {
 object DataRepository extends LazyLogging {
 
   def apply[F[_]: Async](xa: Transactor[F]): DataRepository[F] =
-    (query: String, batchSize: Int) => {
-      logger.info(s"Starting to stream data with query: $query")
+    new DataRepository[F] {
+      override def streamData(query: String, batchSize: Int): Stream[F, SampleData] = {
+        logger.info(s"Starting to stream data with query: $query")
 
-      Fragment
-        .const(query)
-        .query[SampleData]
-        .stream
-        .transact(xa)
-        .chunkN(batchSize)
-        .flatMap(chunk => Stream.chunk(chunk))
+        Fragment
+          .const(query)
+          .query[SampleData]
+          .stream
+          .transact(xa)
+          .chunkN(batchSize)
+          .flatMap(chunk => Stream.chunk(chunk))
+      }
     }
 }