Skip to content

Update stubs #80

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

Open
henribru wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Update stubs #80

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
84 changes: 84 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 @@ -5521,6 +5521,13 @@ 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 create(
self,
*,
Expand Down Expand Up @@ -5697,6 +5704,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 +6660,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 +7125,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):

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

def asyncQuery(
self,
*,
name: str,
body: GoogleCloudAiplatformV1AsyncQueryReasoningEngineRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def create(
self,
*,
Expand Down Expand Up @@ -7185,6 +7234,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 +7641,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
16 changes: 9 additions & 7 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 @@ -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 @@ -1120,6 +1128,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 @@ -8247,7 +8256,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 +8302,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
136 changes: 136 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 @@ -7713,6 +7713,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def events(self) -> EventsResource: ...
def operations(self) -> OperationsResource: ...

def asyncQuery(
self,
*,
name: str,
body: GoogleCloudAiplatformV1beta1AsyncQueryReasoningEngineRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def create(
self,
*,
Expand Down Expand Up @@ -7971,6 +7978,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 SolversResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -8969,6 +9010,7 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def semanticGovernancePolicyEngine(
self,
) -> SemanticGovernancePolicyEngineResource: ...
def skills(self) -> SkillsResource: ...
def solvers(self) -> SolversResource: ...
def specialistPools(self) -> SpecialistPoolsResource: ...
def studies(self) -> StudiesResource: ...
Expand Down Expand Up @@ -10128,6 +10170,13 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def events(self) -> EventsResource: ...
def operations(self) -> OperationsResource: ...

def asyncQuery(
self,
*,
name: str,
body: GoogleCloudAiplatformV1beta1AsyncQueryReasoningEngineRequest = ...,
**kwargs: typing.Any,
) -> GoogleLongrunningOperationHttpRequest: ...
def create(
self,
*,
Expand Down Expand Up @@ -10303,6 +10352,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 SolversResource(googleapiclient.discovery.Resource):
@typing.type_check_only
Expand Down Expand Up @@ -10640,6 +10723,27 @@ class AiplatformResource(googleapiclient.discovery.Resource):
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1GenerateInstanceRubricsResponseHttpRequest: ...

@typing.type_check_only
class VertexExpressResource(googleapiclient.discovery.Resource):
@typing.type_check_only
class EligibilityResource(googleapiclient.discovery.Resource):
def check(
self, **kwargs: typing.Any
) -> (
GoogleCloudAiplatformV1beta1CheckSignUpEligibilityResponseHttpRequest
): ...

def retrieveExpressProject(
self, *, getDefaultApiKey: bool = ..., **kwargs: typing.Any
) -> GoogleCloudAiplatformV1beta1RetrieveExpressProjectResponseHttpRequest: ...
def signUp(
self,
*,
body: GoogleCloudAiplatformV1beta1SignUpRequest = ...,
**kwargs: typing.Any,
) -> GoogleCloudAiplatformV1beta1ExpressProjectHttpRequest: ...
def Eligibility(self) -> EligibilityResource: ...

def new_batch_http_request(
self,
callback: collections.abc.Callable[
Expand Down Expand Up @@ -10700,13 +10804,15 @@ class AiplatformResource(googleapiclient.discovery.Resource):
def semanticGovernancePolicyEngine(
self,
) -> SemanticGovernancePolicyEngineResource: ...
def skills(self) -> SkillsResource: ...
def solvers(self) -> SolversResource: ...
def specialistPools(self) -> SpecialistPoolsResource: ...
def studies(self) -> StudiesResource: ...
def tensorboards(self) -> TensorboardsResource: ...
def trainingPipelines(self) -> TrainingPipelinesResource: ...
def tuningJobs(self) -> TuningJobsResource: ...
def v1beta1(self) -> V1beta1Resource: ...
def vertexExpress(self) -> VertexExpressResource: ...

@typing.type_check_only
class GoogleApiHttpBodyHttpRequest(googleapiclient.http.HttpRequest):
Expand Down Expand Up @@ -10882,6 +10988,16 @@ class GoogleCloudAiplatformV1beta1CachedContentHttpRequest(
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1CachedContent: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1ComputeTokensResponseHttpRequest(
googleapiclient.http.HttpRequest
Expand Down Expand Up @@ -11146,6 +11262,16 @@ class GoogleCloudAiplatformV1beta1ExportTensorboardTimeSeriesDataResponseHttpReq
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1ExportTensorboardTimeSeriesDataResponse: ...

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

@typing.type_check_only
class GoogleCloudAiplatformV1beta1ExtensionHttpRequest(
googleapiclient.http.HttpRequest
Expand Down Expand Up @@ -12594,6 +12720,16 @@ class GoogleCloudAiplatformV1beta1RetrieveContextsResponseHttpRequest(
num_retries: int = 0,
) -> GoogleCloudAiplatformV1beta1RetrieveContextsResponse: ...

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

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