Fix file uploads (#291)

MarkDaoust · web-flow · commit 625e4bd0ef16 · 2024-04-18T13:44:05.000-07:00
diff --git a/google/generativeai/client.py b/google/generativeai/client.py
@@ -73,10 +73,13 @@ def create_file(
         request = self._discovery_api.media().upload(body={"file": file}, media_body=media)
         result = request.execute()
 
+        allowed_keys = set(glm.File.__annotations__)
+
         return glm.File(
             {
                 re.sub("[A-Z]", lambda ch: f"_{ch.group(0).lower()}", key): value
                 for key, value in result["file"].items()
+                if key in allowed_keys
             }
         )
 

Original file line number	Diff line number	Diff line change
`@@ -73,10 +73,13 @@ def create_file(`
`73`	`73`	`request = self._discovery_api.media().upload(body={"file": file}, media_body=media)`
`74`	`74`	`result = request.execute()`
`75`	`75`
	`76`	`+ allowed_keys = set(glm.File.__annotations__)`
	`77`	`+`
`76`	`78`	`return glm.File(`
`77`	`79`	`{`
`78`	`80`	`re.sub("[A-Z]", lambda ch: f"_{ch.group(0).lower()}", key): value`
`79`	`81`	`for key, value in result["file"].items()`
	`82`	`+ if key in allowed_keys`
`80`	`83`	`}`
`81`	`84`	`)`
`82`	`85`