Skip to content

Commit 0c285ea

Browse files
henribruhenribru
authored
Update stubs (#77)
1 parent 8593a40 commit 0c285ea

77 files changed

Lines changed: 1163 additions & 731 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: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4806,6 +4806,17 @@ class AiplatformResource(googleapiclient.discovery.Resource):
48064806
class PublishersResource(googleapiclient.discovery.Resource):
48074807
@typing.type_check_only
48084808
class ModelsResource(googleapiclient.discovery.Resource):
4809+
@typing.type_check_only
4810+
class InvokeResource(googleapiclient.discovery.Resource):
4811+
def invoke(
4812+
self,
4813+
*,
4814+
endpoint: str,
4815+
invokeId: str,
4816+
body: GoogleCloudAiplatformV1InvokeRequest = ...,
4817+
**kwargs: typing.Any,
4818+
) -> GoogleApiHttpBodyHttpRequest: ...
4819+
48094820
def computeTokens(
48104821
self,
48114822
*,
@@ -4883,6 +4894,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
48834894
body: GoogleCloudAiplatformV1StreamRawPredictRequest = ...,
48844895
**kwargs: typing.Any,
48854896
) -> GoogleApiHttpBodyHttpRequest: ...
4897+
def invoke(self) -> InvokeResource: ...
48864898

48874899
def models(self) -> ModelsResource: ...
48884900

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

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3455,6 +3455,9 @@ class GoogleCloudAiplatformV1FunctionResponse(typing_extensions.TypedDict, total
34553455
name: str
34563456
parts: _list[GoogleCloudAiplatformV1FunctionResponsePart]
34573457
response: dict[str, typing.Any]
3458+
scheduling: typing_extensions.Literal[
3459+
"SCHEDULING_UNSPECIFIED", "SILENT", "WHEN_IDLE", "INTERRUPT"
3460+
]
34583461

34593462
@typing.type_check_only
34603463
class GoogleCloudAiplatformV1FunctionResponseBlob(
@@ -7386,6 +7389,7 @@ class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpec(
73867389
developerConnectSource: (
73877390
GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectSource
73887391
)
7392+
imageSpec: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecImageSpec
73897393
inlineSource: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecInlineSource
73907394
pythonSpec: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecPythonSpec
73917395

@@ -7405,6 +7409,12 @@ class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectSo
74057409
GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectConfig
74067410
)
74077411

7412+
@typing.type_check_only
7413+
class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecImageSpec(
7414+
typing_extensions.TypedDict, total=False
7415+
):
7416+
buildArgs: dict[str, typing.Any]
7417+
74087418
@typing.type_check_only
74097419
class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecInlineSource(
74107420
typing_extensions.TypedDict, total=False
@@ -7915,6 +7925,7 @@ class GoogleCloudAiplatformV1Schedule(typing_extensions.TypedDict, total=False):
79157925
lastPauseTime: str
79167926
lastResumeTime: str
79177927
lastScheduledRunResponse: GoogleCloudAiplatformV1ScheduleRunResponse
7928+
maxConcurrentActiveRunCount: str
79187929
maxConcurrentRunCount: str
79197930
maxRunCount: str
79207931
name: str

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

Lines changed: 227 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6117,6 +6117,17 @@ class AiplatformResource(googleapiclient.discovery.Resource):
61176117
class PublishersResource(googleapiclient.discovery.Resource):
61186118
@typing.type_check_only
61196119
class ModelsResource(googleapiclient.discovery.Resource):
6120+
@typing.type_check_only
6121+
class InvokeResource(googleapiclient.discovery.Resource):
6122+
def invoke(
6123+
self,
6124+
*,
6125+
endpoint: str,
6126+
invokeId: str,
6127+
body: GoogleCloudAiplatformV1beta1InvokeRequest = ...,
6128+
**kwargs: typing.Any,
6129+
) -> GoogleApiHttpBodyHttpRequest: ...
6130+
61206131
def computeTokens(
61216132
self,
61226133
*,
@@ -6231,6 +6242,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
62316242
body: GoogleCloudAiplatformV1beta1StreamRawPredictRequest = ...,
62326243
**kwargs: typing.Any,
62336244
) -> GoogleApiHttpBodyHttpRequest: ...
6245+
def invoke(self) -> InvokeResource: ...
62346246

62356247
def models(self) -> ModelsResource: ...
62366248

@@ -6266,6 +6278,53 @@ class AiplatformResource(googleapiclient.discovery.Resource):
62666278
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
62676279
) -> GoogleLongrunningOperationHttpRequest: ...
62686280

6281+
@typing.type_check_only
6282+
class RagDataSchemasResource(googleapiclient.discovery.Resource):
6283+
def batchCreate(
6284+
self,
6285+
*,
6286+
parent: str,
6287+
body: GoogleCloudAiplatformV1beta1BatchCreateRagDataSchemasRequest = ...,
6288+
**kwargs: typing.Any,
6289+
) -> GoogleLongrunningOperationHttpRequest: ...
6290+
def batchDelete(
6291+
self,
6292+
*,
6293+
parent: str,
6294+
body: GoogleCloudAiplatformV1beta1BatchDeleteRagDataSchemasRequest = ...,
6295+
**kwargs: typing.Any,
6296+
) -> GoogleLongrunningOperationHttpRequest: ...
6297+
def create(
6298+
self,
6299+
*,
6300+
parent: str,
6301+
body: GoogleCloudAiplatformV1beta1RagDataSchema = ...,
6302+
ragDataSchemaId: str = ...,
6303+
**kwargs: typing.Any,
6304+
) -> GoogleCloudAiplatformV1beta1RagDataSchemaHttpRequest: ...
6305+
def delete(
6306+
self, *, name: str, **kwargs: typing.Any
6307+
) -> GoogleProtobufEmptyHttpRequest: ...
6308+
def get(
6309+
self, *, name: str, **kwargs: typing.Any
6310+
) -> GoogleCloudAiplatformV1beta1RagDataSchemaHttpRequest: ...
6311+
def list(
6312+
self,
6313+
*,
6314+
parent: str,
6315+
pageSize: int = ...,
6316+
pageToken: str = ...,
6317+
**kwargs: typing.Any,
6318+
) -> GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest: ...
6319+
def list_next(
6320+
self,
6321+
previous_request: GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest,
6322+
previous_response: GoogleCloudAiplatformV1beta1ListRagDataSchemasResponse,
6323+
) -> (
6324+
GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest
6325+
| None
6326+
): ...
6327+
62696328
@typing.type_check_only
62706329
class RagFilesResource(googleapiclient.discovery.Resource):
62716330
@typing.type_check_only
@@ -6300,6 +6359,60 @@ class AiplatformResource(googleapiclient.discovery.Resource):
63006359
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
63016360
) -> GoogleLongrunningOperationHttpRequest: ...
63026361

6362+
@typing.type_check_only
6363+
class RagMetadataResource(googleapiclient.discovery.Resource):
6364+
def batchCreate(
6365+
self,
6366+
*,
6367+
parent: str,
6368+
body: GoogleCloudAiplatformV1beta1BatchCreateRagMetadataRequest = ...,
6369+
**kwargs: typing.Any,
6370+
) -> GoogleLongrunningOperationHttpRequest: ...
6371+
def batchDelete(
6372+
self,
6373+
*,
6374+
parent: str,
6375+
body: GoogleCloudAiplatformV1beta1BatchDeleteRagMetadataRequest = ...,
6376+
**kwargs: typing.Any,
6377+
) -> GoogleLongrunningOperationHttpRequest: ...
6378+
def create(
6379+
self,
6380+
*,
6381+
parent: str,
6382+
body: GoogleCloudAiplatformV1beta1RagMetadata = ...,
6383+
ragMetadataId: str = ...,
6384+
**kwargs: typing.Any,
6385+
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...
6386+
def delete(
6387+
self, *, name: str, **kwargs: typing.Any
6388+
) -> GoogleProtobufEmptyHttpRequest: ...
6389+
def get(
6390+
self, *, name: str, **kwargs: typing.Any
6391+
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...
6392+
def list(
6393+
self,
6394+
*,
6395+
parent: str,
6396+
pageSize: int = ...,
6397+
pageToken: str = ...,
6398+
**kwargs: typing.Any,
6399+
) -> GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest: ...
6400+
def list_next(
6401+
self,
6402+
previous_request: GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest,
6403+
previous_response: GoogleCloudAiplatformV1beta1ListRagMetadataResponse,
6404+
) -> (
6405+
GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest
6406+
| None
6407+
): ...
6408+
def patch(
6409+
self,
6410+
*,
6411+
name: str,
6412+
body: GoogleCloudAiplatformV1beta1RagMetadata = ...,
6413+
**kwargs: typing.Any,
6414+
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...
6415+
63036416
def delete(
63046417
self,
63056418
*,
@@ -6336,6 +6449,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
63366449
| None
63376450
): ...
63386451
def operations(self) -> OperationsResource: ...
6452+
def ragMetadata(self) -> RagMetadataResource: ...
63396453

63406454
def create(
63416455
self,
@@ -6378,6 +6492,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
63786492
**kwargs: typing.Any,
63796493
) -> GoogleLongrunningOperationHttpRequest: ...
63806494
def operations(self) -> OperationsResource: ...
6495+
def ragDataSchemas(self) -> RagDataSchemasResource: ...
63816496
def ragFiles(self) -> RagFilesResource: ...
63826497

63836498
@typing.type_check_only
@@ -6513,6 +6628,42 @@ class AiplatformResource(googleapiclient.discovery.Resource):
65136628

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

6631+
@typing.type_check_only
6632+
class FeedbackEntriesResource(googleapiclient.discovery.Resource):
6633+
@typing.type_check_only
6634+
class OperationsResource(googleapiclient.discovery.Resource):
6635+
def cancel(
6636+
self, *, name: str, **kwargs: typing.Any
6637+
) -> GoogleProtobufEmptyHttpRequest: ...
6638+
def delete(
6639+
self, *, name: str, **kwargs: typing.Any
6640+
) -> GoogleProtobufEmptyHttpRequest: ...
6641+
def get(
6642+
self, *, name: str, **kwargs: typing.Any
6643+
) -> GoogleLongrunningOperationHttpRequest: ...
6644+
def list(
6645+
self,
6646+
*,
6647+
name: str,
6648+
filter: str = ...,
6649+
pageSize: int = ...,
6650+
pageToken: str = ...,
6651+
returnPartialSuccess: bool = ...,
6652+
**kwargs: typing.Any,
6653+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
6654+
def list_next(
6655+
self,
6656+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
6657+
previous_response: GoogleLongrunningListOperationsResponse,
6658+
) -> (
6659+
GoogleLongrunningListOperationsResponseHttpRequest | None
6660+
): ...
6661+
def wait(
6662+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
6663+
) -> GoogleLongrunningOperationHttpRequest: ...
6664+
6665+
def operations(self) -> OperationsResource: ...
6666+
65166667
@typing.type_check_only
65176668
class MemoriesResource(googleapiclient.discovery.Resource):
65186669
@typing.type_check_only
@@ -6941,6 +7092,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
69417092
) -> GoogleIamV1TestIamPermissionsResponseHttpRequest: ...
69427093
def a2a(self) -> A2aResource: ...
69437094
def examples(self) -> ExamplesResource: ...
7095+
def feedbackEntries(self) -> FeedbackEntriesResource: ...
69447096
def memories(self) -> MemoriesResource: ...
69457097
def operations(self) -> OperationsResource: ...
69467098
def sandboxEnvironments(self) -> SandboxEnvironmentsResource: ...
@@ -8356,6 +8508,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):
83568508

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

8511+
@typing.type_check_only
8512+
class FeedbackEntriesResource(googleapiclient.discovery.Resource):
8513+
@typing.type_check_only
8514+
class OperationsResource(googleapiclient.discovery.Resource):
8515+
def cancel(
8516+
self, *, name: str, **kwargs: typing.Any
8517+
) -> GoogleProtobufEmptyHttpRequest: ...
8518+
def delete(
8519+
self, *, name: str, **kwargs: typing.Any
8520+
) -> GoogleProtobufEmptyHttpRequest: ...
8521+
def get(
8522+
self, *, name: str, **kwargs: typing.Any
8523+
) -> GoogleLongrunningOperationHttpRequest: ...
8524+
def list(
8525+
self,
8526+
*,
8527+
name: str,
8528+
filter: str = ...,
8529+
pageSize: int = ...,
8530+
pageToken: str = ...,
8531+
returnPartialSuccess: bool = ...,
8532+
**kwargs: typing.Any,
8533+
) -> GoogleLongrunningListOperationsResponseHttpRequest: ...
8534+
def list_next(
8535+
self,
8536+
previous_request: GoogleLongrunningListOperationsResponseHttpRequest,
8537+
previous_response: GoogleLongrunningListOperationsResponse,
8538+
) -> GoogleLongrunningListOperationsResponseHttpRequest | None: ...
8539+
def wait(
8540+
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
8541+
) -> GoogleLongrunningOperationHttpRequest: ...
8542+
8543+
def operations(self) -> OperationsResource: ...
8544+
83598545
@typing.type_check_only
83608546
class MemoriesResource(googleapiclient.discovery.Resource):
83618547
@typing.type_check_only
@@ -8743,6 +8929,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
87438929
) -> GoogleApiHttpBodyHttpRequest: ...
87448930
def a2a(self) -> A2aResource: ...
87458931
def examples(self) -> ExamplesResource: ...
8932+
def feedbackEntries(self) -> FeedbackEntriesResource: ...
87468933
def memories(self) -> MemoriesResource: ...
87478934
def operations(self) -> OperationsResource: ...
87488935
def sandboxEnvironments(self) -> SandboxEnvironmentsResource: ...
@@ -10328,6 +10515,16 @@ class GoogleCloudAiplatformV1beta1ListRagCorporaResponseHttpRequest(
1032810515
num_retries: int = 0,
1032910516
) -> GoogleCloudAiplatformV1beta1ListRagCorporaResponse: ...
1033010517

10518+
@typing.type_check_only
10519+
class GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest(
10520+
googleapiclient.http.HttpRequest
10521+
):
10522+
def execute(
10523+
self,
10524+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
10525+
num_retries: int = 0,
10526+
) -> GoogleCloudAiplatformV1beta1ListRagDataSchemasResponse: ...
10527+
1033110528
@typing.type_check_only
1033210529
class GoogleCloudAiplatformV1beta1ListRagFilesResponseHttpRequest(
1033310530
googleapiclient.http.HttpRequest
@@ -10338,6 +10535,16 @@ class GoogleCloudAiplatformV1beta1ListRagFilesResponseHttpRequest(
1033810535
num_retries: int = 0,
1033910536
) -> GoogleCloudAiplatformV1beta1ListRagFilesResponse: ...
1034010537

10538+
@typing.type_check_only
10539+
class GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest(
10540+
googleapiclient.http.HttpRequest
10541+
):
10542+
def execute(
10543+
self,
10544+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
10545+
num_retries: int = 0,
10546+
) -> GoogleCloudAiplatformV1beta1ListRagMetadataResponse: ...
10547+
1034110548
@typing.type_check_only
1034210549
class GoogleCloudAiplatformV1beta1ListReasoningEnginesResponseHttpRequest(
1034310550
googleapiclient.http.HttpRequest
@@ -10732,6 +10939,16 @@ class GoogleCloudAiplatformV1beta1RagCorpusHttpRequest(
1073210939
num_retries: int = 0,
1073310940
) -> GoogleCloudAiplatformV1beta1RagCorpus: ...
1073410941

10942+
@typing.type_check_only
10943+
class GoogleCloudAiplatformV1beta1RagDataSchemaHttpRequest(
10944+
googleapiclient.http.HttpRequest
10945+
):
10946+
def execute(
10947+
self,
10948+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
10949+
num_retries: int = 0,
10950+
) -> GoogleCloudAiplatformV1beta1RagDataSchema: ...
10951+
1073510952
@typing.type_check_only
1073610953
class GoogleCloudAiplatformV1beta1RagEngineConfigHttpRequest(
1073710954
googleapiclient.http.HttpRequest
@@ -10750,6 +10967,16 @@ class GoogleCloudAiplatformV1beta1RagFileHttpRequest(googleapiclient.http.HttpRe
1075010967
num_retries: int = 0,
1075110968
) -> GoogleCloudAiplatformV1beta1RagFile: ...
1075210969

10970+
@typing.type_check_only
10971+
class GoogleCloudAiplatformV1beta1RagMetadataHttpRequest(
10972+
googleapiclient.http.HttpRequest
10973+
):
10974+
def execute(
10975+
self,
10976+
http: httplib2.Http | googleapiclient.http.HttpMock | None = None,
10977+
num_retries: int = 0,
10978+
) -> GoogleCloudAiplatformV1beta1RagMetadata: ...
10979+
1075310980
@typing.type_check_only
1075410981
class GoogleCloudAiplatformV1beta1ReadFeatureValuesResponseHttpRequest(
1075510982
googleapiclient.http.HttpRequest

0 commit comments

Comments
 (0)