Skip to content
Merged
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
2 changes: 2 additions & 0 deletions googleapiclient-stubs/_apis/admin/reports_v1/resources.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,8 @@ class ReportsResource(googleapiclient.discovery.Resource):
"contacts",
"takeout",
"graduation",
"voice",
"chrome_sync",
],
actorIpAddress: str = ...,
applicationInfoFilter: str = ...,
Expand Down
102 changes: 102 additions & 0 deletions googleapiclient-stubs/_apis/aiplatform/v1/resources.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -5521,6 +5521,16 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def events(self) -> EventsResource: ...
def operations(self) -> OperationsResource: ...

def asyncQuery(
self,
*,
name: str,
body: GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def cancelAsyncQuery(
self, *, name: str, operationName: str = ..., **kwargs: typing.Any
) -> GoogleCloudAiplatformV1CancelAsyncQueryReasoningEngineResponseHttpRequest: ...
def create(
self,
*,
Expand Down Expand Up @@ -5697,6 +5707,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
) -> GoogleProtobufEmptyHttpRequest: ...
def operations(self) -> OperationsResource: ...

@typing.type_check_only
class SkillsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class OperationsResource(googleapiclient.discovery.Resource):
def cancel(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def delete(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def get(
self, *, name: str, **kwargs: typing.Any
) -> GoogleLongrunningOperationHttpRequest: ...
def list(
self,
*,
name: str,
filter: str = ...,
pageSize: int = ...,
pageToken: str = ...,
returnPartialSuccess: bool = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
def list_next(
self,
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
previous_response: GoogleLongrunningListOperationsResponse,
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
def wait(
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
) -> GoogleLongrunningOperationHttpRequest: ...

def operations(self) -> OperationsResource: ...

@typing.type_check_only
class SpecialistPoolsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -6619,6 +6663,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def ragEngineConfig(self) -> RagEngineConfigResource: ...
def reasoningEngines(self) -> ReasoningEnginesResource: ...
def schedules(self) -> SchedulesResource: ...
def skills(self) -> SkillsResource: ...
def specialistPools(self) -> SpecialistPoolsResource: ...
def studies(self) -> StudiesResource: ...
def tensorboards(self) -> TensorboardsResource: ...
Expand Down Expand Up @@ -7083,6 +7128,18 @@ class AiplatformResource(googleapiclient.discovery.Resource):

def operations(self) -> OperationsResource: ...

def asyncQuery(
self,
*,
name: str,
body: GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def cancelAsyncQuery(
self, *, name: str, operationName: str = ..., **kwargs: typing.Any
) -> (
GoogleCloudAiplatformV1CancelAsyncQueryReasoningEngineResponseHttpRequest
): ...
def create(
self,
*,
Expand Down Expand Up @@ -7185,6 +7242,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):

def operations(self) -> OperationsResource: ...

@typing.type_check_only
class SkillsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class OperationsResource(googleapiclient.discovery.Resource):
def cancel(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def delete(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def get(
self, *, name: str, **kwargs: typing.Any
) -> GoogleLongrunningOperationHttpRequest: ...
def list(
self,
*,
name: str,
filter: str = ...,
pageSize: int = ...,
pageToken: str = ...,
returnPartialSuccess: bool = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
def list_next(
self,
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
previous_response: GoogleLongrunningListOperationsResponse,
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
def wait(
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
) -> GoogleLongrunningOperationHttpRequest: ...

def operations(self) -> OperationsResource: ...

@typing.type_check_only
class SpecialistPoolsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -7558,6 +7649,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def ragEngineConfig(self) -> RagEngineConfigResource: ...
def reasoningEngines(self) -> ReasoningEnginesResource: ...
def schedules(self) -> SchedulesResource: ...
def skills(self) -> SkillsResource: ...
def specialistPools(self) -> SpecialistPoolsResource: ...
def studies(self) -> StudiesResource: ...
def tensorboards(self) -> TensorboardsResource: ...
Expand Down Expand Up @@ -7717,6 +7809,16 @@ class GoogleCloudAiplatformV1CachedContentHttpRequest(googleapiclient.http.HttpR
num_retries: int = 0,
) -> GoogleCloudAiplatformV1CachedContent: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1ComputeTokensResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand Down
22 changes: 14 additions & 8 deletions googleapiclient-stubs/_apis/aiplatform/v1/schemas.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ class CloudAiLargeModelsVisionGenerateVideoExperiments(
numDiffusionSteps: int
promptInputs: CloudAiLargeModelsVisionPromptInputs
requestOriginTag: str
truncateInputVideo: bool
videoTransformMaskGcsUri: str
videoTransformStrength: float

Expand Down Expand Up @@ -353,6 +354,13 @@ class GoogleCloudAiplatformV1AssignNotebookRuntimeRequest(
notebookRuntimeId: str
notebookRuntimeTemplate: str

@typing.type_check_only
class GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest(
typing_extensions.TypedDict, total=False
):
inputGcsUri: str
outputGcsUri: str

@typing.type_check_only
class GoogleCloudAiplatformV1AsyncRetrieveContextsRequest(
typing_extensions.TypedDict, total=False
Expand Down Expand Up @@ -822,6 +830,11 @@ class GoogleCloudAiplatformV1CachedContentUsageMetadata(
totalTokenCount: int
videoDurationSeconds: int

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

@typing.type_check_only
class GoogleCloudAiplatformV1CancelBatchPredictionJobRequest(
typing_extensions.TypedDict, total=False
Expand Down Expand Up @@ -1120,6 +1133,7 @@ class GoogleCloudAiplatformV1CopyModelOperationMetadata(

@typing.type_check_only
class GoogleCloudAiplatformV1CopyModelRequest(typing_extensions.TypedDict, total=False):
customServiceAccount: str
encryptionSpec: GoogleCloudAiplatformV1EncryptionSpec
modelId: str
parentModel: str
Expand Down Expand Up @@ -8187,7 +8201,6 @@ class GoogleCloudAiplatformV1SandboxEnvironmentConnectionInfo(
loadBalancerHostname: str
loadBalancerIp: str
routingToken: str
sandboxHostname: str
sandboxInternalIp: str

@typing.type_check_only
Expand Down Expand Up @@ -8247,7 +8260,6 @@ class GoogleCloudAiplatformV1SandboxEnvironmentTemplate(
"UNSPECIFIED", "PROVISIONING", "ACTIVE", "DEPROVISIONING", "DELETED", "FAILED"
]
updateTime: str
warmPoolConfig: GoogleCloudAiplatformV1SandboxEnvironmentTemplateWarmPoolConfig

@typing.type_check_only
class GoogleCloudAiplatformV1SandboxEnvironmentTemplateCustomContainerEnvironment(
Expand Down Expand Up @@ -8294,12 +8306,6 @@ class GoogleCloudAiplatformV1SandboxEnvironmentTemplateResourceRequirements(
limits: dict[str, typing.Any]
requests: dict[str, typing.Any]

@typing.type_check_only
class GoogleCloudAiplatformV1SandboxEnvironmentTemplateWarmPoolConfig(
typing_extensions.TypedDict, total=False
):
targetInstanceCount: int

@typing.type_check_only
class GoogleCloudAiplatformV1SavedQuery(typing_extensions.TypedDict, total=False):
annotationFilter: str
Expand Down
Loading
Loading