🔧 check modelfile is exists

yusufcanb · Mar 2, 2024 · 16ce0f7 · 16ce0f7
1 parent 2524b93
commit 16ce0f7
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/explain/cli.go b/explain/cli.go
@@ -16,6 +16,22 @@ func (e *Explain) before(_ *cli.Context) error {
 		os.Exit(-1)
 	}
 
+	list, err := e.api.List(context.Background())
+	if err != nil {
+		fmt.Println(shell.Err() + " " + err.Error())
+		fmt.Println(shell.Err() + " Ollama connection failed. Please check your Ollama if it's running or configured correctly.")
+		os.Exit(-1)
+	}
+
+	for _, model := range list.Models {
+		if model.Name == e.modelfile {
+			return nil
+		}
+		fmt.Println(shell.Err() + " " + "tlm explain:7b model not found.\n\nPlease run `tlm deploy` to deploy tlm models first.")
+		os.Exit(-1)
+		return nil
+	}
+
 	return nil
 }
 

diff --git a/suggest/cli.go b/suggest/cli.go
@@ -22,6 +22,22 @@ func (s *Suggest) before(_ *cli.Context) error {
 		os.Exit(-1)
 	}
 
+	list, err := s.api.List(context.Background())
+	if err != nil {
+		fmt.Println(shell.Err() + " " + err.Error())
+		fmt.Println(shell.Err() + " Ollama connection failed. Please check your Ollama if it's running or configured correctly.")
+		os.Exit(-1)
+	}
+
+	for _, model := range list.Models {
+		if model.Name == s.modelfile {
+			return nil
+		}
+		fmt.Println(shell.Err() + " " + "tlm suggest:7b model not found.\n\nPlease run `tlm deploy` to deploy tlm models first.")
+		os.Exit(-1)
+		return nil
+	}
+
 	return nil
 }