/
PredictCodeGenerationFunctionSample.cs
74 lines (65 loc) · 2.4 KB
/
PredictCodeGenerationFunctionSample.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
* 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_function]
using Google.Cloud.AIPlatform.V1;
using System;
using System.Collections.Generic;
using System.Linq;
using Value = Google.Protobuf.WellKnownTypes.Value;
public class PredictCodeGenerationFunctionSample
{
public string PredictFunction(
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 instances = new List<Value>
{
Value.ForStruct(new()
{
Fields =
{
["prefix"] = Value.ForString("Write a C# function that checks if a year is a leap year."),
}
})
};
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_function]