Update client.py (#325)

MarkDaoust · web-flow · commit 50c86affe9fc · 2024-05-08T07:28:10.000+08:00
diff --git a/google/generativeai/client.py b/google/generativeai/client.py
@@ -73,7 +73,7 @@ def create_file(
         request = self._discovery_api.media().upload(body={"file": file}, media_body=media)
         result = request.execute()
 
-        return self.get_file({"name": result["name"]})
+        return self.get_file({"name": result["file"]["name"]})
 
 
 class FileServiceAsyncClient(glm.FileServiceAsyncClient):
