docs(aiplatform): Add PredictCodeGenerationUnitTestSample #2390 (#2401)

aiplatform/api/AIPlatform.Samples.Tests/PredictCodeGenerationUnitTestTest.cs

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Xunit;
+
+[Collection(nameof(AIPlatformFixture))]
+public class PredictCodeGenerationUnitTestTest
+{
+ private readonly AIPlatformFixture _fixture;
+ private readonly PredictCodeGenerationUnitTestSample _sample;
+
+ public PredictCodeGenerationUnitTestTest(AIPlatformFixture fixture)
+ {
+ _fixture = fixture;
+ _sample = new PredictCodeGenerationUnitTestSample();
+ }
+
+ [Fact]
+ public void TestPredictUnitTest()
+ {
+ var response = _sample.PredictUnitTest(_fixture.ProjectId);
+ Assert.NotNull(response);
+ }
+}

aiplatform/api/AIPlatform.Samples/PredictCodeGenerationUnitTestSample.cs

@@ -0,0 +1,102 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// [START aiplatform_sdk_code_generation_unittest]
+
+using Google.Cloud.AIPlatform.V1;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Value = Google.Protobuf.WellKnownTypes.Value;
+
+public class PredictCodeGenerationUnitTestSample
+{
+ public string PredictUnitTest(
+ string projectId = "your-project-id",
+ string locationId = "us-central1",
+ string publisher = "google",
+ string model = "code-bison@001")
+ {
+ // Initialize client that will be used to send requests.
+ // This client only needs to be created once,
+ // and can be reused for multiple requests.
+ var client = new PredictionServiceClientBuilder
+ {
+ Endpoint = $"{locationId}-aiplatform.googleapis.com"
+ }.Build();
+
+ // Configure the parent resource.
+ var endpoint = EndpointName.FromProjectLocationPublisherModel(projectId, locationId, publisher, model);
+
+ var prefix = @"
+Write a unit test for this function:
+ public static bool IsLeapYear(int year)
+ {
+ if (year % 4 == 0)
+ {
+ if (year % 100 == 0)
+ {
+ if (year % 400 == 0)
+ {
+ return true;
+ }
+ else
+ {
+ return false;
+ }
+ }
+ else
+ {
+ return true;
+ }
+ }
+ else
+ {
+ return false;
+ }
+ }";
+
+ var instances = new List<Value>
+ {
+ Value.ForStruct(new()
+ {
+ Fields =
+ {
+ ["prefix"] = Value.ForString(prefix),
+ }
+ })
+ };
+
+ var parameters = Value.ForStruct(new()
+ {
+ Fields =
+ {
+ { "temperature", new Value { NumberValue = 0.5 } },
+ { "maxOutputTokens", new Value { NumberValue = 256 } }
+ }
+ });
+
+ // Make the request.
+ var response = client.Predict(endpoint, instances, parameters);
+
+ // Parse and return the content.
+ var content = response.Predictions.First().StructValue.Fields["content"].StringValue;
+ Console.WriteLine($"Content: {content}");
+ return content;
+ }
+}
+
+// [END aiplatform_sdk_code_generation_unittest]