diff --git a/vibn-agent-runner/fix_genai_runner.js b/vibn-agent-runner/fix_genai_runner.js new file mode 100644 index 00000000..48501ee1 --- /dev/null +++ b/vibn-agent-runner/fix_genai_runner.js @@ -0,0 +1,9 @@ +const fs = require('fs'); + +const file1 = 'src/llm/gemini-chat.ts'; +let code1 = fs.readFileSync(file1, 'utf8'); + +code1 = code1.replace(/thinkingBudgetTokens:/g, 'thinkingBudget:'); +fs.writeFileSync(file1, code1); + +console.log("Fixed thinkingBudget property error for Runner"); diff --git a/vibn-agent-runner/fix_thinking_runner.js b/vibn-agent-runner/fix_thinking_runner.js new file mode 100644 index 00000000..56bf84e1 --- /dev/null +++ b/vibn-agent-runner/fix_thinking_runner.js @@ -0,0 +1,10 @@ +const fs = require('fs'); + +const file1 = 'src/llm/gemini-chat.ts'; +if (fs.existsSync(file1)) { + let code1 = fs.readFileSync(file1, 'utf8'); + code1 = code1.replace(/thinkingBudgetTokens/g, 'thinkingBudget'); + fs.writeFileSync(file1, code1); +} + +console.log("Patched thinkingBudget parameter for Runner"); diff --git a/vibn-agent-runner/src/llm/gemini-chat.ts b/vibn-agent-runner/src/llm/gemini-chat.ts index 3b4cd143..00a4d47a 100644 --- a/vibn-agent-runner/src/llm/gemini-chat.ts +++ b/vibn-agent-runner/src/llm/gemini-chat.ts @@ -114,7 +114,7 @@ export async function callGeminiChat(opts: { } if (opts.includeThoughts) { - config.thinkingConfig = { thinkingBudgetTokens: 1024 }; + config.thinkingConfig = { thinkingBudget: 1024 }; } const fns = toGeminiFunctions(opts.tools ?? []); @@ -174,7 +174,7 @@ export async function* streamGeminiChat(opts: { const config: any = { temperature: opts.temperature ?? 0.7, maxOutputTokens: 8192, - thinkingConfig: { thinkingBudgetTokens: 1024 }, + thinkingConfig: { thinkingBudget: 1024 }, }; if (opts.systemPrompt) {