Update models to use new types

vellum-ai · Aug 18, 2024 · 684f738 · 684f738
1 parent d5a3e41
commit 684f738
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/internal/provider/ml_model/model.go b/internal/provider/ml_model/model.go
@@ -204,20 +204,20 @@ func NewOpenApiPropertyRequest(ctx context.Context, openApiProperty *TfOpenApiPr
 	openApiPropertyType := openApiProperty.Type.ValueString()
 
 	if openApiPropertyType == "number" {
-		f, _ := NewOpenApiNumberPropertyRequest(ctx, openApiProperty.Number)
-		return &vellum.OpenApiPropertyRequest{Type: "number", Number: f}, nil
+		v, _ := NewOpenApiNumberPropertyRequest(ctx, openApiProperty.Number)
+		return &vellum.OpenApiPropertyRequest{OpenApiNumberPropertyRequest: v}, nil
 	} else if openApiPropertyType == "integer" {
-		f, _ := NewOpenApiIntegerPropertyRequest(ctx, openApiProperty.Integer)
-		return &vellum.OpenApiPropertyRequest{Type: "integer", Integer: f}, nil
+		v, _ := NewOpenApiIntegerPropertyRequest(ctx, openApiProperty.Integer)
+		return &vellum.OpenApiPropertyRequest{OpenApiIntegerPropertyRequest: v}, nil
 	} else if openApiPropertyType == "string" {
-		f, _ := NewOpenApiStringPropertyRequest(ctx, openApiProperty.String)
-		return &vellum.OpenApiPropertyRequest{Type: "string", String: f}, nil
+		v, _ := NewOpenApiStringPropertyRequest(ctx, openApiProperty.String)
+		return &vellum.OpenApiPropertyRequest{OpenApiStringPropertyRequest: v}, nil
 	} else if openApiPropertyType == "array" {
-		f, _ := NewOpenApiArrayPropertyRequest(ctx, openApiProperty.Array)
-		return &vellum.OpenApiPropertyRequest{Type: "array", Array: f}, nil
+		v, _ := NewOpenApiArrayPropertyRequest(ctx, openApiProperty.Array)
+		return &vellum.OpenApiPropertyRequest{OpenApiArrayPropertyRequest: v}, nil
 	} else if openApiPropertyType == "object" {
-		f, _ := NewOpenApiObjectPropertyRequest(ctx, openApiProperty.Object)
-		return &vellum.OpenApiPropertyRequest{Type: "object", Object: f}, nil
+		v, _ := NewOpenApiObjectPropertyRequest(ctx, openApiProperty.Object)
+		return &vellum.OpenApiPropertyRequest{OpenApiObjectPropertyRequest: v}, nil
 	}
 
 	return nil, diag.Diagnostics{}
@@ -388,8 +388,8 @@ func NewTfOpenApiPropertyModel(ctx context.Context, property *vellum.OpenApiProp
 	}
 
 	// For now, only support arrays consisting of strings
-	if property.String != nil {
-		stringItems, _ := NewTfOpenApiStringPropertyModel(ctx, property.String)
+	if property.OpenApiStringProperty != nil {
+		stringItems, _ := NewTfOpenApiStringPropertyModel(ctx, property.OpenApiStringProperty)
 		return &TfOpenApiProperty{
 			Type:   types.StringValue("STRING"),
 			String: stringItems,
@@ -466,19 +466,19 @@ func NewTfMlModelTokenizerConfig(ctx context.Context, property *vellum.MlModelTo
 		return nil, diag.Diagnostics{}
 	}
 
-	if property.HuggingFace != nil {
+	if property.HuggingFaceTokenizerConfig != nil {
 		return &TfMlModelTokenizerConfig{
 			Type: types.StringValue("HUGGING_FACE"),
 			HuggingFace: &TfHuggingFaceTokenizerConfig{
-				Name: types.StringValue(property.HuggingFace.Name),
-				Path: types.StringPointerValue(property.HuggingFace.Path),
+				Name: types.StringValue(property.HuggingFaceTokenizerConfig.Name),
+				Path: types.StringPointerValue(property.HuggingFaceTokenizerConfig.Path),
 			},
 		}, nil
-	} else if property.Tiktoken != nil {
+	} else if property.TikTokenTokenizerConfig != nil {
 		return &TfMlModelTokenizerConfig{
 			Type: types.StringValue("TIKTOKEN"),
 			Tiktoken: &TiktokenTokenizerConfig{
-				Name: types.StringValue(property.HuggingFace.Name),
+				Name: types.StringValue(property.TikTokenTokenizerConfig.Name),
 			},
 		}, nil
 	}