Formatting

Author: peterbanda

anthropic-client/src/main/scala/io/cequence/openaiscala/anthropic/service/impl/AwsEventStreamBytesDecoder.scala

@@ -11,15 +11,15 @@ object AwsEventStreamBytesDecoder {
     // eventJson might look like:
     // { ":message-type":"event", ":event-type":"...", "bytes":"base64string" }
 
-    val base64Str = (eventJson \ "bytes").asOpt[String]
-    base64Str match {
-      case Some(encoded) =>
+    (eventJson \ "bytes")
+      .asOpt[String]
+      .map { encoded =>
         val decoded = Base64.getDecoder.decode(encoded)
         Json.parse(decoded)
-      case None =>
+      }
+      .getOrElse(
         // If there's no "bytes" field, return the original JSON (or handle differently)
         eventJson
-    }
+      )
   }
 }
-

anthropic-client/src/main/scala/io/cequence/openaiscala/anthropic/service/impl/AwsEventStreamEventParser.scala

@@ -17,4 +17,3 @@ object AwsEventStreamEventParser {
       None
   }
 }
-

anthropic-client/src/main/scala/io/cequence/openaiscala/anthropic/service/impl/AwsEventStreamFrameDecoder.scala

@@ -14,30 +14,33 @@ class AwsEventStreamFrameDecoder extends GraphStage[FlowShape[ByteString, ByteSt
   override def createLogic(attrs: Attributes): GraphStageLogic = new GraphStageLogic(shape) {
     var buffer = ByteString.empty
 
-    setHandler(in, new InHandler {
-      override def onPush(): Unit = {
-        buffer ++= grab(in)
-        emitFrames()
-      }
-      override def onUpstreamFinish(): Unit = {
-        emitFrames()
-        if (buffer.isEmpty) completeStage()
-        else failStage(new RuntimeException("Truncated frame at stream end"))
+    setHandler(
+      in,
+      new InHandler {
+        override def onPush(): Unit = {
+          buffer ++= grab(in)
+          emitFrames()
+        }
+        override def onUpstreamFinish(): Unit = {
+          emitFrames()
+          if (buffer.isEmpty) completeStage()
+          else failStage(new RuntimeException("Truncated frame at stream end"))
+        }
       }
-    })
+    )
 
-    setHandler(out, new OutHandler {
-      override def onPull(): Unit = {
-        if (!hasBeenPulled(in)) pull(in)
+    setHandler(
+      out,
+      new OutHandler {
+        override def onPull(): Unit = {
+          if (!hasBeenPulled(in)) pull(in)
+        }
       }
-    })
+    )
 
     def emitFrames(): Unit = {
       while (buffer.size >= 4) {
         val totalLength = buffer.iterator.getInt
-        println("buffer size: " + buffer.size)
-        println("total length: " + totalLength)
-        println("buffer:       " + buffer.utf8String)
 
         if (buffer.size < 4 + totalLength) {
           // not enough data yet
@@ -53,4 +56,4 @@ class AwsEventStreamFrameDecoder extends GraphStage[FlowShape[ByteString, ByteSt
       }
     }
   }
-}
+}