Skip to content

Commit a57af7a

Browse files
committed
Merge branch 'development'
2 parents 7704426 + 1bc0d74 commit a57af7a

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

packages/bytebot-agent-cc/src/agent/agent.processor.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,6 @@ export class AgentProcessor {
186186
for await (const message of query({
187187
prompt: task.description,
188188
options: {
189-
continue: true,
190189
abortController: this.abortController,
191190
appendSystemPrompt: AGENT_SYSTEM_PROMPT,
192191
permissionMode: 'bypassPermissions',

packages/bytebot-agent/src/google/google.service.ts

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import {
77
MessageContentBlock,
88
MessageContentType,
99
TextContentBlock,
10+
ThinkingContentBlock,
1011
ToolUseContentBlock,
1112
} from '@bytebot/shared';
1213
import {
@@ -62,6 +63,9 @@ export class GoogleService implements BytebotAgentService {
6263
model,
6364
contents: googleMessages,
6465
config: {
66+
thinkingConfig: {
67+
thinkingBudget: 24576,
68+
},
6569
maxOutputTokens: maxTokens,
6670
systemInstruction: systemPrompt,
6771
tools: useTools
@@ -204,6 +208,13 @@ export class GoogleService implements BytebotAgentService {
204208
});
205209
break;
206210
}
211+
case MessageContentType.Thinking:
212+
parts.push({
213+
text: block.thinking,
214+
thoughtSignature: block.signature,
215+
thought: true,
216+
});
217+
break;
207218
default:
208219
parts.push({
209220
text: JSON.stringify(block),
@@ -259,6 +270,14 @@ export class GoogleService implements BytebotAgentService {
259270
} as TextContentBlock;
260271
}
261272

273+
if (part.thought) {
274+
return {
275+
type: MessageContentType.Thinking,
276+
signature: part.thoughtSignature,
277+
thinking: part.text,
278+
} as ThinkingContentBlock;
279+
}
280+
262281
if (part.functionCall) {
263282
return {
264283
type: MessageContentType.ToolUse,

0 commit comments

Comments
 (0)