Merge pull request #668 from transformerlab/fix/fix-no-ollama-but-still-ollama

Fixed Terminal error if you try to import models and don't have Ollama installed

Author: ParamThakkar123

transformerlab/models/ollamamodel.py

@@ -13,6 +13,9 @@
 async def list_models():
     try:
         ollama_model_library = ollama_models_library_dir()
+    except FileNotFoundError:
+        print("Skipping Ollama models: manifests directory not found")
+        return []
     except Exception as e:
         print("Failed to locate Ollama models library:")
         print(str(e))

transformerlab/routers/tasks.py

@@ -619,8 +619,6 @@ async def get_task_file_content(task_dir: str, file_path: str):
         return {"status": "error", "message": "An error occurred while getting task file content"}
 
 
-
-
 @router.post("/install_from_gallery", summary="Install a task from transformerlab/galleries to local tasks-gallery")
 async def install_task_from_gallery(
     id: str = Form(...),
@@ -958,9 +956,7 @@ async def import_task_from_local_gallery(
                                 remote_info = resp.json()
                                 # Extract uploaded_dir path from response
                                 uploaded_dir = (
-                                    remote_info.get("uploaded_files", {})
-                                    .get("dir_files", {})
-                                    .get("uploaded_dir")
+                                    remote_info.get("uploaded_files", {}).get("dir_files", {}).get("uploaded_dir")
                                 )
                                 if uploaded_dir:
                                     try: