holetron-lab/fleet-memory
1
0
Fork 0
Code Releases Packages Activity Actions

fix(gemini): thought_signature read from wrong object and type in 3.1+ tool calls (#785)

Browse source
This commit is contained in:
Volodymyr Prypeshniuk 2026-03-31 10:09:12 +03:00 committed by GitHub
parent 0096115678
commit 1b5c262a8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
hindsight-api-slim/hindsight_api/engine/providers/gemini_llm.py
View file

@ -7,6 +7,7 @@ This provider supports both:
""" """
import asyncio import asyncio
import base64
import json import json
import logging import logging
import os import os
@ -472,9 +473,10 @@ class GeminiLLM(LLMInterface):
fn_args = parse_llm_json(fn_args_str) fn_args = parse_llm_json(fn_args_str)
thought_signature = tc.get("thought_signature") thought_signature = tc.get("thought_signature")
fc_kwargs: dict[str, Any] = {"name": fn_name, "args": fn_args} fc_kwargs: dict[str, Any] = {"name": fn_name, "args": fn_args}
part_kwargs: dict[str, Any] = {"function_call": genai_types.FunctionCall(**fc_kwargs)}
if thought_signature: if thought_signature:
fc_kwargs["thought_signature"] = thought_signature part_kwargs["thought_signature"] = base64.b64decode(thought_signature)
parts.append(genai_types.Part(function_call=genai_types.FunctionCall(**fc_kwargs))) parts.append(genai_types.Part(**part_kwargs))
gemini_contents.append(genai_types.Content(role="model", parts=parts)) gemini_contents.append(genai_types.Content(role="model", parts=parts))
else: else:
gemini_contents.append(genai_types.Content(role="model", parts=[genai_types.Part(text=content)])) gemini_contents.append(genai_types.Content(role="model", parts=[genai_types.Part(text=content)]))
@ -547,7 +549,10 @@ class GeminiLLM(LLMInterface):
content = part.text content = part.text
if hasattr(part, "function_call") and part.function_call: if hasattr(part, "function_call") and part.function_call:
fc = part.function_call fc = part.function_call
thought_signature = getattr(fc, "thought_signature", None) _raw_ts = getattr(part, "thought_signature", None)
thought_signature = (
base64.b64encode(_raw_ts).decode("ascii") if isinstance(_raw_ts, bytes) else _raw_ts
)
tool_calls.append( tool_calls.append(
LLMToolCall( LLMToolCall(
id=f"gemini_{len(tool_calls)}", id=f"gemini_{len(tool_calls)}",