Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 32 additions & 0 deletions googleapiclient-stubs/_apis/aiplatform/v1/resources.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -2100,6 +2100,14 @@ class AiplatformResource(googleapiclient.discovery.Resource):
deployedModelId: str = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...
def responses(
self,
*,
endpoint: str,
body: GoogleApiHttpBody = ...,
deployedModelId: str = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...

@typing.type_check_only
class OperationsResource(googleapiclient.discovery.Resource):
Expand Down Expand Up @@ -6345,6 +6353,20 @@ class AiplatformResource(googleapiclient.discovery.Resource):
) -> GoogleLongrunningOperationHttpRequest: ...
def operations(self) -> OperationsResource: ...

def askContexts(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1AskContextsRequest = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1AskContextsResponseHttpRequest: ...
def asyncRetrieveContexts(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1AsyncRetrieveContextsRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def augmentPrompt(
self,
*,
Expand Down Expand Up @@ -7353,6 +7375,16 @@ class GoogleCloudAiplatformV1ArtifactHttpRequest(googleapiclient.http.HttpReques
num_retries: int = 0,
) -> GoogleCloudAiplatformV1Artifact: ...

@typing.type_check_only
class GoogleCloudAiplatformV1AskContextsResponseHttpRequest(
googleapiclient.http.HttpRequest
):
def execute(
self,
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
num_retries: int = 0,
) -> GoogleCloudAiplatformV1AskContextsResponse: ...

@typing.type_check_only
class GoogleCloudAiplatformV1AugmentPromptResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand Down
151 changes: 135 additions & 16 deletions googleapiclient-stubs/_apis/aiplatform/v1/schemas.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,17 @@ import typing_extensions

_list = list

@typing.type_check_only
class CloudAiLargeModelsVisionGenerateVideoExperiments(
typing_extensions.TypedDict, total=False
):
humanPose: CloudAiLargeModelsVisionHumanPose
modelName: str
numDiffusionSteps: int
promptInputs: CloudAiLargeModelsVisionPromptInputs
videoTransformMaskGcsUri: str
videoTransformStrength: float

@typing.type_check_only
class CloudAiLargeModelsVisionGenerateVideoResponse(
typing_extensions.TypedDict, total=False
Expand All @@ -18,9 +29,15 @@ class CloudAiLargeModelsVisionGenerateVideoResponseVideo(
typing_extensions.TypedDict, total=False
):
bytesBase64Encoded: str
experimentsMetadata: CloudAiLargeModelsVisionGenerateVideoExperiments
gcsUri: str
mimeType: str

@typing.type_check_only
class CloudAiLargeModelsVisionHumanPose(typing_extensions.TypedDict, total=False):
bodyLandmarksGcsUri: str
faceLandmarksGcsUri: str

@typing.type_check_only
class CloudAiLargeModelsVisionImage(typing_extensions.TypedDict, total=False):
encoding: str
Expand Down Expand Up @@ -60,6 +77,14 @@ class CloudAiLargeModelsVisionNamedBoundingBox(
y1: float
y2: float

@typing.type_check_only
class CloudAiLargeModelsVisionPromptInputs(typing_extensions.TypedDict, total=False):
audioPrompt: str
negativeAudioPrompt: str
negativePrompt: str
promptChunks: _list[str]
transcript: str

@typing.type_check_only
class CloudAiLargeModelsVisionRaiInfo(typing_extensions.TypedDict, total=False):
blockedEntities: _list[str]
Expand Down Expand Up @@ -278,6 +303,20 @@ class GoogleCloudAiplatformV1Artifact(typing_extensions.TypedDict, total=False):
updateTime: str
uri: str

@typing.type_check_only
class GoogleCloudAiplatformV1AskContextsRequest(
typing_extensions.TypedDict, total=False
):
query: GoogleCloudAiplatformV1RagQuery
tools: _list[GoogleCloudAiplatformV1Tool]

@typing.type_check_only
class GoogleCloudAiplatformV1AskContextsResponse(
typing_extensions.TypedDict, total=False
):
contexts: GoogleCloudAiplatformV1RagContexts
response: str

@typing.type_check_only
class GoogleCloudAiplatformV1AssignNotebookRuntimeOperationMetadata(
typing_extensions.TypedDict, total=False
Expand All @@ -293,6 +332,13 @@ class GoogleCloudAiplatformV1AssignNotebookRuntimeRequest(
notebookRuntimeId: str
notebookRuntimeTemplate: str

@typing.type_check_only
class GoogleCloudAiplatformV1AsyncRetrieveContextsRequest(
typing_extensions.TypedDict, total=False
):
query: GoogleCloudAiplatformV1RagQuery
tools: _list[GoogleCloudAiplatformV1Tool]

@typing.type_check_only
class GoogleCloudAiplatformV1Attribution(typing_extensions.TypedDict, total=False):
approximationError: float
Expand Down Expand Up @@ -1941,6 +1987,16 @@ class GoogleCloudAiplatformV1EvaluateDatasetResponse(
aggregationOutput: GoogleCloudAiplatformV1AggregationOutput
outputInfo: GoogleCloudAiplatformV1OutputInfo

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluateDatasetRun(
typing_extensions.TypedDict, total=False
):
checkpointId: str
error: GoogleRpcStatus
evaluateDatasetResponse: GoogleCloudAiplatformV1EvaluateDatasetResponse
evaluationRun: str
operationName: str

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluateInstancesRequest(
typing_extensions.TypedDict, total=False
Expand Down Expand Up @@ -2076,6 +2132,13 @@ class GoogleCloudAiplatformV1EvaluatedAnnotationExplanation(
explanation: GoogleCloudAiplatformV1Explanation
explanationType: str

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationConfig(typing_extensions.TypedDict, total=False):
autoraterConfig: GoogleCloudAiplatformV1AutoraterConfig
inferenceGenerationConfig: GoogleCloudAiplatformV1GenerationConfig
metrics: _list[GoogleCloudAiplatformV1Metric]
outputConfig: GoogleCloudAiplatformV1OutputConfig

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationDataset(
typing_extensions.TypedDict, total=False
Expand All @@ -2087,46 +2150,47 @@ class GoogleCloudAiplatformV1EvaluationDataset(
class GoogleCloudAiplatformV1EvaluationInstance(
typing_extensions.TypedDict, total=False
):
agentData: GoogleCloudAiplatformV1EvaluationInstanceAgentData
agentEvalData: GoogleCloudAiplatformV1EvaluationInstanceAgentData
agentData: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentData
otherData: GoogleCloudAiplatformV1EvaluationInstanceMapInstance
prompt: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
reference: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
response: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
rubricGroups: dict[str, typing.Any]

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentConfig(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig(
typing_extensions.TypedDict, total=False
):
agentId: str
agentType: str
description: str
developerInstruction: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
subAgents: _list[str]
tools: GoogleCloudAiplatformV1EvaluationInstanceAgentConfigTools
tools: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfigTools
toolsText: str

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentConfigTools(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfigTools(
typing_extensions.TypedDict, total=False
):
tool: _list[GoogleCloudAiplatformV1Tool]

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentData(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentData(
typing_extensions.TypedDict, total=False
):
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceAgentConfig
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig
agents: dict[str, typing.Any]
developerInstruction: GoogleCloudAiplatformV1EvaluationInstanceInstanceData
events: GoogleCloudAiplatformV1EvaluationInstanceAgentDataEvents
tools: GoogleCloudAiplatformV1EvaluationInstanceAgentDataTools
events: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataEvents
tools: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataTools
toolsText: str
turns: _list[GoogleCloudAiplatformV1EvaluationInstanceAgentDataConversationTurn]
turns: _list[
GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataConversationTurn
]

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataAgentEvent(
typing_extensions.TypedDict, total=False
):
activeTools: _list[GoogleCloudAiplatformV1Tool]
Expand All @@ -2136,21 +2200,23 @@ class GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent(
stateDelta: dict[str, typing.Any]

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataConversationTurn(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataConversationTurn(
typing_extensions.TypedDict, total=False
):
events: _list[GoogleCloudAiplatformV1EvaluationInstanceAgentDataAgentEvent]
events: _list[
GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataAgentEvent
]
turnId: str
turnIndex: int

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataEvents(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataEvents(
typing_extensions.TypedDict, total=False
):
event: _list[GoogleCloudAiplatformV1Content]

@typing.type_check_only
class GoogleCloudAiplatformV1EvaluationInstanceAgentDataTools(
class GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentDataTools(
typing_extensions.TypedDict, total=False
):
tool: _list[GoogleCloudAiplatformV1Tool]
Expand Down Expand Up @@ -2419,8 +2485,10 @@ class GoogleCloudAiplatformV1EventMetadata(typing_extensions.TypedDict, total=Fa
branch: str
customMetadata: dict[str, typing.Any]
groundingMetadata: GoogleCloudAiplatformV1GroundingMetadata
inputTranscription: GoogleCloudAiplatformV1Transcription
interrupted: bool
longRunningToolIds: _list[str]
outputTranscription: GoogleCloudAiplatformV1Transcription
partial: bool
turnComplete: bool

Expand Down Expand Up @@ -3600,7 +3668,7 @@ class GoogleCloudAiplatformV1GenerateFetchAccessTokenResponse(
class GoogleCloudAiplatformV1GenerateInstanceRubricsRequest(
typing_extensions.TypedDict, total=False
):
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceAgentConfig
agentConfig: GoogleCloudAiplatformV1EvaluationInstanceDeprecatedAgentConfig
contents: _list[GoogleCloudAiplatformV1Content]
location: str
predefinedRubricGenerationSpec: GoogleCloudAiplatformV1PredefinedMetricSpec
Expand Down Expand Up @@ -3836,16 +3904,27 @@ class GoogleCloudAiplatformV1GroundednessSpec(typing_extensions.TypedDict, total

@typing.type_check_only
class GoogleCloudAiplatformV1GroundingChunk(typing_extensions.TypedDict, total=False):
image: GoogleCloudAiplatformV1GroundingChunkImage
maps: GoogleCloudAiplatformV1GroundingChunkMaps
retrievedContext: GoogleCloudAiplatformV1GroundingChunkRetrievedContext
web: GoogleCloudAiplatformV1GroundingChunkWeb

@typing.type_check_only
class GoogleCloudAiplatformV1GroundingChunkImage(
typing_extensions.TypedDict, total=False
):
domain: str
imageUri: str
sourceUri: str
title: str

@typing.type_check_only
class GoogleCloudAiplatformV1GroundingChunkMaps(
typing_extensions.TypedDict, total=False
):
placeAnswerSources: GoogleCloudAiplatformV1GroundingChunkMapsPlaceAnswerSources
placeId: str
route: GoogleCloudAiplatformV1GroundingChunkMapsRoute
text: str
title: str
uri: str
Expand All @@ -3866,6 +3945,14 @@ class GoogleCloudAiplatformV1GroundingChunkMapsPlaceAnswerSourcesReviewSnippet(
reviewId: str
title: str

@typing.type_check_only
class GoogleCloudAiplatformV1GroundingChunkMapsRoute(
typing_extensions.TypedDict, total=False
):
distanceMeters: int
duration: str
encodedPolyline: str

@typing.type_check_only
class GoogleCloudAiplatformV1GroundingChunkRetrievedContext(
typing_extensions.TypedDict, total=False
Expand All @@ -3891,6 +3978,7 @@ class GoogleCloudAiplatformV1GroundingMetadata(
googleMapsWidgetContextToken: str
groundingChunks: _list[GoogleCloudAiplatformV1GroundingChunk]
groundingSupports: _list[GoogleCloudAiplatformV1GroundingSupport]
imageSearchQueries: _list[str]
retrievalMetadata: GoogleCloudAiplatformV1RetrievalMetadata
searchEntryPoint: GoogleCloudAiplatformV1SearchEntryPoint
sourceFlaggingUris: _list[GoogleCloudAiplatformV1GroundingMetadataSourceFlaggingUri]
Expand All @@ -3907,6 +3995,7 @@ class GoogleCloudAiplatformV1GroundingMetadataSourceFlaggingUri(
class GoogleCloudAiplatformV1GroundingSupport(typing_extensions.TypedDict, total=False):
confidenceScores: _list[float]
groundingChunkIndices: _list[int]
renderedParts: _list[int]
segment: GoogleCloudAiplatformV1Segment

@typing.type_check_only
Expand Down Expand Up @@ -3957,6 +4046,11 @@ class GoogleCloudAiplatformV1ImageConfig(typing_extensions.TypedDict, total=Fals
personGeneration: typing_extensions.Literal[
"PERSON_GENERATION_UNSPECIFIED", "ALLOW_ALL", "ALLOW_ADULT", "ALLOW_NONE"
]
prominentPeople: typing_extensions.Literal[
"PROMINENT_PEOPLE_UNSPECIFIED",
"ALLOW_PROMINENT_PEOPLE",
"BLOCK_PROMINENT_PEOPLE",
]

@typing.type_check_only
class GoogleCloudAiplatformV1ImageConfigImageOutputOptions(
Expand Down Expand Up @@ -10274,6 +10368,7 @@ class GoogleCloudAiplatformV1SupervisedTuningDatasetDistributionDatasetBucket(
class GoogleCloudAiplatformV1SupervisedTuningSpec(
typing_extensions.TypedDict, total=False
):
evaluationConfig: GoogleCloudAiplatformV1EvaluationConfig
exportLastCheckpointOnly: bool
hyperParameters: GoogleCloudAiplatformV1SupervisedHyperParameters
trainingDatasetUri: str
Expand Down Expand Up @@ -10537,6 +10632,24 @@ class GoogleCloudAiplatformV1ToolGoogleSearch(typing_extensions.TypedDict, total
"BLOCK_ONLY_EXTREMELY_HIGH",
]
excludeDomains: _list[str]
searchTypes: GoogleCloudAiplatformV1ToolGoogleSearchSearchTypes

@typing.type_check_only
class GoogleCloudAiplatformV1ToolGoogleSearchImageSearch(
typing_extensions.TypedDict, total=False
): ...

@typing.type_check_only
class GoogleCloudAiplatformV1ToolGoogleSearchSearchTypes(
typing_extensions.TypedDict, total=False
):
imageSearch: GoogleCloudAiplatformV1ToolGoogleSearchImageSearch
webSearch: GoogleCloudAiplatformV1ToolGoogleSearchWebSearch

@typing.type_check_only
class GoogleCloudAiplatformV1ToolGoogleSearchWebSearch(
typing_extensions.TypedDict, total=False
): ...

@typing.type_check_only
class GoogleCloudAiplatformV1ToolNameMatchInput(
Expand Down Expand Up @@ -10879,6 +10992,11 @@ class GoogleCloudAiplatformV1TrajectorySingleToolUseSpec(
):
toolName: str

@typing.type_check_only
class GoogleCloudAiplatformV1Transcription(typing_extensions.TypedDict, total=False):
finished: bool
text: str

@typing.type_check_only
class GoogleCloudAiplatformV1Trial(typing_extensions.TypedDict, total=False):
clientId: str
Expand Down Expand Up @@ -10947,6 +11065,7 @@ class GoogleCloudAiplatformV1TuningJob(typing_extensions.TypedDict, total=False)
encryptionSpec: GoogleCloudAiplatformV1EncryptionSpec
endTime: str
error: GoogleRpcStatus
evaluateDatasetRuns: _list[GoogleCloudAiplatformV1EvaluateDatasetRun]
experiment: str
labels: dict[str, typing.Any]
name: str
Expand Down
Loading
Loading