Fix 'ResourceWarning: unclosed <ssl.SSLSocket>' (#422)

MarkDaoust · web-flow · commit 12774d0025c1 · 2024-07-03T09:32:03.000-07:00
Change-Id: I013bd4b4d559913224a49f7a69c5daab78acc157
diff --git a/google/generativeai/client.py b/google/generativeai/client.py
@@ -63,6 +63,7 @@ def _setup_discovery_api(self):
             uri=f"{GENAI_API_DISCOVERY_URL}?version=v1beta&key={api_key}",
         )
         response, content = request.execute()
+        request.http.close()
 
         discovery_doc = content.decode("utf-8")
         self._discovery_api = googleapiclient.discovery.build_from_document(

Original file line number	Diff line number	Diff line change
`@@ -63,6 +63,7 @@ def _setup_discovery_api(self):`
`63`	`63`	`uri=f"{GENAI_API_DISCOVERY_URL}?version=v1beta&key={api_key}",`
`64`	`64`	`)`
`65`	`65`	`response, content = request.execute()`
	`66`	`+ request.http.close()`
`66`	`67`
`67`	`68`	`discovery_doc = content.decode("utf-8")`
`68`	`69`	`self._discovery_api = googleapiclient.discovery.build_from_document(`