fix: update tests to handle new experimental diff option and increase the default confidence to 1

This commit is contained in:
Daniel Riccio
2025-01-15 12:23:51 -05:00
parent 22069e8056
commit 6c8d7f4951
6 changed files with 16 additions and 13 deletions

View File

@@ -7,14 +7,11 @@ import { NewUnifiedDiffStrategy } from './strategies/new-unified'
* @param model The name of the model being used (e.g., 'gpt-4', 'claude-3-opus')
* @returns The appropriate diff strategy for the model
*/
export function getDiffStrategy(model: string, fuzzyMatchThreshold?: number, experimentalDiffStrategy?: boolean): DiffStrategy {
export function getDiffStrategy(model: string, fuzzyMatchThreshold?: number, experimentalDiffStrategy: boolean = false): DiffStrategy {
if (experimentalDiffStrategy) {
// Use the fuzzyMatchThreshold with a minimum of 0.8 (80%)
const threshold = Math.max(fuzzyMatchThreshold ?? 1.0, 0.8)
return new NewUnifiedDiffStrategy(threshold)
return new NewUnifiedDiffStrategy(fuzzyMatchThreshold)
}
// Default to the stable SearchReplaceDiffStrategy
return new SearchReplaceDiffStrategy()
return new SearchReplaceDiffStrategy(fuzzyMatchThreshold)
}
export type { DiffStrategy }

View File

@@ -11,7 +11,7 @@ describe('main', () => {
describe('constructor', () => {
it('should use default confidence threshold when not provided', () => {
const defaultStrategy = new NewUnifiedDiffStrategy()
expect(defaultStrategy['confidenceThreshold']).toBe(0.9)
expect(defaultStrategy['confidenceThreshold']).toBe(1)
})
it('should use provided confidence threshold', () => {

View File

@@ -6,8 +6,8 @@ import { DiffResult, DiffStrategy } from "../../types"
export class NewUnifiedDiffStrategy implements DiffStrategy {
private readonly confidenceThreshold: number
constructor(confidenceThreshold: number = 0.9) {
this.confidenceThreshold = Math.max(confidenceThreshold, 0.8)
constructor(confidenceThreshold: number = 1) {
this.confidenceThreshold = Math.max(confidenceThreshold, 0.8);
}
private parseUnifiedDiff(diff: string): Diff {