Skip to content

Commit 0de2987

Browse files
Ark-kuncopybara-github
authored andcommitted
fix: GenAI - Fixed GenerativeModel.compute_tokens for v1 API
PiperOrigin-RevId: 675711678
1 parent f882657 commit 0de2987

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vertexai/generative_models/_generative_models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2960,7 +2960,7 @@ def _gapic_compute_tokens(
29602960
prediction_resource_name: str,
29612961
contents: List[gapic_content_types.Content],
29622962
) -> gapic_prediction_service_types.CountTokensResponse:
2963-
request = gapic_llm_utility_service_types.ComputeTokensRequest(
2963+
request = types_v1.ComputeTokensRequest(
29642964
endpoint=prediction_resource_name,
29652965
model=prediction_resource_name,
29662966
contents=contents,
@@ -2972,7 +2972,7 @@ async def _gapic_compute_tokens_async(
29722972
prediction_resource_name: str,
29732973
contents: List[gapic_content_types.Content],
29742974
) -> gapic_prediction_service_types.CountTokensResponse:
2975-
request = gapic_llm_utility_service_types.ComputeTokensRequest(
2975+
request = types_v1.ComputeTokensRequest(
29762976
endpoint=prediction_resource_name,
29772977
model=prediction_resource_name,
29782978
contents=contents,

0 commit comments

Comments
 (0)