Updates

Author: djthorpe

cmd/llm/chat.go

@@ -19,6 +19,7 @@ type ChatCmd struct {
 	Model    string `arg:"" help:"Model name"`
 	NoStream bool   `flag:"nostream" help:"Disable streaming"`
 	NoTools  bool   `flag:"nostream" help:"Disable tool calling"`
+	Prompt   string `flag:"prompt" help:"Set the initial user prompt"`
 	System   string `flag:"system" help:"Set the system prompt"`
 }
 
@@ -60,11 +61,17 @@ func (cmd *ChatCmd) Run(globals *Globals) error {
 
 		// Continue looping until end of input
 		for {
-			input, err := globals.term.ReadLine(model.Name() + "> ")
-			if errors.Is(err, io.EOF) {
-				return nil
-			} else if err != nil {
-				return err
+			var input string
+			if cmd.Prompt != "" {
+				input = cmd.Prompt
+				cmd.Prompt = ""
+			} else {
+				input, err = globals.term.ReadLine(model.Name() + "> ")
+				if errors.Is(err, io.EOF) {
+					return nil
+				} else if err != nil {
+					return err
+				}
 			}
 
 			// Ignore empty input

cmd/llm/models.go

@@ -60,7 +60,7 @@ func (*ListAgentsCmd) Run(globals *Globals) error {
 			return fmt.Errorf("No agents found")
 		}
 
-		var agents []string
+		agents := make([]string, 0, len(agent.Agents()))
 		for _, agent := range agent.Agents() {
 			agents = append(agents, agent.Name())
 		}

pkg/mistral/model.go

@@ -12,8 +12,8 @@ import (
 // TYPES
 
 type model struct {
-	*Client
-	meta Model
+	*Client `json:"-"`
+	meta    Model
 }
 
 type Model struct {

pkg/ollama/model.go

@@ -16,7 +16,7 @@ import (
 
 // model is the implementation of the llm.Model interface
 type model struct {
-	*Client
+	*Client `json:"-"`
 	ModelMeta
 }
 
@@ -60,8 +60,12 @@ type PullStatus struct {
 ///////////////////////////////////////////////////////////////////////////////
 // STRINGIFY
 
+func (m model) MarshalJSON() ([]byte, error) {
+	return json.Marshal(m.ModelMeta)
+}
+
 func (m model) String() string {
-	data, err := json.MarshalIndent(m.ModelMeta, "", "  ")
+	data, err := json.MarshalIndent(m, "", "  ")
 	if err != nil {
 		return err.Error()
 	}