File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
vertexai/generative_models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments