update unittest case

Signed-off-by: dongjiang <dongjiang1989@126.com>

Author: dongjiang1989

go/core/internal/controller/translator/agent/adk_api_translator_test.go

@@ -1413,3 +1413,102 @@ func Test_AdkApiTranslator_SandboxAgent_BYOEmitsSandbox(t *testing.T) {
 	require.False(t, sawDeploy)
 	require.False(t, sawService, "sandbox runtime must not include Service; agent-sandbox owns it")
 }
+
+func Test_AdkApiTranslator_AnthropicBaseURLEnvVar(t *testing.T) {
+	scheme := schemev1.Scheme
+	require.NoError(t, v1alpha2.AddToScheme(scheme))
+
+	namespace := "test-ns"
+
+	tests := []struct {
+		name            string
+		baseURL         string
+		wantEnvVar      bool
+		wantEnvVarValue string
+	}{
+		{
+			name:       "empty baseURL does not inject env var",
+			baseURL:    "",
+			wantEnvVar: false,
+		},
+		{
+			name:            "custom baseURL injects ANTHROPIC_BASE_URL env var",
+			baseURL:         "https://proxy.internal/anthropic/v1",
+			wantEnvVar:      true,
+			wantEnvVarValue: "https://proxy.internal/anthropic/v1",
+		},
+		{
+			name:            "default Anthropic URL still injects env var",
+			baseURL:         "https://api.anthropic.com",
+			wantEnvVar:      true,
+			wantEnvVarValue: "https://api.anthropic.com",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ns := &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: namespace}}
+
+			modelConfig := &v1alpha2.ModelConfig{
+				ObjectMeta: metav1.ObjectMeta{Name: "anthropic-model", Namespace: namespace},
+				Spec: v1alpha2.ModelConfigSpec{
+					Provider:     v1alpha2.ModelProviderAnthropic,
+					Model:        "claude-3-sonnet-20240229",
+					APIKeySecret: "anthropic-secret",
+					Anthropic: &v1alpha2.AnthropicConfig{
+						BaseURL: tt.baseURL,
+					},
+				},
+			}
+
+			agent := &v1alpha2.Agent{
+				ObjectMeta: metav1.ObjectMeta{Name: "test-agent", Namespace: namespace},
+				Spec: v1alpha2.AgentSpec{
+					Type:        v1alpha2.AgentType_Declarative,
+					Description: "Test Agent",
+					Declarative: &v1alpha2.DeclarativeAgentSpec{
+						SystemMessage: "You are a test agent",
+						ModelConfig:   "anthropic-model",
+					},
+				},
+			}
+
+			kubeClient := fake.NewClientBuilder().
+				WithScheme(scheme).
+				WithObjects(ns, modelConfig, agent).
+				Build()
+
+			defaultModel := types.NamespacedName{Namespace: namespace, Name: "anthropic-model"}
+			trans := translator.NewAdkApiTranslator(kubeClient, defaultModel, nil, "", nil)
+
+			outputs, err := translator.TranslateAgent(context.Background(), trans, agent)
+			require.NoError(t, err)
+			require.NotNil(t, outputs)
+
+			// Find the deployment and check env vars
+			var deployment *appsv1.Deployment
+			for _, obj := range outputs.Manifest {
+				if d, ok := obj.(*appsv1.Deployment); ok {
+					deployment = d
+					break
+				}
+			}
+			require.NotNil(t, deployment)
+
+			var found bool
+			for _, envVar := range deployment.Spec.Template.Spec.Containers[0].Env {
+				if envVar.Name == "ANTHROPIC_BASE_URL" {
+					found = true
+					assert.Equal(t, tt.wantEnvVarValue, envVar.Value)
+					break
+				}
+			}
+
+			if tt.wantEnvVar {
+				assert.True(t, found, "expected ANTHROPIC_BASE_URL env var to be injected")
+			} else {
+				assert.False(t, found, "did not expect ANTHROPIC_BASE_URL env var")
+			}
+		})
+	}
+}

python/packages/kagent-adk/tests/unittests/models/test_anthropic.py

@@ -35,6 +35,23 @@ def test_client_uses_base_url(self):
             _ = llm._anthropic_client
             assert mock_anthropic.call_args.kwargs["base_url"] == "https://proxy.internal/anthropic"
 
+    def test_client_falls_back_to_env_base_url(self):
+        llm = KAgentAnthropicLlm(model="claude-3-sonnet-20240229")
+        with mock.patch.dict("os.environ", {"ANTHROPIC_BASE_URL": "https://env-proxy.internal/anthropic"}):
+            with mock.patch("kagent.adk.models._anthropic.AsyncAnthropic") as mock_anthropic:
+                mock_anthropic.return_value = mock.MagicMock(spec=AsyncAnthropic)
+                _ = llm._anthropic_client
+                assert mock_anthropic.call_args.kwargs["base_url"] == "https://env-proxy.internal/anthropic"
+
+    def test_explicit_base_url_takes_precedence_over_env(self):
+        llm = KAgentAnthropicLlm(model="claude-3-sonnet-20240229", base_url="https://explicit.internal/anthropic")
+        with mock.patch.dict("os.environ", {"ANTHROPIC_BASE_URL": "https://env.internal/anthropic"}):
+            with mock.patch("kagent.adk.models._anthropic.AsyncAnthropic") as mock_anthropic:
+                mock_anthropic.return_value = mock.MagicMock(spec=AsyncAnthropic)
+                _ = llm._anthropic_client
+                # Explicit base_url wins over env
+                assert mock_anthropic.call_args.kwargs["base_url"] == "https://explicit.internal/anthropic"
+
     def test_client_uses_extra_headers(self):
         llm = KAgentAnthropicLlm(model="claude-3-sonnet-20240229", extra_headers={"X-Org": "test-org"})
         with mock.patch("kagent.adk.models._anthropic.AsyncAnthropic") as mock_anthropic: