🐛 fixes #24 and #25

yusufcanb · Apr 14, 2024 · 0763640 · 0763640
1 parent 261b103
commit 0763640
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/explain/api.go b/explain/api.go
@@ -52,5 +52,6 @@ func (e *Explain) StreamExplanationFor(mode, prompt string) error {
 	if err != nil {
 		fmt.Println("Error during generation:", err)
 	}
+	fmt.Printf("\n")
 	return nil
 }
diff --git a/explain/explain.go b/explain/explain.go
@@ -10,8 +10,8 @@ import (
 var modelFile string
 
 type Explain struct {
-	api *ollama.Client
-
+	api       *ollama.Client
+	version   string
 	tag       string
 	modelfile string
 }
@@ -26,5 +26,5 @@ func (e *Explain) Modelfile() string {
 
 func New(api *ollama.Client, version string) *Explain {
 	modelfileName := fmt.Sprintf("tlm:%s-e", version)
-	return &Explain{api: api, tag: modelfileName, modelfile: modelFile}
+	return &Explain{api: api, tag: modelfileName, modelfile: modelFile, version: version}
 }
diff --git a/suggest/cli.go b/suggest/cli.go
@@ -102,7 +102,7 @@ func (s *Suggest) action(c *cli.Context) error {
 	if form.action == Explain {
 		fmt.Println(shell.SuccessMessage("┃ > ") + "Explaining..." + "\n")
 
-		exp := explain.New(s.api, "")
+		exp := explain.New(s.api, s.version)
 		err = exp.StreamExplanationFor(Stable, form.command)
 		if err != nil {
 			return err

diff --git a/suggest/suggest.go b/suggest/suggest.go
@@ -11,6 +11,7 @@ var suggestModelfile string
 
 type Suggest struct {
 	api       *ollama.Client
+	version   string
 	tag       string
 	modelfile string
 }
@@ -25,5 +26,5 @@ func (s *Suggest) Modelfile() string {
 
 func New(api *ollama.Client, version string) *Suggest {
 	tag := fmt.Sprintf("tlm:%s-s", version)
-	return &Suggest{api: api, tag: tag, modelfile: suggestModelfile}
+	return &Suggest{api: api, tag: tag, modelfile: suggestModelfile, version: version}
 }