github.com/anthropics/anthropic-sdk-go@v1.70.1/betamodel_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 TestBetaModelGetWithOptionalParams(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.Beta.Models.Get(
    29  		context.TODO(),
    30  		"model_id",
    31  		anthropic.BetaModelGetParams{
    32  			Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
    33  			WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
    34  		},
    35  	)
    36  	if err != nil {
    37  		var apierr *anthropic.Error
    38  		if errors.As(err, &apierr) {
    39  			t.Log(string(apierr.DumpRequest(true)))
    40  		}
    41  		t.Fatalf("err should be nil: %s", err.Error())
    42  	}
    43  }
    44  
    45  func TestBetaModelListWithOptionalParams(t *testing.T) {
    46  	baseURL := "http://localhost:4010"
    47  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    48  		baseURL = envURL
    49  	}
    50  	if !testutil.CheckTestServer(t, baseURL) {
    51  		return
    52  	}
    53  	client := anthropic.NewClient(
    54  		option.WithBaseURL(baseURL),
    55  		option.WithAPIKey("my-anthropic-api-key"),
    56  	)
    57  	_, err := client.Beta.Models.List(context.TODO(), anthropic.BetaModelListParams{
    58  		AfterID:     anthropic.String("after_id"),
    59  		BeforeID:    anthropic.String("before_id"),
    60  		Limit:       anthropic.Int(1),
    61  		Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
    62  		WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
    63  	})
    64  	if err != nil {
    65  		var apierr *anthropic.Error
    66  		if errors.As(err, &apierr) {
    67  			t.Log(string(apierr.DumpRequest(true)))
    68  		}
    69  		t.Fatalf("err should be nil: %s", err.Error())
    70  	}
    71  }