gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix Gemini response parsing to avoid warnings from non-text parts

Browse source
This commit is contained in:
yangdx 2025-11-07 04:00:37 +08:00
parent ea141e2779
commit 8c27555358
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lightrag/llm/gemini.py
View file

@ -115,9 +115,12 @@ def _format_history_messages(history_messages: list[dict[str, Any]] | None) -> s
def _extract_response_text(response: Any) -> str:
if getattr(response, "text", None):
return response.text
"""
Extract text content from Gemini response, avoiding warnings about non-text parts.
Always extracts text manually from parts to avoid triggering warnings when
non-text parts (like 'thought_signature') are present in the response.
"""
candidates = getattr(response, "candidates", None)
if not candidates:
return ""
@ -127,6 +130,7 @@ def _extract_response_text(response: Any) -> str:
if not getattr(candidate, "content", None):
continue
for part in getattr(candidate.content, "parts", []):
# Only extract text parts to avoid non-text content like thought_signature
text = getattr(part, "text", None)
if text:
parts.append(text)
@ -191,9 +195,8 @@ async def gemini_complete_if_cache(
usage = getattr(chunk, "usage_metadata", None)
if usage is not None:
usage_container["usage"] = usage
text_piece = getattr(chunk, "text", None) or _extract_response_text(
chunk
)
# Always use manual extraction to avoid warnings about non-text parts
text_piece = _extract_response_text(chunk)
if text_piece:
loop.call_soon_threadsafe(queue.put_nowait, text_piece)
loop.call_soon_threadsafe(queue.put_nowait, None)