LiteLlm with SSE streaming records the answer twice in gen_ai.output.messages
Les mainteneurs répondent en général sous 4 jours
Évaluation
- Difficulté
- 2/5
- Temps estimé
- 1-3 heures
- Accessibilité débutants
- 78/100
- Type d'issue
- Bug
- Clarté
- Clairement spécifiée
- Activité
- Active
- Stack technique
- python
- Domaine
- backend, observability
Piste de recherche
Commencez par src/google/adk/telemetry/_instrumentation.py, en particulier record_llm_response, et comparez la façon dont le code traite les réponses partielles diffusées en streaming par rapport à la réponse agrégée. Lisez ensuite le code de streaming de LiteLlm dans src/google/adk/models/lite_llm.py, autour des lignes liées. Reproduisez le problème avec l’exemple StreamingMode.SSE fourni et vérifiez que la sortie capturée contient un seul message de réponse complet, correspondant à StreamingMode.NONE.
Rédigé par le modèle d'indexation à partir du texte de l'issue.
Description
🔴 Required Information
Describe the Bug:
with StreamingMode.SSE and a LiteLlm model the generate_content span ends up with every streamed chunk in gen_ai.output.messages and then the full answer again at the end. so the answer shows up twice and all the chunks have an empty finish_reason. without streaming its one message like expected. only visible with message content capture on (OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT)
looks like the text chunks get yielded as partial responses with no finish reason (lite_llm.py#L4475-L4485) and only the final aggregated response gets one (#L4556-L4558). record_llm_response keeps recording until it sees a finish reason (_instrumentation.py#L568-L595) so every chunk plus the aggregate goes in. since its in the LiteLlm wrapper i'd expect real providers to do the same but i only tested with litellm's mock_response
Steps to Reproduce:
pip install google-adk litellm- run this (no api key needed, it uses litellm's
mock_response)
import asyncio
import json
import os
os.environ["OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT"] = "SPAN_ONLY"
from opentelemetry import trace
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import SimpleSpanProcessor
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
spans = InMemorySpanExporter()
provider = TracerProvider()
provider.add_span_processor(SimpleSpanProcessor(spans))
trace.set_tracer_provider(provider)
from google.adk.agents import Agent
from google.adk.agents.run_config import RunConfig, StreamingMode
from google.adk.models.lite_llm import LiteLlm
from google.adk.runners import InMemoryRunner
from google.genai import types
async def run(mode):
spans.clear()
model = LiteLlm(model="openai/gpt-4o", mock_response="Oslo is 4C and cloudy today.", api_key="unused")
runner = InMemoryRunner(agent=Agent(name="a", model=model), app_name="p")
session = await runner.session_service.create_session(app_name="p", user_id="u")
msg = types.Content(role="user", parts=[types.Part(text="weather?")])
async for _ in runner.run_async(user_id="u", session_id=session.id, new_message=msg,
run_config=RunConfig(streaming_mode=mode)):
pass
await runner.close()
span = next(s for s in spans.get_finished_spans() if s.name.startswith("generate_content"))
messages = json.loads(span.attributes["gen_ai.output.messages"])
print(f"{mode.name}: {len(messages)} output message(s)")
for m in messages:
print(" ", repr(m.get("finish_reason")), "".join(p.get("content", "") for p in m["parts"]))
asyncio.run(run(StreamingMode.NONE))
asyncio.run(run(StreamingMode.SSE))
Expected Behavior:
one output message with the full answer, same as without streaming
Observed Behavior:
NONE: 1 output message(s)
'stop' Oslo is 4C and cloudy today.
SSE: 11 output message(s)
'' Osl
'' o i
'' s 4
'' C a
'' nd
'' clo
'' udy
'' to
'' day
'' .
'stop' Oslo is 4C and cloudy today.
Environment Details:
- ADK Library Version (pip show google-adk): 2.12.0 (main at 77a74f0)
- Desktop OS: macOS
- Python Version (python -V): 3.12.15
Model Information:
- Are you using LiteLLM: Yes
- Which model is being used: openai/gpt-4o via litellm
mock_response
🟡 Optional Information
Regression:
havent checked older versions
maybe skip the partial chunks when recording output messages and keep just the aggregate? happy to send a PR if that sounds right
- Langage dominant
- Python
- Étoiles
- 21.9k
- Forks
- 4.2k
- Merge moyen
- 3 j 16 h
- PR mergées (30 j)
- 6
Préparer son environnement
- Aucun Dockerfile ni fichier Docker Compose
- Propose un modèle de pull request
- Lire le guide de contribution
Par où commencer
- Lisez l'issue en entier, puis le guide de contribution du projet.
- Signalez en commentaire que vous la prenez — cela évite que deux personnes fassent le même travail.
- Forkez le dépôt et travaillez sur une branche.
- Ouvrez une pull request qui référence le numéro de l'issue.
Autres issues de google/adk-python
-
gen_ai.choice log reports gen_ai.system=gemini for OpenAI and Groq models through LiteLlmPeut-être pris @llalitkumarrr l’a pris il y a 1 jour. Ouvertemodels
Difficulté 2/5 1-3 heures Accessibilité débutants 82/100
google/adk-python#7631 · 1 personne assignée ·
Les mainteneurs répondent en général sous 4 jours
-
Update opentelemetry-api and opentelemetry-sdk to 1.44.0Peut-être pris @llalitkumarrr l’a pris il y a 15 jours. Ouverterequest clarification tracing
Difficulté 2/5 1-3 heures Accessibilité débutants 70/100
google/adk-python#7433 · 5 commentaires · 1 personne assignée ·
Les mainteneurs répondent en général sous 4 jours
-
CI Mypy Check flags an existing streaming_utils.py error as new because the PR run reuses the baseline's mypy cachePeut-être pris @DeanChensj l’a pris il y a 17 jours. Ouverteneeds review
Difficulté 2/5 1-3 heures Accessibilité débutants 72/100
google/adk-python#7409 · 1 commentaire · 2 personnes assignées ·
Les mainteneurs répondent en général sous 4 jours
-
A2aAgentExecutor sends the raw exception text to the A2A caller when the run failsPeut-être pris @sanketpatil06 l’a pris il y a 18 jours. Ouvertea2a request clarification
Difficulté 2/5 1-3 heures Accessibilité débutants 84/100
google/adk-python#7385 · 2 commentaires · 1 personne assignée ·
Les mainteneurs répondent en général sous 4 jours
-
Please support mermaid 12 (inbuild elk) in `adk web`Peut-être pris @sanketpatil06 l’a pris il y a 18 jours. Ouverteneeds review web
Difficulté 2/5 1-3 heures Accessibilité débutants 62/100
google/adk-python#7381 · 1 commentaire · 1 personne assignée ·
Les mainteneurs répondent en général sous 4 jours
Toutes les issues de google/adk-python
Issues similaires
-
bug
Difficulté 1/5 Moins d'une heure Accessibilité débutants 88/100
Les mainteneurs répondent en général sous 1 jour
-
Difficulté 2/5 1-3 heures Accessibilité débutants 68/100
Les mainteneurs répondent en général sous 1 jour
-
documentation
Difficulté 2/5 1-3 heures Accessibilité débutants 84/100
sergey-miryanov/gcmon#230 ·
Les mainteneurs répondent en général sous 1 jour
-
Difficulté 2/5 1-3 heures Accessibilité débutants 78/100
Azure-Samples/azure-search-openai-demo#3249 ·
Les mainteneurs répondent en général sous 1 jour
-
[BUG] Failed archive extraction deletes the dataset cachePeut-être pris @dajiaohuang l’a pris aujourd’hui. Ouverte
Difficulté 2/5 1-3 heures Accessibilité débutants 78/100
alibaba/GraphScope#4674 ·