Update src/tools/tools.py
Browse files- src/tools/tools.py +6 -0
src/tools/tools.py
CHANGED
|
@@ -107,6 +107,12 @@ def preprocess_files(files: List[str]) -> Dict[str, Dict[str, Any]]:
|
|
| 107 |
elif file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
|
| 108 |
info["type"] = "image"
|
| 109 |
info["suggested_tool"] = "vision_qa_gemma"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 110 |
else:
|
| 111 |
info["type"] = "unknown"
|
| 112 |
info["suggested_tool"] = "analyze_txt_file (fallback)"
|
|
|
|
| 107 |
elif file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
|
| 108 |
info["type"] = "image"
|
| 109 |
info["suggested_tool"] = "vision_qa_gemma"
|
| 110 |
+
elif file_ext in [".mp3"]:
|
| 111 |
+
info["type"] = "audio"
|
| 112 |
+
info["suggested_tool"] = "transcribe_audio"
|
| 113 |
+
elif file_ext in [".mp4"]:
|
| 114 |
+
info["type"] = "video"
|
| 115 |
+
info["suggested_tool"] = "video_qa_gemma"
|
| 116 |
else:
|
| 117 |
info["type"] = "unknown"
|
| 118 |
info["suggested_tool"] = "analyze_txt_file (fallback)"
|