add support to upload file-like object

Author: Hamza-nabil

google/generativeai/files.py

@@ -21,6 +21,7 @@
 import logging
 from google.generativeai import protos
 from itertools import islice
+from io import IOBase
 
 from google.generativeai.types import file_types
 
@@ -32,7 +33,7 @@
 
 
 def upload_file(
-    path: str | pathlib.Path | os.PathLike,
+    path: str | pathlib.Path | os.PathLike | IOBase,
     *,
     mime_type: str | None = None,
     name: str | None = None,
@@ -42,7 +43,7 @@ def upload_file(
     """Calls the API to upload a file using a supported file service.
 
     Args:
-        path: The path to the file to be uploaded.
+        path: The path to the file or a file-like object (e.g., BytesIO) to be uploaded.
         mime_type: The MIME type of the file. If not provided, it will be
             inferred from the file extension.
         name: The name of the file in the destination (e.g., 'files/sample-image').
@@ -57,17 +58,22 @@ def upload_file(
     """
     client = get_default_file_client()
 
-    path = pathlib.Path(os.fspath(path))
+    if not isinstance(path, IOBase):
+        path = pathlib.Path(os.fspath(path))
+
+        if display_name is None:
+            display_name = path.name
+
+        if mime_type is None:
+            mime_type, _ = mimetypes.guess_type(path)
 
     if mime_type is None:
-        mime_type, _ = mimetypes.guess_type(path)
+        # Guess failed or IOBase, use octet-stream.
+        mime_type = 'application/octet-stream'
 
     if name is not None and "/" not in name:
         name = f"files/{name}"
 
-    if display_name is None:
-        display_name = path.name
-
     response = client.create_file(
         path=path, mime_type=mime_type, name=name, display_name=display_name, resumable=resumable
     )