与模型对话

源码

main.go

package main

import (
    "context"
    "fmt"
    "log"

    "github.com/ollama/ollama/api"
)

func main() {
    client, err := api.ClientFromEnvironment()
    if err != nil {
        log.Fatal(err)
    }

    messages := []api.Message{
        api.Message{
            Role:    "system",
            Content: "Provide very brief, concise responses",
        },
        api.Message{
            Role:    "user",
            Content: "Name some unusual animals",
        },
        api.Message{
            Role:    "assistant",
            Content: "Monotreme, platypus, echidna",
        },
        api.Message{
            Role:    "user",
            Content: "which of these is the most dangerous?",
        },
    }

    ctx := context.Background()
    req := &api.ChatRequest{
        Model:    "llama2",
        Messages: messages,
    }

    respFunc := func(resp api.ChatResponse) error {
        fmt.Print(resp.Message.Content)
        return nil
    }

    err = client.Chat(ctx, req, respFunc)
    if err != nil {
        log.Fatal(err)
    }
}

仓库地址

bash-comparemodels