Spaces:
Running
Running
Commit
·
8ae1091
1
Parent(s):
5d665be
mimetype for other files are supported
Browse files- src/manager/manager.py +3 -1
src/manager/manager.py
CHANGED
|
@@ -18,6 +18,7 @@ from src.tools.default_tools.memory_manager import MemoryManager
|
|
| 18 |
from pathlib import Path
|
| 19 |
from google.genai.errors import APIError
|
| 20 |
import backoff
|
|
|
|
| 21 |
|
| 22 |
logger = logging.getLogger(__name__)
|
| 23 |
handler = logging.StreamHandler(sys.stdout)
|
|
@@ -179,10 +180,11 @@ class GeminiManager:
|
|
| 179 |
path = message["content"][0]
|
| 180 |
try:
|
| 181 |
image_bytes = open(path, "rb").read()
|
|
|
|
| 182 |
parts = [
|
| 183 |
types.Part.from_bytes(
|
| 184 |
data=image_bytes,
|
| 185 |
-
mime_type=
|
| 186 |
),
|
| 187 |
]
|
| 188 |
except Exception as e:
|
|
|
|
| 18 |
from pathlib import Path
|
| 19 |
from google.genai.errors import APIError
|
| 20 |
import backoff
|
| 21 |
+
import mimetypes
|
| 22 |
|
| 23 |
logger = logging.getLogger(__name__)
|
| 24 |
handler = logging.StreamHandler(sys.stdout)
|
|
|
|
| 180 |
path = message["content"][0]
|
| 181 |
try:
|
| 182 |
image_bytes = open(path, "rb").read()
|
| 183 |
+
mime_type, _ = mimetypes.guess_type(path)
|
| 184 |
parts = [
|
| 185 |
types.Part.from_bytes(
|
| 186 |
data=image_bytes,
|
| 187 |
+
mime_type=mime_type
|
| 188 |
),
|
| 189 |
]
|
| 190 |
except Exception as e:
|