Skip to content

Commit a267d43

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Increase default timeout to 300 seconds for ask_contexts and async_retrieve_contexts in VertexRagServiceClient.
PiperOrigin-RevId: 892413253
1 parent 3502e7e commit a267d43

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

  • google/cloud
    • aiplatform_v1beta1/services/vertex_rag_service
    • aiplatform_v1/services/vertex_rag_service

google/cloud/aiplatform_v1/services/vertex_rag_service/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1176,7 +1176,7 @@ def ask_contexts(
11761176
parent: Optional[str] = None,
11771177
query: Optional[vertex_rag_service.RagQuery] = None,
11781178
retry: OptionalRetry = gapic_v1.method.DEFAULT,
1179-
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
1179+
timeout: Union[float, object] = 300.0,
11801180
metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
11811181
) -> vertex_rag_service.AskContextsResponse:
11821182
r"""Agentic Retrieval Ask API for RAG.
@@ -1300,7 +1300,7 @@ def async_retrieve_contexts(
13001300
parent: Optional[str] = None,
13011301
query: Optional[vertex_rag_service.RagQuery] = None,
13021302
retry: OptionalRetry = gapic_v1.method.DEFAULT,
1303-
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
1303+
timeout: Union[float, object] = 300.0,
13041304
metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
13051305
) -> operation.Operation:
13061306
r"""Asynchronous API to retrieves relevant contexts for a

google/cloud/aiplatform_v1beta1/services/vertex_rag_service/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1176,7 +1176,7 @@ def ask_contexts(
11761176
parent: Optional[str] = None,
11771177
query: Optional[vertex_rag_service.RagQuery] = None,
11781178
retry: OptionalRetry = gapic_v1.method.DEFAULT,
1179-
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
1179+
timeout: Union[float, object] = 300.0,
11801180
metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
11811181
) -> vertex_rag_service.AskContextsResponse:
11821182
r"""Agentic Retrieval Ask API for RAG.
@@ -1300,7 +1300,7 @@ def async_retrieve_contexts(
13001300
parent: Optional[str] = None,
13011301
query: Optional[vertex_rag_service.RagQuery] = None,
13021302
retry: OptionalRetry = gapic_v1.method.DEFAULT,
1303-
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
1303+
timeout: Union[float, object] = 300.0,
13041304
metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
13051305
) -> operation.Operation:
13061306
r"""Asynchronous API to retrieves relevant contexts for a

0 commit comments

Comments
 (0)