github.com/anthropics/anthropic-sdk-go@v1.70.1/betamessagebatch_test.go

// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

package anthropic_test

import (
	"context"
	"errors"
	"os"
	"testing"

	"github.com/anthropics/anthropic-sdk-go"
	"github.com/anthropics/anthropic-sdk-go/internal/testutil"
	"github.com/anthropics/anthropic-sdk-go/option"
	"github.com/anthropics/anthropic-sdk-go/shared/constant"
)

func TestBetaMessageBatchNewWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := anthropic.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("my-anthropic-api-key"),
	)
	_, err := client.Beta.Messages.Batches.New(context.TODO(), anthropic.BetaMessageBatchNewParams{
		Requests: []anthropic.BetaMessageBatchNewParamsRequest{{
			CustomID: "my-custom-id-1",
			Params: anthropic.BetaMessageBatchNewParamsRequestParams{
				MaxTokens: 1024,
				Messages: []anthropic.BetaMessageParam{{
					Content: []anthropic.BetaContentBlockParamUnion{{
						OfText: &anthropic.BetaTextBlockParam{
							Text: "x",
							CacheControl: anthropic.BetaCacheControlEphemeralParam{
								TTL: anthropic.BetaCacheControlEphemeralTTLTTL5m,
							},
							Citations: []anthropic.BetaTextCitationParamUnion{{
								OfCharLocation: &anthropic.BetaCitationCharLocationParam{
									CitedText:      "The grass is green. The sky is blue.",
									DocumentIndex:  0,
									DocumentTitle:  anthropic.String("x"),
									EndCharIndex:   0,
									StartCharIndex: 0,
								},
							}},
						},
					}},
					Role:    anthropic.BetaMessageParamRoleUser,
					ClearAt: anthropic.BetaMessageParamClearAtNextUserMessage,
					OutputConfig: anthropic.BetaSystemMessageOutputConfigParam{
						Effort: anthropic.BetaSystemMessageOutputConfigEffortLow,
					},
				}},
				Model: anthropic.ModelClaudeOpus5,
				CacheControl: anthropic.BetaCacheControlEphemeralParam{
					TTL: anthropic.BetaCacheControlEphemeralTTLTTL5m,
				},
				Container: anthropic.BetaMessageBatchNewParamsRequestParamsContainerUnion{
					OfContainers: &anthropic.BetaContainerParams{
						ID: anthropic.String("id"),
						Skills: []anthropic.BetaSkillParams{{
							SkillID: "pdf",
							Type:    anthropic.BetaSkillParamsTypeAnthropic,
							Version: anthropic.String("latest"),
						}},
					},
				},
				ContextManagement: anthropic.BetaContextManagementConfigParam{
					Edits: []anthropic.BetaContextManagementConfigEditUnionParam{{
						OfClearToolUses20250919: &anthropic.BetaClearToolUses20250919EditParam{
							ClearAtLeast: anthropic.BetaInputTokensClearAtLeastParam{
								Value: 0,
							},
							ClearToolInputs: anthropic.BetaClearToolUses20250919EditClearToolInputsUnionParam{
								OfBool: anthropic.Bool(true),
							},
							ExcludeTools: []string{"string"},
							Keep: anthropic.BetaToolUsesKeepParam{
								Value: 0,
							},
							Trigger: anthropic.BetaClearToolUses20250919EditTriggerUnionParam{
								OfInputTokens: &anthropic.BetaInputTokensTriggerParam{
									Value: 1,
								},
							},
						},
					}},
				},
				Diagnostics: anthropic.BetaDiagnosticsParam{
					PreviousMessageID: anthropic.String("previous_message_id"),
				},
				FallbackCreditToken: anthropic.BetaMessageBatchNewParamsRequestParamsFallbackCreditTokenUnion{
					OfString: anthropic.String("x"),
				},
				Fallbacks: anthropic.BetaFallbacksParamUnion{
					OfDefault: constant.ValueOf[constant.Default](),
				},
				InferenceGeo: anthropic.String("inference_geo"),
				MCPServers: []anthropic.BetaRequestMCPServerURLDefinitionParam{{
					Name:               "name",
					URL:                "url",
					AuthorizationToken: anthropic.String("authorization_token"),
					ToolConfiguration: anthropic.BetaRequestMCPServerToolConfigurationParam{
						AllowedTools: []string{"string"},
						Enabled:      anthropic.Bool(true),
					},
				}},
				Metadata: anthropic.BetaMetadataParam{
					UserID: anthropic.String("13803d75-b4b5-4c3e-b2a2-6f21399b021b"),
				},
				OutputConfig: anthropic.BetaOutputConfigParam{
					Effort: anthropic.BetaOutputConfigEffortLow,
					Format: anthropic.BetaJSONOutputFormatParam{
						Schema: map[string]any{
							"foo": "bar",
						},
					},
					TaskBudget: anthropic.BetaTokenTaskBudgetParam{
						Total:     1024,
						Remaining: anthropic.Int(0),
					},
				},
				OutputFormat: anthropic.BetaJSONOutputFormatParam{
					Schema: map[string]any{
						"foo": "bar",
					},
				},
				ServiceTier:   "auto",
				Speed:         "standard",
				StopSequences: []string{"string"},
				Stream:        anthropic.Bool(false),
				System: []anthropic.BetaTextBlockParam{{
					Text: "Today's date is 2024-06-01.",
					CacheControl: anthropic.BetaCacheControlEphemeralParam{
						TTL: anthropic.BetaCacheControlEphemeralTTLTTL5m,
					},
					Citations: []anthropic.BetaTextCitationParamUnion{{
						OfCharLocation: &anthropic.BetaCitationCharLocationParam{
							CitedText:      "The grass is green. The sky is blue.",
							DocumentIndex:  0,
							DocumentTitle:  anthropic.String("x"),
							EndCharIndex:   0,
							StartCharIndex: 0,
						},
					}},
				}},
				Temperature: anthropic.Float(1),
				Thinking: anthropic.BetaThinkingConfigParamUnion{
					OfAdaptive: &anthropic.BetaThinkingConfigAdaptiveParam{
						BlockBinding: anthropic.BetaThinkingBlockBindingParam{
							PrefixMismatchBehavior: anthropic.BetaThinkingPrefixMismatchBehaviorError,
						},
						Display: anthropic.BetaThinkingConfigAdaptiveDisplaySummarized,
					},
				},
				ToolChoice: anthropic.BetaToolChoiceUnionParam{
					OfAuto: &anthropic.BetaToolChoiceAutoParam{
						DisableParallelToolUse: anthropic.Bool(true),
					},
				},
				Tools: []anthropic.BetaToolUnionParam{{
					OfTool: &anthropic.BetaToolParam{
						InputSchema: anthropic.BetaToolInputSchemaParam{
							Properties: map[string]any{
								"location": "bar",
								"unit":     "bar",
							},
							Required: []string{"location"},
						},
						Name:           "name",
						AllowedCallers: []string{"direct"},
						CacheControl: anthropic.BetaCacheControlEphemeralParam{
							TTL: anthropic.BetaCacheControlEphemeralTTLTTL5m,
						},
						DeferLoading:        anthropic.Bool(true),
						Description:         anthropic.String("Get the current weather in a given location"),
						EagerInputStreaming: anthropic.Bool(true),
						InputExamples: []map[string]any{{
							"foo": "bar",
						}},
						Strict: anthropic.Bool(true),
						Type:   anthropic.BetaToolTypeCustom,
					},
				}},
				TopK: anthropic.Int(5),
				TopP: anthropic.Float(0.7),
			},
		}},
		Betas:         []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
		UserProfileID: anthropic.String("anthropic-user-profile-id"),
		WorkspaceID:   anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
	})
	if err != nil {
		var apierr *anthropic.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestBetaMessageBatchGetWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := anthropic.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("my-anthropic-api-key"),
	)
	_, err := client.Beta.Messages.Batches.Get(
		context.TODO(),
		"message_batch_id",
		anthropic.BetaMessageBatchGetParams{
			Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
			WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
		},
	)
	if err != nil {
		var apierr *anthropic.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestBetaMessageBatchListWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := anthropic.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("my-anthropic-api-key"),
	)
	_, err := client.Beta.Messages.Batches.List(context.TODO(), anthropic.BetaMessageBatchListParams{
		AfterID:     anthropic.String("after_id"),
		BeforeID:    anthropic.String("before_id"),
		Limit:       anthropic.Int(1),
		Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
		WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
	})
	if err != nil {
		var apierr *anthropic.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestBetaMessageBatchDeleteWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := anthropic.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("my-anthropic-api-key"),
	)
	_, err := client.Beta.Messages.Batches.Delete(
		context.TODO(),
		"message_batch_id",
		anthropic.BetaMessageBatchDeleteParams{
			Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
			WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
		},
	)
	if err != nil {
		var apierr *anthropic.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestBetaMessageBatchCancelWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := anthropic.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("my-anthropic-api-key"),
	)
	_, err := client.Beta.Messages.Batches.Cancel(
		context.TODO(),
		"message_batch_id",
		anthropic.BetaMessageBatchCancelParams{
			Betas:       []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
			WorkspaceID: anthropic.String("wrkspc_011CZkZaBF1tNoB5wlCeusgy"),
		},
	)
	if err != nil {
		var apierr *anthropic.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}