Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchani.git" did not exist on "4bacf402fb0dfe03943330b6fa2f56fc4b90af17"
Commit fad00a85 authored by Bruce MacDonald's avatar Bruce MacDonald
Browse files

stop running model on interactive exit

parent 62be2050
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
cmd/interactive.go
View file @ fad00a85
...@@ -475,6 +475,14 @@ func generateInteractive(cmd *cobra.Command, opts runOptions) error { ...@@ -475,6 +475,14 @@ func generateInteractive(cmd *cobra.Command, opts runOptions) error {
usage() usage()
} }
case strings.HasPrefix(line, "/exit"), strings.HasPrefix(line, "/bye"): case strings.HasPrefix(line, "/exit"), strings.HasPrefix(line, "/bye"):
// stop the running model
client, err := api.ClientFromEnvironment()
if err != nil {
return err
}
client.Chat(cmd.Context(), &api.ChatRequest{Model: opts.Model, KeepAlive: &api.Duration{}}, func(resp api.ChatResponse) error {
return nil
})
return nil return nil
case strings.HasPrefix(line, "/"): case strings.HasPrefix(line, "/"):
args := strings.Fields(line) args := strings.Fields(line)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment