Fix error when streaming from Gemini includes only executable_code or code_execution_result (#2719)

Author: binaryCrossEntropy

pydantic_ai_slim/pydantic_ai/models/google.py

@@ -559,6 +559,10 @@ async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:
                     )
                     if maybe_event is not None:  # pragma: no branch
                         yield maybe_event
+                elif part.executable_code is not None:
+                    pass
+                elif part.code_execution_result is not None:
+                    pass
                 else:
                     assert part.function_response is not None, f'Unexpected part: {part}'  # pragma: no cover
 

tests/models/cassettes/test_google/test_google_model_builtin_code_execution_stream.yaml

@@ -0,0 +1,67 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '287'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: what is 65465-6544 * 65464-6+1.02255
+        role: user
+      generationConfig: {}
+      systemInstruction:
+        parts:
+        - text: Be concise and always use Python to do calculations no matter how small.
+        role: user
+      tools:
+      - codeExecution: {}
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:streamGenerateContent?alt=sse
+  response:
+    body:
+      string: "data: {\"candidates\": [{\"content\": {\"parts\": [{\"executableCode\": {\"language\": \"PYTHON\",\"code\":
+        \"result = 65465 - 6544 * 65464 - 6 + 1.02255\\nprint(result)\\n\"}}],\"role\": \"model\"}}],\"usageMetadata\": {\"promptTokenCount\":
+        46,\"totalTokenCount\": 46,\"promptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\": 46}]},\"modelVersion\":
+        \"gemini-2.0-flash\",\"responseId\": \"_oizaOUTzpCEwg_H-YnYAg\"}\r\n\r\ndata: {\"candidates\": [{\"content\": {\"parts\":
+        [{\"codeExecutionResult\": {\"outcome\": \"OUTCOME_OK\",\"output\": \"-428330955.97745\\n\"}}],\"role\": \"model\"}}],\"usageMetadata\":
+        {\"promptTokenCount\": 46,\"totalTokenCount\": 46,\"promptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\":
+        46}]},\"modelVersion\": \"gemini-2.0-flash\",\"responseId\": \"_oizaOUTzpCEwg_H-YnYAg\"}\r\n\r\ndata: {\"candidates\":
+        [{\"content\": {\"parts\": [{\"text\": \"The\"}],\"role\": \"model\"}}],\"usageMetadata\": {\"promptTokenCount\":
+        46,\"totalTokenCount\": 46,\"promptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\": 46}]},\"modelVersion\":
+        \"gemini-2.0-flash\",\"responseId\": \"_oizaOUTzpCEwg_H-YnYAg\"}\r\n\r\ndata: {\"candidates\": [{\"content\": {\"parts\":
+        [{\"text\": \" answer is -428330955.9774\"}],\"role\": \"model\"}}],\"usageMetadata\": {\"promptTokenCount\": 46,\"totalTokenCount\":
+        46,\"promptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\": 46}]},\"modelVersion\": \"gemini-2.0-flash\",\"responseId\":
+        \"_oizaOUTzpCEwg_H-YnYAg\"}\r\n\r\ndata: {\"candidates\": [{\"content\": {\"parts\": [{\"text\": \"5.\\n\"}],\"role\":
+        \"model\"},\"finishReason\": \"STOP\"}],\"usageMetadata\": {\"promptTokenCount\": 44,\"candidatesTokenCount\": 77,\"totalTokenCount\":
+        221,\"promptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\": 44}],\"candidatesTokensDetails\": [{\"modality\":
+        \"TEXT\",\"tokenCount\": 77}],\"toolUsePromptTokenCount\": 100,\"toolUsePromptTokensDetails\": [{\"modality\": \"TEXT\",\"tokenCount\":
+        100}]},\"modelVersion\": \"gemini-2.0-flash\",\"responseId\": \"_oizaOUTzpCEwg_H-YnYAg\"}\r\n\r\n"
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-disposition:
+      - attachment
+      content-type:
+      - text/event-stream
+      server-timing:
+      - gfet4t7; dur=5315
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    status:
+      code: 200
+      message: OK
+version: 1

tests/models/test_google.py

@@ -230,6 +230,25 @@ async def test_google_model_stream(allow_model_requests: None, google_provider:
     assert data == snapshot('The capital of France is Paris.\n')
 
 
+async def test_google_model_builtin_code_execution_stream(
+    allow_model_requests: None,
+    google_provider: GoogleProvider,
+):
+    """Test Gemini streaming only code execution result or executable_code."""
+    model = GoogleModel('gemini-2.0-flash', provider=google_provider)
+    agent = Agent(
+        model=model,
+        system_prompt='Be concise and always use Python to do calculations no matter how small.',
+        builtin_tools=[CodeExecutionTool()],
+    )
+    event_parts: list[str] = []
+    async with agent.run_stream(user_prompt='what is 65465-6544 * 65464-6+1.02255') as result:
+        async for chunk in result.stream_text():
+            event_parts.append(chunk)
+
+    assert event_parts == snapshot(['The answer is -428330955.97745.\n'])
+
+
 async def test_google_model_retry(allow_model_requests: None, google_provider: GoogleProvider):
     model = GoogleModel('gemini-2.5-pro-preview-03-25', provider=google_provider)
     agent = Agent(