github.com/anthropics/anthropic-sdk-go@v1.70.1/completion_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  	"errors"
     8  	"os"
     9  	"testing"
    10  
    11  	"github.com/anthropics/anthropic-sdk-go"
    12  	"github.com/anthropics/anthropic-sdk-go/internal/testutil"
    13  	"github.com/anthropics/anthropic-sdk-go/option"
    14  )
    15  
    16  func TestCompletionNewWithOptionalParams(t *testing.T) {
    17  	baseURL := "http://localhost:4010"
    18  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    19  		baseURL = envURL
    20  	}
    21  	if !testutil.CheckTestServer(t, baseURL) {
    22  		return
    23  	}
    24  	client := anthropic.NewClient(
    25  		option.WithBaseURL(baseURL),
    26  		option.WithAPIKey("my-anthropic-api-key"),
    27  	)
    28  	_, err := client.Completions.New(context.TODO(), anthropic.CompletionNewParams{
    29  		MaxTokensToSample: 256,
    30  		Model:             anthropic.ModelClaudeFable5_1,
    31  		Prompt:            "\n\nHuman: Hello, world!\n\nAssistant:",
    32  		Metadata: anthropic.MetadataParam{
    33  			UserID: anthropic.String("13803d75-b4b5-4c3e-b2a2-6f21399b021b"),
    34  		},
    35  		StopSequences: []string{"string"},
    36  		Temperature:   anthropic.Float(1),
    37  		TopK:          anthropic.Int(5),
    38  		TopP:          anthropic.Float(0.7),
    39  		Betas:         []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
    40  		WorkspaceID:   anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
    41  	})
    42  	if err != nil {
    43  		var apierr *anthropic.Error
    44  		if errors.As(err, &apierr) {
    45  			t.Log(string(apierr.DumpRequest(true)))
    46  		}
    47  		t.Fatalf("err should be nil: %s", err.Error())
    48  	}
    49  }