Handy fun 'getMetadata(key: String)' introduce for Match data holder.

cequence-io · May 16, 2023 · c2b2e8c · c2b2e8c
1 parent a122ba1
commit c2b2e8c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pinecone-core/src/main/scala/io/cequence/pineconescala/domain/response/QueryResponse.scala b/pinecone-core/src/main/scala/io/cequence/pineconescala/domain/response/QueryResponse.scala
@@ -12,7 +12,10 @@ case class QueryResponse(
 case class Match(
   id: String,
   score: Double,
-  values: Seq[Double], // not returned when includeValues = false
+  values: Seq[Double],                   // not returned when includeValues = false
   sparseValues: Option[SparseVector],
-  metadata: Option[Map[String, String]] // not returned when includeMetadata = true
-)
+  metadata: Option[Map[String, String]]  // not returned when includeMetadata = false
+) {
+  val metadataUnwrapped : Map[String, String] = metadata.getOrElse(Map[String, String]())
+  def getMetadata(key: String): Option[String] = metadataUnwrapped.get(key)
+}