Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,72 +1,20 @@ | ||
| github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= | ||
| github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8= | ||
| github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw= | ||
| github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= | ||
| github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM= | ||
| github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y= | ||
| github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= | ||
| github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= | ||
| golang.design/x/clipboard v0.7.0 h1:4Je8M/ys9AJumVnl8m+rZnIvstSnYj1fvzqYrU3TXvo= | ||
| golang.design/x/clipboard v0.7.0/go.mod h1:PQIvqYO9GP29yINEfsEn5zSQKAz3UgXmZKzDA6dnq2E= | ||
| golang.org/x/exp/shiny v0.0.0-20241009180824-f66d83c29e7c h1:jTMrjjZRcSH/BDxWhXCP6OWsfVgmnwI7J+F4/nyVXaU= | ||
| golang.org/x/exp/shiny v0.0.0-20241009180824-f66d83c29e7c/go.mod h1:3F+MieQB7dRYLTmnncoFbb1crS5lfQoTfDgQy6K4N0o= | ||
| golang.org/x/image v0.14.0 h1:tNgSxAFe3jC4uYqvZdTr84SZoM1KfwdC9SKIFrLjFn4= | ||
| golang.org/x/image v0.14.0/go.mod h1:HUYqC05R2ZcZ3ejNQsIHQDQiwWM4JBqmm6MKANTp4LE= | ||
| golang.org/x/mobile v0.0.0-20231127183840-76ac6878050a h1:sYbmY3FwUWCBTodZL1S3JUuOvaW6kM2o+clDzzDNBWg= | ||
| golang.org/x/mobile v0.0.0-20231127183840-76ac6878050a/go.mod h1:Ede7gF0KGoHlj822RtphAHK1jLdrcuRBZg0sF1Q+SPc= | ||
| golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4= | ||
| golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||
| gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,55 @@ | ||
| package cmd | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "path/filepath" | ||
|
|
||
| "github.com/gphorvath/grimoire/src/config" | ||
| "github.com/spf13/cobra" | ||
| "golang.design/x/clipboard" | ||
| ) | ||
|
|
||
| var copyCmd = &cobra.Command{ | ||
| Use: "copy [prompt name]", | ||
| Short: "Copy the prompt to the clipboard", | ||
| Args: cobra.ExactArgs(1), | ||
| Run: runCopyCmd, | ||
| } | ||
|
|
||
| func init() { | ||
| rootCmd.AddCommand(copyCmd) | ||
| } | ||
|
|
||
| func runCopyCmd(cmd *cobra.Command, args []string) { | ||
| baseDir := config.GetPromptDir() | ||
| filename := args[0] + ".md" | ||
|
|
||
| dir, err := findFileDir(baseDir, filename) | ||
| if err != nil { | ||
| fmt.Fprintf(os.Stderr, "Error: %v\n", err) | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| if dir == "" { | ||
| fmt.Fprintf(os.Stderr, "Error: prompt not found\n") | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| filePath := filepath.Join(dir, filename) | ||
|
|
||
| // Init returns an error if the package is not ready for use. | ||
| err = clipboard.Init() | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| // Copy the contents of the file to the clipboard | ||
| content, err := os.ReadFile(filePath) | ||
| if err != nil { | ||
| fmt.Fprintf(os.Stderr, "Error: %v\n", err) | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| clipboard.Write(clipboard.FmtText, content) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| package cmd | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
|
|
||
| "github.com/gphorvath/grimoire/src/config" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| type OllamaRequest struct { | ||
| Model string `json:"model"` | ||
| Prompt string `json:"prompt"` | ||
| Stream bool `json:"stream"` | ||
| } | ||
|
|
||
| type OllamaResponse struct { | ||
| Response string `json:"response"` | ||
| } | ||
|
|
||
| var ( | ||
| completeCmd = &cobra.Command{ | ||
| Use: "complete [prompt]", | ||
| Short: "Get completion from Ollama", | ||
| Args: cobra.ExactArgs(1), | ||
| RunE: runComplete, | ||
| } | ||
| ) | ||
|
|
||
| func init() { | ||
| rootCmd.AddCommand(completeCmd) | ||
| } | ||
|
|
||
| func runComplete(cmd *cobra.Command, args []string) error { | ||
| prompt := args[0] | ||
|
|
||
| // Create request body | ||
| reqBody := OllamaRequest{ | ||
| Model: config.OllamaModel, | ||
| Prompt: prompt, | ||
| Stream: config.OllamaStream, | ||
| } | ||
|
|
||
| jsonData, err := json.Marshal(reqBody) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| url := config.OllamaURL + "/generate" | ||
|
|
||
| // Make POST request to Ollama API | ||
| resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonData)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| // Handle streaming response | ||
| decoder := json.NewDecoder(resp.Body) | ||
| for { | ||
| var ollamaResp OllamaResponse | ||
| if err := decoder.Decode(&ollamaResp); err != nil { | ||
| if err == io.EOF { | ||
| fmt.Println() | ||
| break | ||
| } | ||
| return err | ||
| } | ||
| fmt.Print(ollamaResp.Response) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters