Added anthropic (claude) chat

Author: djthorpe

cmd/api/anthropic.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/djthorpe/go-tablewriter"
+	"github.com/mutablelogic/go-client"
+	"github.com/mutablelogic/go-client/pkg/anthropic"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// GLOBALS
+
+var (
+	anthropicName   = "claude"
+	anthropicClient *anthropic.Client
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// LIFECYCLE
+
+func anthropicRegister(flags *Flags) {
+	// Register flags required
+	flags.String(anthropicName, "anthropic-api-key", "${ANTHROPIC_API_KEY}", "API Key")
+
+	flags.Register(Cmd{
+		Name:        anthropicName,
+		Description: "Interact with Claude, from https://www.anthropic.com/api",
+		Parse:       anthropicParse,
+		Fn: []Fn{
+			{Name: "chat", Call: anthropicChat, Description: "Chat with Claude", MinArgs: 1},
+		},
+	})
+}
+
+func anthropicParse(flags *Flags, opts ...client.ClientOpt) error {
+	apiKey := flags.GetString("anthropic-api-key")
+	if apiKey == "" {
+		return fmt.Errorf("missing -anthropic-api-key flag")
+	}
+	if client, err := anthropic.New(flags.GetString("anthropic-api-key"), opts...); err != nil {
+		return err
+	} else {
+		anthropicClient = client
+	}
+
+	// Return success
+	return nil
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// METHODS
+
+func anthropicChat(w *tablewriter.Writer, args []string) error {
+	// Request -> Response
+	message := anthropic.NewMessage("user", strings.Join(args, " "))
+	responses, err := anthropicClient.Messages(message)
+	if err != nil {
+		return err
+	}
+
+	// Write table
+	return w.Write(responses)
+}

cmd/api/cmd.go

@@ -48,7 +48,8 @@ func (c *Cmd) Get(args []string) (*Fn, []string, error) {
 	// Check number of arguments
 	if fn.MinArgs != 0 && nargs < fn.MinArgs {
 		return nil, nil, fmt.Errorf("not enough arguments for %q", fn.Name)
-	} else if nargs > fn.MaxArgs {
+	}
+	if fn.MaxArgs != 0 && nargs > fn.MaxArgs {
 		return nil, nil, fmt.Errorf("too many arguments for %q", fn.Name)
 	}
 

cmd/api/main.go

@@ -17,6 +17,7 @@ func main() {
 	ipifyRegister(flags)
 	bwRegister(flags)
 	newsapiRegister(flags)
+	anthropicRegister(flags)
 
 	// Parse
 	if err := flags.Parse(os.Args[1:]); errors.Is(err, ErrHelp) {

pkg/anthropic/client.go

@@ -0,0 +1,43 @@
+/*
+anthropic implements an API client for anthropic (https://docs.anthropic.com/en/api/getting-started)
+*/
+package anthropic
+
+import (
+	// Packages
+	"github.com/mutablelogic/go-client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type Client struct {
+	*client.Client
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// GLOBALS
+
+const (
+	endPoint            = "https://api.anthropic.com/v1"
+	defaultVersion      = "2023-06-01"
+	defaultMessageModel = "claude-3-haiku-20240307"
+	defaultMaxTokens    = 4096
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// LIFECYCLE
+
+// Create a new client
+func New(ApiKey string, opts ...client.ClientOpt) (*Client, error) {
+	// Create client
+	opts = append(opts, client.OptEndpoint(endPoint))
+	opts = append(opts, client.OptHeader("x-api-key", ApiKey), client.OptHeader("anthropic-version", defaultVersion))
+	client, err := client.New(opts...)
+	if err != nil {
+		return nil, err
+	}
+
+	// Return the client
+	return &Client{client}, nil
+}

pkg/anthropic/client_test.go

@@ -0,0 +1,31 @@
+package anthropic_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client"
+	anthropic "github.com/mutablelogic/go-client/pkg/anthropic"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_client_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := anthropic.New(GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+	t.Log(client)
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// ENVIRONMENT
+
+func GetApiKey(t *testing.T) string {
+	key := os.Getenv("ANTHROPIC_API_KEY")
+	if key == "" {
+		t.Skip("ANTHROPIC_API_KEY not set")
+		t.SkipNow()
+	}
+	return key
+}

pkg/anthropic/message.go

@@ -0,0 +1,63 @@
+package anthropic
+
+import (
+	// Packages
+	"github.com/mutablelogic/go-client"
+	schema "github.com/mutablelogic/go-client/pkg/openai/schema"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type reqMessage struct {
+	Model     string           `json:"model"`
+	Messages  []schema.Message `json:"messages,omitempty"`
+	MaxTokens int              `json:"max_tokens,omitempty"`
+}
+
+type respMessage struct {
+	Id           string           `json:"id"`
+	Type         string           `json:"type,omitempty"`
+	Role         string           `json:"role"`
+	Model        string           `json:"model"`
+	StopReason   string           `json:"stop_reason"`
+	StopSequence string           `json:"stop_sequence"`
+	Content      []schema.Content `json:"content"`
+	Usage        struct {
+		InputTokens int `json:"input_tokens"`
+		OuputTokens int `json:"output_tokens"`
+	} `json:"usage"`
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// PUBLIC METHODS
+
+func NewMessage(role, text string) schema.Message {
+	return schema.Message{
+		Role:    role,
+		Content: text,
+	}
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// Send a structured list of input messages with text and/or image content,
+// and the model will generate the next message in the conversation.
+func (c *Client) Messages(message schema.Message) ([]schema.Content, error) {
+	var request reqMessage
+	var response respMessage
+
+	request.Model = defaultMessageModel
+	request.MaxTokens = defaultMaxTokens
+	request.Messages = []schema.Message{message}
+
+	// Return the response
+	if payload, err := client.NewJSONRequest(request); err != nil {
+		return nil, err
+	} else if err := c.Do(payload, &response, client.OptPath("messages")); err != nil {
+		return nil, err
+	} else {
+		return response.Content, nil
+	}
+}

pkg/anthropic/message_test.go

@@ -0,0 +1,23 @@
+package anthropic_test
+
+import (
+	"os"
+	"testing"
+
+	opts "github.com/mutablelogic/go-client"
+	anthropic "github.com/mutablelogic/go-client/pkg/anthropic"
+	schema "github.com/mutablelogic/go-client/pkg/openai/schema"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_message_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := anthropic.New(GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+	_, err = client.Messages(schema.Message{
+		Role:    "user",
+		Content: "What is the weather today",
+	})
+	assert.NoError(err)
+}

pkg/openai/schema/chat.go

@@ -12,3 +12,8 @@ type MessageChoice struct {
 	Index        int    `json:"index"`
 	FinishReason string `json:"finish_reason"`
 }
+
+type Content struct {
+	Type string `json:"type,width:4"`
+	Text string `json:"text,wrap,width:60"`
+}