Skip to content

Commit b9af1f1

Browse files
henribruhenribru
committed
Update stubs
1 parent 0c285ea commit b9af1f1

78 files changed

Lines changed: 3422 additions & 219 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎googleapiclient-stubs/_apis/aiplatform/v1/resources.pyi‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2100,6 +2100,14 @@ class AiplatformResource(googleapiclient.discovery.Resource):
21002100
deployedModelId: str = ...,
21012101
**kwargs: typing.Any,
21022102
) -> GoogleApiHttpBodyHttpRequest: ...
2103+
def responses(
2104+
self,
2105+
*,
2106+
endpoint: str,
2107+
body: GoogleApiHttpBody = ...,
2108+
deployedModelId: str = ...,
2109+
**kwargs: typing.Any,
2110+
) -> GoogleApiHttpBodyHttpRequest: ...
21032111

21042112
@typing.type_check_only
21052113
class OperationsResource(googleapiclient.discovery.Resource):
@@ -6345,6 +6353,20 @@ class AiplatformResource(googleapiclient.discovery.Resource):
63456353
) -> GoogleLongrunningOperationHttpRequest: ...
63466354
def operations(self) -> OperationsResource: ...
63476355

6356+
def askContexts(
6357+
self,
6358+
*,
6359+
parent: str,
6360+
body: GoogleCloudAiplatformV1AskContextsRequest = ...,
6361+
**kwargs: typing.Any,
6362+
) -> GoogleCloudAiplatformV1AskContextsResponseHttpRequest: ...
6363+
def asyncRetrieveContexts(
6364+
self,
6365+
*,
6366+
parent: str,
6367+
body: GoogleCloudAiplatformV1AsyncRetrieveContextsRequest = ...,
6368+
**kwargs: typing.Any,
6369+
) -> GoogleLongrunningOperationHttpRequest: ...
63486370
def augmentPrompt(
63496371
self,
63506372
*,
@@ -7353,6 +7375,16 @@ class GoogleCloudAiplatformV1ArtifactHttpRequest(googleapiclient.http.HttpReques
73537375
num_retries: int = 0,
73547376
) -> GoogleCloudAiplatformV1Artifact: ...
73557377

7378+
@typing.type_check_only
7379+
class GoogleCloudAiplatformV1AskContextsResponseHttpRequest(
7380+
googleapiclient.http.HttpRequest
7381+
):
7382+
def execute(
7383+
self,
7384+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
7385+
num_retries: int = 0,
7386+
) -> GoogleCloudAiplatformV1AskContextsResponse: ...
7387+
73567388
@typing.type_check_only
73577389
class GoogleCloudAiplatformV1AugmentPromptResponseHttpRequest(
73587390
googleapiclient.http.HttpRequest

‎googleapiclient-stubs/_apis/aiplatform/v1/schemas.pyi‎

Lines changed: 135 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,17 @@ import typing_extensions
44

55
_list = list
66

7+
@typing.type_check_only
8+
class CloudAiLargeModelsVisionGenerateVideoExperiments(
9+
typing_extensions.TypedDict, total=False
10+
):
11+
humanPose: CloudAiLargeModelsVisionHumanPose
12+
modelName: str
13+
numDiffusionSteps: int
14+
promptInputs: CloudAiLargeModelsVisionPromptInputs
15+
videoTransformMaskGcsUri: str
16+
videoTransformStrength: float
17+
718
@typing.type_check_only
819
class CloudAiLargeModelsVisionGenerateVideoResponse(
920
typing_extensions.TypedDict, total=False
@@ -18,9 +29,15 @@ class CloudAiLargeModelsVisionGenerateVideoResponseVideo(
1829
typing_extensions.TypedDict, total=False
1930
):
2031
bytesBase64Encoded: str
32+
experimentsMetadata: CloudAiLargeModelsVisionGenerateVideoExperiments
2133
gcsUri: str
2234
mimeType: str
2335

36+
@typing.type_check_only
37+
class CloudAiLargeModelsVisionHumanPose(typing_extensions.TypedDict, total=False):
38+
bodyLandmarksGcsUri: str
39+
faceLandmarksGcsUri: str
40+
2441
@typing.type_check_only
2542
class CloudAiLargeModelsVisionImage(typing_extensions.TypedDict, total=False):
2643
encoding: str
@@ -60,6 +77,14 @@ class CloudAiLargeModelsVisionNamedBoundingBox(
6077
y1: float
6178
y2: float
6279

80+
@typing.type_check_only
81+
class CloudAiLargeModelsVisionPromptInputs(typing_extensions.TypedDict, total=False):
82+
audioPrompt: str
83+
negativeAudioPrompt: str
84+
negativePrompt: str
85+
promptChunks: _list[str]
86+
transcript: str
87+
6388
@typing.type_check_only
6489
class CloudAiLargeModelsVisionRaiInfo(typing_extensions.TypedDict, total=False):
6590
blockedEntities: _list[str]
@@ -278,6 +303,20 @@ class GoogleCloudAiplatformV1Artifact(typing_extensions.TypedDict, total=False):
278303
updateTime: str
279304
uri: str
280305

306+
@typing.type_check_only
307+
class GoogleCloudAiplatformV1AskContextsRequest(
308+
typing_extensions.TypedDict, total=False
309+
):
310+
query: GoogleCloudAiplatformV1RagQuery
311+
tools: _list[GoogleCloudAiplatformV1Tool]
312+
313+
@typing.type_check_only
314+
class GoogleCloudAiplatformV1AskContextsResponse(
315+
typing_extensions.TypedDict, total=False
316+
):
317+
contexts: GoogleCloudAiplatformV1RagContexts
318+
response: str
319+
281320
@typing.type_check_only
282321
class GoogleCloudAiplatformV1AssignNotebookRuntimeOperationMetadata(
283322
typing_extensions.TypedDict, total=False
@@ -293,6 +332,13 @@ class GoogleCloudAiplatformV1AssignNotebookRuntimeRequest(
293332
notebookRuntimeId: str
294333
notebookRuntimeTemplate: str
295334

335+
@typing.type_check_only
336+
class GoogleCloudAiplatformV1AsyncRetrieveContextsRequest(
337+
typing_extensions.TypedDict, total=False
338+
):
339+
query: GoogleCloudAiplatformV1RagQuery
340+
tools: _list[GoogleCloudAiplatformV1Tool]
341+
296342
@typing.type_check_only
297343
class GoogleCloudAiplatformV1Attribution(typing_extensions.TypedDict, total=False):
298344
approximationError: float
@@ -1941,6 +1987,16 @@ class GoogleCloudAiplatformV1EvaluateDatasetResponse(
19411987
aggregationOutput: GoogleCloudAiplatformV1AggregationOutput
19421988
outputInfo: GoogleCloudAiplatformV1OutputInfo
19431989

1990+
@typing.type_check_only
1991+
class GoogleCloudAiplatformV1EvaluateDatasetRun(
1992+
typing_extensions.TypedDict, total=False
1993+
):
1994+
checkpointId: str
1995+
error: GoogleRpcStatus
1996+
evaluateDatasetResponse: GoogleCloudAiplatformV1EvaluateDatasetResponse
1997+
evaluationRun: str
1998+
operationName: str
1999+
19442000
@typing.type_check_only
19452001
class GoogleCloudAiplatformV1EvaluateInstancesRequest(
19462002
typing_extensions.TypedDict, total=False
@@ -2076,6 +2132,13 @@ class GoogleCloudAiplatformV1EvaluatedAnnotationExplanation(
20762132
explanation: GoogleCloudAiplatformV1Explanation
20772133
explanationType: str
20782134

2135+
@typing.type_check_only
2136+
class GoogleCloudAiplatformV1EvaluationConfig(typing_extensions.TypedDict, total=False):
2137+
autoraterConfig: GoogleCloudAiplatformV1AutoraterConfig
2138+
inferenceGenerationConfig: GoogleCloudAiplatformV1GenerationConfig
2139+
metrics: _list[GoogleCloudAiplatformV1Metric]
2140+
outputConfig: GoogleCloudAiplatformV1OutputConfig
2141+
20792142
@typing.type_check_only
20802143
class GoogleCloudAiplatformV1EvaluationDataset(
20812144
typing_extensions.TypedDict, total=False
@@ -2087,46 +2150,47 @@ class GoogleCloudAiplatformV1EvaluationDataset(
20872150
class GoogleCloudAiplatformV1EvaluationInstance(
20882151
typing_extensions.TypedDict, total=False
20892152
):
2090-
agentData: GoogleCloudAiplatformV1EvaluationInstanceAgentData
2091-
agentEvalData: GoogleCloudAiplatformV1EvaluationInstanceAgentData
2153+
agentData: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentData
20922154
otherData: GoogleCloudAiplatformV1EvaluationInstanceMapInstance
20932155
prompt: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
20942156
reference: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
20952157
response: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
20962158
rubricGroups: dict[str, typing.Any]
20972159

20982160
@typing.type_check_only
2099-
class GoogleCloudAiplatformV1EvaluationInstanceAgentConfig(
2161+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig(
21002162
typing_extensions.TypedDict, total=False
21012163
):
21022164
agentId: str
21032165
agentType: str
21042166
description: str
21052167
developerInstruction: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
21062168
subAgents: _list[str]
2107-
tools: GoogleCloudAiplatformV1EvaluationInstanceAgentConfigTools
2169+
tools: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfigTools
21082170
toolsText: str
21092171

21102172
@typing.type_check_only
2111-
class GoogleCloudAiplatformV1EvaluationInstanceAgentConfigTools(
2173+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfigTools(
21122174
typing_extensions.TypedDict, total=False
21132175
):
21142176
tool: _list[GoogleCloudAiplatformV1Tool]
21152177

21162178
@typing.type_check_only
2117-
class GoogleCloudAiplatformV1EvaluationInstanceAgentData(
2179+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentData(
21182180
typing_extensions.TypedDict, total=False
21192181
):
2120-
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceAgentConfig
2182+
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig
21212183
agents: dict[str, typing.Any]
21222184
developerInstruction: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
2123-
events: GoogleCloudAiplatformV1EvaluationInstanceAgentDataEvents
2124-
tools: GoogleCloudAiplatformV1EvaluationInstanceAgentDataTools
2185+
events: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataEvents
2186+
tools: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataTools
21252187
toolsText: str
2126-
turns: _list[GoogleCloudAiplatformV1EvaluationInstanceAgentDataConversationTurn]
2188+
turns: _list[
2189+
GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataConversationTurn
2190+
]
21272191

21282192
@typing.type_check_only
2129-
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent(
2193+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataAgentEvent(
21302194
typing_extensions.TypedDict, total=False
21312195
):
21322196
activeTools: _list[GoogleCloudAiplatformV1Tool]
@@ -2136,21 +2200,23 @@ class GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent(
21362200
stateDelta: dict[str, typing.Any]
21372201

21382202
@typing.type_check_only
2139-
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataConversationTurn(
2203+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataConversationTurn(
21402204
typing_extensions.TypedDict, total=False
21412205
):
2142-
events: _list[GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent]
2206+
events: _list[
2207+
GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataAgentEvent
2208+
]
21432209
turnId: str
21442210
turnIndex: int
21452211

21462212
@typing.type_check_only
2147-
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataEvents(
2213+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataEvents(
21482214
typing_extensions.TypedDict, total=False
21492215
):
21502216
event: _list[GoogleCloudAiplatformV1Content]
21512217

21522218
@typing.type_check_only
2153-
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataTools(
2219+
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataTools(
21542220
typing_extensions.TypedDict, total=False
21552221
):
21562222
tool: _list[GoogleCloudAiplatformV1Tool]
@@ -2419,8 +2485,10 @@ class GoogleCloudAiplatformV1EventMetadata(typing_extensions.TypedDict, total=Fa
24192485
branch: str
24202486
customMetadata: dict[str, typing.Any]
24212487
groundingMetadata: GoogleCloudAiplatformV1GroundingMetadata
2488+
inputTranscription: GoogleCloudAiplatformV1Transcription
24222489
interrupted: bool
24232490
longRunningToolIds: _list[str]
2491+
outputTranscription: GoogleCloudAiplatformV1Transcription
24242492
partial: bool
24252493
turnComplete: bool
24262494

@@ -3600,7 +3668,7 @@ class GoogleCloudAiplatformV1GenerateFetchAccessTokenResponse(
36003668
class GoogleCloudAiplatformV1GenerateInstanceRubricsRequest(
36013669
typing_extensions.TypedDict, total=False
36023670
):
3603-
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceAgentConfig
3671+
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig
36043672
contents: _list[GoogleCloudAiplatformV1Content]
36053673
location: str
36063674
predefinedRubricGenerationSpec: GoogleCloudAiplatformV1PredefinedMetricSpec
@@ -3836,16 +3904,27 @@ class GoogleCloudAiplatformV1GroundednessSpec(typing_extensions.TypedDict, total
38363904

38373905
@typing.type_check_only
38383906
class GoogleCloudAiplatformV1GroundingChunk(typing_extensions.TypedDict, total=False):
3907+
image: GoogleCloudAiplatformV1GroundingChunkImage
38393908
maps: GoogleCloudAiplatformV1GroundingChunkMaps
38403909
retrievedContext: GoogleCloudAiplatformV1GroundingChunkRetrievedContext
38413910
web: GoogleCloudAiplatformV1GroundingChunkWeb
38423911

3912+
@typing.type_check_only
3913+
class GoogleCloudAiplatformV1GroundingChunkImage(
3914+
typing_extensions.TypedDict, total=False
3915+
):
3916+
domain: str
3917+
imageUri: str
3918+
sourceUri: str
3919+
title: str
3920+
38433921
@typing.type_check_only
38443922
class GoogleCloudAiplatformV1GroundingChunkMaps(
38453923
typing_extensions.TypedDict, total=False
38463924
):
38473925
placeAnswerSources: GoogleCloudAiplatformV1GroundingChunkMapsPlaceAnswerSources
38483926
placeId: str
3927+
route: GoogleCloudAiplatformV1GroundingChunkMapsRoute
38493928
text: str
38503929
title: str
38513930
uri: str
@@ -3866,6 +3945,14 @@ class GoogleCloudAiplatformV1GroundingChunkMapsPlaceAnswerSourcesReviewSnippet(
38663945
reviewId: str
38673946
title: str
38683947

3948+
@typing.type_check_only
3949+
class GoogleCloudAiplatformV1GroundingChunkMapsRoute(
3950+
typing_extensions.TypedDict, total=False
3951+
):
3952+
distanceMeters: int
3953+
duration: str
3954+
encodedPolyline: str
3955+
38693956
@typing.type_check_only
38703957
class GoogleCloudAiplatformV1GroundingChunkRetrievedContext(
38713958
typing_extensions.TypedDict, total=False
@@ -3891,6 +3978,7 @@ class GoogleCloudAiplatformV1GroundingMetadata(
38913978
googleMapsWidgetContextToken: str
38923979
groundingChunks: _list[GoogleCloudAiplatformV1GroundingChunk]
38933980
groundingSupports: _list[GoogleCloudAiplatformV1GroundingSupport]
3981+
imageSearchQueries: _list[str]
38943982
retrievalMetadata: GoogleCloudAiplatformV1RetrievalMetadata
38953983
searchEntryPoint: GoogleCloudAiplatformV1SearchEntryPoint
38963984
sourceFlaggingUris: _list[GoogleCloudAiplatformV1GroundingMetadataSourceFlaggingUri]
@@ -3907,6 +3995,7 @@ class GoogleCloudAiplatformV1GroundingMetadataSourceFlaggingUri(
39073995
class GoogleCloudAiplatformV1GroundingSupport(typing_extensions.TypedDict, total=False):
39083996
confidenceScores: _list[float]
39093997
groundingChunkIndices: _list[int]
3998+
renderedParts: _list[int]
39103999
segment: GoogleCloudAiplatformV1Segment
39114000

39124001
@typing.type_check_only
@@ -3957,6 +4046,11 @@ class GoogleCloudAiplatformV1ImageConfig(typing_extensions.TypedDict, total=Fals
39574046
personGeneration: typing_extensions.Literal[
39584047
"PERSON_GENERATION_UNSPECIFIED", "ALLOW_ALL", "ALLOW_ADULT", "ALLOW_NONE"
39594048
]
4049+
prominentPeople: typing_extensions.Literal[
4050+
"PROMINENT_PEOPLE_UNSPECIFIED",
4051+
"ALLOW_PROMINENT_PEOPLE",
4052+
"BLOCK_PROMINENT_PEOPLE",
4053+
]
39604054

39614055
@typing.type_check_only
39624056
class GoogleCloudAiplatformV1ImageConfigImageOutputOptions(
@@ -10274,6 +10368,7 @@ class GoogleCloudAiplatformV1SupervisedTuningDatasetDistributionDatasetBucket(
1027410368
class GoogleCloudAiplatformV1SupervisedTuningSpec(
1027510369
typing_extensions.TypedDict, total=False
1027610370
):
10371+
evaluationConfig: GoogleCloudAiplatformV1EvaluationConfig
1027710372
exportLastCheckpointOnly: bool
1027810373
hyperParameters: GoogleCloudAiplatformV1SupervisedHyperParameters
1027910374
trainingDatasetUri: str
@@ -10537,6 +10632,24 @@ class GoogleCloudAiplatformV1ToolGoogleSearch(typing_extensions.TypedDict, total
1053710632
"BLOCK_ONLY_EXTREMELY_HIGH",
1053810633
]
1053910634
excludeDomains: _list[str]
10635+
searchTypes: GoogleCloudAiplatformV1ToolGoogleSearchSearchTypes
10636+
10637+
@typing.type_check_only
10638+
class GoogleCloudAiplatformV1ToolGoogleSearchImageSearch(
10639+
typing_extensions.TypedDict, total=False
10640+
): ...
10641+
10642+
@typing.type_check_only
10643+
class GoogleCloudAiplatformV1ToolGoogleSearchSearchTypes(
10644+
typing_extensions.TypedDict, total=False
10645+
):
10646+
imageSearch: GoogleCloudAiplatformV1ToolGoogleSearchImageSearch
10647+
webSearch: GoogleCloudAiplatformV1ToolGoogleSearchWebSearch
10648+
10649+
@typing.type_check_only
10650+
class GoogleCloudAiplatformV1ToolGoogleSearchWebSearch(
10651+
typing_extensions.TypedDict, total=False
10652+
): ...
1054010653

1054110654
@typing.type_check_only
1054210655
class GoogleCloudAiplatformV1ToolNameMatchInput(
@@ -10879,6 +10992,11 @@ class GoogleCloudAiplatformV1TrajectorySingleToolUseSpec(
1087910992
):
1088010993
toolName: str
1088110994

10995+
@typing.type_check_only
10996+
class GoogleCloudAiplatformV1Transcription(typing_extensions.TypedDict, total=False):
10997+
finished: bool
10998+
text: str
10999+
1088211000
@typing.type_check_only
1088311001
class GoogleCloudAiplatformV1Trial(typing_extensions.TypedDict, total=False):
1088411002
clientId: str
@@ -10947,6 +11065,7 @@ class GoogleCloudAiplatformV1TuningJob(typing_extensions.TypedDict, total=False)
1094711065
encryptionSpec: GoogleCloudAiplatformV1EncryptionSpec
1094811066
endTime: str
1094911067
error: GoogleRpcStatus
11068+
evaluateDatasetRuns: _list[GoogleCloudAiplatformV1EvaluateDatasetRun]
1095011069
experiment: str
1095111070
labels: dict[str, typing.Any]
1095211071
name: str

0 commit comments

Comments
 (0)