mirror of
https://github.com/apple/sourcekit-lsp.git
synced 2026-03-02 18:23:24 +01:00
5709e1a864
This adds a sourcekitd plugin that drives the code completion requests. It also includes a `CompletionScoring` module that’s used to rank code completion results based on their contextual match, allowing us to show more relevant code completion results at the top.
35 lines
1.2 KiB
Swift
35 lines
1.2 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2024 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import CompletionScoring
|
|
import Foundation
|
|
|
|
package struct WeightedChoices<T: Sendable>: Sendable {
|
|
package typealias WeightedChoice = (likelihood: Double, value: T)
|
|
private var choices: [T] = []
|
|
|
|
package init(_ choices: [WeightedChoice]) {
|
|
precondition(choices.hasContent)
|
|
let smallest = choices.map(\.likelihood).min()!
|
|
let samples = 1.0 / (smallest / 2.0) + 1
|
|
for choice in choices {
|
|
precondition(choice.likelihood > 0)
|
|
precondition(choice.likelihood <= 1.0)
|
|
self.choices.append(contentsOf: Array(repeating: choice.value, count: Int(choice.likelihood * samples)))
|
|
}
|
|
}
|
|
|
|
package func select(using randomness: inout RepeatableRandomNumberGenerator) -> T {
|
|
choices.randomElement(using: &randomness)!
|
|
}
|
|
}
|