chore: use Pydantic to generate OpenAPI schema

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 96
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-602ce64aa309cc15fa09388d99c9f298795686fc37605237cbc03c39d29aabf6.yml
-openapi_spec_hash: fc6995247b2555e8660bc9291eb10415
-config_hash: e8a35d9d37cb4774b4b0fe1b167dc156
+configured_endpoints: 103
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-2b99a80543f8bc8fa164167693c214651ac8e710f4726fb5869183b4d6c71a03.yml
+openapi_spec_hash: a5632057f5e4d956a71c20a79c0d879c
+config_hash: 0017f6c419cbbf7b949f9b2842917a79

README.md

@@ -53,13 +53,11 @@ import (
 
 func main() {
 	client := llamastackclient.NewClient()
-	response, err := client.Models.Register(context.TODO(), llamastackclient.ModelRegisterParams{
-		ModelID: "model_id",
-	})
+	models, err := client.Models.List(context.TODO())
 	if err != nil {
 		panic(err.Error())
 	}
-	fmt.Printf("%+v\n", response.Identifier)
+	fmt.Printf("%+v\n", models)
 }
 
 ```
@@ -302,6 +300,7 @@ _, err := client.Chat.Completions.New(context.TODO(), llamastackclient.ChatCompl
 			Content: llamastackclient.ChatCompletionNewParamsMessageUserContentUnion{
 				OfString: llamastackclient.String("string"),
 			},
+			Role: "user",
 		},
 	}},
 	Model: "model",
@@ -338,6 +337,7 @@ client.Chat.Completions.New(
 				Content: llamastackclient.ChatCompletionNewParamsMessageUserContentUnion{
 					OfString: llamastackclient.String("string"),
 				},
+				Role: "user",
 			},
 		}},
 		Model: "model",
@@ -404,6 +404,7 @@ client.Chat.Completions.New(
 				Content: llamastackclient.ChatCompletionNewParamsMessageUserContentUnion{
 					OfString: llamastackclient.String("string"),
 				},
+				Role: "user",
 			},
 		}},
 		Model: "model",
@@ -428,6 +429,7 @@ completion, err := client.Chat.Completions.New(
 				Content: llamastackclient.ChatCompletionNewParamsMessageUserContentUnion{
 					OfString: llamastackclient.String("string"),
 				},
+				Role: "user",
 			},
 		}},
 		Model: "model",

alphabenchmark.go

@@ -10,7 +10,6 @@ package llamastackclient
 
 import (
 	"context"
-	"encoding/json"
 	"errors"
 	"fmt"
 	"net/http"
@@ -21,7 +20,6 @@ import (
 	"github.com/llamastack/llama-stack-client-go/option"
 	"github.com/llamastack/llama-stack-client-go/packages/param"
 	"github.com/llamastack/llama-stack-client-go/packages/respjson"
-	"github.com/llamastack/llama-stack-client-go/shared/constant"
 )
 
 // AlphaBenchmarkService contains methods and other services that help with
@@ -79,28 +77,44 @@ func (r *AlphaBenchmarkService) Register(ctx context.Context, body AlphaBenchmar
 	return
 }
 
+// Unregister a benchmark.
+//
+// Deprecated: deprecated
+func (r *AlphaBenchmarkService) Unregister(ctx context.Context, benchmarkID string, opts ...option.RequestOption) (err error) {
+	opts = slices.Concat(r.Options, opts)
+	opts = append([]option.RequestOption{option.WithHeader("Accept", "")}, opts...)
+	if benchmarkID == "" {
+		err = errors.New("missing required benchmark_id parameter")
+		return
+	}
+	path := fmt.Sprintf("v1alpha/eval/benchmarks/%s", benchmarkID)
+	err = requestconfig.ExecuteNewRequest(ctx, http.MethodDelete, path, nil, nil, opts...)
+	return
+}
+
 // A benchmark resource for evaluating model performance.
 type Benchmark struct {
-	// Identifier of the dataset to use for the benchmark evaluation
-	DatasetID  string `json:"dataset_id,required"`
+	DatasetID string `json:"dataset_id,required"`
+	// Unique identifier for this resource in llama stack
 	Identifier string `json:"identifier,required"`
-	// Metadata for this evaluation task
-	Metadata   map[string]BenchmarkMetadataUnion `json:"metadata,required"`
-	ProviderID string                            `json:"provider_id,required"`
-	// List of scoring function identifiers to apply during evaluation
+	// ID of the provider that owns this resource
+	ProviderID       string   `json:"provider_id,required"`
 	ScoringFunctions []string `json:"scoring_functions,required"`
-	// The resource type, always benchmark
-	Type               constant.Benchmark `json:"type,required"`
-	ProviderResourceID string             `json:"provider_resource_id"`
+	// Metadata for this evaluation task
+	Metadata map[string]any `json:"metadata"`
+	// Unique identifier for this resource in the provider
+	ProviderResourceID string `json:"provider_resource_id,nullable"`
+	// Any of "benchmark".
+	Type BenchmarkType `json:"type"`
 	// JSON contains metadata for fields, check presence with [respjson.Field.Valid].
 	JSON struct {
 		DatasetID          respjson.Field
 		Identifier         respjson.Field
-		Metadata           respjson.Field
 		ProviderID         respjson.Field
 		ScoringFunctions   respjson.Field
-		Type               respjson.Field
+		Metadata           respjson.Field
 		ProviderResourceID respjson.Field
+		Type               respjson.Field
 		ExtraFields        map[string]respjson.Field
 		raw                string
 	} `json:"-"`
@@ -112,57 +126,11 @@ func (r *Benchmark) UnmarshalJSON(data []byte) error {
 	return apijson.UnmarshalRoot(data, r)
 }
 
-// BenchmarkMetadataUnion contains all possible properties and values from [bool],
-// [float64], [string], [[]any].
-//
-// Use the methods beginning with 'As' to cast the union to one of its variants.
-//
-// If the underlying value is not a json object, one of the following properties
-// will be valid: OfBool OfFloat OfString OfAnyArray]
-type BenchmarkMetadataUnion struct {
-	// This field will be present if the value is a [bool] instead of an object.
-	OfBool bool `json:",inline"`
-	// This field will be present if the value is a [float64] instead of an object.
-	OfFloat float64 `json:",inline"`
-	// This field will be present if the value is a [string] instead of an object.
-	OfString string `json:",inline"`
-	// This field will be present if the value is a [[]any] instead of an object.
-	OfAnyArray []any `json:",inline"`
-	JSON       struct {
-		OfBool     respjson.Field
-		OfFloat    respjson.Field
-		OfString   respjson.Field
-		OfAnyArray respjson.Field
-		raw        string
-	} `json:"-"`
-}
-
-func (u BenchmarkMetadataUnion) AsBool() (v bool) {
-	apijson.UnmarshalRoot(json.RawMessage(u.JSON.raw), &v)
-	return
-}
-
-func (u BenchmarkMetadataUnion) AsFloat() (v float64) {
-	apijson.UnmarshalRoot(json.RawMessage(u.JSON.raw), &v)
-	return
-}
-
-func (u BenchmarkMetadataUnion) AsString() (v string) {
-	apijson.UnmarshalRoot(json.RawMessage(u.JSON.raw), &v)
-	return
-}
+type BenchmarkType string
 
-func (u BenchmarkMetadataUnion) AsAnyArray() (v []any) {
-	apijson.UnmarshalRoot(json.RawMessage(u.JSON.raw), &v)
-	return
-}
-
-// Returns the unmodified JSON received from the API
-func (u BenchmarkMetadataUnion) RawJSON() string { return u.JSON.raw }
-
-func (r *BenchmarkMetadataUnion) UnmarshalJSON(data []byte) error {
-	return apijson.UnmarshalRoot(data, r)
-}
+const (
+	BenchmarkTypeBenchmark BenchmarkType = "benchmark"
+)
 
 type ListBenchmarksResponse struct {
 	Data []Benchmark `json:"data,required"`
@@ -181,18 +149,12 @@ func (r *ListBenchmarksResponse) UnmarshalJSON(data []byte) error {
 }
 
 type AlphaBenchmarkRegisterParams struct {
-	// The ID of the benchmark to register.
-	BenchmarkID string `json:"benchmark_id,required"`
-	// The ID of the dataset to use for the benchmark.
-	DatasetID string `json:"dataset_id,required"`
-	// The scoring functions to use for the benchmark.
-	ScoringFunctions []string `json:"scoring_functions,omitzero,required"`
-	// The ID of the provider benchmark to use for the benchmark.
+	BenchmarkID         string            `json:"benchmark_id,required"`
+	DatasetID           string            `json:"dataset_id,required"`
+	ScoringFunctions    []string          `json:"scoring_functions,omitzero,required"`
 	ProviderBenchmarkID param.Opt[string] `json:"provider_benchmark_id,omitzero"`
-	// The ID of the provider to use for the benchmark.
-	ProviderID param.Opt[string] `json:"provider_id,omitzero"`
-	// The metadata to use for the benchmark.
-	Metadata map[string]AlphaBenchmarkRegisterParamsMetadataUnion `json:"metadata,omitzero"`
+	ProviderID          param.Opt[string] `json:"provider_id,omitzero"`
+	Metadata            map[string]any    `json:"metadata,omitzero"`
 	paramObj
 }
 
@@ -203,34 +165,3 @@ func (r AlphaBenchmarkRegisterParams) MarshalJSON() (data []byte, err error) {
 func (r *AlphaBenchmarkRegisterParams) UnmarshalJSON(data []byte) error {
 	return apijson.UnmarshalRoot(data, r)
 }
-
-// Only one field can be non-zero.
-//
-// Use [param.IsOmitted] to confirm if a field is set.
-type AlphaBenchmarkRegisterParamsMetadataUnion struct {
-	OfBool     param.Opt[bool]    `json:",omitzero,inline"`
-	OfFloat    param.Opt[float64] `json:",omitzero,inline"`
-	OfString   param.Opt[string]  `json:",omitzero,inline"`
-	OfAnyArray []any              `json:",omitzero,inline"`
-	paramUnion
-}
-
-func (u AlphaBenchmarkRegisterParamsMetadataUnion) MarshalJSON() ([]byte, error) {
-	return param.MarshalUnion(u, u.OfBool, u.OfFloat, u.OfString, u.OfAnyArray)
-}
-func (u *AlphaBenchmarkRegisterParamsMetadataUnion) UnmarshalJSON(data []byte) error {
-	return apijson.UnmarshalRoot(data, u)
-}
-
-func (u *AlphaBenchmarkRegisterParamsMetadataUnion) asAny() any {
-	if !param.IsOmitted(u.OfBool) {
-		return &u.OfBool.Value
-	} else if !param.IsOmitted(u.OfFloat) {
-		return &u.OfFloat.Value
-	} else if !param.IsOmitted(u.OfString) {
-		return &u.OfString.Value
-	} else if !param.IsOmitted(u.OfAnyArray) {
-		return &u.OfAnyArray
-	}
-	return nil
-}

alphabenchmark_test.go

@@ -76,10 +76,8 @@ func TestAlphaBenchmarkRegisterWithOptionalParams(t *testing.T) {
 		BenchmarkID:      "benchmark_id",
 		DatasetID:        "dataset_id",
 		ScoringFunctions: []string{"string"},
-		Metadata: map[string]llamastackclient.AlphaBenchmarkRegisterParamsMetadataUnion{
-			"foo": {
-				OfBool: llamastackclient.Bool(true),
-			},
+		Metadata: map[string]any{
+			"foo": "bar",
 		},
 		ProviderBenchmarkID: llamastackclient.String("provider_benchmark_id"),
 		ProviderID:          llamastackclient.String("provider_id"),
@@ -92,3 +90,24 @@ func TestAlphaBenchmarkRegisterWithOptionalParams(t *testing.T) {
 		t.Fatalf("err should be nil: %s", err.Error())
 	}
 }
+
+func TestAlphaBenchmarkUnregister(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 := llamastackclient.NewClient(
+		option.WithBaseURL(baseURL),
+	)
+	err := client.Alpha.Benchmarks.Unregister(context.TODO(), "benchmark_id")
+	if err != nil {
+		var apierr *llamastackclient.Error
+		if errors.As(err, &apierr) {
+			t.Log(string(apierr.DumpRequest(true)))
+		}
+		t.Fatalf("err should be nil: %s", err.Error())
+	}
+}