From 227b37d76867f91604abaec934bae5ede472a07b Mon Sep 17 00:00:00 2001
From: Google Colaboratory Team <colaboratory-team@google.com>
Date: Mon, 1 Apr 2024 11:54:58 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 620911635
---
 google/colab/_import_hooks/_generativeai.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/google/colab/_import_hooks/_generativeai.py b/google/colab/_import_hooks/_generativeai.py
index ca10d1f8..5923d91f 100644
--- a/google/colab/_import_hooks/_generativeai.py
+++ b/google/colab/_import_hooks/_generativeai.py
@@ -61,6 +61,8 @@ def load_module(self, fullname):
 
         def fetch(request):
           path = request.path
+          if request.query:
+            path = f'{path}?{request.query}'
           method = request.method
           headers = json.dumps(dict(request.headers))
           body = repr(request.body.decode('utf-8')) if request.body else 'null'