/**
 * @author: dn-jinmin/dn-jinmin
 * @doc:
 */

package main

import (
	"fmt"
	"gitee.com/dn-jinmin/claude"
)

var (
	apikey = "sk-6wWCfobHHEX9n2QuGA4yi0jouBQ46DkQpL9XPgaAbuEd71c3"
	url    = "https://api.aiproxy.io"
)

func main() {
	client, err := claude.NewClient(apikey, claude.WithClientBaseURL(url))
	if err != nil {
		panic(err)
	}

	req := &claude.CompletionRequest{
		Prompt: "\n\nHuman: hello world!\n\nAssistant: ",
		Model:  claude.ClaudeV2,
	}
	resp, err := client.Complete(req)
	if err != nil {
		panic(err)
	}

	fmt.Println(resp)
}