getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-19 16:03:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: preserve Gemini thought signatures for tool calls

Browse source
This commit is contained in:
Nico Bailon 2025-12-20 07:12:37 -08:00
parent 1294239a7a
commit 81e820d01a
1 changed files with 27 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

27
packages/ai/src/providers/openai-completions.ts
View file

@ -262,6 +262,20 @@ export const streamOpenAICompletions: StreamFunction<"openai-completions"> = (
} }
} }
} }
const reasoningDetails = (choice.delta as any).reasoning_details;
if (reasoningDetails && Array.isArray(reasoningDetails)) {
for (const detail of reasoningDetails) {
if (detail.type === "reasoning.encrypted" && detail.id && detail.data) {
const matchingToolCall = output.content.find(
(b) => b.type === "toolCall" && b.id === detail.id,
) as ToolCall | undefined;
if (matchingToolCall) {
matchingToolCall.thoughtSignature = JSON.stringify(detail);
}
}
}
}
} }
} }
@ -487,6 +501,19 @@ function convertMessages(
arguments: JSON.stringify(tc.arguments), arguments: JSON.stringify(tc.arguments),
}, },
})); }));
const reasoningDetails = toolCalls
.filter((tc) => tc.thoughtSignature)
.map((tc) => {
try {
return JSON.parse(tc.thoughtSignature!);
} catch {
return null;
}
})
.filter(Boolean);
if (reasoningDetails.length > 0) {
(assistantMsg as any).reasoning_details = reasoningDetails;
}
} }
// Skip assistant messages that have no content and no tool calls. // Skip assistant messages that have no content and no tool calls.
// Mistral explicitly requires "either content or tool_calls, but not none". // Mistral explicitly requires "either content or tool_calls, but not none".