github.com/anthropics/anthropic-sdk-go@v1.70.1/usage_test.go (about)

     1  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
     2  
     3  package anthropic_test
     4  
     5  import (
     6  	"context"
     7  	"os"
     8  	"testing"
     9  
    10  	"github.com/anthropics/anthropic-sdk-go"
    11  	"github.com/anthropics/anthropic-sdk-go/internal/testutil"
    12  	"github.com/anthropics/anthropic-sdk-go/option"
    13  )
    14  
    15  func TestUsage(t *testing.T) {
    16  	baseURL := "http://localhost:4010"
    17  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    18  		baseURL = envURL
    19  	}
    20  	if !testutil.CheckTestServer(t, baseURL) {
    21  		return
    22  	}
    23  	client := anthropic.NewClient(
    24  		option.WithBaseURL(baseURL),
    25  		option.WithAPIKey("my-anthropic-api-key"),
    26  	)
    27  	message, err := client.Messages.New(context.TODO(), anthropic.MessageNewParams{
    28  		MaxTokens: 1024,
    29  		Messages: []anthropic.MessageParam{{
    30  			Content: []anthropic.ContentBlockParamUnion{{
    31  				OfText: &anthropic.TextBlockParam{
    32  					Text: "x",
    33  				},
    34  			}},
    35  			Role: anthropic.MessageParamRoleUser,
    36  		}},
    37  		Model: anthropic.ModelClaudeSonnet5,
    38  	})
    39  	if err != nil {
    40  		t.Fatalf("err should be nil: %s", err.Error())
    41  	}
    42  	t.Logf("%+v\n", message.Content)
    43  }