Add cassettes

Author: DouweM

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -2132,6 +2132,15 @@ async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:
             elif isinstance(chunk, responses.ResponseMcpCallCompletedEvent):
                 pass  # there's nothing we need to do here
 
+            elif isinstance(chunk, responses.ResponseFileSearchCallCompletedEvent):
+                pass  # there's nothing we need to do here
+
+            elif isinstance(chunk, responses.ResponseFileSearchCallSearchingEvent):
+                pass  # there's nothing we need to do here
+
+            elif isinstance(chunk, responses.ResponseFileSearchCallInProgressEvent):
+                pass  # there's nothing we need to do here
+
             else:  # pragma: no cover
                 warnings.warn(
                     f'Handling of this event type is not yet implemented. Please report on our GitHub: {chunk}',

tests/json_body_serializer.py

@@ -15,6 +15,7 @@
 
 FILTERED_HEADER_PREFIXES = ['anthropic-', 'cf-', 'x-']
 FILTERED_HEADERS = {'authorization', 'date', 'request-id', 'server', 'user-agent', 'via', 'set-cookie', 'api-key'}
+ALLOWED_HEADERS = {'x-goog-upload-url', 'x-goog-upload-status'}  # required for test_google_model_file_search_tool
 
 
 class LiteralDumper(Dumper):
@@ -55,7 +56,9 @@ def serialize(cassette_dict: Any):  # pragma: lax no cover
             headers = {k: v for k, v in headers.items() if k not in FILTERED_HEADERS}
             # filter headers by prefix
             headers = {
-                k: v for k, v in headers.items() if not any(k.startswith(prefix) for prefix in FILTERED_HEADER_PREFIXES)
+                k: v
+                for k, v in headers.items()
+                if not any(k.startswith(prefix) for prefix in FILTERED_HEADER_PREFIXES) or k in ALLOWED_HEADERS
             }
             # update headers on source object
             data['headers'] = headers

tests/models/cassettes/test_google/test_google_model_file_search_tool.yaml

@@ -26,18 +26,18 @@ interactions:
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=455
+      - gfet4t7; dur=373
       transfer-encoding:
       - chunked
       vary:
       - Origin
       - X-Origin
       - Referer
     parsed_body:
-      createTime: '2025-11-14T13:06:44.036465Z'
+      createTime: '2025-11-20T23:37:08.933211Z'
       displayName: test-file-search-store
-      name: fileSearchStores/testfilesearchstore-yc4nqyu6bkpo
-      updateTime: '2025-11-14T13:06:44.036465Z'
+      name: fileSearchStores/testfilesearchstore-s6zmrh92ulpr
+      updateTime: '2025-11-20T23:37:08.933211Z'
     status:
       code: 200
       message: OK
@@ -58,7 +58,7 @@ interactions:
     method: POST
     parsed_body:
       mimeType: text/plain
-    uri: https://generativelanguage.googleapis.com/upload/v1beta/fileSearchStores/testfilesearchstore-yc4nqyu6bkpo:uploadToFileSearchStore
+    uri: https://generativelanguage.googleapis.com/upload/v1beta/fileSearchStores/testfilesearchstore-s6zmrh92ulpr:uploadToFileSearchStore
   response:
     body:
       string: ''
@@ -69,6 +69,10 @@ interactions:
       - '0'
       content-type:
       - text/plain; charset=utf-8
+      x-goog-upload-status:
+      - active
+      x-goog-upload-url:
+      - https://generativelanguage.googleapis.com/upload/v1beta/fileSearchStores/testfilesearchstore-s6zmrh92ulpr:uploadToFileSearchStore?upload_id=AOCedOHMyp8-WH0roPvEsby85i78ZlUxHR4ZVhKtN6xXYFFUkHedk2H5w7cIOz4M9p4LeWPKM46woeaM3jNrhq6vfgJmt4ds3RnrIiO5LEsa3g&upload_protocol=resumable
     status:
       code: 200
       message: OK
@@ -86,7 +90,7 @@ interactions:
       host:
       - generativelanguage.googleapis.com
     method: POST
-    uri: https://generativelanguage.googleapis.com/upload/v1beta/fileSearchStores/testfilesearchstore-yc4nqyu6bkpo:uploadToFileSearchStore?upload_id=AOCedOGMFY9IXAkbC4iGTlw4gkakgbCCpXweCb8TJzUJMegDkFGYkygsxUAtdzCgKeOt3GTgFDmt9ynY1SQsUcVSSI2sK2rSi6i8RrGC7y276ls&upload_protocol=resumable
+    uri: https://generativelanguage.googleapis.com/upload/v1beta/fileSearchStores/testfilesearchstore-s6zmrh92ulpr:uploadToFileSearchStore?upload_id=AOCedOHMyp8-WH0roPvEsby85i78ZlUxHR4ZVhKtN6xXYFFUkHedk2H5w7cIOz4M9p4LeWPKM46woeaM3jNrhq6vfgJmt4ds3RnrIiO5LEsa3g&upload_protocol=resumable
   response:
     headers:
       alt-svc:
@@ -99,13 +103,15 @@ interactions:
       - Origin
       - X-Origin
       - Referer
+      x-goog-upload-status:
+      - final
     parsed_body:
-      name: fileSearchStores/testfilesearchstore-yc4nqyu6bkpo/upload/operations/7k9hlotn1ptc-zy3bg6jfttw0
+      name: fileSearchStores/testfilesearchstore-s6zmrh92ulpr/upload/operations/k742l26km1p8-jjheny21pwon
       response:
         '@type': type.googleapis.com/google.ai.generativelanguage.v1main.UploadToFileSearchStoreResponse
-        documentName: fileSearchStores/testfilesearchstore-yc4nqyu6bkpo/documents/7k9hlotn1ptc-zy3bg6jfttw0
+        documentName: fileSearchStores/testfilesearchstore-s6zmrh92ulpr/documents/k742l26km1p8-jjheny21pwon
         mimeType: text/plain
-        parent: fileSearchStores/testfilesearchstore-yc4nqyu6bkpo
+        parent: fileSearchStores/testfilesearchstore-s6zmrh92ulpr
         sizeBytes: '79'
     status:
       code: 200
@@ -119,7 +125,7 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '274'
+      - '344'
       content-type:
       - application/json
       host:
@@ -128,40 +134,186 @@ interactions:
     parsed_body:
       contents:
       - parts:
-        - text: What is the capital of France according to my files?
+        - text: What is the capital of France?
         role: user
       generationConfig:
         responseModalities:
         - TEXT
+      systemInstruction:
+        parts:
+        - text: You are a helpful assistant.
+        role: user
+      tools:
+      - fileSearch:
+          file_search_store_names:
+          - fileSearchStores/testfilesearchstore-s6zmrh92ulpr
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '2105'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=11428
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - content:
+          parts:
+          - text: |-
+              The capital of France is Paris.
+
+              Paris is a major global center for art, fashion, gastronomy, and culture. A famous landmark in the city is the Eiffel Tower. The city is also known for its 19th-century cityscape, which is crisscrossed by wide boulevards and the River Seine.
+          role: model
+        finishReason: STOP
+        groundingMetadata:
+          groundingChunks:
+          - retrievedContext:
+              fileSearchStore: fileSearchStores/testfilesearchstore-s6zmrh92ulpr
+              text: Paris is the capital of France. The Eiffel Tower is a famous landmark in Paris.
+          - retrievedContext:
+              fileSearchStore: fileSearchStores/testfilesearchstore-s6zmrh92ulpr
+              text: Paris is the capital of France. The Eiffel Tower is a famous landmark in Paris.
+          groundingSupports:
+          - groundingChunkIndices:
+            - 1
+            segment:
+              endIndex: 30
+              text: The capital of France is Paris
+          - groundingChunkIndices:
+            - 1
+            segment:
+              endIndex: 156
+              startIndex: 107
+              text: A famous landmark in the city is the Eiffel Tower
+        index: 0
+      modelVersion: gemini-2.5-pro
+      responseId: NKYfaZHSO4udqtsP5YSmuAs
+      usageMetadata:
+        candidatesTokenCount: 100
+        promptTokenCount: 15
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 15
+        thoughtsTokenCount: 879
+        toolUsePromptTokenCount: 1485
+        toolUsePromptTokensDetails:
+        - modality: TEXT
+          tokenCount: 1485
+        totalTokenCount: 2479
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '738'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: What is the capital of France?
+        role: user
+      - parts:
+        - text: |-
+            The capital of France is Paris.
+
+            Paris is a major global center for art, fashion, gastronomy, and culture. A famous landmark in the city is the Eiffel Tower. The city is also known for its 19th-century cityscape, which is crisscrossed by wide boulevards and the River Seine.
+        role: model
+      - parts:
+        - text: Tell me about the Eiffel Tower.
+        role: user
+      generationConfig:
+        responseModalities:
+        - TEXT
+      systemInstruction:
+        parts:
+        - text: You are a helpful assistant.
+        role: user
       tools:
       - fileSearch:
           file_search_store_names:
-          - fileSearchStores/testfilesearchstore-yc4nqyu6bkpo
+          - fileSearchStores/testfilesearchstore-s6zmrh92ulpr
     uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent
   response:
     headers:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '133'
+      - '2782'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=7289
+      - gfet4t7; dur=8799
       transfer-encoding:
       - chunked
       vary:
       - Origin
       - X-Origin
       - Referer
     parsed_body:
-      error:
-        code: 503
-        message: The model is overloaded. Please try again later.
-        status: UNAVAILABLE
+      candidates:
+      - content:
+          parts:
+          - text: |-
+              The Eiffel Tower is a famous landmark located in Paris, the capital of France. It is one of the most recognizable structures in the world.
+
+              Here are some key facts about the Eiffel Tower:
+
+              *   **Construction:** It was designed by the engineer Gustave Eiffel, and his company built it between 1887 and 1889. It was created as the entrance arch for the 1889 Exposition Universelle (World's Fair), which celebrated the 100th anniversary of the French Revolution.
+              *   **Design and Material:** The tower is made of wrought iron and is a masterpiece of structural engineering. Its open-lattice design was innovative for its time and was chosen for its wind resistance and stability.
+              *   **Height:** The Eiffel Tower stands at a height of 330 meters (1,083 feet), including its antennas. For about 41 years, it held the title of the tallest man-made structure in the world until the Chrysler Building in New York City was completed in 1930.
+              *   **Visiting:** The tower has three levels for visitors. The first two levels can be reached by stairs or elevators and feature restaurants and shops. The top level, accessible only by elevator, offers panoramic views of Paris.
+              *   **Cultural Impact:** Initially criticized by some of France's leading artists and intellectuals for its design, the Eiffel Tower has since become a global cultural icon of France and a symbol of Paris itself. It is one of the most visited paid monuments in the world.
+          role: model
+        finishReason: STOP
+        groundingMetadata:
+          groundingChunks:
+          - retrievedContext:
+              fileSearchStore: fileSearchStores/testfilesearchstore-s6zmrh92ulpr
+              text: Paris is the capital of France. The Eiffel Tower is a famous landmark in Paris.
+          groundingSupports:
+          - groundingChunkIndices:
+            - 0
+            segment:
+              endIndex: 78
+              text: The Eiffel Tower is a famous landmark located in Paris, the capital of France.
+        index: 0
+      modelVersion: gemini-2.5-pro
+      responseId: PqYfadjYC8eHqtsPicvToAs
+      usageMetadata:
+        candidatesTokenCount: 339
+        promptTokenCount: 88
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 88
+        thoughtsTokenCount: 167
+        toolUsePromptTokenCount: 270
+        toolUsePromptTokensDetails:
+        - modality: TEXT
+          tokenCount: 270
+        totalTokenCount: 864
     status:
-      code: 503
-      message: Service Unavailable
+      code: 200
+      message: OK
 - request:
     body: ''
     headers:
@@ -176,7 +328,7 @@ interactions:
       host:
       - generativelanguage.googleapis.com
     method: DELETE
-    uri: https://generativelanguage.googleapis.com/v1beta/fileSearchStores/testfilesearchstore-yc4nqyu6bkpo?force=True
+    uri: https://generativelanguage.googleapis.com/v1beta/fileSearchStores/testfilesearchstore-s6zmrh92ulpr?force=True
   response:
     headers:
       alt-svc:
@@ -186,7 +338,7 @@ interactions:
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=858
+      - gfet4t7; dur=528
       transfer-encoding:
       - chunked
       vary: