feat: support files

Author: wanlin31

internal/store/store.go

@@ -112,8 +112,16 @@ func readBody(req *http.Request) (map[string]any, error) {
 	}
 	err = json.Unmarshal(body, &resultMap)
 	if err != nil {
-		log.Fatalf("Error unmarshaling JSON: %v", err)
-		return nil, err
+		contentType := http.DetectContentType(body)
+		isBinary := strings.HasPrefix(contentType, "image/") ||
+			strings.HasPrefix(contentType, "video/") ||
+			strings.HasPrefix(contentType, "audio/") ||
+			contentType == "application/pdf" ||
+			contentType == "application/octet-stream"
+		if !isBinary {
+			log.Printf("Error unmarshaling JSON: %v", err)
+			return nil, err
+		}
 	}
 	// Restore the request body for further use.
 	req.Body = io.NopCloser(bytes.NewBuffer(body))

sdks/python/sample/data/scones.jpg

@@ -5,3 +5,19 @@ endpoints:
     source_type: http
     source_port: 17080
     health: /healthz
+  - target_host: generativelanguage.googleapis.com
+    target_port: 443
+    source_port: 14530
+    source_type: http
+    target_type: https
+    health: /healthz
+    redact_request_headers:
+      - X-Goog-Api-Key
+      - Authorization
+      - X-Goog-User-Project
+      - User-Agent
+      - X-Goog-Api-Client
+    response_header_replacements:
+      - header: X-Goog-Upload-Url
+        regex: "^https://generativelanguage.googleapis.com/"
+        replace: "http://localhost:14530/"

sdks/python/sample/test-data/config/test-server-config.yml

@@ -0,0 +1,56 @@
+{
+  "recordID": "684fa789a83c255a1b4046883aeeaa5788e00d54e2a710b19c92ff9dfbe57df3",
+  "interactions": [
+    {
+      "request": {
+        "method": "POST",
+        "url": "/upload/v1beta/files?upload_id=AAwnv3LPK9LarxPC5n6wRUV-HO-V5B0PImh3xDrc8DxZ5zWLPeB2gp4tFfV9UEKE_VBy4NgXy-0FMx6DhpAJ8__eUBHtSpmZw19AqV-d0d_Kcw\u0026upload_protocol=resumable",
+        "request": "POST /upload/v1beta/files?upload_id=AAwnv3LPK9LarxPC5n6wRUV-HO-V5B0PImh3xDrc8DxZ5zWLPeB2gp4tFfV9UEKE_VBy4NgXy-0FMx6DhpAJ8__eUBHtSpmZw19AqV-d0d_Kcw\u0026upload_protocol=resumable HTTP/1.1",
+        "headers": {
+          "Accept": "*/*",
+          "Accept-Encoding": "gzip, deflate",
+          "Connection": "keep-alive",
+          "Content-Length": "394671",
+          "X-Goog-Upload-Command": "upload, finalize",
+          "X-Goog-Upload-Offset": "0"
+        },
+        "bodySegments": [
+          null
+        ],
+        "previousRequest": "b4d6e60a9b97e7b98c63df9308728c5c88c0b40c398046772c63447b94608b4d",
+        "serverAddress": "generativelanguage.googleapis.com",
+        "port": 443,
+        "protocol": "https"
+      },
+      "shaSum": "f63bd502f1ea60968b95fa4361861017b5dd9d55d77cbb0f595aaaff40acdcda",
+      "response": {
+        "statusCode": 200,
+        "headers": {
+          "Content-Length": "505",
+          "Content-Type": "application/json; charset=UTF-8",
+          "Date": "Tue, 16 Sep 2025 17:07:29 GMT",
+          "Server": "UploadServer",
+          "Vary": "Origin, X-Origin, Referer",
+          "X-Goog-Upload-Status": "final",
+          "X-Guploader-Uploadid": "AAwnv3LPK9LarxPC5n6wRUV-HO-V5B0PImh3xDrc8DxZ5zWLPeB2gp4tFfV9UEKE_VBy4NgXy-0FMx6DhpAJ8__eUBHtSpmZw19AqV-d0d_Kcw"
+        },
+        "bodySegments": [
+          {
+            "file": {
+              "createTime": "2025-09-16T17:07:29.717643Z",
+              "expirationTime": "2025-09-18T17:07:29.501913838Z",
+              "mimeType": "image/jpeg",
+              "name": "files/0kkj8nf2zgqq",
+              "sha256Hash": "NzQ1MDVjZDc1OTJiMTBiYzJkNGU5YzRhYTQyOTg2NmRlNzNmNDAzZGIxMWVkZDk1YjViNDAwNGFlYTdlMTM3Ng==",
+              "sizeBytes": "394671",
+              "source": "UPLOADED",
+              "state": "ACTIVE",
+              "updateTime": "2025-09-16T17:07:29.717643Z",
+              "uri": "https://generativelanguage.googleapis.com/v1beta/files/0kkj8nf2zgqq"
+            }
+          }
+        ]
+      }
+    }
+  ]
+}

sdks/python/sample/test-data/recordings/684fa789a83c255a1b4046883aeeaa5788e00d54e2a710b19c92ff9dfbe57df3.json

@@ -0,0 +1,56 @@
+{
+  "recordID": "97139812273710c0557f818dd79cb51326a639ce83aef69013473dddb0c76ce3",
+  "interactions": [
+    {
+      "request": {
+        "method": "POST",
+        "url": "/upload/v1beta/files?upload_id=AAwnv3JuIG4MalPkarKR89aTKomHOXHoOu8hImQXUp_NlrDZPJ6QSkswH5Gx0fgqZXx_c-HlUED2aOhm5RrDn7uYRIEjFICy-0ltpjuw-g2ThQ\u0026upload_protocol=resumable",
+        "request": "POST /upload/v1beta/files?upload_id=AAwnv3JuIG4MalPkarKR89aTKomHOXHoOu8hImQXUp_NlrDZPJ6QSkswH5Gx0fgqZXx_c-HlUED2aOhm5RrDn7uYRIEjFICy-0ltpjuw-g2ThQ\u0026upload_protocol=resumable HTTP/1.1",
+        "headers": {
+          "Accept": "*/*",
+          "Accept-Encoding": "gzip, deflate",
+          "Connection": "keep-alive",
+          "Content-Length": "394671",
+          "X-Goog-Upload-Command": "upload, finalize",
+          "X-Goog-Upload-Offset": "0"
+        },
+        "bodySegments": [
+          null
+        ],
+        "previousRequest": "b4d6e60a9b97e7b98c63df9308728c5c88c0b40c398046772c63447b94608b4d",
+        "serverAddress": "generativelanguage.googleapis.com",
+        "port": 443,
+        "protocol": "https"
+      },
+      "shaSum": "f7f17a7869a6d53c61fcd70426b43579fb9c113f2e9e09a8f1a81e57ff09ca05",
+      "response": {
+        "statusCode": 200,
+        "headers": {
+          "Content-Length": "505",
+          "Content-Type": "application/json; charset=UTF-8",
+          "Date": "Tue, 16 Sep 2025 17:06:34 GMT",
+          "Server": "UploadServer",
+          "Vary": "Origin, X-Origin, Referer",
+          "X-Goog-Upload-Status": "final",
+          "X-Guploader-Uploadid": "AAwnv3JuIG4MalPkarKR89aTKomHOXHoOu8hImQXUp_NlrDZPJ6QSkswH5Gx0fgqZXx_c-HlUED2aOhm5RrDn7uYRIEjFICy-0ltpjuw-g2ThQ"
+        },
+        "bodySegments": [
+          {
+            "file": {
+              "createTime": "2025-09-16T17:06:34.492915Z",
+              "expirationTime": "2025-09-18T17:06:34.125123298Z",
+              "mimeType": "image/jpeg",
+              "name": "files/heb0ucobclos",
+              "sha256Hash": "NzQ1MDVjZDc1OTJiMTBiYzJkNGU5YzRhYTQyOTg2NmRlNzNmNDAzZGIxMWVkZDk1YjViNDAwNGFlYTdlMTM3Ng==",
+              "sizeBytes": "394671",
+              "source": "UPLOADED",
+              "state": "ACTIVE",
+              "updateTime": "2025-09-16T17:06:34.492915Z",
+              "uri": "https://generativelanguage.googleapis.com/v1beta/files/heb0ucobclos"
+            }
+          }
+        ]
+      }
+    }
+  ]
+}

sdks/python/sample/test-data/recordings/97139812273710c0557f818dd79cb51326a639ce83aef69013473dddb0c76ce3.json

@@ -29,18 +29,18 @@
           "Cache-Control": "max-age=0, private, must-revalidate",
           "Content-Encoding": "gzip",
           "Content-Language": "en-US",
-          "Content-Security-Policy": "default-src 'none'; base-uri 'self'; child-src github.githubassets.com github.com/assets-cdn/worker/ github.com/assets/ gist.github.com/assets-cdn/worker/; connect-src 'self' uploads.github.com www.githubstatus.com collector.github.com raw.githubusercontent.com api.github.com github-cloud.s3.amazonaws.com github-production-repository-file-5c1aeb.s3.amazonaws.com github-production-upload-manifest-file-7fdce7.s3.amazonaws.com github-production-user-asset-6210df.s3.amazonaws.com *.rel.tunnels.api.visualstudio.com wss://*.rel.tunnels.api.visualstudio.com objects-origin.githubusercontent.com copilot-proxy.githubusercontent.com proxy.individual.githubcopilot.com proxy.business.githubcopilot.com proxy.enterprise.githubcopilot.com *.actions.githubusercontent.com wss://*.actions.githubusercontent.com productionresultssa0.blob.core.windows.net/ productionresultssa1.blob.core.windows.net/ productionresultssa2.blob.core.windows.net/ productionresultssa3.blob.core.windows.net/ productionresultssa4.blob.core.windows.net/ productionresultssa5.blob.core.windows.net/ productionresultssa6.blob.core.windows.net/ productionresultssa7.blob.core.windows.net/ productionresultssa8.blob.core.windows.net/ productionresultssa9.blob.core.windows.net/ productionresultssa10.blob.core.windows.net/ productionresultssa11.blob.core.windows.net/ productionresultssa12.blob.core.windows.net/ productionresultssa13.blob.core.windows.net/ productionresultssa14.blob.core.windows.net/ productionresultssa15.blob.core.windows.net/ productionresultssa16.blob.core.windows.net/ productionresultssa17.blob.core.windows.net/ productionresultssa18.blob.core.windows.net/ productionresultssa19.blob.core.windows.net/ github-production-repository-image-32fea6.s3.amazonaws.com github-production-release-asset-2e65be.s3.amazonaws.com insights.github.com wss://alive.github.com wss://alive-staging.github.com api.githubcopilot.com api.individual.githubcopilot.com api.business.githubcopilot.com api.enterprise.githubcopilot.com github.githubassets.com edge.fullstory.com rs.fullstory.com; font-src github.githubassets.com; form-action 'self' github.com gist.github.com copilot-workspace.githubnext.com objects-origin.githubusercontent.com; frame-ancestors 'none'; frame-src viewscreen.githubusercontent.com notebooks.githubusercontent.com www.youtube-nocookie.com; img-src 'self' data: blob: github.githubassets.com media.githubusercontent.com camo.githubusercontent.com identicons.github.com avatars.githubusercontent.com private-avatars.githubusercontent.com github-cloud.s3.amazonaws.com objects.githubusercontent.com release-assets.githubusercontent.com secured-user-images.githubusercontent.com/ user-images.githubusercontent.com/ private-user-images.githubusercontent.com opengraph.githubassets.com marketplace-screenshots.githubusercontent.com/ copilotprodattachments.blob.core.windows.net/github-production-copilot-attachments/ github-production-user-asset-6210df.s3.amazonaws.com customer-stories-feed.github.com spotlights-feed.github.com objects-origin.githubusercontent.com *.githubusercontent.com images.ctfassets.net/8aevphvgewt8/; manifest-src 'self'; media-src github.com user-images.githubusercontent.com/ secured-user-images.githubusercontent.com/ private-user-images.githubusercontent.com github-production-user-asset-6210df.s3.amazonaws.com gist.github.com github.githubassets.com assets.ctfassets.net/8aevphvgewt8/ videos.ctfassets.net/8aevphvgewt8/; script-src github.githubassets.com; style-src 'unsafe-inline' github.githubassets.com; upgrade-insecure-requests; worker-src github.githubassets.com github.com/assets-cdn/worker/ github.com/assets/ gist.github.com/assets-cdn/worker/",
+          "Content-Security-Policy": "default-src 'none'; base-uri 'self'; child-src github.githubassets.com github.com/assets-cdn/worker/ github.com/assets/ gist.github.com/assets-cdn/worker/; connect-src 'self' uploads.github.com www.githubstatus.com collector.github.com raw.githubusercontent.com api.github.com github-cloud.s3.amazonaws.com github-production-repository-file-5c1aeb.s3.amazonaws.com github-production-upload-manifest-file-7fdce7.s3.amazonaws.com github-production-user-asset-6210df.s3.amazonaws.com *.rel.tunnels.api.visualstudio.com wss://*.rel.tunnels.api.visualstudio.com github.githubassets.com objects-origin.githubusercontent.com copilot-proxy.githubusercontent.com proxy.individual.githubcopilot.com proxy.business.githubcopilot.com proxy.enterprise.githubcopilot.com *.actions.githubusercontent.com wss://*.actions.githubusercontent.com productionresultssa0.blob.core.windows.net/ productionresultssa1.blob.core.windows.net/ productionresultssa2.blob.core.windows.net/ productionresultssa3.blob.core.windows.net/ productionresultssa4.blob.core.windows.net/ productionresultssa5.blob.core.windows.net/ productionresultssa6.blob.core.windows.net/ productionresultssa7.blob.core.windows.net/ productionresultssa8.blob.core.windows.net/ productionresultssa9.blob.core.windows.net/ productionresultssa10.blob.core.windows.net/ productionresultssa11.blob.core.windows.net/ productionresultssa12.blob.core.windows.net/ productionresultssa13.blob.core.windows.net/ productionresultssa14.blob.core.windows.net/ productionresultssa15.blob.core.windows.net/ productionresultssa16.blob.core.windows.net/ productionresultssa17.blob.core.windows.net/ productionresultssa18.blob.core.windows.net/ productionresultssa19.blob.core.windows.net/ github-production-repository-image-32fea6.s3.amazonaws.com github-production-release-asset-2e65be.s3.amazonaws.com insights.github.com wss://alive.github.com wss://alive-staging.github.com api.githubcopilot.com api.individual.githubcopilot.com api.business.githubcopilot.com api.enterprise.githubcopilot.com edge.fullstory.com rs.fullstory.com; font-src github.githubassets.com; form-action 'self' github.com gist.github.com copilot-workspace.githubnext.com objects-origin.githubusercontent.com; frame-ancestors 'none'; frame-src viewscreen.githubusercontent.com notebooks.githubusercontent.com www.youtube-nocookie.com; img-src 'self' data: blob: github.githubassets.com media.githubusercontent.com camo.githubusercontent.com identicons.github.com avatars.githubusercontent.com private-avatars.githubusercontent.com github-cloud.s3.amazonaws.com objects.githubusercontent.com release-assets.githubusercontent.com secured-user-images.githubusercontent.com/ user-images.githubusercontent.com/ private-user-images.githubusercontent.com opengraph.githubassets.com marketplace-screenshots.githubusercontent.com/ copilotprodattachments.blob.core.windows.net/github-production-copilot-attachments/ github-production-user-asset-6210df.s3.amazonaws.com customer-stories-feed.github.com spotlights-feed.github.com objects-origin.githubusercontent.com *.githubusercontent.com images.ctfassets.net/8aevphvgewt8/; manifest-src 'self'; media-src github.com user-images.githubusercontent.com/ secured-user-images.githubusercontent.com/ private-user-images.githubusercontent.com github-production-user-asset-6210df.s3.amazonaws.com gist.github.com github.githubassets.com assets.ctfassets.net/8aevphvgewt8/ videos.ctfassets.net/8aevphvgewt8/; script-src github.githubassets.com; style-src 'unsafe-inline' github.githubassets.com; upgrade-insecure-requests; worker-src github.githubassets.com github.com/assets-cdn/worker/ github.com/assets/ gist.github.com/assets-cdn/worker/",
           "Content-Type": "text/html; charset=utf-8",
-          "Date": "Mon, 15 Sep 2025 17:58:03 GMT",
-          "Etag": "W/\"cd488b2ea2ae9702b6652dfcb4294b9f\"",
+          "Date": "Tue, 16 Sep 2025 17:07:20 GMT",
+          "Etag": "W/\"fbfb80938c80947c01efcd9b07d550e6\"",
           "Referrer-Policy": "origin-when-cross-origin, strict-origin-when-cross-origin",
           "Server": "github.com",
-          "Set-Cookie": "_gh_sess=3YXTN2Kn9fZ548MFymDvF4bmU3CVT1fhBeYO4rhAseW2l4rkmT0wzpD6sxgKPmt6uiXpcanTYQoSk0His54Z5ZVw1bPeaynC5BS5XzhvbMHfO4jT%2Ffg7udqglbBw1lmCRyr34q2WlFb%2F%2BM5aF3ErLwjPQUIHzxN1MH12fLFOxbh919Qvxs%2FEc%2FEdCQkb6a5wSvhYSihZXyGtDSQkRN%2FtE74aZZzOhn9hhHo68SeDmE47VCWIRm%2Bk1MGP3C6YiXwHjHF3T0%2Bu4OupJqUgKvKkbw%3D%3D--q7RzlJy371g%2FhXQu--2ZImD096TJjWHHRIqjJi%2Bg%3D%3D; Path=/; HttpOnly; Secure; SameSite=Lax, _octo=GH1.1.1334543195.1757959087; Path=/; Domain=github.com; Expires=Tue, 15 Sep 2026 17:58:07 GMT; Secure; SameSite=Lax, logged_in=no; Path=/; Domain=github.com; Expires=Tue, 15 Sep 2026 17:58:07 GMT; HttpOnly; Secure; SameSite=Lax",
+          "Set-Cookie": "_gh_sess=5uG%2FiPcnYWHHlvxfs4Jg5qhwVdzXE9yF8Vny6wIYl8DlBINY0uD22%2Fa9GyjhTLc8kOv7pJyW%2FmQ1i00xnIHt6qRJnmXFlsPo%2BmfSKrSMcZP77525zGMvcYc6XyzzJixzr5GB4bsOTLNmArmPEMFeLqTFQSEzDPsLdFkJdcm1tshI11cFcOGhX83qmCHn1htpVVDD9xAUU64YzMud2jrxAjOdg2hO1xLfe3J1x%2BjNJhmh8rnakdwZQCmIXzpX3R6XyRitDDk%2F4Y%2FMvUME8dkiRg%3D%3D--NNWFRZ4hSwlY9waY--iCcRp2MPy2pzsEGvoWJ14A%3D%3D; Path=/; HttpOnly; Secure; SameSite=Lax, _octo=GH1.1.1388852089.1758042448; Path=/; Domain=github.com; Expires=Wed, 16 Sep 2026 17:07:28 GMT; Secure; SameSite=Lax, logged_in=no; Path=/; Domain=github.com; Expires=Wed, 16 Sep 2026 17:07:28 GMT; HttpOnly; Secure; SameSite=Lax",
           "Strict-Transport-Security": "max-age=31536000; includeSubdomains; preload",
           "Vary": "X-PJAX, X-PJAX-Container, Turbo-Visit, Turbo-Frame, X-Requested-With, Accept-Language,Accept-Encoding, Accept, X-Requested-With",
           "X-Content-Type-Options": "nosniff",
           "X-Frame-Options": "deny",
-          "X-Github-Request-Id": "D872:2A9986:920E617:96857F7:68C853AF",
+          "X-Github-Request-Id": "DA2B:1A28F1:44A2F33:462F309:68C99950",
           "X-Xss-Protection": "0"
         }
       }