Skip to content

Commit 2d43715

Browse files
henribruhenribru
committed
Update stubs
1 parent a0a3cb7 commit 2d43715

98 files changed

Lines changed: 2975 additions & 495 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: 84 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5521,6 +5521,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
55215521
def events(self) -> EventsResource: ...
55225522
def operations(self) -> OperationsResource: ...
55235523

5524+
def asyncQuery(
5525+
self,
5526+
*,
5527+
name: str,
5528+
body: GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest = ...,
5529+
**kwargs: typing.Any,
5530+
) -> GoogleLongrunningOperationHttpRequest: ...
55245531
def create(
55255532
self,
55265533
*,
@@ -5697,6 +5704,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
56975704
) -> GoogleProtobufEmptyHttpRequest: ...
56985705
def operations(self) -> OperationsResource: ...
56995706

5707+
@typing.type_check_only
5708+
class SkillsResource(googleapiclient.discovery.Resource):
5709+
@typing.type_check_only
5710+
class OperationsResource(googleapiclient.discovery.Resource):
5711+
def cancel(
5712+
self, *, name: str, **kwargs: typing.Any
5713+
) -> GoogleProtobufEmptyHttpRequest: ...
5714+
def delete(
5715+
self, *, name: str, **kwargs: typing.Any
5716+
) -> GoogleProtobufEmptyHttpRequest: ...
5717+
def get(
5718+
self, *, name: str, **kwargs: typing.Any
5719+
) -> GoogleLongrunningOperationHttpRequest: ...
5720+
def list(
5721+
self,
5722+
*,
5723+
name: str,
5724+
filter: str = ...,
5725+
pageSize: int = ...,
5726+
pageToken: str = ...,
5727+
returnPartialSuccess: bool = ...,
5728+
**kwargs: typing.Any,
5729+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
5730+
def list_next(
5731+
self,
5732+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
5733+
previous_response: GoogleLongrunningListOperationsResponse,
5734+
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
5735+
def wait(
5736+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
5737+
) -> GoogleLongrunningOperationHttpRequest: ...
5738+
5739+
def operations(self) -> OperationsResource: ...
5740+
57005741
@typing.type_check_only
57015742
class SpecialistPoolsResource(googleapiclient.discovery.Resource):
57025743
@typing.type_check_only
@@ -6619,6 +6660,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
66196660
def ragEngineConfig(self) -> RagEngineConfigResource: ...
66206661
def reasoningEngines(self) -> ReasoningEnginesResource: ...
66216662
def schedules(self) -> SchedulesResource: ...
6663+
def skills(self) -> SkillsResource: ...
66226664
def specialistPools(self) -> SpecialistPoolsResource: ...
66236665
def studies(self) -> StudiesResource: ...
66246666
def tensorboards(self) -> TensorboardsResource: ...
@@ -7083,6 +7125,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
70837125

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

7128+
def asyncQuery(
7129+
self,
7130+
*,
7131+
name: str,
7132+
body: GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest = ...,
7133+
**kwargs: typing.Any,
7134+
) -> GoogleLongrunningOperationHttpRequest: ...
70867135
def create(
70877136
self,
70887137
*,
@@ -7185,6 +7234,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
71857234

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

7237+
@typing.type_check_only
7238+
class SkillsResource(googleapiclient.discovery.Resource):
7239+
@typing.type_check_only
7240+
class OperationsResource(googleapiclient.discovery.Resource):
7241+
def cancel(
7242+
self, *, name: str, **kwargs: typing.Any
7243+
) -> GoogleProtobufEmptyHttpRequest: ...
7244+
def delete(
7245+
self, *, name: str, **kwargs: typing.Any
7246+
) -> GoogleProtobufEmptyHttpRequest: ...
7247+
def get(
7248+
self, *, name: str, **kwargs: typing.Any
7249+
) -> GoogleLongrunningOperationHttpRequest: ...
7250+
def list(
7251+
self,
7252+
*,
7253+
name: str,
7254+
filter: str = ...,
7255+
pageSize: int = ...,
7256+
pageToken: str = ...,
7257+
returnPartialSuccess: bool = ...,
7258+
**kwargs: typing.Any,
7259+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
7260+
def list_next(
7261+
self,
7262+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
7263+
previous_response: GoogleLongrunningListOperationsResponse,
7264+
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
7265+
def wait(
7266+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
7267+
) -> GoogleLongrunningOperationHttpRequest: ...
7268+
7269+
def operations(self) -> OperationsResource: ...
7270+
71887271
@typing.type_check_only
71897272
class SpecialistPoolsResource(googleapiclient.discovery.Resource):
71907273
@typing.type_check_only
@@ -7558,6 +7641,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
75587641
def ragEngineConfig(self) -> RagEngineConfigResource: ...
75597642
def reasoningEngines(self) -> ReasoningEnginesResource: ...
75607643
def schedules(self) -> SchedulesResource: ...
7644+
def skills(self) -> SkillsResource: ...
75617645
def specialistPools(self) -> SpecialistPoolsResource: ...
75627646
def studies(self) -> StudiesResource: ...
75637647
def tensorboards(self) -> TensorboardsResource: ...

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

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ class CloudAiLargeModelsVisionGenerateVideoExperiments(
1616
numDiffusionSteps: int
1717
promptInputs: CloudAiLargeModelsVisionPromptInputs
1818
requestOriginTag: str
19+
truncateInputVideo: bool
1920
videoTransformMaskGcsUri: str
2021
videoTransformStrength: float
2122

@@ -353,6 +354,13 @@ class GoogleCloudAiplatformV1AssignNotebookRuntimeRequest(
353354
notebookRuntimeId: str
354355
notebookRuntimeTemplate: str
355356

357+
@typing.type_check_only
358+
class GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest(
359+
typing_extensions.TypedDict, total=False
360+
):
361+
inputGcsUri: str
362+
outputGcsUri: str
363+
356364
@typing.type_check_only
357365
class GoogleCloudAiplatformV1AsyncRetrieveContextsRequest(
358366
typing_extensions.TypedDict, total=False
@@ -1120,6 +1128,7 @@ class GoogleCloudAiplatformV1CopyModelOperationMetadata(
11201128

11211129
@typing.type_check_only
11221130
class GoogleCloudAiplatformV1CopyModelRequest(typing_extensions.TypedDict, total=False):
1131+
customServiceAccount: str
11231132
encryptionSpec: GoogleCloudAiplatformV1EncryptionSpec
11241133
modelId: str
11251134
parentModel: str
@@ -8247,7 +8256,6 @@ class GoogleCloudAiplatformV1SandboxEnvironmentTemplate(
82478256
"UNSPECIFIED", "PROVISIONING", "ACTIVE", "DEPROVISIONING", "DELETED", "FAILED"
82488257
]
82498258
updateTime: str
8250-
warmPoolConfig: GoogleCloudAiplatformV1SandboxEnvironmentTemplateWarmPoolConfig
82518259

82528260
@typing.type_check_only
82538261
class GoogleCloudAiplatformV1SandboxEnvironmentTemplateCustomContainerEnvironment(
@@ -8294,12 +8302,6 @@ class GoogleCloudAiplatformV1SandboxEnvironmentTemplateResourceRequirements(
82948302
limits: dict[str, typing.Any]
82958303
requests: dict[str, typing.Any]
82968304

8297-
@typing.type_check_only
8298-
class GoogleCloudAiplatformV1SandboxEnvironmentTemplateWarmPoolConfig(
8299-
typing_extensions.TypedDict, total=False
8300-
):
8301-
targetInstanceCount: int
8302-
83038305
@typing.type_check_only
83048306
class GoogleCloudAiplatformV1SavedQuery(typing_extensions.TypedDict, total=False):
83058307
annotationFilter: str

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

Lines changed: 136 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7713,6 +7713,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
77137713
def events(self) -> EventsResource: ...
77147714
def operations(self) -> OperationsResource: ...
77157715

7716+
def asyncQuery(
7717+
self,
7718+
*,
7719+
name: str,
7720+
body: GoogleCloudAiplatformV1beta1AsyncQueryReasoningEngineRequest = ...,
7721+
**kwargs: typing.Any,
7722+
) -> GoogleLongrunningOperationHttpRequest: ...
77167723
def create(
77177724
self,
77187725
*,
@@ -7971,6 +7978,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
79717978

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

7981+
@typing.type_check_only
7982+
class SkillsResource(googleapiclient.discovery.Resource):
7983+
@typing.type_check_only
7984+
class OperationsResource(googleapiclient.discovery.Resource):
7985+
def cancel(
7986+
self, *, name: str, **kwargs: typing.Any
7987+
) -> GoogleProtobufEmptyHttpRequest: ...
7988+
def delete(
7989+
self, *, name: str, **kwargs: typing.Any
7990+
) -> GoogleProtobufEmptyHttpRequest: ...
7991+
def get(
7992+
self, *, name: str, **kwargs: typing.Any
7993+
) -> GoogleLongrunningOperationHttpRequest: ...
7994+
def list(
7995+
self,
7996+
*,
7997+
name: str,
7998+
filter: str = ...,
7999+
pageSize: int = ...,
8000+
pageToken: str = ...,
8001+
returnPartialSuccess: bool = ...,
8002+
**kwargs: typing.Any,
8003+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
8004+
def list_next(
8005+
self,
8006+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
8007+
previous_response: GoogleLongrunningListOperationsResponse,
8008+
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
8009+
def wait(
8010+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
8011+
) -> GoogleLongrunningOperationHttpRequest: ...
8012+
8013+
def operations(self) -> OperationsResource: ...
8014+
79748015
@typing.type_check_only
79758016
class SolversResource(googleapiclient.discovery.Resource):
79768017
@typing.type_check_only
@@ -8969,6 +9010,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
89699010
def semanticGovernancePolicyEngine(
89709011
self,
89719012
) -> SemanticGovernancePolicyEngineResource: ...
9013+
def skills(self) -> SkillsResource: ...
89729014
def solvers(self) -> SolversResource: ...
89739015
def specialistPools(self) -> SpecialistPoolsResource: ...
89749016
def studies(self) -> StudiesResource: ...
@@ -10128,6 +10170,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
1012810170
def events(self) -> EventsResource: ...
1012910171
def operations(self) -> OperationsResource: ...
1013010172

10173+
def asyncQuery(
10174+
self,
10175+
*,
10176+
name: str,
10177+
body: GoogleCloudAiplatformV1beta1AsyncQueryReasoningEngineRequest = ...,
10178+
**kwargs: typing.Any,
10179+
) -> GoogleLongrunningOperationHttpRequest: ...
1013110180
def create(
1013210181
self,
1013310182
*,
@@ -10303,6 +10352,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
1030310352

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

10355+
@typing.type_check_only
10356+
class SkillsResource(googleapiclient.discovery.Resource):
10357+
@typing.type_check_only
10358+
class OperationsResource(googleapiclient.discovery.Resource):
10359+
def cancel(
10360+
self, *, name: str, **kwargs: typing.Any
10361+
) -> GoogleProtobufEmptyHttpRequest: ...
10362+
def delete(
10363+
self, *, name: str, **kwargs: typing.Any
10364+
) -> GoogleProtobufEmptyHttpRequest: ...
10365+
def get(
10366+
self, *, name: str, **kwargs: typing.Any
10367+
) -> GoogleLongrunningOperationHttpRequest: ...
10368+
def list(
10369+
self,
10370+
*,
10371+
name: str,
10372+
filter: str = ...,
10373+
pageSize: int = ...,
10374+
pageToken: str = ...,
10375+
returnPartialSuccess: bool = ...,
10376+
**kwargs: typing.Any,
10377+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
10378+
def list_next(
10379+
self,
10380+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
10381+
previous_response: GoogleLongrunningListOperationsResponse,
10382+
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
10383+
def wait(
10384+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
10385+
) -> GoogleLongrunningOperationHttpRequest: ...
10386+
10387+
def operations(self) -> OperationsResource: ...
10388+
1030610389
@typing.type_check_only
1030710390
class SolversResource(googleapiclient.discovery.Resource):
1030810391
@typing.type_check_only
@@ -10640,6 +10723,27 @@ class AiplatformResource(googleapiclient.discovery.Resource):
1064010723
**kwargs: typing.Any,
1064110724
) -> GoogleCloudAiplatformV1beta1GenerateInstanceRubricsResponseHttpRequest: ...
1064210725

10726+
@typing.type_check_only
10727+
class VertexExpressResource(googleapiclient.discovery.Resource):
10728+
@typing.type_check_only
10729+
class EligibilityResource(googleapiclient.discovery.Resource):
10730+
def check(
10731+
self, **kwargs: typing.Any
10732+
) -> (
10733+
GoogleCloudAiplatformV1beta1CheckSignUpEligibilityResponseHttpRequest
10734+
): ...
10735+
10736+
def retrieveExpressProject(
10737+
self, *, getDefaultApiKey: bool = ..., **kwargs: typing.Any
10738+
) -> GoogleCloudAiplatformV1beta1RetrieveExpressProjectResponseHttpRequest: ...
10739+
def signUp(
10740+
self,
10741+
*,
10742+
body: GoogleCloudAiplatformV1beta1SignUpRequest = ...,
10743+
**kwargs: typing.Any,
10744+
) -> GoogleCloudAiplatformV1beta1ExpressProjectHttpRequest: ...
10745+
def Eligibility(self) -> EligibilityResource: ...
10746+
1064310747
def new_batch_http_request(
1064410748
self,
1064510749
callback: collections.abc.Callable[
@@ -10700,13 +10804,15 @@ class AiplatformResource(googleapiclient.discovery.Resource):
1070010804
def semanticGovernancePolicyEngine(
1070110805
self,
1070210806
) -> SemanticGovernancePolicyEngineResource: ...
10807+
def skills(self) -> SkillsResource: ...
1070310808
def solvers(self) -> SolversResource: ...
1070410809
def specialistPools(self) -> SpecialistPoolsResource: ...
1070510810
def studies(self) -> StudiesResource: ...
1070610811
def tensorboards(self) -> TensorboardsResource: ...
1070710812
def trainingPipelines(self) -> TrainingPipelinesResource: ...
1070810813
def tuningJobs(self) -> TuningJobsResource: ...
1070910814
def v1beta1(self) -> V1beta1Resource: ...
10815+
def vertexExpress(self) -> VertexExpressResource: ...
1071010816

1071110817
@typing.type_check_only
1071210818
class GoogleApiHttpBodyHttpRequest(googleapiclient.http.HttpRequest):
@@ -10882,6 +10988,16 @@ class GoogleCloudAiplatformV1beta1CachedContentHttpRequest(
1088210988
num_retries: int = 0,
1088310989
) -> GoogleCloudAiplatformV1beta1CachedContent: ...
1088410990

10991+
@typing.type_check_only
10992+
class GoogleCloudAiplatformV1beta1CheckSignUpEligibilityResponseHttpRequest(
10993+
googleapiclient.http.HttpRequest
10994+
):
10995+
def execute(
10996+
self,
10997+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
10998+
num_retries: int = 0,
10999+
) -> GoogleCloudAiplatformV1beta1CheckSignUpEligibilityResponse: ...
11000+
1088511001
@typing.type_check_only
1088611002
class GoogleCloudAiplatformV1beta1ComputeTokensResponseHttpRequest(
1088711003
googleapiclient.http.HttpRequest
@@ -11146,6 +11262,16 @@ class GoogleCloudAiplatformV1beta1ExportTensorboardTimeSeriesDataResponseHttpReq
1114611262
num_retries: int = 0,
1114711263
) -> GoogleCloudAiplatformV1beta1ExportTensorboardTimeSeriesDataResponse: ...
1114811264

11265+
@typing.type_check_only
11266+
class GoogleCloudAiplatformV1beta1ExpressProjectHttpRequest(
11267+
googleapiclient.http.HttpRequest
11268+
):
11269+
def execute(
11270+
self,
11271+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
11272+
num_retries: int = 0,
11273+
) -> GoogleCloudAiplatformV1beta1ExpressProject: ...
11274+
1114911275
@typing.type_check_only
1115011276
class GoogleCloudAiplatformV1beta1ExtensionHttpRequest(
1115111277
googleapiclient.http.HttpRequest
@@ -12594,6 +12720,16 @@ class GoogleCloudAiplatformV1beta1RetrieveContextsResponseHttpRequest(
1259412720
num_retries: int = 0,
1259512721
) -> GoogleCloudAiplatformV1beta1RetrieveContextsResponse: ...
1259612722

12723+
@typing.type_check_only
12724+
class GoogleCloudAiplatformV1beta1RetrieveExpressProjectResponseHttpRequest(
12725+
googleapiclient.http.HttpRequest
12726+
):
12727+
def execute(
12728+
self,
12729+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
12730+
num_retries: int = 0,
12731+
) -> GoogleCloudAiplatformV1beta1RetrieveExpressProjectResponse: ...
12732+
1259712733
@typing.type_check_only
1259812734
class GoogleCloudAiplatformV1beta1RetrieveMemoriesResponseHttpRequest(
1259912735
googleapiclient.http.HttpRequest

0 commit comments

Comments
 (0)