Skip to content

Commit ba77302

Browse files
a-mauricea-maurice
authored
Initial VertexAI API surface (#1179)
* Initial VertexAI API surface * Comment out IAsyncEnumerable for now * Add missing guids
1 parent 0b96ab2 commit ba77302

16 files changed

+846
-2
lines changed

‎unity_packer/guids.json‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1401,6 +1401,9 @@
14011401
"Firebase/Editor/FirebaseMessaging_version-12.5.0_manifest.txt": "fdf5d2bb0f9d4f0ba42e38b1f5c6b102",
14021402
"Firebase/Editor/FirebaseRemoteConfig_version-12.5.0_manifest.txt": "2d1a5b051f1a452e9eb8afaa9799e48f",
14031403
"Firebase/Editor/FirebaseStorage_version-12.5.0_manifest.txt": "94536946ac944e639dd690b9828a5314",
1404+
"Firebase/Editor/FirebaseVertexAI_version-12.5.0_manifest.txt": "aff68d4dbbcf4a65892205825ae01dfa",
1405+
"Firebase/Plugins/Firebase.VertexAI.dll": "7c23ae13d705434895cb14976926117b",
1406+
"Firebase/Plugins/Firebase.VertexAI.pdb": "01edb10da18a4975af2cc8f7a6de5e5d",
14041407
"Firebase/Plugins/x86_64/FirebaseCppApp-12_5_0.bundle": "60898de491fb4a168588dfa7b0c24942",
14051408
"Firebase/Plugins/x86_64/FirebaseCppApp-12_5_0.dll": "f25775ec30db485e86377f6898623aca",
14061409
"Firebase/Plugins/x86_64/FirebaseCppApp-12_5_0.so": "ca2abf1d7ed145289e902d62c88972d4",

‎vertexai/CMakeLists.txt‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,20 @@
1616

1717
# Vertex AI in Firebase CSharp files
1818
set(firebase_vertexai_src
19+
src/Candidate.cs
20+
src/Chat.cs
21+
src/Citation.cs
22+
src/CountTokensResponse.cs
23+
src/FunctionCalling.cs
24+
src/GenerateContentResponse.cs
25+
src/GenerationConfig.cs
26+
src/GenerativeModel.cs
27+
src/ModelContent.cs
28+
src/RequestOptions.cs
29+
src/Safety.cs
30+
src/Schema.cs
1931
src/VertexAI.cs
32+
src/VertexAIException.cs
2033
)
2134

2235
unity_pack_documentation_sources(vertexai

‎vertexai/src/Candidate.cs‎

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System;
18+
using System.Collections.Generic;
19+
20+
namespace Firebase.VertexAI {
21+
22+
public enum FinishReason {
23+
Unknown,
24+
Stop,
25+
MaxTokens,
26+
Safety,
27+
Recitation,
28+
Other,
29+
Blocklist,
30+
ProhibitedContent,
31+
SPII,
32+
MalformedFunctionCall,
33+
}
34+
35+
public readonly struct Candidate {
36+
public ModelContent Content { get; }
37+
public IEnumerable<SafetyRating> SafetyRatings { get; }
38+
public FinishReason? FinishReason { get; }
39+
public CitationMetadata? CitationMetadata { get; }
40+
}
41+
42+
}

‎vertexai/src/Chat.cs‎

Lines changed: 76 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System;
18+
using System.Collections.Generic;
19+
using System.Threading.Tasks;
20+
21+
namespace Firebase.VertexAI {
22+
23+
public class Chat {
24+
25+
public IEnumerable<ModelContent> History { get; }
26+
27+
// Note: The generation functions are the same as the ones in GenerativeModel
28+
29+
public Task<GenerateContentResponse> GenerateContentAsync(
30+
params ModelContent[] content) {
31+
throw new NotImplementedException();
32+
}
33+
public Task<GenerateContentResponse> GenerateContentAsync(
34+
string text) {
35+
throw new NotImplementedException();
36+
}
37+
public Task<GenerateContentResponse> GenerateContentAsync(
38+
IEnumerable<ModelContent> content) {
39+
throw new NotImplementedException();
40+
}
41+
42+
// The build logic isn't able to resolve IAsyncEnumerable for some reason, even
43+
// though it is usable in Unity 2021.3. Will need to investigate further.
44+
/*
45+
public IAsyncEnumerable<GenerateContentResponse> GenerateContentStreamAsync(
46+
params ModelContent[] content) {
47+
throw new NotImplementedException();
48+
}
49+
public IAsyncEnumerable<GenerateContentResponse> GenerateContentStreamAsync(
50+
string text) {
51+
throw new NotImplementedException();
52+
}
53+
public IAsyncEnumerable<GenerateContentResponse> GenerateContentStreamAsync(
54+
IEnumerable<ModelContent> content) {
55+
throw new NotImplementedException();
56+
}
57+
*/
58+
59+
public Task<CountTokensResponse> CountTokensAsync(
60+
params ModelContent[] content) {
61+
throw new NotImplementedException();
62+
}
63+
public Task<CountTokensResponse> CountTokensAsync(
64+
string text) {
65+
throw new NotImplementedException();
66+
}
67+
public Task<CountTokensResponse> CountTokensAsync(
68+
IEnumerable<ModelContent> content) {
69+
throw new NotImplementedException();
70+
}
71+
72+
// Note: No public constructor, get one through GenerativeModel.StartChat
73+
74+
}
75+
76+
}

‎vertexai/src/Citation.cs‎

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System.Collections.Generic;
18+
19+
namespace Firebase.VertexAI {
20+
21+
public readonly struct CitationMetadata {
22+
public IEnumerable<Citation> Citations { get; }
23+
24+
// Hidden constructor, users don't need to make this
25+
}
26+
27+
public readonly struct Citation {
28+
public int StartIndex { get; }
29+
public int EndIndex { get; }
30+
public System.Uri Uri { get; }
31+
public string Title { get; }
32+
public string License { get; }
33+
public System.DateTime? PublicationDate { get; }
34+
35+
// Hidden constructor, users don't need to make this
36+
}
37+
38+
}

‎vertexai/src/CountTokensResponse.cs‎

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
namespace Firebase.VertexAI {
18+
19+
public readonly struct CountTokensResponse {
20+
public int TotalTokens { get; }
21+
public int? TotalBillableCharacters { get; }
22+
23+
// Hidden constructor, users don't need to make this
24+
}
25+
26+
}

‎vertexai/src/FunctionCalling.cs‎

Lines changed: 70 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,70 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System;
18+
using System.Collections.Generic;
19+
20+
namespace Firebase.VertexAI {
21+
22+
public readonly struct FunctionDeclaration {
23+
// No public properties, on purpose since it is meant for user input only
24+
25+
public FunctionDeclaration(string name, string description,
26+
IDictionary<string, Schema> parameters,
27+
IEnumerable<string> optionalParameters = null) {
28+
throw new NotImplementedException();
29+
}
30+
}
31+
32+
public readonly struct Tool {
33+
// No public properties, on purpose since it is meant for user input only
34+
35+
public Tool(params FunctionDeclaration[] functionDeclarations) {
36+
throw new NotImplementedException();
37+
}
38+
public Tool(IEnumerable<FunctionDeclaration> functionDeclarations) {
39+
throw new NotImplementedException();
40+
}
41+
}
42+
43+
public readonly struct ToolConfig {
44+
// No public properties, on purpose since it is meant for user input only
45+
46+
public ToolConfig(FunctionCallingConfig? functionCallingConfig = null) {
47+
throw new NotImplementedException();
48+
}
49+
}
50+
51+
public readonly struct FunctionCallingConfig {
52+
// No public properties, on purpose since it is meant for user input only
53+
54+
public static FunctionCallingConfig Auto() {
55+
throw new NotImplementedException();
56+
}
57+
58+
public static FunctionCallingConfig Any(params string[] allowedFunctionNames) {
59+
throw new NotImplementedException();
60+
}
61+
public static FunctionCallingConfig Any(IEnumerable<string> allowedFunctionNames) {
62+
throw new NotImplementedException();
63+
}
64+
65+
public static FunctionCallingConfig None() {
66+
throw new NotImplementedException();
67+
}
68+
}
69+
70+
}

‎vertexai/src/GenerateContentResponse.cs‎

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System.Collections.Generic;
18+
19+
namespace Firebase.VertexAI {
20+
21+
public readonly struct GenerateContentResponse {
22+
public IEnumerable<Candidate> Candidates { get; }
23+
public PromptFeedback? PromptFeedback { get; }
24+
public UsageMetadata? UsageMetadata { get; }
25+
26+
// Helper properties
27+
// The response's content as text, if it exists
28+
public string Text { get; }
29+
30+
// Returns function calls found in any Parts of the first candidate of the response, if any.
31+
public IEnumerable<ModelContent.FunctionCallPart> FunctionCalls { get; }
32+
}
33+
34+
public enum BlockReason {
35+
Unknown,
36+
Safety,
37+
Other,
38+
Blocklist,
39+
ProhibitedContent,
40+
}
41+
42+
public readonly struct PromptFeedback {
43+
public BlockReason? BlockReason { get; }
44+
public string BlockReasonMessage { get; }
45+
public IEnumerable<SafetyRating> SafetyRatings { get; }
46+
47+
// Hidden constructor, users don't need to make this
48+
}
49+
50+
public readonly struct UsageMetadata {
51+
public int PromptTokenCount { get; }
52+
public int CandidatesTokenCount { get; }
53+
public int TotalTokenCount { get; }
54+
55+
// Hidden constructor, users don't need to make this
56+
}
57+
58+
}

‎vertexai/src/GenerationConfig.cs‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
using System;
18+
19+
namespace Firebase.VertexAI {
20+
21+
public readonly struct GenerationConfig {
22+
23+
public GenerationConfig(
24+
float? temperature = null,
25+
float? topP = null,
26+
float? topK = null,
27+
int? candidateCount = null,
28+
int? maxOutputTokens = null,
29+
float? prescencePenalty = null,
30+
float? frequencyPenalty = null,
31+
string[] stopSequences = null,
32+
string responseMimeType = null,
33+
Schema responseSchema = null) { throw new NotImplementedException(); }
34+
}
35+
36+
}

0 commit comments

Comments
 (0)