simplify api client

This commit is contained in:
Michael Yang
2023-07-06 15:02:10 -07:00
parent c4b9e84945
commit b0e63bfb4c
2 changed files with 21 additions and 25 deletions

View File

@@ -36,10 +36,7 @@ func RunRun(cmd *cobra.Command, args []string) error {
}
func pull(model string) error {
client, err := NewAPIClient()
if err != nil {
return err
}
client := api.NewClient()
var bar *progressbar.ProgressBar
return client.Pull(
@@ -68,10 +65,7 @@ func RunGenerate(_ *cobra.Command, args []string) error {
}
func generate(model string, prompts ...string) error {
client, err := NewAPIClient()
if err != nil {
return err
}
client := api.NewClient()
for _, prompt := range prompts {
client.Generate(context.Background(), &api.GenerateRequest{Model: model, Prompt: prompt}, func(resp api.GenerateResponse) error {
@@ -121,12 +115,6 @@ func RunServer(_ *cobra.Command, _ []string) error {
return server.Serve(ln)
}
func NewAPIClient() (*api.Client, error) {
return &api.Client{
URL: "http://localhost:11434",
}, nil
}
func NewCLI() *cobra.Command {
log.SetFlags(log.LstdFlags | log.Lshortfile)