Skip to content

Commit c4097e4

Browse files
author
Joseph Smith
committed
Silly formatting bits
1 parent ce45879 commit c4097e4

1 file changed

Lines changed: 16 additions & 3 deletions

File tree

media-api/app/lib/elasticsearch/ElasticSearch.scala

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -267,9 +267,18 @@ class ElasticSearch(
267267
.size(k)
268268
}
269269

270-
private case class HybridResult(id: String, lexicalScore: Double, semanticScore: Double, image: SourceWrapper[Image])
270+
private case class HybridResult(
271+
id: String,
272+
lexicalScore: Double,
273+
semanticScore: Double,
274+
image: SourceWrapper[Image]
275+
)
276+
271277
private case object HybridResult {
272-
def resolveHitAndFillInSemanticScore(hit: SearchHit, queryEmbedding: List[Double]): Option[HybridResult] =
278+
def resolveHitAndFillInSemanticScore(
279+
hit: SearchHit,
280+
queryEmbedding: List[Double]
281+
): Option[HybridResult] =
273282
resolveHit(hit).map { image =>
274283
val semanticScore = image.instance.embedding
275284
.flatMap(_.cohereEmbedV4)
@@ -283,7 +292,11 @@ class ElasticSearch(
283292
HybridResult(hit.id, lexicalScore = hit.score, semanticScore = semanticScore, image = image)
284293
}
285294

286-
def combineScoresAndGetTopK(results: List[HybridResult], vecWeight: Double, k: Int)(implicit logMarker: LogMarker): List[HybridResult] = {
295+
def combineScoresAndGetTopK(
296+
results: List[HybridResult],
297+
vecWeight: Double,
298+
k: Int
299+
)(implicit logMarker: LogMarker): List[HybridResult] = {
287300
val dedupedResults = results.distinctBy(_.id)
288301
logger.info(logMarker, s"${dedupedResults.length} deduped results")
289302

0 commit comments

Comments
 (0)