Skip to content

Update stubs #77

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
henribru merged 1 commit into from
Mar 11, 2026
Merged

Update stubs #77

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
12 changes: 12 additions & 0 deletions googleapiclient-stubs/_apis/aiplatform/v1/resources.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4806,6 +4806,17 @@ class AiplatformResource(googleapiclient.discovery.Resource):
class PublishersResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class ModelsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class InvokeResource(googleapiclient.discovery.Resource):
def invoke(
self,
*,
endpoint: str,
invokeId: str,
body: GoogleCloudAiplatformV1InvokeRequest = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...

def computeTokens(
self,
*,
Expand Down Expand Up @@ -4883,6 +4894,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
body: GoogleCloudAiplatformV1StreamRawPredictRequest = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...
def invoke(self) -> InvokeResource: ...

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

Expand Down
11 changes: 11 additions & 0 deletions googleapiclient-stubs/_apis/aiplatform/v1/schemas.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3455,6 +3455,9 @@ class GoogleCloudAiplatformV1FunctionResponse(typing_extensions.TypedDict, total
name: str
parts: _list[GoogleCloudAiplatformV1FunctionResponsePart]
response: dict[str, typing.Any]
scheduling: typing_extensions.Literal[
"SCHEDULING_UNSPECIFIED", "SILENT", "WHEN_IDLE", "INTERRUPT"
]

@typing.type_check_only
class GoogleCloudAiplatformV1FunctionResponseBlob(
Expand Down Expand Up @@ -7386,6 +7389,7 @@ class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpec(
developerConnectSource: (
GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectSource
)
imageSpec: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecImageSpec
inlineSource: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecInlineSource
pythonSpec: GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecPythonSpec

Expand All @@ -7405,6 +7409,12 @@ class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectSo
GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecDeveloperConnectConfig
)

@typing.type_check_only
class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecImageSpec(
typing_extensions.TypedDict, total=False
):
buildArgs: dict[str, typing.Any]

@typing.type_check_only
class GoogleCloudAiplatformV1ReasoningEngineSpecSourceCodeSpecInlineSource(
typing_extensions.TypedDict, total=False
Expand Down Expand Up @@ -7915,6 +7925,7 @@ class GoogleCloudAiplatformV1Schedule(typing_extensions.TypedDict, total=False):
lastPauseTime: str
lastResumeTime: str
lastScheduledRunResponse: GoogleCloudAiplatformV1ScheduleRunResponse
maxConcurrentActiveRunCount: str
maxConcurrentRunCount: str
maxRunCount: str
name: str
Expand Down
227 changes: 227 additions & 0 deletions googleapiclient-stubs/_apis/aiplatform/v1beta1/resources.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6117,6 +6117,17 @@ class AiplatformResource(googleapiclient.discovery.Resource):
class PublishersResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class ModelsResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class InvokeResource(googleapiclient.discovery.Resource):
def invoke(
self,
*,
endpoint: str,
invokeId: str,
body: GoogleCloudAiplatformV1beta1InvokeRequest = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...

def computeTokens(
self,
*,
Expand Down Expand Up @@ -6231,6 +6242,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
body: GoogleCloudAiplatformV1beta1StreamRawPredictRequest = ...,
**kwargs: typing.Any,
) -> GoogleApiHttpBodyHttpRequest: ...
def invoke(self) -> InvokeResource: ...

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

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

@typing.type_check_only
class RagDataSchemasResource(googleapiclient.discovery.Resource):
def batchCreate(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1BatchCreateRagDataSchemasRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def batchDelete(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1BatchDeleteRagDataSchemasRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def create(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1RagDataSchema = ...,
ragDataSchemaId: str = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1RagDataSchemaHttpRequest: ...
def delete(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def get(
self, *, name: str, **kwargs: typing.Any
) -> GoogleCloudAiplatformV1beta1RagDataSchemaHttpRequest: ...
def list(
self,
*,
parent: str,
pageSize: int = ...,
pageToken: str = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest: ...
def list_next(
self,
previous_request: GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest,
previous_response: GoogleCloudAiplatformV1beta1ListRagDataSchemasResponse,
) -> (
GoogleCloudAiplatformV1beta1ListRagDataSchemasResponseHttpRequest
| None
): ...

@typing.type_check_only
class RagFilesResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -6300,6 +6359,60 @@ class AiplatformResource(googleapiclient.discovery.Resource):
self, *, name: str, timeout: str = ..., **kwargs: typing.Any
) -> GoogleLongrunningOperationHttpRequest: ...

@typing.type_check_only
class RagMetadataResource(googleapiclient.discovery.Resource):
def batchCreate(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1BatchCreateRagMetadataRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def batchDelete(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1BatchDeleteRagMetadataRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def create(
self,
*,
parent: str,
body: GoogleCloudAiplatformV1beta1RagMetadata = ...,
ragMetadataId: str = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...
def delete(
self, *, name: str, **kwargs: typing.Any
) -> GoogleProtobufEmptyHttpRequest: ...
def get(
self, *, name: str, **kwargs: typing.Any
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...
def list(
self,
*,
parent: str,
pageSize: int = ...,
pageToken: str = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest: ...
def list_next(
self,
previous_request: GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest,
previous_response: GoogleCloudAiplatformV1beta1ListRagMetadataResponse,
) -> (
GoogleCloudAiplatformV1beta1ListRagMetadataResponseHttpRequest
| None
): ...
def patch(
self,
*,
name: str,
body: GoogleCloudAiplatformV1beta1RagMetadata = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1RagMetadataHttpRequest: ...

def delete(
self,
*,
Expand Down Expand Up @@ -6336,6 +6449,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
| None
): ...
def operations(self) -> OperationsResource: ...
def ragMetadata(self) -> RagMetadataResource: ...

def create(
self,
Expand Down Expand Up @@ -6378,6 +6492,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def operations(self) -> OperationsResource: ...
def ragDataSchemas(self) -> RagDataSchemasResource: ...
def ragFiles(self) -> RagFilesResource: ...

@typing.type_check_only
Expand Down Expand Up @@ -6513,6 +6628,42 @@ class AiplatformResource(googleapiclient.discovery.Resource):

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

@typing.type_check_only
class FeedbackEntriesResource(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 MemoriesResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -6941,6 +7092,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
) -> GoogleIamV1TestIamPermissionsResponseHttpRequest: ...
def a2a(self) -> A2aResource: ...
def examples(self) -> ExamplesResource: ...
def feedbackEntries(self) -> FeedbackEntriesResource: ...
def memories(self) -> MemoriesResource: ...
def operations(self) -> OperationsResource: ...
def sandboxEnvironments(self) -> SandboxEnvironmentsResource: ...
Expand Down Expand Up @@ -8356,6 +8508,40 @@ class AiplatformResource(googleapiclient.discovery.Resource):

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

@typing.type_check_only
class FeedbackEntriesResource(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 MemoriesResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -8743,6 +8929,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
) -> GoogleApiHttpBodyHttpRequest: ...
def a2a(self) -> A2aResource: ...
def examples(self) -> ExamplesResource: ...
def feedbackEntries(self) -> FeedbackEntriesResource: ...
def memories(self) -> MemoriesResource: ...
def operations(self) -> OperationsResource: ...
def sandboxEnvironments(self) -> SandboxEnvironmentsResource: ...
Expand Down Expand Up @@ -10328,6 +10515,16 @@ class GoogleCloudAiplatformV1beta1ListRagCorporaResponseHttpRequest(
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1ListRagCorporaResponse: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1ListRagFilesResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand All @@ -10338,6 +10535,16 @@ class GoogleCloudAiplatformV1beta1ListRagFilesResponseHttpRequest(
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1ListRagFilesResponse: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1ListReasoningEnginesResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand Down Expand Up @@ -10732,6 +10939,16 @@ class GoogleCloudAiplatformV1beta1RagCorpusHttpRequest(
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1RagCorpus: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1RagEngineConfigHttpRequest(
googleapiclient.http.HttpRequest
Expand All @@ -10750,6 +10967,16 @@ class GoogleCloudAiplatformV1beta1RagFileHttpRequest(googleapiclient.http.HttpRe
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1RagFile: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1ReadFeatureValuesResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand Down
Loading
Loading