Skip to content

Commit 9f2f0ad

Browse files
sararobcopybara-github
authored andcommitted
chore: fix issue where unreferenced enums weren't getting generated into types
PiperOrigin-RevId: 825574934
1 parent 39bb9df commit 9f2f0ad

3 files changed

Lines changed: 1 addition & 27 deletions

File tree

tests/unit/vertexai/genai/test_genai_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,4 +92,4 @@ async def test_call_aclose_async_client(self):
9292
project=_TEST_PROJECT, location=_TEST_LOCATION
9393
).aio
9494
await async_client.aclose()
95-
mock_aclose.assert_called_once()
95+
mock_aclose.assert_called()

vertexai/_genai/types/__init__.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -587,9 +587,6 @@
587587
from .common import OptimizeResponseEndpointDict
588588
from .common import OptimizeResponseEndpointOrDict
589589
from .common import OptimizeResponseOrDict
590-
from .common import OptimizerMethodPlaceholder
591-
from .common import OptimizerMethodPlaceholderDict
592-
from .common import OptimizerMethodPlaceholderOrDict
593590
from .common import OptimizeTarget
594591
from .common import PairwiseChoice
595592
from .common import PairwiseMetricInput
@@ -1746,9 +1743,6 @@
17461743
"PromptOptimizerConfig",
17471744
"PromptOptimizerConfigDict",
17481745
"PromptOptimizerConfigOrDict",
1749-
"OptimizerMethodPlaceholder",
1750-
"OptimizerMethodPlaceholderDict",
1751-
"OptimizerMethodPlaceholderOrDict",
17521746
"ApplicableGuideline",
17531747
"ApplicableGuidelineDict",
17541748
"ApplicableGuidelineOrDict",

vertexai/_genai/types/common.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -12081,26 +12081,6 @@ class PromptOptimizerConfigDict(TypedDict, total=False):
1208112081
PromptOptimizerConfigOrDict = Union[PromptOptimizerConfig, PromptOptimizerConfigDict]
1208212082

1208312083

12084-
class OptimizerMethodPlaceholder(_common.BaseModel):
12085-
"""Placeholder class to generate OptimizerMethod enum in common.py."""
12086-
12087-
method: Optional[PromptOptimizerMethod] = Field(
12088-
default=None, description="""The method for optimizing multiple prompts."""
12089-
)
12090-
12091-
12092-
class OptimizerMethodPlaceholderDict(TypedDict, total=False):
12093-
"""Placeholder class to generate OptimizerMethod enum in common.py."""
12094-
12095-
method: Optional[PromptOptimizerMethod]
12096-
"""The method for optimizing multiple prompts."""
12097-
12098-
12099-
OptimizerMethodPlaceholderOrDict = Union[
12100-
OptimizerMethodPlaceholder, OptimizerMethodPlaceholderDict
12101-
]
12102-
12103-
1210412084
class ApplicableGuideline(_common.BaseModel):
1210512085
"""Applicable guideline for the optimize_prompt method."""
1210612086

0 commit comments

Comments
 (0)