updating test-cel

Signed-off-by: Aaron Choo <achoo30@bloomberg.net>
envoyproxy · Dec 12, 2024 · e200702 · e200702
1 parent c625ebe
commit e200702
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/api/v1alpha1/api.go b/api/v1alpha1/api.go
@@ -152,11 +152,14 @@ const (
 // +kubebuilder:object:root=true
 
 // LLMProviderPolicy specifies the provider specific configuration.
-//
-// This is a provider specific-configuration, e.g.AWS Bedrock, Azure etc.
 type LLMProviderPolicy struct {
 	metav1.TypeMeta   `json:",inline"`
 	metav1.ObjectMeta `json:"metadata,omitempty"`
+	Spec              LLMProviderPolicySpec `json:"spec,omitempty"`
+}
+
+// LLMProviderPolicySpec specifies a provider (e.g.AWS Bedrock, Azure etc.) specific-configuration like auth
+type LLMProviderPolicySpec struct {
 	// BackendRefs lists the LLMBackends that this provider policy will apply
 	// The namespace is "local", i.e. the same namespace as the LLMRoute.
 	//

diff --git a/tests/cel-validation/main_test.go b/tests/cel-validation/main_test.go
@@ -153,7 +153,7 @@ func TestLLMProviderPolicy(t *testing.T) {
 			data, err := tests.ReadFile(path.Join("testdata/llmproviderpolicy", tc.name))
 			require.NoError(t, err)
 
-			llmRoute := &aigv1a1.LLMRoute{}
+			llmRoute := &aigv1a1.LLMProviderPolicy{}
 			err = yaml.UnmarshalStrict(data, llmRoute)
 			require.NoError(t, err)
 

diff --git a/tests/cel-validation/testdata/llmproviderpolicy/basic.yaml b/tests/cel-validation/testdata/llmproviderpolicy/basic.yaml
@@ -3,4 +3,5 @@ kind: LLMProviderPolicy
 metadata:
   name: dog-provider-policy
   namespace: default
-type: AWSBedrock
+spec:
+  type: AWSBedrock
diff --git a/tests/cel-validation/testdata/llmproviderpolicy/unknown_provider.yaml b/tests/cel-validation/testdata/llmproviderpolicy/unknown_provider.yaml
@@ -3,4 +3,5 @@ kind: LLMProviderPolicy
 metadata:
   name: dog-provider-policy
   namespace: default
-type: UnknownType
+spec:
+  type: UnknownType