Skip to content

Commit 0fb8eea

Browse files
committed
fix typechecks for scripts/
1 parent ea5a45f commit 0fb8eea

File tree

9 files changed

+23
-16
lines changed

9 files changed

+23
-16
lines changed

evals/git-evals/gen-evals.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
import { execSync } from 'child_process'
22
import fs from 'fs'
33
import path from 'path'
4-
import { mapLimit } from 'async'
54

5+
import { disableLiveUserInputCheck } from '@codebuff/backend/live-user-inputs'
66
import { promptAiSdk } from '@codebuff/backend/llm-apis/vercel-ai-sdk/ai-sdk'
77
import { models } from '@codebuff/common/old-constants'
8+
import { mapLimit } from 'async'
89

910
import { extractRepoNameFromUrl, setupTestRepo } from './setup-test-repo'
10-
import { disableLiveUserInputCheck } from '@codebuff/backend/live-user-inputs'
1111

1212
import type { EvalData, EvalInput, FileState, EvalCommit } from './types'
1313
const SPEC_GENERATION_PROMPT = `Given a set of file changes and an optional description, write a clear specification describing WHAT needs to be implemented.
@@ -132,6 +132,7 @@ File Changes:\n${fileContext}`
132132
fingerprintId,
133133
userInputId,
134134
userId: undefined,
135+
logger: console,
135136
})
136137

137138
// Extract spec from <spec></spec> tags

evals/git-evals/judge-git-eval.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,7 @@ export async function judgeEvalRun(evalRun: EvalRunLog) {
193193
userInputId: generateCompactId(),
194194
userId: undefined,
195195
timeout: 10 * 60 * 1000, // 10 minute timeout
196+
logger: console,
196197
}).catch((error) => {
197198
console.warn(`Judge ${index + 1} failed:`, error)
198199
return null

evals/git-evals/pick-commits.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,14 @@
33
import { execFileSync } from 'child_process'
44
import fs from 'fs'
55
import path from 'path'
6-
import { mapLimit } from 'async'
76

7+
import { disableLiveUserInputCheck } from '@codebuff/backend/live-user-inputs'
88
import { promptAiSdkStructured } from '@codebuff/backend/llm-apis/vercel-ai-sdk/ai-sdk'
99
import { models } from '@codebuff/common/old-constants'
10+
import { mapLimit } from 'async'
1011
import { z } from 'zod/v4'
1112

1213
import { extractRepoNameFromUrl, setupTestRepo } from './setup-test-repo'
13-
import { disableLiveUserInputCheck } from '@codebuff/backend/live-user-inputs'
1414

1515
// Types for commit data
1616
export interface CommitDiff {
@@ -379,6 +379,7 @@ async function screenCommitsWithGpt5(
379379
fingerprintId,
380380
userInputId,
381381
userId: undefined,
382+
logger: console,
382383
})
383384

384385
// Handle empty or invalid response

evals/git-evals/post-eval-analysis.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,5 +188,6 @@ export async function analyzeEvalResults(
188188
userInputId: generateCompactId(),
189189
userId: undefined,
190190
timeout: 10 * 60 * 1000, // 10 minute timeout
191+
logger: console,
191192
})
192193
}

evals/git-evals/run-git-evals.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,7 @@ Explain your reasoning in detail. Do not ask Codebuff to git commit changes.`,
151151
userInputId: generateCompactId(),
152152
userId: undefined,
153153
timeout: 5 * 60_000, // 5 minute timeout
154+
logger: console,
154155
})
155156
} catch (agentError) {
156157
throw new Error(

scripts/ft-file-selection/grade-traces.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ async function gradeTraces({ logger }: { logger: Logger }) {
4646
batch.map(async (traceAndRelabels) => {
4747
try {
4848
console.log(`Grading trace ${traceAndRelabels.trace.id}`)
49-
const result = await gradeRun(traceAndRelabels)
49+
const result = await gradeRun({ ...traceAndRelabels, logger })
5050
return {
5151
traceId: traceAndRelabels.trace.id,
5252
status: 'success',

scripts/ft-file-selection/relabel-for-offline-scoring.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@ async function relabelTraceForModel(
191191
userInputId: 'relabel-offline-scoring',
192192
userId: TEST_USER_ID,
193193
maxOutputTokens: 1000,
194+
logger: console,
194195
})
195196

196197
const newRelabel: Relabel = {

scripts/ft-file-selection/relabel-traces.ts

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -67,22 +67,22 @@ async function runTraces() {
6767
fingerprintId: 'relabel-trace-run',
6868
userInputId: 'relabel-trace-run',
6969
userId: TEST_USER_ID,
70+
logger: console,
7071
})
7172
} else {
72-
output = await promptFlashWithFallbacks(
73-
messagesWithSystem({
73+
output = await promptFlashWithFallbacks({
74+
messages: messagesWithSystem({
7475
messages: messages as Message[],
7576
system: system as System,
7677
}),
77-
{
78-
model:
79-
model as typeof models.openrouter_gemini2_5_pro_preview,
80-
clientSessionId: 'relabel-trace-run',
81-
fingerprintId: 'relabel-trace-run',
82-
userInputId: 'relabel-trace-run',
83-
userId: 'relabel-trace-run',
84-
},
85-
)
78+
model:
79+
model as typeof models.openrouter_gemini2_5_pro_preview,
80+
clientSessionId: 'relabel-trace-run',
81+
fingerprintId: 'relabel-trace-run',
82+
userInputId: 'relabel-trace-run',
83+
userId: 'relabel-trace-run',
84+
logger: console,
85+
})
8686
}
8787

8888
// Create relabel record

scripts/get-changelog.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ Start your response with a heading using ### (three hashes) and organize the con
172172
model: models.openrouter_claude_sonnet_4,
173173
userId: undefined,
174174
chargeUser: false,
175+
logger: console,
175176
})
176177

177178
// Clean up the AI response

0 commit comments

Comments
 (0)